%![c@sdZddlZddlZddlZddlZddlZddlmZddlmZddlm Z ddlm Z ddlm Z ddlm Z dd lm Z dd lmZd Zd Zd ZejdkrddlmZdZn!ejdZdZdZe jZdZde jfdYZdZdde dZ!dZ"dZ#ddZ$dS(sThe 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. iN(tbases(t exceptions(tmanager(tmodutils(t raw_building(t rebuilder(tnodes(tutilt__s#@cCst|ddtjS(Nstexec(tcompilet_astt PyCF_ONLY_AST(tstring((s.lib/python2.7/site-packages/astroid/builder.pyt_parse(sii(tdetect_encodingcCs_t|d}t|jd}WdQXt|dddd|}|j}|||fS(Ntrbitrtnewlinetencoding(topenRtreadlinetNonetread(tfilenamet byte_streamRtstreamtdata((s.lib/python2.7/site-packages/astroid/builder.pytopen_source_file/s  s\s*#+.*coding[:=]\s*([-\w.]+)cCs`|jdrdSxF|jddd D].}tj|}|dk r*|jdSq*WdS(sEget encoding from a python file as string or return None if not foundssUTF-8s iiN(t startswithtsplitt _ENCODING_RGXtmatchRtgroup(R tlineR ((s.lib/python2.7/site-packages/astroid/builder.pyt_guess_encoding9s cCs4t|d}|j}t|}|||fS(sget data for parsing a filetU(RRR#(RRRR((s.lib/python2.7/site-packages/astroid/builder.pyREs  cCsNy|j}Wntk r#n'X|rJ|td|DkrJtStS(Ncss|]}|jVqdS(N(tvalue(t.0tslot((s.lib/python2.7/site-packages/astroid/builder.pys Vs(tslotstNotImplementedErrortsettFalsetTrue(tnodetattrnameR(((s.lib/python2.7/site-packages/astroid/builder.pyt_can_assign_attrPs "tAstroidBuildercBsheZdZd edZd dZd dZdd dZdZ dZ dZ d Z RS( sClass 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. cCs/tt|j|pt|_||_dS(N(tsuperR0t__init__tMANAGERt_managert_apply_transforms(tselfRtapply_transforms((s.lib/python2.7/site-packages/astroid/builder.pyR2escCsd}t|dd}|dk rtjjtj|\}}|dkrtjj|dr|j|d|}qn|dkr|j |d|d|}|j r|j j |}qn|S( s/Build an astroid from a living module instance.t__file__s.pys.pycs.pyotmodnametpathN(s.pys.pycs.pyo( RtgetattrtosR:tsplitextRt_path_from_filenametexistst file_buildt inspect_buildR5R4tvisit_transforms(R6tmoduleR9R-R:tpath_text((s.lib/python2.7/site-packages/astroid/builder.pyt module_buildjs !"  c CsZyt|\}}}Wntk rV}tjtjdd|d|d|nmttfk r}tjtjdd|d|d|n-t k rtjtjdd|nX||d kr+ydj t j |}Wq+tk r'tjjtjj|d }q+Xn|j|||}|j||SWd QXd S( sxBuild astroid from a source code file (i.e. from an ast) *path* is expected to be a python source file s#Unable to load file {path}: {error}R9R:terrorsBPython 3 encoding specification error or unknown encoding: {error}s.Wrong or no encoding specified for {filename}.Rt.iN(RtIOErrorRtreraiseRtAstroidBuildingErrort SyntaxErrort LookupErrortAstroidSyntaxErrort UnicodeErrorRtjoinRtmodpath_from_filet ImportErrorR<R:R=tbasenamet _data_buildt _post_build(R6R:R9RRRtexcRC((s.lib/python2.7/site-packages/astroid/builder.pyR@|s,      )tcCs7|j|||}|jd|_|j|dS(s&Build astroid from source code string.sutf-8(RTtencodet file_bytesRU(R6RR9R:RC((s.lib/python2.7/site-packages/astroid/builder.pyt string_buildscCs||_|jj|xZ|jD]O}|jdkrex*|jD]\}}|jj|qBWn|j|q#Wx|j D]}|j |qW|j r|jj |}n|S(s@Handles encoding and delayed nodes after a module has been builtt __future__( t file_encodingR4t cache_modulet_import_from_nodesR9tnamestfuture_importstaddtadd_from_names_to_localst_delayed_assattrtdelayed_assattrR5RB(R6RCRt from_nodetsymbolt_tdelayed((s.lib/python2.7/site-packages/astroid/builder.pyRUs  c Cs$yt|d}WnJtttfk r`}tjtjdd|d|d|d|nX|d k rt j j |}nd}|j dr|d }t }n4|d k ot j jt j j|d d k}tj|j}|j||||} |j| _|j| _| S( s3Build tree node from data and add some informationss s#Parsing Python code failed: {error}tsourceR9R:RGss .__init__iiR2N(Rt TypeErrort ValueErrorRLRRJRRNRR<R:tabspathtendswithR,R=RSRt TreeRebuilderR4t visit_moduleR^Rc( R6RR9R:R-RVt node_filetpackagetbuilderRC((s.lib/python2.7/site-packages/astroid/builder.pyRTs$     4  csdfd}x|jD]\}}|dkry|j}Wntjk rcq"nXx}|jD]3}|jj||||jjj|qqWq"|jj|p||||jjj|p|q"WdS(sdStore imported names to the locals Resort the locals if coming from a delayed node cSs|jS(N(t fromlineno(R-((s.lib/python2.7/site-packages/astroid/builder.pytscs|jddS(Ntkey(tsort(tmy_list(t _key_func(s.lib/python2.7/site-packages/astroid/builder.pyt sort_localsst*N( R_tdo_import_moduleRRKt public_namestparentt set_localtscopetlocals(R6R-Rytnametasnametimported((Rxs.lib/python2.7/site-packages/astroid/builder.pyRbs  !cCs_yD|j}x1|jjD] }|tjkr:qnyy|jtjkr||j}|j }t ||j swqn6t |tjrwn|j r|j }n |j}Wntk rqnX|j|j g}||krqn|jdkr2|r2|djjdkr2|jd|q|j|qWWntjk rZnXdS(s^Visit a AssAttr node This adds name to locals and handle members definition. R2iN(tframetexprtinferRt Uninferablet __class__RtInstancet_proxiedtinstance_attrsR/R.t isinstancet is_functionRtAttributeErrort setdefaultRtinserttappendRtInferenceError(R6R-Rtinferredtiattrstvalues((s.lib/python2.7/site-packages/astroid/builder.pyRds6         N( t__name__t __module__t__doc__RR,R2RFR@RZRURTRbRd(((s.lib/python2.7/site-packages/astroid/builder.pyR0[s     cCstj|ddd|dtS(NtdocRWR:Rq(RtModuleR,(RR:((s.lib/python2.7/site-packages/astroid/builder.pytbuild_namespace_package_module sRWcCs=tj|}tdtd|}|j|d|d|S(svParses 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. RR7R9R:(ttextwraptdedentR0R3RZ(tcodet module_nameR:R7Rr((s.lib/python2.7/site-packages/astroid/builder.pytparse s   ccs*t|tjrt|jtjr|jjtkr|jd}|j|_x|jj D]}t |j|}t|t t frxXt |D]%\}}||kr|||>> 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(RRtExprR%(R-((s.lib/python2.7/site-packages/astroid/builder.pyt_extractsiRii( Rt splitlineststripRmt_STATEMENT_SELECTORRRRtextendRtbodytlen( RRRtrequested_linesRR"ttreet extractedR-((s.lib/python2.7/site-packages/astroid/builder.pyt extract_nodefs"6  (ii(%RtreR<tsysRR tastroidRRRRRRRRRRRt version_infottokenizeRRR RR#tAstroidManagerR3R/tInspectBuilderR0RRR,RRRR(((s.lib/python2.7/site-packages/astroid/builder.pyt s<            & #