B A!p\> @sdZddlZddlmZmZddlZddlZddlZddlZddl Z ddl m Z m Z m Z mZmZmZmZmZmZddlZddlZddlmZddlmZddlmZejZejejejej fZ!ej"ddkZ#e#sd Z$nd Z$d d d dhZ%dZ&dZ'dZ(dZ)dZ*dZ+dZ,dZ-dZ.dddddddZ/dd e/0DZ1e2e1Z3Gd!d"d"e4Z5d#d$Z6ej7j8e9d%d&d'Z:ej7j8e ej7j8d%d(d)Z;ej7j8e e9e eej7j8e9d%d.d/Z?ej@AZd0ZBej7j8e9d%d1d2ZCed?ZHej7j8ej7j8e9d@dAdBZIej7j8ej7j8d%dCdDZJej7j8esrc@s eZdZdS)NoSuchArgumentErrorN)__name__ __module__ __qualname__r{r{r{rrsrcCs2|}x|r t|jtjs |j}qW|o0||jjkS)z=Returns true if node is inside the name of an except handler.) isinstanceparentastroid ExceptHandlerr~)nodecurrentr{r{ris_inside_excepts r)rreturncCs.|j}x"|dk r(t|tjr dS|j}qWdS)z*Return true if given node is inside lambdaNTF)rrrLambda)rrr{r{ris_inside_lambdas    rccsFt|tjtjfr could be considered as a builtin defined by python )rSPECIAL_BUILTINS)r~r{r{rr!sr)var_nodevarnamescopercst|tjr^xL|jD]B}t|tjr1sz&is_defined_in_scope..Tr~F)rrZIfbodyranyZtargetsZNonlocalnamesCOMP_NODE_TYPESZFornodes_of_classrr~ZWithitemsZ parent_ofr FunctionDefargsZ is_argumentZ default_valueris_defined_in_scopeZ NoDefaultrr)rrrrZass_noderZidsr{)rrr'sH             r)rrcCs|j}|j}x|r*t|||r"dS|j}qW|}|}|j}xr|r|j|krx"|tjD]}|j|kr`dSq`Wx2|tj tj fD]}|dd|j DkrdSqW|}qDWdS)zreturn True if the variable node is defined by a parent node (list, set, dict, or generator comprehension, lambda) or in a previous sibling node on the same line (statement_defining ; statement_using) TcSsg|]}|dp|dqS)ryrr{)r|r~r{r{r ksz%is_defined_before..F) r~rrZ statementZprevious_siblingrrrr ImportFromImportr)rrZ_nodestmtlinenoZ assign_nodeZimp_noder{r{ris_defined_beforeWs$    rcCsR|}t|tjtjfrNx2|jjD]&}x |tjD]}||kr6dSq6Wq$WdS)zbreturn true if the given Name node is used in function or lambda default argument's value TF) rrrrrrdefaultsrName)rrZ default_nodeZdefault_name_noder{r{ris_default_argumentqs rcCsL|j}x@|dk rFt|tjr dS|jsP|j}qWdS)z5return true if the name is used in function decoratorNTF) rrrZ DecoratorsZ is_statementrrZComprehensionScopeListComp)rrr{r{ris_func_decorator~s   r)framerrcCsFy |j}Wntk rdSXx |D]}||tjkr&dSq&WdS)zjreturn True if `frame` is an astroid.Class node with `node` in the subtree of its bases attribute FT)rAttributeErrorrrr)rrrbaser{r{ris_ancestor_names  rcCs*x$|r$t|tjtjtjfr$|j}qW|S)zLreturn the higher parent which is not an AssignName, Tuple or List node )rrrrr r)rr{r{r assign_parents r) class_noder~rcCs2x,|D] }||kr t||tjr dSq WdS)z=return True if is a method overridden from an ancestorTF) ancestorsrrr)rr~ancestorr{r{roverrides_a_methodsr)messagesrcsfdd}|S)z@decorator to store messages that are handled by a checker methodcs |_|S)N)Z checks_msgs)func)rr{rstore_messagessz&check_messages..store_messagesr{)rrr{)rrcheck_messagess rc@seZdZdZdS)IncompleteFormatStringz:A format string ended in the middle of a format specifier.N)rrr__doc__r{r{r{rrsrc@seZdZdZddZdS)UnsupportedFormatCharacterzXA format character in a format string is not one of the supported format characters.cCst||||_dS)N) Exceptionrindex)selfrr{r{rrs z#UnsupportedFormatCharacter.__init__N)rrrrrr{r{r{rrsr) format_stringrc st}t}g}d}fdd}d}x|tkr|}|dkr||\}}d}|dkrd} ||\}}|} x:| dkr|dkr| d7} n|dkr| d8} ||\}}qvW|d} | | }x|d kr||\}}qW|d kr|d7}||\}}nx|tjkr||\}}qW|d krp||\}}|d krP|d7}||\}}n x|tjkrn||\}}qRW|d kr||\}}trd } nd} || krt||r|||||<n|dkr|d7}|||d7}q(W||||fS)a'Parses a format string, returning a tuple of (keys, num_args), where keys is the set of mapping keys in the format string, and num_args is the number of arguments required by the format string. Raises IncompleteFormatString or UnsupportedFormatCharacter if a parse error occurs.rcs$|d7}|tkrt||fS)Nry)lenr)i)rr{r next_chars z&parse_format_string..next_char%N(ry)z#0- +*.ZhlLzdiouxXeEfFgGcrs%azdiouxXeEfFgGcrs%) setdictrstringZdigitsPY3Kraddappend) rrZ key_typesZ pos_typesZnum_argsrrcharkeyZdepthZ key_startZkey_endflagsr{)rrparse_format_stringsh                   r)rcCs*y t|Stk r$tYnXdS)N)_stringZformatter_field_name_split ValueErrorr)rr{r{rsplit_format_field_namess rc cst}yj||}xZ|D]R}tdd|ddDr:q|d}|d}|V|rxt|D] }|Vq^WqWWnJtk r}z,|jddrdVd VdSt|Wdd}~XYnXdS) zy Given a format string, return an iterator of all the valid format fields. It handles nested fields as well. css|]}|dkVqdS)Nr{)r|itemr{r{rr"sz(collect_string_fields..ryNrzrzcannot switch from manual1) rZ Formatterparseallcollect_string_fieldsrr startswithr)rZ formatterZ parseiteratorresultr~nestedZfieldexcr{r{rrs$  rc Csg}d}t}xt|D]}|rYnX|rj|jrjx|jD]}|j|krR|jSqRWtdS)a Returns the specified argument from a function call. :param astroid.Call call_node: Node representing a function call to check. :param int position: position of the argument. :param str keyword: the keyword of the argument. :returns: The node representing the argument, None if the argument is not found. :rtype: astroid.Name :raises ValueError: if both position and keyword are None. :raises NoSuchArgumentError: if no argument at the provided position or with the provided keyword. Nz2Must specify at least one of: position or keyword.)rr IndexErrorkeywordsargvaluer)rrrrr{r{rget_argument_from_call~s     r cCsD|jdkr|jtkrdSt|ds*dStdd|jddDS)zV Return true if the given class node is subclass of exceptions.Exception. )r BaseExceptionTrFcss|]}t|VqdS)N)inherit_from_std_ex)r|rr{r{rrsz&inherit_from_std_ex..)Zrecurs)r~rEXCEPTIONS_MODULEhasattrrr)rr{r{rr s   r )handlerrcs>ddt|ts|f}fdd|D}|js4dS||S)ar Check if the given exception handler catches the given error_type. The *handler* parameter is a node, representing an ExceptHandler node. The *error_type* can be an exception, such as AttributeError, the name of an exception, or it can be a tuple of errors. The function will return True if the handler catches any of the given errors. cSst|ts|jS|S)N)rrr)errorr{r{rstringify_errors z&error_of_type..stringify_errorcsh|] }|qSr{r{)r|r)rr{r sz error_of_type..T)rtupletypeZcatch)r error_typeZexpected_errorsr{)rr error_of_types  rc CsV|js dSxF|jjD]:}t|tjs&qyt|r4dSWqtjk rLYqXqWdS)zA Detect if the given function node is decorated with a property. FT) decoratorsnodesrrr_is_property_decoratorInferenceError)r decoratorr{r{rdecorated_with_propertys  r)rrcCslxf|D]Z}t|tjr |jtkr6|jdkr6dSx,|D] }|jdkr@|jtkr@dSq@Wq WdS)NpropertyTF)inferrrrrr~rr)rinferedrr{r{rrs   r)rqnamesrc s`|jr|jjng}xH|D]@}y"tfdd|Dr.TF)rrrrrr)rrrZdecorator_noder{)rrdecorated_withs  r i)maxsize)ris_abstract_cbrcCs|dkrtttd}i}yt|}Wn(tk r<iStjk rPiSXx|D]}x|D]}|}t |tj rt |}|s|j |krf||j =qft |tj s|j |kr||j =t |tj rf||}|r|||j <qf|sf|j |krf||j =qfWqXW|S)a Get the unimplemented abstract methods for the given *node*. A method can be considered abstract if the callback *is_abstract_cb* returns a ``True`` value. The check defaults to verifying that a method is decorated with abstract methods. The function will work only for new-style classes. For old-style classes, it will simply return an empty dictionary. For the rest of them, it will return a dictionary of abstract method names and their inferred objects. N)r)rr  ABC_METHODSreversedmroNotImplementedErrorrZ ResolveErrorvaluesrr safe_inferr~r)rr"Zvisitedr%robjrabstractr{r{runimplemented_abstract_methodss8        r+cCsF|}tjtjf}x|r*t|j|s*|j}qW|rBt|j|rB|jSdS)zDReturn the ExceptHandler or the TryExcept node in which the node is.N)rr TryExceptrr)rrZignoresr{r{rfind_try_except_wrapper_node+s  r-cCspt|}|sdSt|tjr.|jj}|jj}ntj dd|jD}|j}t dd|D}t |t }|pn|S)z8Check if the given node is from a fallback import block.Fcss|] }|jVqdS)N)r)r|rr{r{rrDsz)is_from_fallback_block..css |]}t|tjtjfVqdS)N)rrrr)r|Z import_noder{r{rrIs) r-rrrrrhandlers itertoolschain from_iterabler"_except_handlers_ignores_exception ImportError)rcontextZ other_bodyr.Zhas_fallback_importsZignores_import_errorr{r{ris_from_fallback_block9s    r5)r.rcCstt|fd}tt||S)N)r)rrrmap)r. exceptionrr{r{rr2Psr2cs,t|}t|tjr(fdd|jDSdS)a]Return the collections of handlers handling the exception in arguments. Args: node (astroid.NodeNG): A node that is potentially wrapped in a try except. exception (builtin.Exception or str): exception or name of the exception. Returns: list: the collection of handlers that are handling the exception or None. csg|]}t|r|qSr{)r)r|r)r7r{rrgsz*get_exception_handlers..N)r-rrr,r.)rr7r4r{)r7rget_exception_handlersWs  r8cCst|}t|tjS)aCheck if the node is directly under a Try/Except statement. (but not under an ExceptHandler!) Args: node (astroid.Raise): the node raising the exception. Returns: bool: True if the node is inside a try/except statement, False otherwise. )r-rrr,)rr4r{r{ris_node_inside_try_exceptls r9cCst||}|sdSt|S)zCheck if the node is in a TryExcept which handles the given exception. If the exception is not given, the function is going to look for bare excepts. F)r8r)rr7Zmanaging_handlersr{r{rnode_ignores_exceptionzs r:cCs4x.|D]"}|j|kr |jddr dSq WdS)zVreturn true if the given class node should be considered as an abstract class F)Zpass_is_abstractT)r}rr is_abstract)rmethodr{r{rclass_is_abstracts  r=)rattrrcCsRy||}Wntjk r$dSX|d}t|tjrNt|jjtjrNdSdS)NFrT)rrZ NotFoundErrorrrrrConst)rr>Z attributesfirstr{r{r_supports_protocol_methods rAcCstjtjtjtjf}t||S)N)rrSetCompDictComp GeneratorExpr)rZcomprehensionsr{r{ris_comprehensions rE)rrcCst|tot|tS)N)rAGETITEM_METHOD KEYS_METHOD)rr{r{r_supports_mapping_protocolsrHcCs t|tS)N)rACONTAINS_METHOD)rr{r{r"_supports_membership_test_protocolsrJcCst|tpt|tS)N)rA ITER_METHODrF)rr{r{r_supports_iteration_protocols rLcCs t|tS)N)rA AITER_METHOD)rr{r{r"_supports_async_iteration_protocolsrNcCs t|tS)N)rArF)rr{r{r_supports_getitem_protocolsrOcCs t|tS)N)rASETITEM_METHOD)rr{r{r_supports_setitem_protocolsrQcCs t|tS)N)rADELITEM_METHOD)rr{r{r_supports_delitem_protocolsrScCs<|}|d}|d}|dp.|d}|p:|p:|S)NZmixinr*r)lowerrr)r~ZlnameZis_mixinr;Zis_baser{r{r_is_abstract_class_names   rUcCsPxJ|dk rJt|tjrBt|r"dSt|dd}|dk rBt|rBdS|j}qWdS)NTr~F)rrrr=rrUr)rr~r{r{ris_inside_abstract_classs    rV)rprotocol_callbackrcCst|tjr4t|sdS|}|dk r4||r4dSt|tjrdt|sLdS|rXdS||rddSt|tjrt|j tjrt|j r|j }||SdS)NTF) rrrhas_known_bases metaclassZ BaseInstanceZhas_dynamic_getattr_basesZProxyZ_proxied)rrWmetar{r{r_supports_protocols(    r\F)r check_asyncrcCs|r t}nt}t||S)N)rNrLr\)rr]Zprotocol_checkr{r{r is_iterablesr^cCs t|tS)N)r\rH)rr{r{r is_mappingsr_cCst|t}|pt|S)N)r\rJr^)rZ supportedr{r{rsupports_membership_test s r`cCs$t|tjrt|trdSt|tS)NT)rrrrACLASS_GETITEM_METHODr\rO)rr{r{rsupports_getitems  rbcCs t|tS)N)r\rQ)rr{r{rsupports_setitemsrccCs t|tS)N)r\rS)rr{r{rsupports_delitemsrdcCsjy|j|d}t|}Wntjk r.dSXy t|dStjk rRdStk rd|SXdS)zReturn the inferred value for the given node. Return None if inference failed or if there is some ambiguity (more than one node has been inferred). )r4N)rnextrr StopIteration)rr4Zinferitrr{r{rr(s   r()rrcCsly|jStk rYnXxD|jD]:}t||d}t|tjrT||ksTt||ds$d|_dSq$Wd|_dS)z=Return true if all base classes of a class could be inferred.)r4FT)Z_all_bases_knownrrr(rrrrX)rr4rrr{r{rrX6s    rXcCs4|dkp2t|tjr|jdkp2t|tjo2|jdkS)NNone)rrr?rrr~)rr{r{ris_noneJsrhcCsrt}yDx>|D]2}|tjkst|r*q||t|dkrdSqWWntjk r`dSX|rn|SdS)zReturn the inferred type for `node` If there is more than one possible type, or if inferred type is Uninferable or None, return None ryN) rrrZ Uninferablerhrrrpop)rtypesZvar_typer{r{r node_typeRs   rkc Csd}t|tjsdS|jr"|jjng}xz|D]r}t|tjs>q,|j}t|tjr,|jdkr\q,yt |j }Wntj k rw,YnXt|tjr,t ||Sq,WdS)z>Check if the given function node is a singledispatch function.)zfunctools.singledispatchzsingledispatch.singledispatchFregister)rrrrrCallrZ Attributerrerrrr )rZsingledispatch_qnamesrrrZfunc_defr{r{r(is_registered_in_singledispatch_functiongs"    rncCsnt|ddrt|jdSt|ddr4t|jdSt|ddrNt|jdSt|ddrht|jdS|jS)z Get the last lineno of the given node. For a simple statement this will just be node.lineno, but for a node that has child statements (e.g. a method) this will be the lineno of the last child statement recursively. finalbodyForelser.r)rget_node_last_linenororqr.rr)rr{r{rrrs    rrcCs:d}|}|j|}|o8t|dtjo8|djdkS)z;Check if the postponed evaluation of annotations is enabledZ annotationsrZ __future__)rlocalsgetrrrmodname)rr~modulerr{r{ris_postponed_evaluation_enableds  rw)rrrc Cs^tdd||fDsdSx>|D]2}ytj||r.FT)rrrZhelpersZ is_subtyper )rrrr{r{ris_subclass_ofs rx)NN)N)F)N)N)rr functoolsrrr/rrsysrtypingrrrrrr r r r rrZastroid.exceptionsr rrZrrrrrBrCrDr version_inforr r#rKrMZ NEXT_METHODrFrarPrRrIrGZ_SPECIAL_METHODS_PARAMSrZSPECIAL_METHODS_PARAMSrZ PYMETHODSrrrZ node_classesZNodeNGboolrrrrrr__dict__copyrrrrrrrrrrrrrrrrrrrrrrmrr r rrrrrr r+r,r-r5r2r8rr9r:rr=rArErHrJrLrNrOrQrSrUrVr\r^r_r`rbrcrdr(rXrhrrkrnrrrwrxr{r{r{rs,    (=     ,    *L""0   :        !