B S\>@sdZddlZddlZddlmZddlmZddlmZddlm Z ddlm Z ddlm Z dd lm Z dd lm Z dd lmZdd lmZd ZdZe ZddZddZGddde jZddZd!ddZddZddZd"dd ZdS)#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. N)detect_encoding)_parse)bases) exceptions)manager)modutils) raw_building) rebuilder)nodes)util__z#@c CsFt|d}t|jd}WdQRXt|dd|d}|}|||fS)Nrbrr)newlineencoding)openrreadlineread)filenameZ byte_streamrstreamdatar.lib/python3.7/site-packages/astroid/builder.pyopen_source_file/s  rcCs@y |}Wntk r YnX|r<|dd|Dkr=sz#_can_assign_attr..FT)slotsNotImplementedError)nodeattrnamerrrr_can_assign_attr7s r!cs`eZdZdZdfdd ZdddZddd Zdd d Zd dZddZ ddZ ddZ Z S)AstroidBuilderaClass for building an astroid tree from source code or from a live module. The param *manager* specifies the manager class which should be used. If no manager is given, then the default one will be used. The param *apply_transforms* determines if the transforms should be applied after the tree was built from source or from a live object, by default being True. NTcs"tt||pt|_||_dS)N)superr"__init__MANAGER_manager_apply_transforms)selfrapply_transforms) __class__rrr$Ms zAstroidBuilder.__init__cCsd}t|dd}|dk rVtjt|\}}|dkrVtj|drV||d|}|dkr|j|||d}|j r|j |}|S)z/Build an astroid from a living module instance.N__file__)z.pyz.pycz.pyoz.py)modnamepath) getattrosr-splitextrZ_path_from_filenameexists file_buildZ inspect_buildr'r&visit_transforms)r(moduler,rr-Zpath_Zextrrr module_buildRs  zAstroidBuilder.module_buildc Cs*yt|\}}}Wntk rH}ztjd|||d|Wdd}~XYnjttfk r}ztjd|||d|Wdd}~XYn2tk r}ztjd|d|Wdd}~XYnX|j|dkrydt |}Wn.t k rt j t j |d}YnX||||}|||SQRXdS) zxBuild 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})r,r-errorNzBPython 3 encoding specification error or unknown encoding: {error}z.Wrong or no encoding specified for {filename}.)r.r)rIOErrorrAstroidBuildingError SyntaxError LookupErrorAstroidSyntaxError UnicodeErrorjoinrZmodpath_from_file ImportErrorr/r-r0basename _data_build _post_build)r(r-r,rrrexcr4rrrr2ds8 zAstroidBuilder.file_buildcCs&||||}|d|_||dS)z&Build astroid from source code string.zutf-8)rAencodeZ file_bytesrB)r(rr,r-r4rrr string_builds zAstroidBuilder.string_buildcCs||_|j|xB|jD]8}|jdkrHx|jD]\}}|j|q0W||qWx|j D]}| |q^W|j r|j |}|S)z@Handles encoding and delayed nodes after a module has been builtZ __future__) file_encodingr& cache_module_import_from_nodesr,namesZfuture_importsaddadd_from_names_to_locals_delayed_assattrdelayed_assattrr'r3)r(r4rZ from_nodeZsymbol_ZdelayedrrrrBs     zAstroidBuilder._post_buildc Csyt|d}Wn>tttfk rN}ztjd||||d|Wdd}~XYnX|dk rftj|}nd}| dr|dd}d}n$|dk otj tj |d d k}t |j}|||||} |j| _|j| _| S) z3Build tree node from data and add some informations z#Parsing Python code failed: {error})sourcer,r-r6Nzz .__init__iTrr$)r TypeError ValueErrorr:rr<r/r-abspathendswithr0r@r Z TreeRebuilderr&Z visit_modulerIrM) r(rr,r-rrCZ node_filepackagebuilderr4rrrrAs.   zAstroidBuilder._data_buildc sddfdd}x|jD]\}}|dkry |}Wntjk rPwYnXx^|D]&}|j||||jj|q\Wq|j|p||||jj|p|qWdS)zdStore imported names to the locals Resort the locals if coming from a delayed node cSs|jS)N) fromlineno)rrrrsz9AstroidBuilder.add_from_names_to_locals..cs|jddS)N)key)sort)Zmy_list) _key_funcrr sort_localssz.sort_locals*N) rJZdo_import_modulerr9Z public_namesparentZ set_localZscopelocals)r(rr]nameZasnameZimportedr)r\rrLs  z'AstroidBuilder.add_from_names_to_localsc Csy|}x|jD]}|tjkr&qyP|jtjkrP|j}|j }t ||j stwn$t |tjr`wn|j rn|j }n|j}Wntk rwYnX||j g}||krq|jdkr|r|djdkr|d|q||qWWntjk rYnXdS)z^Visit a AssAttr node This adds name to locals and handle members definition. r$rN)frameexprZinferr Z Uninferabler*rZInstanceZ_proxiedZinstance_attrsr!r isinstanceZ is_functionr`AttributeError setdefaultrainsertappendrZInferenceError)r(rrbZinferredZiattrsvaluesrrrrNs8      zAstroidBuilder.delayed_assattr)NT)N)N)rDN) __name__ __module__ __qualname____doc__r$r5r2rFrBrArLrN __classcell__rr)r*rr"Bs  & r"cCstj|d|ddS)NrDT)docr-rV)r ZModule)rar-rrrbuild_namespace_package_modulesrprDTcCs&t|}tt|d}|j|||dS)avParses 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. )rr))r,r-)textwrapdedentr"r%rF)code module_namer-r)rWrrrparses  ruccst|tjrt|jtjr|jjtkr|jd}|j|_xh|jj D]\}t |j|}t|t t frx>> 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. :rtype: astroid.bases.NodeNG, or a list of nodes. cSst|tjr|jS|S)N)rdr ZExprr)rrrr_extracts zextract_node.._extract)rtz"Empty tree, cannot extract from itcsg|]}t|qSr)r)rr)treerr sz extract_node..csg|] }|qSrr)rr)rrrrsr) r| splitlinesstriprU_STATEMENT_SELECTORrhruZbodyrSextendrlen)rsrtZrequested_linesrrZ extractedr)rrr extract_node`s$7  r)rDNT)rD)rmr/rqtokenizerZ astroid._astrZastroidrrrrrr r r rxrZAstroidManagerr%rr!ZInspectBuilderr"rprurrrrrrrs0           B '#