
    {-jL                     v    d dl mZmZ d dlmZ ddlZddlZ G d de          Z G d de          Zd	 Z	d
 Z
dS )   )	AstToGAst	GAstToAst)gast    Nc                   d   e Zd Zej        j        dk     rd Zej        j        dk     r	d Zd Zd Z	ej        j        dk     r-d Z
d	 Zd
 Zd Zd Zd Zd Zd Zd Zd Zd Zd Zd Zd Zd Zd Zd Zd Zej        j        dk     rd Zdej        j        cxk    rdk     rn d S d Zd Zd Zd S d S )!
Ast3ToGAst
   c                     t          j        |                     |j                  |                     |j                            }d x|_        |_        d x|_        |_        |S N)	r   alias_visitnameasnamelineno
col_offset
end_linenoend_col_offsetselfnodenew_nodes      V/var/www/html/banglarbhumi/venv/lib/python3.11/site-packages/paddle/utils/gast/ast3.pyvisit_aliaszAst3ToGAst.visit_alias)   s[    zDI&&DK(( H 598HOh1<@@H("9O    	   c                     t          j        |                     |j                  t          j                              }t          j        ||          S r   )r   Tupler   dimsLoadcopy_locationr   s      r   visit_ExtSlicezAst3ToGAst.visit_ExtSlice4   s:    z$++di"8"8$)++FFH%h555r   c                 6    |                      |j                  S r   )r   value)r   r   s     r   visit_IndexzAst3ToGAst.visit_Index8   s    ;;tz***r   c                     t          j        |                     |j                  |                     |j                  d           }t          j        ||           d x|_        |_        |S r   )r   Assignr   targetsr#   r    r   r   r   s      r   visit_AssignzAst3ToGAst.visit_Assign;   sb    {DL))DJ'' H x...<@@H("9Or      c                 `    t          j        |                     |j                  g           }|S r   )r   Moduler   bodyr   s      r   visit_ModulezAst3ToGAst.visit_ModuleH   s&    {4;;ty#9#92>>HOr   c                 `    t          j        |j        d           }t          j        ||          S r   )r   Constantnr    r   s      r   	visit_NumzAst3ToGAst.visit_NumL   0    } H %h555r   c                     t          j        t          d           }t          j        ||           d x|_        |_        |S r   )r   r/   Ellipsisr    r   r   r   s      r   visit_EllipsiszAst3ToGAst.visit_EllipsisS   sC    } H x...<@@H("9Or   c                 `    t          j        |j        d           }t          j        ||          S r   r   r/   sr    r   s      r   	visit_StrzAst3ToGAst.visit_Str\   r2   r   c                 `    t          j        |j        d           }t          j        ||          S r   r7   r   s      r   visit_ByteszAst3ToGAst.visit_Bytesc   r2   r   c           	      N   t          j        |                     |j                  |                     |j                  |                     |j                  |                     |j                  |                     |j                  g           }t          j        ||          S r   	r   ClassDefr   r   baseskeywordsr,   decorator_listr    r   s      r   visit_ClassDefzAst3ToGAst.visit_ClassDefj       }DI&&DJ''DM**DI&&D/00 H %h555r   c           	      P   t          j        |                     |j                  |                     |j                  |                     |j                  |                     |j                  |                     |j                  d g           }t          j        ||          S r   )	r   FunctionDefr   r   argsr,   rA   returnsr    r   s      r   visit_FunctionDefzAst3ToGAst.visit_FunctionDefu   s    'DI&&DI&&DI&&D/00DL)) H %h555r   c           	      P   t          j        |                     |j                  |                     |j                  |                     |j                  |                     |j                  |                     |j                  d g           }t          j        ||          S r   )	r   AsyncFunctionDefr   r   rF   r,   rA   rG   r    r   s      r   visit_AsyncFunctionDefz!Ast3ToGAst.visit_AsyncFunctionDef   s    ,DI&&DI&&DI&&D/00DL)) H %h555r   c                    t          j        |                     |j                  |                     |j                  |                     |j                  |                     |j                  d           }t          j        ||          S r   )r   Forr   targetiterr,   orelser    r   s      r   	visit_ForzAst3ToGAst.visit_For   so    xDK((DI&&DI&&DK(( H %h555r   c                    t          j        |                     |j                  |                     |j                  |                     |j                  |                     |j                  d           }t          j        ||          S r   )r   AsyncForr   rN   rO   r,   rP   r    r   s      r   visit_AsyncForzAst3ToGAst.visit_AsyncFor   so    }DK((DI&&DI&&DK(( H %h555r   c                     t          j        |                     |j                  |                     |j                  d           }t          j        ||          S r   )r   Withr   itemsr,   r    r   s      r   
visit_WithzAst3ToGAst.visit_With   sM    yDJ''DI&& H
 %h555r   c                     t          j        |                     |j                  |                     |j                  d           }t          j        ||          S r   )r   	AsyncWithr   rW   r,   r    r   s      r   visit_AsyncWithzAst3ToGAst.visit_AsyncWith   sM    ~DJ''DI&& H
 %h555r   c                    t           j        j        dk     r|j        rWt	          j        |                     |j                  t	          j                              }t	          j        ||           |g}ng }|j	        rKt	          j
        d |                     |j	                            }t	          j        ||j	                   |g}ng }ng x}}t	          j        |                     |j                  |                     |j                  |z   |                     |j                  |z             }t	          j        ||          S )N   )sysversion_infominorstarargsr   Starredr   r   r    kwargskeywordCallfuncrF   r@   )r   r   starstarredkwrc   r   s          r   
visit_CallzAst3ToGAst.visit_Call   s   %))= !<DM(B(BDIKKPPD&tT222#fGG G;  dDKK,D,DEEB&r4;777 TFFFF#%%&yDI&&DI&&0DM**V3 H
 %h555r   c                     |j         t          j        d d           }n=|j         du rt          j        dd           }n|j         du rt          j        dd           }t          j        ||          S )NTF)r#   r   r/   r    r   s      r   visit_NameConstantzAst3ToGAst.visit_NameConstant   sn    z!=t44t##=t44u$$=55%h555r   c                    t          j        |                     |j                  g |                     |j                  |                     |j                  |                     |j                  |                     |j                  |                     |j                            }t          j	        ||          S r   )
r   	argumentsr   rF   vararg
kwonlyargskw_defaultskwargdefaultsr    r   s      r   visit_argumentszAst3ToGAst.visit_arguments   s    ~DI&&DK((DO,,D,--DJ''DM** H %h555r   c                     t          j        |j        |                     |j                  d d           }t          j        ||          S r   )r   Nameidr   ctxastr    r   s      r   
visit_NamezAst3ToGAst.visit_Name   sD    9GKK!!	
 
  4000r   c                    t           j        j        dk     rd }n|                     |j                  }t          j        |j        t          j                    |                     |j	                  |          }t          j        ||          S Nr)   )r^   r_   r`   r   type_commentr   rv   argParam
annotationry   r    )r   r   	extra_argr   s       r   	visit_argzAst3ToGAst.visit_arg   sv    !A%%IID$566I9HJLLKK((	
 
  4000r   c           	      F   |j         rt          j        |                     |j                  t          j        |j         t          j                    d d           |                     |j                            }t          j	        ||          S | 
                    |          S r   )r   r   ExceptHandlerr   typerv   Storer,   ry   r    generic_visitr   s      r   visit_ExceptHandlerzAst3ToGAst.visit_ExceptHandler   s    9 	,)DI&&	$)TZ\\4>>DI&& H
 $Xt444%%d+++r      c                     t          j        |                     |j                  |                     |j                  |                     |j                  d          }t          j        ||          S )Nr   )rN   rO   ifsis_async)r   comprehensionr   rN   rO   r   ry   r    r   s      r   visit_comprehensionzAst3ToGAst.visit_comprehension  sd    ){{4;//[[++KK))	  H $Xt444r      c           	      N   t          j        |                     |j                  |                     |j                  |                     |j                  |                     |j                  |                     |j                  g           }t          j        ||          S r   r=   r   s      r   rB   zAst3ToGAst.visit_ClassDef  rC   r   c           
         t          j        |                     |j                  |                     |j                  |                     |j                  |                     |j                  |                     |j                  |                     |j                  g           }t          j	        ||          S r   )
r   rE   r   r   rF   r,   rA   rG   r}   r    r   s      r   rH   zAst3ToGAst.visit_FunctionDef  s    'DI&&DI&&DI&&D/00DL))D-.. H %h555r   c           
         t          j        |                     |j                  |                     |j                  |                     |j                  |                     |j                  |                     |j                  |                     |j                  g           }t          j	        ||          S r   )
r   rJ   r   r   rF   r,   rA   rG   r}   r    r   s      r   rK   z!Ast3ToGAst.visit_AsyncFunctionDef%  s    ,DI&&DI&&DI&&D/00DL))D-.. H %h555r   N)__name__
__module____qualname__r^   r_   r`   r   r!   r$   r(   r-   r1   r5   r9   r;   rB   rH   rK   rQ   rT   rX   r[   rj   rl   rt   rz   r   r   r    r   r   r   r   &   s       
""	 	 	 !!	6 	6 	6	+ 	+ 	+		 		 		 !!	 	 		6 	6 	6	 	 		6 	6 	6	6 	6 	6		6 		6 		6
	6 
	6 
	6
	6 
	6 
	6	6 	6 	6	6 	6 	6	6 	6 	6	6 	6 	6	6 	6 	62	6 	6 	6
	6 
	6 
	61 1 11 1 1	, 	, 	, !!	5 	5 	5 	C"''''R''''''		6 		6 		6
	6 
	6 
	6
	6 
	6 
	6 
	6 
	63 ('r   r   c                      e Zd Zej        j        dk     rd Zej        j        dk     rd Zd Zej        j        dk     rd Z	d Z
d	 Zd
 Zd Zej        j        dk     rd Zd Zn%ej        j        dk     rd Zd Zd Zd Zd Zd Zd Zdej        j        cxk    rdk     rn nd Zdej        j        cxk    rdk     r	n nd Zd Zd ZdS )
GAstToAst3r	   c                     t          j        |                     |j                  |                     |j                            }|S r   )ry   r   r   r   r   r   s      r   r   zGAstToAst3.visit_alias5  s;    yDI&&DK(@(@ H Or   r   c                    d t          |j        t          j                  rt	          d |j        j        D                       r>t          j        fd|                     |j        j                  D                       }nt          j        |                     |j        j                  t          j	                              }t          j
        ||j                   t          j        |          }n# |                     |j                            }t          j
        ||j                   t          j        |                     |j                  ||                     |j                            }t          j
        ||          S )Nc                 b    t          | t          j                  r| S t          j        |           S r   )
isinstancery   SliceIndex)r8   s    r   adjust_slicez0GAstToAst3.visit_Subscript.<locals>.adjust_slice>  s)    a++ (H9Q<<'r   c              3   J   K   | ]}t          |t          j                  V  d S r   )r   r   slice).0elts     r   	<genexpr>z-GAstToAst3.visit_Subscript.<locals>.<genexpr>E  s.      NNsz#tz22NNNNNNr   c                 &    g | ]} |          S r   r   )r   xr   s     r   
<listcomp>z.GAstToAst3.visit_Subscript.<locals>.<listcomp>G  s!    OOOQaOOOr   )r   r   r   r   anyeltsry   ExtSlicer   r   r    r   	Subscriptr#   rx   )r   r   	new_slicer#   r   r   s        @r   visit_SubscriptzGAstToAst3.visit_Subscript=  sT   ( ( ( $*dj11 
BNNdjoNNNNN 1 #OOOO$++djo2N2NOOO! !II  Idkk$*/&B&BCHJJOOE%eTZ888 #	% 0 0II(LTZ)@)@AA	i444}DJ''DH%% H
 $Xt444r   c                     t          j        |                     |j                  |                     |j                            }t          j        ||          S r   )ry   r&   r   r'   r#   r    r   s      r   r(   zGAstToAst3.visit_AssignX  sJ    :KK%%KK
##
 

  4000r   r)   c                 ^    t          j        |                     |j                            }|S r   )ry   r+   r   r,   r   s      r   r-   zGAstToAst3.visit_Moduleb  s$    z$++di"8"899HOr   c                 6   |j         t          j        |j                   }n|j         t          u rt          j                    }nt	          |j         t
                    rt          j        |j                   }nt	          |j         t          t          t          f          rt          j	        |j                   }nMt	          |j         t                    rt          j        |j                   }nt          j        |j                   }t          j        ||          S r   )r#   ry   NameConstantr4   r   boolintfloatcomplexNumstrStrBytesr    r   s      r   visit_ConstantzGAstToAst3.visit_Constantf  s    z!+DJ77x''<>>DJ-- 1+DJ77DJeW(=>> 174:..DJ,, 174:..9TZ00$Xt444r   c                 8   |d S t           j        j        dk     rt                      }n|                     |j                  f}t          j        |                     |j                  |                     |j	                  g|R  }t          j
        ||          S r|   )r^   r_   r`   tupler   r}   ry   r~   rw   r   r    r   r   
extra_argsr   s       r   	_make_argzGAstToAst3._make_argu  s    <4!A%%JJ++d&788:J7KK  $++do">">
AK
 
 
  4000r   c                     t          j        |                     |j                  |                     |j                            }t          j        ||          S r   )ry   rv   r   rw   rx   r    r   s      r   rz   zGAstToAst3.visit_Name  sJ    8KK  KK!!
 
  4000r   c                    |j         ret          j        |                     |j                  |j         j        |                     |j                            }t          j        ||          S |                     |          S r   )	r   ry   r   r   r   rw   r,   r    r   r   s      r   r   zGAstToAst3.visit_ExceptHandler  sn    9 	,(DI&&	dkk$)6L6L H $Xt444%%d+++r   r]   c           	      Z   |j         rGt          |j         d         t          j                  r"|j         d d         }|j         d         j        }n	|j         }d }|j        r4|j        d         j        "|j        d d         }|j        d         j        }n	|j        }d }t          j        | 	                    |j
                  | 	                    |          | 	                    |          | 	                    |          | 	                    |                    }t          j        ||          S )N)rF   r   r   rb   r#   r@   r~   ry   re   r   rf   r    )r   r   rF   ra   r@   rc   r   s          r   rj   zGAstToAst3.visit_Call  s	   y  Z	"t|DD  y"~9R=.y} r!2!6!>="-r*0=xDI&&D!!H%%H%%F## H $Xt444r   c           	      |   |                      |           t          j        |                     |j                  |                     |j                  |                     |j                  |                     |j                  |                     |j                  d d           }t          j	        ||          S )N)r   r?   r@   r,   rA   ra   rc   )
r   ry   r>   r   r   r?   r@   r,   rA   r    r   s      r   rB   zGAstToAst3.visit_ClassDef  s    t$$$|[[++kk$*--T]33[[++#{{4+>??  H $Xt444r   c           	      L   t          j        |                     |j                  |                     |j                  |                     |j                  |                     |j                  |                     |j                            }t          j        ||          S r   )	ry   rE   r   r   rF   r,   rA   rG   r    r   s      r   rH   zGAstToAst3.visit_FunctionDef  s~    DI&&DI&&DI&&D/00DL)) H $Xt444r   c           	      L   t          j        |                     |j                  |                     |j                  |                     |j                  |                     |j                  |                     |j                            }t          j        ||          S r   )	ry   rJ   r   r   rF   r,   rA   rG   r    r   s      r   rK   z!GAstToAst3.visit_AsyncFunctionDef  s    +DI&&DI&&DI&&D/00DL)) H $Xt444r   c                    t          j        |                     |j                  |                     |j                  |                     |j                  |                     |j                            }t          j        ||          S r   )ry   rM   r   rN   rO   r,   rP   r    r   s      r   rQ   zGAstToAst3.visit_For  sl    wDK((DI&&DI&&DK((	 H $Xt444r   c                    t          j        |                     |j                  |                     |j                  |                     |j                  |                     |j                  d           }t          j        ||          S r   )ry   rS   r   rN   rO   r,   rP   r    r   s      r   rT   zGAstToAst3.visit_AsyncFor  so    |DK((DI&&DI&&DK(( H $Xt444r   c                     t          j        |                     |j                  |                     |j                            }t          j        ||          S r   )ry   rV   r   rW   r,   r    r   s      r   rX   zGAstToAst3.visit_With  sJ    xDJ''DI&& H $Xt444r   c                     t          j        |                     |j                  |                     |j                            }t          j        ||          S r   )ry   rZ   r   rW   r,   r    r   s      r   r[   zGAstToAst3.visit_AsyncWith  sJ    }DJ''DI&& H $Xt444r   c                     t          j        |                     |j                  |                     |j                  |                     |j                            }t          j        ||          S r   )ry   re   r   rf   rF   r@   r    r   s      r   rj   zGAstToAst3.visit_Call  s[    xDI&&DI&&DM** H
 $Xt444r   r   c           	      L   t          j        |                     |j                  |                     |j                  |                     |j                  |                     |j                  |                     |j                            }t          j        ||          S r   )	ry   r>   r   r   r?   r@   r,   rA   r    r   s      r   rB   zGAstToAst3.visit_ClassDef  s~    |DI&&DJ''DM**DI&&D/00 H $Xt444r   c           
      ~   t          j        |                     |j                  |                     |j                  |                     |j                  |                     |j                  |                     |j                  |                     |j                            }t          j	        ||          S r   )
ry   rE   r   r   rF   r,   rA   rG   r}   r    r   s      r   rH   zGAstToAst3.visit_FunctionDef  s    DI&&DI&&DI&&D/00DL))D-.. H $Xt444r   c           
      ~   t          j        |                     |j                  |                     |j                  |                     |j                  |                     |j                  |                     |j                  |                     |j                            }t          j	        ||          S r   )
ry   rJ   r   r   rF   r,   rA   rG   r}   r    r   s      r   rK   z!GAstToAst3.visit_AsyncFunctionDef  s    +DI&&DI&&DI&&D/00DL))D-.. H $Xt444r   c                                           |j                   fd|j        D                                  |j                                        |j                                       |j                  g}t          j        j	        dk    r6t          j         fd|j        D              fd|j        D             g|R  }n#t          j         fd|j        D             g|R  }|S )Nc                 :    g | ]}                     |          S r   r   r   r0   r   s     r   r   z.GAstToAst3.visit_arguments.<locals>.<listcomp>   s%    8881T^^A888r   r)   c                 :    g | ]}                     |          S r   r   )r   r~   r   s     r   r   z.GAstToAst3.visit_arguments.<locals>.<listcomp>'  s%    AAA$$AAAr   c                 :    g | ]}                     |          S r   r   r   s     r   r   z.GAstToAst3.visit_arguments.<locals>.<listcomp>(  %    666q""666r   c                 :    g | ]}                     |          S r   r   r   s     r   r   z.GAstToAst3.visit_arguments.<locals>.<listcomp>-  r   r   )r   ro   rp   r   rq   rr   rs   r^   r_   r`   ry   rn   posonlyargsrF   r   s   `   r   rt   zGAstToAst3.visit_arguments  s   NN4;''8888888KK())NN4:&&KK&&

 !Q&&}AAAA0@AAA6666DI666   HH }6666DI6669C  H r   N)r   r   r   r^   r_   r`   r   r   r(   r-   r   r   rz   r   rj   rB   rH   rK   rQ   rT   rX   r[   rt   r   r   r   r   r   2  s       
""	 	 	 !!	5 	5 	561 1 1 !!	 	 		5 	5 	51 1 11 1 1, , , !!	5 	5 	50	5 	5 	5 	5 
		!	#	#	5 	5 	5	5 	5 	5	5 	5 	5	5 	5 	5	5 	5 	5	5 	5 	5	5 	5 	5 	C"''''R'''''	5 	5 	5 	C"''''R'''''		5 		5 		5		5 		5 		5    r   r   c                 D    t                                          |           S r   )r   visitr   s    r   ast_to_gastr   2      <<d###r   c                 D    t                                          |           S r   )r   r   r   s    r   gast_to_astr   6  r   r   )astnr   r    r   ry   r^   r   r   r   r   r   r   r   <module>r      s   @ ' & & & & & & &       



 



I6 I6 I6 I6 I6 I6 I6 I6X} } } } } } } }@$ $ $$ $ $ $ $r   