-Ph*%dZddlZddlmZddlmZmZddlmZddl m Z ddl m Z dd l mZmZdd lmZdd l mZdd l mZdd l mZGddejZ ddZGdde ZGddeZGdde ZdZejdjZdZGddZ Gdde Z!dS) z. Support for parsing strings into code trees. N)StringIO) PyrexScannerStringSourceDescriptor) ModuleScope) PyrexTypes)VisitorTransform)Node StatListNode)NameNode)Parsing)Main) UtilNodesceZdZddZddZdS) StringParseContextNFcl|g}|i}tj||||d||_dS)N3)cpplanguage_level)rContext__init__ module_name)selfnameinclude_directoriescompiler_directivesrs \/var/www/html/test/jupyter/venv/lib/python3.11/site-packages/Cython/Compiler/TreeFragment.pyrzStringParseContext.__init__sM  &"$   &"$  d$79LRUfijjjrTcZ||jdfvrtdt|d|S)NcythonzBNot yet supporting any cimports/includes from string code snippets) parent_modulecontext)rAssertionErrorr)rr from_moduleposneed_pxdabsolute_fallbackrelative_imports r find_modulezStringParseContext.find_module!s9 t/: : : !eff f;dDIIIIr)NNF)NNrTF)__name__ __module__ __qualname__rr)rrrrsB    JJJJJJrrFc|t|}t|ts Jdd}|} ||ddf}t||} |rd| _|| |d} t |} t| | || || } tj | }|'tj | d| | }| |_ d|_ ntj | || }| |_ |S) a Utility method to parse a (unicode) string of code. This is mostly used for internal Cython compiler purposes (creating code snippets that transforms should emit, as well as unit testing). code - a unicode string containing Cython (module-level) code name - a descriptive name for the code source (to use in error messages etc.) in_utility_code - used to suppress some messages from utility code. False by default because some generated code snippets like properties and dataclasses probably want to see those messages. RETURNS The tree, i.e. a ModuleNode. The ModuleNode's scope attribute is set to the scope used when parsing. Nz!unicode code snippets only pleasezUTF-8rrTF)r%r&)source_encodingscoper" initial_pos)allow_struct_enum_decorator)ctx)levelr3)r isinstancestrrin_utility_coder)rrr Ctxp_moduler0is_pxdp_code)rcodepxdsr4r1r"r2r7encodingr code_sourcer0bufscannerr3trees rparse_from_stringsrC's&$T** dC EE"EEEEHKQl (t44K+&* #    u  M ME 4..C3 x"g[RRRG +2M N N NC }KSAAA  ~gU<<<DJ KrceZdZdZdS) TreeCopierc`||S|}|||SN) clone_node visitchildrenrnodecs r visit_NodezTreeCopier.visit_Node^s4 <K!!A   q ! ! !HrN)r*r+r,rMr-rrrErE]s#rrEc(eZdZfdZfdZxZS)ApplyPositionAndCopycVt||_dSrGsuperrr%rr% __class__s rrzApplyPositionAndCopy.__init__h$ rcbt|}|j|_|SrG)rRrMr%)rrKcopyrTs rrMzApplyPositionAndCopy.visit_Nodels(ww!!$''8 rr*r+r,rrM __classcell__rTs@rrOrOgsQrrOcDeZdZdZdZfdZdZdZdZdZ dZ xZ S) TemplateTransforma Makes a copy of a template tree while doing substitutions. A dictionary "substitutions" should be passed in when calling the transform; mapping names to replacement nodes. Then replacement happens like this: - If an ExprStatNode contains a single NameNode, whose name is a key in the substitutions dictionary, the ExprStatNode is replaced with a copy of the tree given in the dictionary. It is the responsibility of the caller that the replacement node is a valid statement. - If a single NameNode is otherwise encountered, it is replaced if its name is listed in the substitutions dictionary in the same way. It is the responsibility of the caller to make sure that the replacement nodes is a valid expression. Also a list "temps" should be passed. Any names listed will be transformed into anonymous, temporary names. Currently supported for tempnames is: NameNode (various function and class definition nodes etc. should be added to this) Each replacement node gets the position of the substituted node recursively applied to every member node. rcx||_||_i}g}|D]O}txjdz c_t jt j}|||<||P||_ t |} |r*t j | ||| } | S)Nr)tempsbody) substitutionsr%r\temp_name_counterr TempHandlerpy_object_typeappendtempmaprR__call__TempsBlockNodeget_pos) rrKr`r^r%re temphandlestemphandleresultrTs rrfzTemplateTransform.__call__s*  ' 'D  / /1 4 / /)**CDDF"GDM   v & & & & !!$''  ;-dll4.@.@4?39;;;F rc,|jr|jS|jSrG)r%rrKs rrhzTemplateTransform.get_poss 8 8O8Orc|dS|}|j |j|_|||SrG)rHr%rIrJs rrMzTemplateTransform.visit_NodesE <4!!Ax#   q ! ! !Hrc|j|}|(|j}||j}t||S||SrG)r`getr%rOrM)rrKkeysubr%s rtry_substitutionz"TemplateTransform.try_substitutionsZ $$S)) ?(C{$(C,',,S11 1??4(( (rc|j|j}|r(|||S|||jSrG)rerqrrefrhrt)rrK temphandles rvisit_NameNodez TemplateTransform.visit_NameNodesW\%%di00  :>>$,,t"4"455 5((ty99 9rct|jtr |||jjS||SrG)r5exprr rtrrMrns rvisit_ExprStatNodez$TemplateTransform.visit_ExprStatNodesD di * * )((ty~>> >??4(( (r) r*r+r,__doc__rarfrhrMrtrxr{rYrZs@rr\r\rs6$ ))):::)))))))rr\c0t|SrG)rE)rKs rcopy_code_treer~s :<<  rz^ *crd|D}|r'td|Dfd|D}|S)zQStrips empty lines and common indentation from the list of strings given in linescBg|]}|dk|S))strip.0xs r z'strip_common_indent..s% 1 1 11bQrclg|]1}tt|d2S)r)len _match_indentgrouprs rrz'strip_common_indent..s6GGGA]1--33A6677GGGrc$g|] }|d SrGr-)rr minindents rrz'strip_common_indent..s!...19::...r)min)linesrs @rstrip_common_indentrsZ 2 1 1 1 1E /GGGGGHH ....... Lrc$eZdZddZdZddZdS) TreeFragmentNc|i}|g}|g}|sd}t|trd}||} i} |D]\} } || | | <t|| | ||x} }||j}t|t st | j|g}|D]}|||}||_n.fmts&uzz*=aggenn*M*MNNNr)r4r1)r%statsz3Unrecognized code format (accepts unicode and Node)) r5r6itemsrCr_r r%rootr NotImplementedError ValueErrorr^)rr<rr=r^pipeliner4r1rfmt_codefmt_pxdsrrvaluemodt transforms rrzTreeFragment.__init__sV <D =E  H %$D dC  T N N Ns4yyHH"jjll + + U #E  (x\ghhh hC!}Fa.. 9 SWQC888% ! ! $IaLLDII d # # T ,)+++DIIRSS S rc*t|jSrG)r~r)rs rrWzTreeFragment.copysdi(((rcb|i}|g}t|j||j|z|S)N)r`r^r%)r\rr^)rnodesr^r%s r substitutezTreeFragment.substitutesM =E =E" ""4938+/:+=SJJJ Jr)NNNNNN)NNN)r*r+r,rrWrr-rrrrsR!!!!F)))JJJJJJrrc$eZdZfdZdZxZS)SetPosTransformcVt||_dSrGrQrSs rrzSetPosTransform.__init__rUrcH|j|_|||SrG)r%rIrns rrMzSetPosTransform.visit_Nodes$8 4    rrXrZs@rrrsGrr)NNNNFF)"r|reiorScanningrrSymtabrrrVisitorr Nodesr r ExprNodesr r rrrrrCrErOr\r~compilematchrrrrr-rrrsS   ::::::::%%%%%%%%%%%%%% J J J J J J J JGKAF',3333l!:W)W)W)W)W)(W)W)W)t 5!!' .J.J.J.J.J.J.J.Jb&r