BVh3dZddlZddlZddlZddlZddlZddlZddlZddlZddl Z ddl Z ddl m Z ddl m Z ddlmZej dZdZdZej(dk\reZ ej(d k\reZej,ZneZej.Zej1d Zej4d Zd Zd ZdZdZdZ dZ!dZ"ddZ#dZ$ddZ%y)z/Converting code to AST. Adapted from Tangent. N)errors) inspect_utils) tf_inspect ) )r __future__z\s*c&|jddS)z7Removes any backslash line continuations from the code.z\ r)replace) code_strings W/home/dcms/DCMS/lib/python3.12/site-packages/tensorflow/python/autograph/pyct/parser.py_unfold_continuationsr:s   VR ((ct|}tjtj|j }d}g} |D]}|j | |D]p}|\}}}}}|tjk(r|}t|}nF|tjtjtjtjfvsnd}n|s|St|}d|v} t|D]\\} }|\}}}}}|tjk(s"d|vr| sd|vr| stj dt||k\r||d}||f|| <^tj"|} g} t%|j'd| j'dD]\} }t)j*t,| j/}t)j*t,|j/}t|t|kDr| t|t|z d}n| }| j |dj1| } | S#tj$rYwxYw)z9Dedents a code so that its first line starts at row zero.Nr  z:code mixing tabs and spaces for indentation is not allowedr)rtokenizegenerate_tokensioStringIOreadlineappend TokenErrorINDENTlenNLNEWLINESTRINGCOMMENT enumeraterUnsupportedLanguageElementError untokenizezipsplitrematch_LEADING_WHITESPACEgroupjoin)r token_genblock_indentationtokenstoktok_type tok_string_ block_levelfirst_indent_uses_tabsinew_code dedented_codelinenew_lineoriginal_indent new_indent dedented_lines r dedent_blockr<?sf&k2+&&r{{;'?'H'HI) &  mmC  c$'!Hj!Q8??"$)*k  X%%x8H8HJ J     %&+#44&! )fa$'!Hj!Q8??" * !7j )?44 HJ J ZK ' - Z(fQi )  (( -K--d3X^^D5IJ(ndHhh2D9??AO-x8>>@J ?c*o-3/#j/ABCmm'(YY} %( /k      sH??IIcZtj|r t|S tj|}t|}td|D}dj||fz}t|t||fS#t$r }t j d|d|d}~wwxYw)aReturns the AST and source code of given entity. Args: entity: Any, Python function/method/class future_features: Iterable[Text], future features to use (e.g. 'print_statement'). See https://docs.python.org/2/reference/simple_stmts.html#future Returns: gast.AST, Text: the parsed AST node; the source code that was parsed to generate the AST (including any prefixes that this function may have added). z$Unable to locate the source code of a=. Note that functions defined in certain environments, like the interactive Python shell, do not expose their source code. If that is the case, you should define them in a .py source file. If you are certain the code is graph-compatible, wrap the call using @tf.autograph.experimental.do_not_convert. Original error: Nc3>K|]}dj|yw)zfrom __future__ import {}N)format).0names r zparse_entity..s#L37!((.Lsr) preamble_len) rislambda _parse_lambdagetimmediatesourceOSErrorrInaccessibleSourceCodeErrorr<tupler*parser)entityfuture_featuresoriginal_sourceesourcefuture_statementss r parse_entityrQsF#    L#66v>O  (&L;JLL 99&&2 3& vC$8 96 AA L  , , .vh7G HIc  K LLLsB B* B%%B*ctj|D]5}t|dd}| ||z |_t|dd}|,||z |_7||dz |}t|dd}| |dd||d<t|dd} | ,t j d|d } | | jd } | |d | d|d <d j|D cgc]} | jc} } || fScc} w) zCReturns a clean node and source code without indenting and context.linenoN end_linenoend_col_offset col_offsetz(?jqM*+.JqMyyj9!((*9:* z :sC)c||yt|tjr |jSt|tsJ|SN) isinstancerYNameidstr)r_s r _arg_namerns6 \ dii 77N D#   +rctj|}td|jjD}|t|jk7ry|jt |jj k7ry|jt |jjk7rytd|jjD}|t|jk7ryy)z.Returns True is node fits the argspec of func.c32K|]}t|ywrirnr@args rrBz(_node_matches_argspec..s=sIcN=Fc32K|]}t|ywrirqrrs rrBz(_node_matches_argspec..sIS)C.IrtT) rgetfullargspecrIargsvarargsrnvarargvarkwkwarg kwonlyargs)r_funcarg_spec node_argsnode_kwonlyargss r_node_matches_argspecrs & &t ,(=diinn==)% &&  499#3#344  ^^y11 IDII4H4HII/h1122  rc tj|}tj|}|jj}t j ||j}dj|}t|dd}g}|D]%}t|d||kr|j|%ng} |D]-}| jdtj|D/g} | D]} td} } tj| D]C}t!| t|d| } t|d| }t|dd}||}t#| |} E| |cxkr| ksrnu| j| | | ft%| d k(r| \\}} } t'||| | S| sCd j| Dcgc] }t)|c}}t+j,d |d || Dcgc]}t/|d|s|}}t%|d k(r|\\}} } t'||| | Sd jd t1|D}t+j,d |d|cc}wcc}w)a&Returns the AST and source code of given lambda function. Args: lam: types.LambdaType, Python function/method/class Returns: gast.AST, Text: the parsed AST node; the source code that was parsed to generate the AST (including any prefixes that this function may have added). rrFrC single_noderSc3VK|]!}t|tjs|#ywri)rjrYLambda)r@rcs rrBz _parse_lambda..s$BjDKK&@Bs))rTNrUrz#could not parse the source code of z*: no matching AST found among candidates: c3dK|](\}\}}}dj|t|d*yw)z Match {}: {} F)include_encoding_markerN)r?unparse)r@r4r_r1s rrBz _parse_lambda..7s61 !\dAqq'$"NO1s.0z: found multiple definitions with identical signatures at the location. This error may be avoided by defining each lambda on a single line and with unique argument names. The matching definitions were: )inspect getmodule getsourcefile__code__co_firstlineno linecachegetlines__dict__r*rJr[rextendrYrZMAX_SIZEminmaxrrgrrr"rr!)lammodfdef_liner`rO all_nodes search_nodesr_ lambda_nodes candidateslnrarbrcrSrTl lambda_codesvmatchess rrErEs ## C ! \\ ( ((   Q -% 775>&F>),  d tXx(H4$   ,BdB99T?BBB *  *b1$D YYr] wq(D1 2dq(D)f1lD1j   v d x4T4() * _$T4 D%t 44 99,?Qgaj?@L  0 0 -cU355AN D EE # G1&;AaD#&FQ G' G\Q!T4 D%t 44 II1&w/1 1' ..+C51ABI K LL@ Hs$II1Ictj|}|j}|r||d}|r-t|dk7rt dj ||dS|S)aReturns the AST of given piece of code. Args: src: Text preamble_len: Int, indicates leading nodes in the parsed AST which should be dropped. single_node: Bool, whether `src` is assumed to be represented by exactly one AST node. Returns: ast.AST NrUz!expected exactly one node, got {}r)rYrJbodyr ValueErrorr?)srcrCr module_nodenodess rrJrJBs^ 3+   % ,- E 5zQ :AA%H II 8O ,rct|jz}t|td} t |t j stdj||jS)zReturns the AST of given identifier. Args: src: A piece of code that represents a single Python expression Returns: A gast.AST object. Raises: ValueError: if src does not consist of a single Expression. Trz.expected exactly one node of type Expr, got {}) STANDARD_PREAMBLEstriprJSTANDARD_PREAMBLE_LENrjrYExprrr?value)rr_s rparse_expressionrZs[ CIIK'# s!6D I$ dDII &  : A A$ G II rc~t|ttfs|f}g}|r|jd|D]}t|tj rt j |}n|}tturtj||jtj|jdj|S)aReturns the source code of given AST. Args: node: The code to compile, as an AST object. indentation: Unused, deprecated. The returning code will always be indented at 4 spaces. include_encoding_marker: Bool, whether to include a comment on the first line to explicitly specify UTF-8 encoding. Returns: code: The source code generated from the AST object source_mapping: A mapping between the user and AutoGraph generated code. z# coding=utf-8r) rjlistrIrrYAST gast_to_ast astunparseastfix_missing_locationsrrr*)r_ indentationrcodesrcast_ns rrrms D4- ( 7D % LL!" 4a!TXXq!eeS & LL##E*00234 5 r)rT)NT)&__doc__rrrrr&systextwraprrrY tensorflow.python.autograph.pyctrrtensorflow.python.utilrdedent PY2_PREAMBLE PY3_PREAMBLEr version_informaxsizemaxintcountrcompiler(rr<rQrgrnrrErJrrrrrs     3:-x   v*t" [[(" ZZ()// =!bjj() AH!BHD,VLt0&r