7gIdZddlmZddlZddlZddlZddlZddlZddlZddl m Z m Z ddl m Z ddlmZddlmZmZmZmZmZmZddlmZmZdd lmZdd lmZmZmZdd l m!Z!d Z"d Z#erejHdde%ddZ&ddZ'GddejPZ)ddZ* d ddZ+d dZ,d!dZ-d"d#dZ.d"d$dZ/ d% d&dZ0y)'zThe AstroidBuilder makes astroid from living object and / or from _ast. The builder is not thread safe and can't be used to parse different sources at the same time. ) annotationsN)IteratorSequence) TextIOWrapper)detect_encoding)basesmodutilsnodes raw_building rebuilderutil) ParserModuleget_parser_module) PY312_PLUS)AstroidBuildingErrorAstroidSyntaxErrorInferenceError)AstroidManager__z#@ignorezinvalid escape sequencect|d5}t|jd}dddt|d}|j}|||fS#1swY,xYw)Nrbr)newlineencoding)openrreadlineread)filename byte_streamrstreamdatas /lib/python3.12/site-packages/astroid/builder.pyopen_source_filer#*sW h ";#7#78;  (D8 .pyc.pyo.pyrBmodnamer?)getattr__name__ get_source string_buildosr?splitextr _path_from_filenameexists file_build inspect_buildr3_managervisit_transforms isinstancer Module) r7modulerDr)r?loadersourcepath_exts r" module_buildzAstroidBuilder.module_buildPsvz40t4 0G&&w/F((t(D  <%%fgD%ID%%}}55d;$ --- r;cH t|\}}}|5|% djtj|}|j|||\}}|j!|||cdddS#t$r}td||||d}~wttf$r}t d||||d}~wt $r}td||d}~wwxYw#t$rBtjjtjj|d}YwxYw#1swYyxYw) zyBuild astroid from a source code file (i.e. from an ast). *path* is expected to be a python source file z#Unable to load file {path}: {error})rDr?errorNzBPython 3 encoding specification error or unknown encoding: {error}z.Wrong or no encoding specified for {filename}.)r.r)r#OSErrorr SyntaxError LookupErrorr UnicodeErrorjoinr modpath_from_file ImportErrorrIr?rJbasename _data_build _post_build) r7r?rDr rr!excrSbuilders r"rMzAstroidBuilder.file_buildms6  %5d%; "FHd*J!hhx'A'A$'GHG#..tWdCOFG##FGX>V) &6     [) $     &@4  #J gg..rww/?/?/EFqIGJ Vs]A-D$C (D- C6BCB(( C4CC ADDDDD!c|j|||\}}|jd|_|j||dS)z&Build astroid from source code string.zutf-8)rdencode file_bytesre)r7r!rDr?rSrgs r"rHzAstroidBuilder.string_buildsB**4$? KK099r;c||_|jj||jD]Q}|jdk(r/|j D] \}}|j j|"|j|S|jD]}|j||jr|jj|}|S)zAHandles encoding and delayed nodes after a module has been built. __future__) file_encodingrO cache_module_import_from_nodesrDnamesfuture_importsaddadd_from_names_to_locals_delayed_assattrdelayed_assattrr3rP)r7rSrgr from_nodesymbol_delayeds r"rezAstroidBuilder._post_builds ( ""6* 33I  L0!*IFA))--f5"1  ) )) 4 4 //G   )0  ! !]]33F;F r;c t|d|\}}| t j j|}nd}|jdr|dd}d}nH|duxrBt j jt j j|d d k(}tj|j||} | j||||} | | fS#tttf$r}t d|||||d}~wwxYw) z4Build tree node from data and add some informations.T) type_commentsrDz#Parsing Python code failed: {error})rUrDr?rZNzz .__init__irr2) _parse_string TypeError ValueErrorr]rrIr?abspathendswithrJrcr TreeRebuilderrO visit_module) r7r!rDr?r) parser_modulerf node_filepackagergrSs r"rdzAstroidBuilder._data_builds "/D'# D-  -II   K (crlGGD NGG$$RWW%5%5d%;._key_funcs??'a 'r;c*|jy)N)key)sort)my_listrs r" sort_localsz.sort_localss LLYL 'r;*N)r) nodes.NodeNGreturnint)rzlist[nodes.NodeNG]rNone)parentrpdo_import_moduler public_names set_localscopelocals)r7r)rnameasnameimportedrs @r"rsz'AstroidBuilder.add_from_names_to_localss  ( ({{{ JJLD&s{#446H%113DKK))$5 1 1 3 : :4 @A4 %%fnd;DKK--/66v~FG',sC55 DDcddlm} |jjD]}t |t j r t|tj|jhvr/|j}|j}t||jslvt |tjrt |tjt j fr|j r |j}n |j"}|j'|jg}||vr|j)|y#t$$rY$wxYw#t*$rYywxYw)z_Visit a AssAttr node. This adds name to locals and handle members definition. r)objectsN)astroidrexprinferrQr UninferableBasetyperInstanceExceptionInstance_proxiedinstance_attrsr,r*Proxy is_functionrAttributeError setdefaultappendr)r7r)rinferrediattrsvaluess r"ruzAstroidBuilder.delayed_assattrs $  IIOO-h(<(<=H~%..':S:S)TT#+#4#4!)!8!8/$--H$#Henn=!#Hu{{D d#7.*&   sZ9E AD:E D:2E 3*D:E %D:6E : EE EE EE)NT)r8zAstroidManager | Noner9boolrrr0)rSztypes.ModuleTyperD str | Noner nodes.Module)r?strrDrrr)N)r!rrDrr?rrr)rSrrgzrebuilder.TreeRebuilderrrrr)r!rrDrr?rrz,tuple[nodes.Module, rebuilder.TreeRebuilder])r)znodes.ImportFromrr)r)znodes.AssignAttrrr) rF __module__ __qualname____doc__r2rXrMrHrerdrsru __classcell__)r:s@r"r.r.>sOS2,2GK2 2?C&1; :$?N@D::"%:2<: :"-DPS *"%-7 5BH4%r;r.c\tj||d}|jgd|S)NT)r?r)bodydoc_node)r rRpostinit)rr?rSs r"build_namespace_package_modulers) \\$T4 8F OOdO+ Mr;c~tj|}tt|}|j |||S)awParses a source string in order to obtain an astroid AST from it. :param str code: The code for the module. :param str module_name: The name for the module, if any :param str path: The path for the module :param bool apply_transforms: Apply the transforms for the give code. Use it if you don't want the default transforms to be applied. )r8r9rC)textwrapdedentr.rrH)code module_namer?r9rgs r"parsers? ??4 D 3CG   k  EEr;c#rKt|tjrt|jtjr|jj t k(r|jd}|jsJ|j|_|jjD]b}t|j|}t|trt|D]\}}||us |||<G||usLt|j||d|y|jD]}t|Ed{y7w)aFind expressions in a call to _TRANSIENT_FUNCTION and extract them. The function walks the AST recursively to search for expressions that are wrapped into a call to _TRANSIENT_FUNCTION. If it finds such an expression, it completely removes the function call node from the tree, replacing it by the wrapped expression inside the parent. :param node: An astroid node. :type node: astroid.bases.NodeNG :yields: The sequence of wrapped expressions on the modified tree expression can be found. rN)rQr CallfuncNamer_TRANSIENT_FUNCTIONargsr_astroid_fieldsrElist enumeratesetattr get_children_extract_expressions)r) real_exprrchildidxcompound_childs r"rr/s 4$ tyy%** - IINN1 1IIaL {{{;;  KK//DDKK.E%&+4U+;'C%-%.c ,<$ T950&&(E+E2 2 2) 2sCD7! D7.?D7-D5.D7ct|tjtjtjfr |j }n |j }||k(r|S|jD]}t||}|s|cSy)aExtracts the statement on a specific line from an AST. If the line number of node matches line, it will be returned; otherwise its children are iterated and the function is called recursively. :param node: An astroid node. :type node: astroid.bases.NodeNG :param line: The line number of the statement to extract. :type line: int :returns: The statement on the line, or None if no statement for the line can be found. :rtype: astroid.bases.NodeNG or None N) rQr ClassDef FunctionDef MatchCaserlinenor_find_statement_by_line)r)line node_linerresults r"rrWsp$):):EOOLM OO KK D ""$(5 M% r;cLdd}g}t|jD]=\}}|jjts*|j |dz?t ||}|js tdg}|r|Dcgc]}t||}}|jt||s|j |jd|Dcgc] }|| }}|Dcgc]}|| } }t| dk(r| dS| Scc}wcc}wcc}w)ahParses some Python code as a module and extracts a designated AST node. Statements: To extract one or more statement nodes, append #@ to the end of the line Examples: >>> def x(): >>> def y(): >>> return 1 #@ The return statement will be extracted. >>> class X(object): >>> def meth(self): #@ >>> pass The function object 'meth' will be extracted. Expressions: To extract arbitrary expressions, surround them with the fake function call __(...). After parsing, the surrounded expression will be returned and the whole AST (accessible via the returned node's parent attribute) will look like the function call was never there in the first place. Examples: >>> a = __(1) The const node will be extracted. >>> def x(d=__(foo.bar)): pass The node containing the default argument will be extracted. >>> def foo(a, b): >>> return 0 < __(len(a)) < b The node containing the function call 'len' will be extracted. If no statements or expressions are selected, the last toplevel statement will be returned. If the selected statement is a discard statement, (i.e. an expression turned into a statement), the wrapped expression is returned instead. For convenience, singleton lists are unpacked. :param str code: A piece of Python code that is parsed as a module. Will be passed through textwrap.dedent first. :param str module_name: The name of the module. :returns: The designated node from the parse tree, or a list of nodes. cRt|tjr |jS|Sr0)rQr Exprr&rs r"_extractzextract_node.._extracts dEJJ '::  r;)rz"Empty tree, cannot extract from itr)r)nodes.NodeNG | Nonerr) r splitlinesstripr_STATEMENT_SELECTORrrrr~rextendrlen) rrrrequested_linesrrtree extractedr)extracted_without_nones r" extract_noderzs#l "$Ot01 T ::< !4 5  " "37 +2 ; /D 99=>>+-IETU_T,T48_ U)$/0 2',56ID$II6/8MytD $$TD[$F =   Gs) B A BB )rrrztuple[TextIOWrapper, str, str])r)znodes.ClassDefr*rrr)rrr?z Sequence[str]rr)rNT) rrrrr?rr9rrr)r)rrzIterator[nodes.NodeNG])r)rrrrr)r)rrrrrz!nodes.NodeNG | list[nodes.NodeNG])rrrrrr)TN)r!rr{rrDrrztuple[ast.Module, ParserModule])1rrlrastrIrrtypeswarningscollections.abcrriortokenizerrrr r r r r astroid._astrr astroid.constrastroid.exceptionsrrrastroid.managerrrrfilterwarnings SyntaxWarningr#r,r4r.rrrrrrr|r;r"rs  #  .$II8$WW*  HH&?O"-R\00Rj! F FF F F  F,%3P FS"lBF! !"!4>!$!r;