Mhj:dZddlZ ddlmZn#e$r ddlmZYnwxYwddlmZddlm Z m Z m Z m Z m Z mZddlmZddlmZegdZed d gezZegd ezZegd Zed dgZGddZGddZGddee ZGddeZGddee ZGddee ZGddee ZGdde eZ Gdd eZ!Gd!d"eZ"Gd#d$eZ#Gd%d&eZ$Gd'd(e$Z%Gd)d*e$Z&Gd+d,eZ'Gd-d.eZ(Gd/d0eZ)Gd1d2Z*Gd3d4ee*Z+Gd5d6ee*Z,Gd7d8eeZ-Gd9d:e-Z.Gd;de-Z0Gd?d@e0Z1dAZ2GdBdCe0Z3GdDdEe3Z4GdFdGeZ5GdHdIe5Z6GdJdKe5Z7GdLdMe5Z8GdNdOe5Z9GdPdQe5Z:GdRdSeZ;GdTdUe;Z<GdVdWe;Z=GdXdYeZ>GdZd[e>Z?Gd\d]e>Z@Gd^d_e>ZAGd`daeZBdbZCGdcddeeZDGdedfeZEGdgdheZFGdidjeZGeGZHGdkdleZIdS)ma This is the syntax tree for Python 3 syntaxes. The classes represent syntax elements like functions and imports. All of the nodes can be traced back to the `Python grammar file `_. If you want to know how a tree is structured, just analyse that file (for each Python version it's a bit different). There's a lot of logic here that makes it easier for Jedi (and other libraries) to deal with a Python syntax tree. By using :py:meth:`parso.tree.NodeOrLeaf.get_code` on a module, you can get back the 1-to-1 representation of the input given to the parser. This is important if you want to refactor a parser tree. >>> from parso import parse >>> parser = parse('import os') >>> module = parser.get_root_node() >>> module Any subclasses of :class:`Scope`, including :class:`Module` has an attribute :attr:`iter_imports `: >>> list(module.iter_imports()) [] Changes to the Python Grammar ----------------------------- A few things have changed when looking at Python grammar files: - :class:`Param` does not exist in Python grammar files. It is essentially a part of a ``parameters`` node. |parso| splits it up to make it easier to analyse parameters. However this just makes it easier to deal with the syntax tree, it doesn't actually change the valid syntax. - A few nodes like `lambdef` and `lambdef_nocond` have been merged in the syntax tree to make it easier to do deal with them. Parser Tree Classes ------------------- N)Mapping)Tuple)NodeBaseNodeLeaf ErrorNode ErrorLeafsearch_ancestor) split_prefix) split_lines)if_stmt while_stmtfor_stmttry_stmt with_stmt async_stmtsuiter simple_stmt)rr decorated async_funcdef) expr_stmt sync_comp_forrr import_name import_fromparamdel_stmtnamedexpr_testrrceZdZdZdZdS)DocstringMixinc|jdkr|jd}n|jdvrA|j|jddz}|jdkr |jd}n7|j}|jj}||}|sdS||dz }|jdkr |jd}|jd kr|SdS) zN Returns the string leaf of a docstring. e.g. ``r'''foo'''``. file_inputrfuncdefclassdef:rNrstring)typechildrenindexparent)selfnodercr+s Q/var/www/html/test/jupyter/venv/lib/python3.11/site-packages/parso/python/tree.py get_doc_nodezDocstringMixin.get_doc_nodeJs 9 $ $=#DD Y1 1 1=!4!4S!9!9A!=>DyG##}Q'+K"+AGGK((E tUQYz PythonMixin.get_name_of_positionis  " "A!T"" "6V## x(L(L(L(L19(L(L(L(L(LHHH//99%!MMM&tr2N)r3r4r5__doc__r6r>r r2r0r8r8cs4I     r2r8ceZdZdZdZdZdS) PythonLeafr cFt||SN)r get_start_pos_of_prefixr-s r0 _split_prefixzPythonLeaf._split_prefix|sD$">">"@"@AAAr2c|}|(|jdkr|jdvr|}|0t|j}|jt |z dzdfS|jS)z[ Basically calls :py:meth:`parso.tree.NodeOrLeaf.get_start_pos_of_prefix`. N error_leaf)INDENTDEDENT ERROR_DEDENTr'r)get_previous_leafr) token_typer prefixlinelenr=)r- previous_leafliness r0rFz"PythonLeaf.get_start_pos_of_prefixs ..00  $);|)K)K!,0TTT);;==M   ,,E9s5zz)A-q0 0$$r2N)r3r4r5r6rHrFr r2r0rCrCys:IBBB%%%%%r2rCcBeZdZdZdZedeeeffdZdS)_LeafWithoutNewlinesz. Simply here to optimize performance. r returncH|j|jt|jzfSrE)rQcolumnrRvaluerGs r0r=z_LeafWithoutNewlines.end_possy$+DJ777r2N) r3r4r5rAr6propertyrintr=r r2r0rVrVsRI 8sCx888X888r2rVceZdZdZdS)PythonBaseNoder Nr3r4r5r6r r2r0r^r^IIIr2r^ceZdZdZdS) PythonNoder Nr_r r2r0rbrbr`r2rbceZdZdZdS)PythonErrorNoder Nr_r r2r0rdrdr`r2rdceZdZdZdS)PythonErrorLeafr Nr_r r2r0rfrfr`r2rfceZdZdZdZdZdS) EndMarkerr endmarkercldt|jdt|jd|jdS)N)r)r3reprrPr=rGs r0__repr__zEndMarker.__repr__s: JJ   dk!2!2!2!2DLLL  r2N)r3r4r5r6r)rnr r2r0rhrhs-I D     r2rhc eZdZdZdZdZdZdS)Newlinez&Contains NEWLINE and ENDMARKER tokens.r newlinec\dt|jdt|jdS)Nrk: rl)r)r3rmrZrGs r0rnzNewline.__repr__s.!$ZZ000$tz2B2B2B2BCCr2N)r3r4r5rAr6r)rnr r2r0rprps800I DDDDDDr2rpc0eZdZdZdZdZdZd dZd dZdS) Nameza A string. Sometimes it is important to know if the string belongs to a name or not. r:r c bdt|jd|jd|jd|jd S)Nrkrs@,rl)r)r3rZrQrYrGs r0rnz Name.__repr__s9#'::#6#6#6 #'999dkkk; ;r2Fc2||duS)z< Returns True if the name is being defined. )include_setitemN)get_definitionr-rzs r0 is_definitionzName.is_definitions ""?"CC4OOr2c<|j}|j}|dvr||jkr|SdS|dkr!|dkr|jSdS|Q|jdkrdS|jtvr-|||vr|S|r|jt vr|SdS|j}|QdS)z Returns None if there's no definition for a name. :param import_name_always: Specifies if an import name is always a definition. Normally foo in `from foo import bar` is not a definition. r#N except_clauseasr)r,r)r:get_previous_sibling_GET_DEFINITION_TYPESget_defined_names_IMPORTS)r-import_name_alwaysrzr.type_s r0r{zName.get_definitions{  + + +ty   4 O # #((**d22{"4yG##ty111411/BBBBK% $)x*?*?Kt;Dtr2NF)FF) r3r4r5rAr)r6rnr}r{r r2r0ruruse DI;;;PPPP r2ruceZdZdZdS)Literalr Nr_r r2r0rrr`r2rceZdZdZdZdS)Numbernumberr Nr3r4r5r)r6r r2r0rrs DIIIr2rc2eZdZdZdZedZdZdS)Stringr(r c\tjd|jdS)Nz \w*(?=[\'"])r)rematchrZgrouprGs r0 string_prefixzString.string_prefixs$x44::1===r2ctjd|jtj}|ddt |d S)Nz('{3}|"{3}|'|")(.*)$)flagsr')rsearchrZDOTALLrrR)r-rs r0 _get_payloadzString._get_payloadsW ' J)    {{1~~3EKKNN 3 33344r2N)r3r4r5r)r6r[rrr r2r0rrsF DI >>X>55555r2rceZdZdZdZdZdS) FStringStringx f-strings contain f-string expressions and normal python strings. These are the string parts of f-strings. fstring_stringr Nr3r4r5rAr)r6r r2r0rrs$ DIIIr2rceZdZdZdZdZdS) FStringStartr fstring_startr Nrr r2r0rrs$ DIIIr2rceZdZdZdZdZdS) FStringEndr fstring_endr Nrr r2r0rr s$ DIIIr2rceZdZdZdZdZdS)_StringComparisonMixinr cJt|tr |j|kS||uS)ze Make comparisons with strings easy. Improves the readability of the parser. )r;strrZr-others r0__eq__z_StringComparisonMixin.__eq__,s- eS ! ! ':& &u}r2c*t|jSrE)hashrZrGs r0__hash__z_StringComparisonMixin.__hash__6DJr2N)r3r4r5r6rrr r2r0rr)s7I     r2rceZdZdZdZdS)Operatoroperatorr Nrr r2r0rr:s DIIIr2rceZdZdZdZdS)Keywordkeywordr Nrr r2r0rr?s DIIIr2rcJeZdZdZdZfdZdZdZdZdZ dZ d Z xZ S) Scopez Super class for the parser tree, which represents the state of a python text file. A Scope is either a function, class or lambda. r cJt|dSrEsuper__init__r-r* __class__s r0rzScope.__init__L! """""r2c,|dS)z9 Returns a generator of `funcdef` nodes. r$_search_in_scoperGs r0 iter_funcdefszScope.iter_funcdefsOs$$Y///r2c,|dS)z: Returns a generator of `classdef` nodes. r%rrGs r0iter_classdefszScope.iter_classdefsUs$$Z000r2c.|ddS)zO Returns a generator of `import_name` and `import_from` nodes. rrrrGs r0 iter_importszScope.iter_imports[s$$]MBBBr2c2fd|jS)Nc3xK|D]3}|jvr|V|jtvr|jEd{V4dSrE)r)_FUNC_CONTAINERSr*)r*elementnamesscans r0rz$Scope._search_in_scope..scanbsj# 6 6<5((!MMM<#333#tG$4555555555  6 6r2r*)r-rrs `@r0rzScope._search_in_scopeas6 6 6 6 6 6 6tDM"""r2c|jdS)zD Returns the part that is executed by the function. rrGs r0 get_suitezScope.get_suiteks}R  r2c  |jj}n#t$rd}YnwxYwdt|jd|d|jdd|jdd S)Nrkrsrwr-rl)r:rZAttributeErrorr)r3r<r=)r-r:s r0rnzScope.__repr__qsy 9?DD   DDD $(::#6#6#6#'>!#4#4#4dl1oooG Gs  ) r3r4r5rAr6rrrrrrrn __classcell__rs@r0rrDs I#####000 111 CCC ###!!! GGGGGGGr2rc6eZdZdZdZdZfdZdZdZxZ S)Modulez The top scope, which is always a module. Depending on the underlying parser this may be a full module or just a part of a module. ) _used_namesr"cXt|d|_dSrE)rrrrs r0rzModule.__init__s) """r2c#K|D]d}|jdkrW|jdkrL|D]7}d|D}t |dkr|ddkr |dV8edS)zU :return: A list of future import names. :rtype: list of str rrcg|] }|j Sr )rZ).0r:s r0 z4Module._iter_future_import_names..s999DTZ999r2r __future__r'N)rr)level get_pathsrR)r-imppathrs r0_iter_future_import_namesz Module._iter_future_import_namess$$&& ' 'Cx=((SY!^^MMOO''D99D999E5zzQ58|+C+C#Ah  ' 'r2cp|j'ifd|t|_|jS)z Returns all the :class:`Name` leafs that exist in this module. This includes both definitions and references of names. Nc |j}|D] }|dS#t$rB|jdkr3|jg}||YdSYdSwxYw)Nr:)r*rr) setdefaultrZappend)r.r*childarrdctrecurses r0rz&Module.get_used_names..recurses'#}H "*'''' &)))yF**!nnTZ<< 4((((((+**)sAA('A()rUsedNamesMapping)r-rrs @@r0get_used_nameszModule.get_used_namessZ   #C ' ' ' ' ' ' GDMMM/44D r2) r3r4r5rAr6r)rrrrrs@r0rr{sl !I D     '''        r2rceZdZdZdZdS) Decorator decoratorr Nrr r2r0rrs DIIIr2rc.eZdZdZedZdZdS) ClassOrFuncr c|jdS)zR Returns the `Name` leaf that defines the function or class name. r'rrGs r0r:zClassOrFunc.names }Qr2c|j}|jdkr|j}|jdkr7|jdjdkr|jdjS|jddSgS)z4 :rtype: list of :class:`Decorator` rrr decoratorsNr')r,r)r*)r-rs r0get_decoratorszClassOrFunc.get_decoratorsskK >_ , ,!(I >[ ( (!!$)\99 )!,55 )"1"--Ir2N)r3r4r5r6r[r:rr r2r0rrsAI   X r2rc0eZdZdZdZdZfdZdZxZS)Classz> Used to store the parsed contents of a python class. r%r cJt|dSrErrs r0rzClass.__init__rr2ch|jddkrdS|jddkrdS|jdS)z Returns the `arglist` node that defines the super classes. It returns None if there are no arguments. r(N)rrGs r0get_super_arglistzClass.get_super_arglists? = s " "4}Q3&&t}Q''r2) r3r4r5rAr)r6rrrrs@r0rrs] DI##### ( ( ( ( ( ( (r2rc |d}n#t$rgcYSwxYw|jdvrt|g|gS|dkr|gS|jdkr|g}n|j}g}d}t |dgzdD]\}}||dkrz|||}|rn|ddkrt |dks|ddks |ddkr|D] } || _ ||z }n#|t|||}|S) a `argslist_list` is a list that can contain an argslist as a first item, but most not. It's basically the items between the parameter brackets (which is at most one item). This function modifies the parser structure. It generates `Param` objects from the normal ast. Those param objects do not exist in a normal ast, but make the evaluation of the ast tree so much easier. You could also say that this function replaces the argslist node with a list of Param objects. r)r:fpdef*tfpdefNr'rx/) IndexErrorr)Paramr* enumeraterRr,r) r, argslist_listfirstr* new_childrenstartendrparam_childrenps r0_create_paramsrska    z&&&ugv&&'' #w : ! !wHH~H #Hv$5q99 JC} !)%)!4!  %a(C//!$^!4!4!9!9$21$5$<$<-a0C77!/..A'-AHH$6 $++E.&,I,IJJJEs  czeZdZdZdZdZfdZdZdZe dZ dZ d Z d Z d Ze d ZxZS) Functionan Used to store the parsed contents of a python function. Children:: 0. 1. 2. parameter list (including open-paren and close-paren s) 3. or 5. 4. or 6. Node() representing function body 3. -> (if annotation is also present) 4. annotation (if present) r$r ct||jd}|jdd}td|Dst |||jdd<dSdS)Nrr'rc3@K|]}t|tVdSrEr;rrrs r0 z$Function.__init__..-,MM:eU++MMMMMMr2)rrr*anyr)r-r* parametersparameters_childrenrs r0rzFunction.__init__'s """]1% (1!B$7MM9LMMMMM X(6zCV(W(WJ " % % % X Xr2c&|jdjS)NrrrGs r0_get_param_nodeszFunction._get_param_nodes0s}Q((r2c>d|DS)z. Returns a list of `Param()`. c(g|]}|jdk |S)r)r))rrs r0rz'Function.get_params..7s$HHHaaf6G6G6G6G6Gr2)rrGs r0 get_paramszFunction.get_params3s%IH40022HHHHr2c|jdSNr'rrGs r0r:z Function.name9}Qr2c.fd|jS)z6 Returns a generator of `yield_expr`. c3K|D]^}|jdvr |j}|Ed{V&#t$r,|jdkr|jjdkr |jVn|VY[wxYwdS)N)r%r$lambdefyield yield_expr)r)r*rrZr,)r*rnested_childrenrs r0rz'Function.iter_yield_exprs..scanAs# 5 5<#EEE 5&-&6O $tO4444444444&***}//">.,>>").0000")MMM * 5 5s,3A"!A"rr-rs @r0iter_yield_exprszFunction.iter_yield_exprs=s0 5 5 5 5 5 tDM"""r2c.fd|jS)z7 Returns a generator of `return_stmt`. c3K|D]K}|jdks|jdkr|jdkr|V|jtvr|jEd{VLdS)N return_stmtrrWr)rZ_RETURN_STMT_CONTAINERSr*r*rrs r0rz(Function.iter_return_stmts..scanWs# 6 6<=00"<944(9R9R!MMM<#:::#tG$4555555555  6 6r2rr*s @r0iter_return_stmtszFunction.iter_return_stmtsS0 6 6 6 6 6tDM"""r2c.fd|jS)zi Returns a generator of `raise_stmt`. Includes raise statements inside try-except blocks c3K|D]K}|jdks|jdkr|jdkr|V|jtvr|jEd{VLdS)N raise_stmtrraiser/r1s r0rz'Function.iter_raise_stmts..scanes# 6 6<<//"<944'9Q9Q!MMM<#:::#tG$4555555555  6 6r2rr*s @r0iter_raise_stmtszFunction.iter_raise_stmtsar3r2cJt|dduS)zK :return bool: Checks if a function is a generator or not. N)nextr+rGs r0 is_generatorzFunction.is_generatoros%D))++T22$>>r2c |jddkr |jdS|jddksJdS#t$rYdSwxYw)zW Returns the test node after `->` or `None` if there is no annotation. rz->r&N)r*rrGs r0 annotationzFunction.annotationuse  }Q4''}Q''=#s****4   44 s55 AA)r3r4r5rAr)r6rrr r[r:r+r2r8r;r>rrs@r0rrs   DIXXXXX)))III   X ###, # # # # # #???   X     r2rcbeZdZdZdZdZfdZedZdZ edZ dZ xZ S) Lambdaz Lambdas are basically trimmed functions, so give it the same interface. Children:: 0. *. for each argument x -2. -1. Node() representing body r&r ctt|||jdd}t d|Dst |||jdd<dSdS)Nr'c3@K|]}t|tVdSrErrs r0rz"Lambda.__init__..rr2)rrrr*rr)r-r*rrs r0rzLambda.__init__s h&&x000"mAbD1MM9LMMMMM L"07J"K"KDM!B$    L Lr2c td)zN Raises an AttributeError. Lambdas don't have a defined name. zlambda is not named.)rrGs r0r:z Lambda.names 3444r2c |jddS)Nr'rBrrGs r0rzLambda._get_param_nodess}QrT""r2cdS)zA Returns `None`, lambdas don't have annotations. Nr rGs r0r>zLambda.annotations tr2c2d|jjd|jdS)Nrkrwrl)rr3r<rGs r0rnzLambda.__repr__s  N333T^^^DDr2) r3r4r5rAr)r6rr[r:rr>rnrrs@r0r@r@s   DILLLLL55X5 ###X EEEEEEEr2r@ceZdZdZdS)Flowr Nr_r r2r0rIrIr`r2rIc(eZdZdZdZdZdZdZdS)IfStmtr r c#jKt|jD]\}}|dvr|j|dzVdS)z E.g. returns all the `test` nodes that are named as x, below: if x: pass elif x: pass )elififr'N)rr*)r-ir/s r0get_test_nodeszIfStmt.get_test_nodessRdm,, + +DAqN""mAE**** + +r2c|j}tt|D]}|j|kr||jkrdS|cS dS)z Searches for the branch in which the node is and returns the corresponding test node (see function above). However if the node is in the test node itself and not in the suite return None. N)r<reversedlistrPr=)r-r.r< check_nodes r0get_corresponding_test_nodez"IfStmt.get_corresponding_test_nodesv N "4(;(;(=(=#>#>?? & &J#i//z11144&%%% 0 & &r2cL|jD]}|dkr|j|jkrdSdS)z; Checks if a node is defined after `else`. elseTF)r*r<)r-r.r/s r0is_node_after_elsezIfStmt.is_node_after_elses:  AF{{>AK//445r2N)r3r4r5r)r6rPrUrXr r2r0rKrKsK DI + + +&&&     r2rKceZdZdZdZdS) WhileStmtrr Nrr r2r0rZrZ DIIIr2rZc$eZdZdZdZdZddZdS)ForStmtrr c|jdS)zE Returns the input node ``y`` from: ``for x in y:``. rrrGs r0 get_testlistzForStmt.get_testlists}Qr2Fc8t|jd|Sr"_defined_namesr*r|s r0rzForStmt.get_defined_namesdmA.@@@r2Nr)r3r4r5r)r6r_rr r2r0r]r]sG DI   AAAAAAr2r]ceZdZdZdZdZdS)TryStmtrr c#hK|jD]'}|jdkr|jdV|dkrdV(dS)z Returns the ``test`` nodes found in ``except_clause`` nodes. Returns ``[None]`` for except clauses without an exception given. rr'exceptNr*r)r-r.s r0get_except_clause_testszTryStmt.get_except_clause_testss[ M  DyO++mA&&&&&!!   r2N)r3r4r5r)r6rjr r2r0reres- DI     r2rec$eZdZdZdZddZdZdS)WithStmtrr Fcg}|jdddD]+}|jdkr|t|jd|z },|S)z} Returns the a list of `Name` that the with statement defines. The defined names are set after `as`. r'rBr with_item)r*r)rb)r-rzrrns r0rzWithStmt.get_defined_namessV qAv. P PI~,, (:1(=OOO r2ch|d}|td|jdS)Nrnz2The name is not actually part of a with statement.r)r ValueErrorr*)r-r:r.s r0get_test_node_from_namez WithStmt.get_test_node_from_names6##K00 <QRR R}Qr2Nr)r3r4r5r)r6rrqr r2r0rlrlsA DI         r2rlc$eZdZdZdZdZdZdS)Importr c ||}n#t$rYnwxYw|D](}||vr"|d||dzcS)t d)zo The path is the list of names that leads to the searched name. :return list of Name: Nr'z+Name should be defined in the import itself)_aliasesKeyErrorrr+rp)r-r:rs r0get_path_for_namezImport.get_path_for_names  ==??4(DD    D NN$$ 3 3Dt||1TZZ--112222FGGGs  **cdS)NFr rGs r0 is_nestedzImport.is_nested/sur2c$|jddkS)NrrrrGs r0is_star_importzImport.is_star_import2s}R C''r2N)r3r4r5r6rwryr{r r2r0rsrssIIHHH"(((((r2rscLeZdZdZdZd dZdZdZedZ dZ d Z d S) ImportFromrr Fc>d|DS)z Returns the a list of `Name` that the import defines. The defined names are set after `import` or in case an alias - `as` - is present that name is returned. cg|] \}}|p| Sr r rr:aliass r0rz0ImportFrom.get_defined_names..@s HHH+$ HHHr2)_as_name_tuplesr|s r0rzImportFrom.get_defined_names:s% IH1E1E1G1GHHHHr2cXtd|DS)z-Mapping from alias to its corresponding name.c3(K|] \}}|||fVdSrEr rs r0rz&ImportFrom._aliases..Ds;**kdE(DM((((**r2)dictrrGs r0ruzImportFrom._aliasesBs;**T5I5I5K5K***** *r2c~|jddD]}|dvrn |jdkr|jdddS|dkrgS|gS)Nr'.z... dotted_namerimportrh)r-ns r0get_from_nameszImportFrom.get_from_namesGseqrr"  A $$% 6] " ":ccc? " (]]I3Jr2cfd}|jddD]}|dvr|t|jz }|S)&The level parameter of ``__import__``.rr'Nrr*rRrZ)r-rrs r0rzImportFrom.levelRsJqrr"  AL  QW% r2c#K|jd}|dkr|jd}n|dkrdS|jdkr|jddd}n|g}|D]&}|jdkr|dfV|jdddV'dS)NrrrBrimport_as_namesrr:rh)r-lastas_namesas_names r0rzImportFrom._as_name_tuples]s}R  3;;=$DD S[[ F 9) ) )}SSqS)HHvH , ,G|v%%tm####&sss+++++  , ,r2c||jddkrgSfd|DS)z The import paths defined in an import statement. Typically an array like this: ``[, ]``. :return list of list of Name: rrc"g|] \}}|gz Sr r )rr:rdotteds r0rz(ImportFrom.get_paths..ys#JJJKD%$JJJr2)rr*r)r-rs @r0rzImportFrom.get_pathsnsU$$&& =  # #8OJJJJ43G3G3I3IJJJJr2Nr) r3r4r5r)r6rrurr[rrrr r2r0r}r}6s DIIIII***   X,,," K K K K Kr2r}cPeZdZdZdZdZd dZedZdZ dZ d Z d Z d S) ImportNamezBFor ``import_name`` nodes. Covers normal imports without ``from``.rr Fc>d|DS)z Returns the a list of `Name` that the import defines. The defined names is always the first name after `import` or in case an alias - `as` - is present that name is returned. c(g|]\}}|p|dS)rr rrrs r0rz0ImportName.get_defined_names..s%LLL[T5 aLLLr2_dotted_as_namesr|s r0rzImportName.get_defined_namess% MLD4I4I4K4KLLLLr2cdS)rrr rGs r0rzImportName.levels qr2c>d|DS)Ncg|]\}}|Sr r rs r0rz(ImportName.get_paths..s@@@u@@@r2rrGs r0rzImportName.get_pathss"@@(=(=(?(?@@@@r2c# K|jd}|jdkr|jddd}n|g}|D]Q}|jdkr|jd}|jd}nd}|jdkr|g|fV=|jddd|fVRdS)z9Generator of (list(path), alias) where alias may be None.r'dotted_as_namesNrdotted_as_namerr:rh)r-rrrrs r0rzImportName._dotted_as_namess-*  #4 4 4&/!4HH'(H 3 3G|///(+!*1-|v%%i&&&&&&sss+U22222 3 3r2cXtd|DS)z This checks for the special case of nested imports, without aliases and from statement:: import foo.bar cBg|]\}}|t|dkdSr")rRrs r0rz(ImportName.is_nested..s6888;4}TQ)6r2)boolrrGs r0ryzImportName.is_nesteds=88$*?*?*A*A88899 9r2cXtd|DS)z= :return list of Name: Returns all the alias c34K|]\}}|||dfVdS)Nrr rs r0rz&ImportName._aliases..s@**+$(DH%((((**r2)rrrGs r0ruzImportName._aliasess=**9N9N9P9P***** *r2Nr) r3r4r5rAr)r6rr[rrrryrur r2r0rr|sLL DIMMMMXAAA333(999*****r2rcJeZdZdZdZedZedZddZdS) KeywordStatementz For the following statements: `assert`, `del`, `global`, `nonlocal`, `raise`, `return`, `yield`. `pass`, `continue` and `break` are not in there, because they are just simple keywords and the parser reduces it to a keyword. r cd|jzS)z Keyword statements start with the keyword and end with `_stmt`. You can crosscheck this with the Python grammar. z%s_stmt)rrGs r0r)zKeywordStatement.types 4<''r2c&|jdjSNr)r*rZrGs r0rzKeywordStatement.keywords}Q%%r2Fc~|j}|dkrt|jd|S|dvr|jdddSgS)Ndelr')globalnonlocalr)rrbr*)r-rzrs r0rz"KeywordStatement.get_defined_namessO, e  !$-"2ODD D , , ,=A& & r2Nr) r3r4r5rAr6r[r)rrr r2r0rrsmI ((X(&&X&r2rc(eZdZdZedZdS) AssertStmtr c|jdSr"rrGs r0 assertionzAssertStmt.assertionr#r2N)r3r4r5r6r[rr r2r0rrs2I   X   r2rceZdZdZdZdS) GlobalStmtr c"|jdddS)Nr'rrrGs r0get_global_nameszGlobalStmt.get_global_namess}QTT""r2N)r3r4r5r6rr r2r0rrs(I#####r2rceZdZdZdS) ReturnStmtr Nr_r r2r0rrr`r2rceZdZdZdZdS) YieldExprr(r Nrr r2r0rrr[r2rcg}|jdvr(|jdddD]}|t||z }n |jdvr|t|jd|z }n|jdvr|jddkr|jd }|jd d kr!||jdn|jd d krd|rb|jddd D]Q}|jd kr"||jdn#|jdkr||nRn|||S)zk A helper function to find the defined names in statements, for loops and list comprehensions. )testlist_star_expr testlist_compexprlisttestlistNr)atom star_exprr')power atom_exprrB**rrr[trailerr:)r)r*rbr)currentrzrrrr.s r0rbrbs| E|VVV%ccc* < .s^   dmAE*000&t}Q'7II10 0000r2r)r*r)rbrangerR)r-rzrs`` r0rzExprStmt.get_defined_namess =  K / /"4=#3_EEE     1c$-0014a88     r2c|jd}|jdkr3t|jdkr|jd}n |jd}|S)z*Returns the right-hand-side of the equals.rrr=rr'r*r)rRris r0get_rhszExprStmt.get_rhssO}R  9 # #4=!!Q&&}Q'}Q' r2c#K|jd}|jdkr't|jdkrdS|jd}|V|jdddEd{VdS)zZ Returns a generator of `+=`, `=`, etc. or None if there is no operation. r'rrNrrr-r s r0yield_operatorszExprStmt.yield_operators's{ a  : $ $5>""a''N1%E =A&&&&&&&&&&r2Nr)r3r4r5r)r6rrrr r2r0rr sP DI     ' ' ' ' 'r2rceZdZdZddZdS) NamedExprrFc8t|jd|Srrar|s r0rzNamedExpr.get_defined_names9rcr2Nr)r3r4r5r)rr r2r0rr6s3 DAAAAAAr2rceZdZdZdZdfd ZedZedZedZ dZ ed Z dd Z ed Z d Zdfd ZdZxZS)rz It's a helper class that makes business logic with params much easier. The Python grammar defines no ``param`` node. It defines it in a different way that is not really suited to working with parameters. rNcXt|||_dSrE)rrr,)r-r*r,rs r0rzParam.__init__Es& """ r2cP|jd}|dvrt|jSdS)zc Is `0` in case of `foo`, `1` in case of `*foo` or `2` in case of `**foo`. rrrrrs r0 star_countzParam.star_countIs0  a  K  u{## #qr2c|jddk} |jdt|z dkr|jdt|z SdS#t$rYdSwxYw)z{ The default is the test node that appears after the `=`. Is `None` in case no default is present. rrxrBrN)r*r\r)r- has_commas r0defaultz Param.defaultTsy M"%,  }R#i..01S88}R#i..%89998   44 s=A A! A!c|}|jdkr<|jddksJt|jdksJ|jd}|SdS)zz The default is the test node that appears after `:`. Is `None` in case no annotation is present. rr'r&rrN)_tfpdefr)r*rR)r-rr>s r0r>zParam.annotationash  ;( " "?1%,,,,v''1,,,,+J 4r2cTt|jddv}|j|S)z1 tfpdef: see e.g. grammar36.txt. rr)r\r*)r-offsets r0rz Param._tfpdefps*T]1%455}V$$r2c|jdkr|jdS|S)z/ The `Name` leaf of the param. rr)rr)r*rGs r0r:z Param.namews> <<>> ( * *<<>>*1- -<<>> !r2Fc|jgSrE)r:r|s r0rzParam.get_defined_namess  {r2c:|jj|} |jjd}||kr|dz}n#t$rYnwxYw |jjd}||kr|dz}n#t$rYnwxYw|dz S)zB Property for the positional index of a paramter. rrrr')r,r*r+rp)r-r+keyword_only_indexs r0position_indexzParam.position_indexs  $**400 !%!5!;!;C!@!@ )))     D  !%!5!;!;C!@!@ )))     D qys#*A AA*B BBc.|ddS)z= Returns the function/lambda of a parameter. r$r&)r rGs r0get_parent_functionzParam.get_parent_functions##Iy999r2Tc|r!t|S|j}|ddkr |dd}|||S)z Like all the other get_code functions, but includes the param `include_comma`. :param include_comma bool: If enabled includes the comma in the string output. rrxN)include_prefix)rget_coder*_get_code_for_children)r-r include_commar*rs r0rzParam.get_codesk  477##N33 3= B<3  }H** )+   r2c|jdnd|jz}dt|jdt ||zdS)Nrz=%srkrsrl)rrr)r3rr)r-rs r0rnzParam.__repr__s\ ,""%$,:O:O:Q:Q2Q!$ZZ000#dllnn2E2E2O2O2OPPr2rEr)TT)r3r4r5rAr)rr[rrr>rr:rrrrrnrrs@r0rr=s; DX  X   X %%%""X"X*:::       $QQQQQQQr2rceZdZdZdZddZdS) SyncCompForrr Fc8t|jd|S)zN Returns the a list of `Name` that the comprehension defines. r'rar|s r0rzSyncCompFor.get_defined_namess dmA.@@@r2Nr)r3r4r5r)r6rr r2r0rrs8 DIAAAAAAr2rc6eZdZdZdZdZdZdZdZdZ dS) rzO This class exists for the sole purpose of creating an immutable dict. c||_dSrE_dict)r-rs r0rzUsedNamesMapping.__init__s  r2c|j|SrEr)r-keys r0 __getitem__zUsedNamesMapping.__getitem__sz#r2c*t|jSrE)rRrrGs r0__len__zUsedNamesMapping.__len__s4:r2c*t|jSrE)iterrrGs r0__iter__zUsedNamesMapping.__iter__rr2c t|SrE)idrGs r0rzUsedNamesMapping.__hash__s $xxr2c ||uSrEr rs r0rzUsedNamesMapping.__eq__s u}r2N) r3r4r5rArrrrrrr r2r0rrsx   r2r)JrArcollections.abcr ImportError collectionstypingr parso.treerrrrr r parso.python.prefixr parso.utilsr set_FLOW_CONTAINERSr0rrrrr8rCrVr^rbrdrfrhrprurrrrrrrrrrrrrrrrr@rIrKrZr]rerlrsr}rrrrrrrbrrrrCompForrr r2r0rs **X $'''''''$$$########$RRRRRRRRRRRRRRRR,,,,,,######3<<<==#w 677:JJ3::: 3 }- . .8,%%%%%d%%%088888:888[(dk9i     $   DDDDDjDDD11111 111hjW 55555W555"J:        "#%; "$: 4G4G4G4G4GNN4G4G4Gn3 3 3 3 3 U3 3 3 l %6(((((K(((0***Zjjjjj{jjjZ*E*E*E*E*EX*E*E*EZ>*****T***Z A A A A Ad A A A     d         t   .(((((^(((6CKCKCKCKCKCKCKCKL8*8*8*8*8*8*8*8*v~>     !   #####!###! :('('('('('~~('('('VAAAAAAAAwQwQwQwQwQNwQwQwQt A A A A A. A A A ws