o 2g3@s0UdZddlmZddlmZmZddlZddlZddl Zddl m Z ddl m Z mZmZmZmZmZmZddlmZddlmZdd lmZmZmZmZmZmZmZm Z m!Z!m"Z"d hZ#d e$d <hd Z%d e$d<dZ&d e$d<GdddeZ' d+d,ddZ(d-ddZ)d.d d!Z*d/d#d$Z+d0d&d'Z,d1d)d*Z-dS)2z__ge____gt____le____lt___ORDERING_METHODSzfunctools.partialPARTIALc@seZdZUded<ded<dS) _MethodInfobool is_staticrtypeN)__name__ __module__ __qualname____annotations__r(r(N/home/garg/my-data/venv/lib/python3.10/site-packages/mypy/plugins/functools.pyr "s  r Fctxmypy.plugin.ClassDefContextauto_attribs_defaultr!returnc st|s|jd|jdStfddd}|}|s"dSt|}|jd}|}|jj|jdkrRt |jj}t |t rM|j ddd ksRttj}tD]}|spttd ||d tg} t|j|j|| |qTdS) zFAdd dunder methods to classes decorated with functools.total_ordering.zNNo ordering operation defined when using "functools.total_ordering": < > <= >=Tcs|du|fSNr()kcomparison_methodsr(r)4sz9functools_total_ordering_maker_callback..)keyz builtins.bool.r!otherN)_analyze_classapifailreasonmax_find_other_type named_typer#ret_typer isinstancernamesplitrrimplementation_artifactrgetr r rr cls) r*r,root root_method other_type bool_typer>proper_ret_type additional_opargsr(r0r)'functools_total_ordering_maker_callback's4    rLmethodrcCsz|jrdnd}d}d}t|jj|jjD]\}}|r)||kr$|}n|d7}q|tkr1|}nq|dur;ttj S|S)z?Find the type of the ``other`` argument in a comparison method.rN) r"zipr# arg_kinds arg_types is_positionalrrrrB)rM first_arg_pos cur_pos_arg other_argarg_kindarg_typer(r(r)r<Ns   r<dict[str, _MethodInfo | None]cCsi}|jjjddD]G}tD]B}||jvrQ||vrQ|j|j}t|tr5t|jt r5t |j |j||<qt|t rMt |j}t|t rMt |j|||<qd||<qq |S)zMAnalyze the class body, its parents, and return the comparison methods found.Nr5)rDinfomrornamesnoder?r r#rr r"r ris_staticmethod)r*r1rDr@r\ proper_typer(r(r)r7ds"    r7mypy.plugin.FunctionContextcCsvt|jtjjs |jSt|jdkr|jSt|jddkr!|jStt|jddt r0|jSt ||jdddS)z6Infer a more precise return type for functools.partialrrN)callee) r?r8mypychecker TypeCheckerdefault_return_typelenrQrrhandle_partial_with_calleer*r(r(r)partial_new_callback|sriracs&tjtjjs jStt|}tr!tfdd|j DSjj |jd}|dur1jS| }jj d}|sMjt|jg}d}n |j}djj d<d}g}gg}gt} tjddddD]:\} } t| D]1\} } | | vrqv| | || j| | |j| | j| | qvqnt||j|jfd d d }tt|jD]\} }|| sqd d t|Dq|jdd|jD|fdd|jDd}|jdkr| jt!jdd||tj"t!r j"j#ndd}| j"jj$j%||||d\}}|s-|jj d<t|}t|t&s:jS|r[t|j}t|t'rO|j(j)tkrRjS|j|jdd}g}g}g}t|D]\} }t*|jt*|jkr|j| }tj+|s|j| }n|j| }|r|j| t,j-t,j.fvr||j| ||||j| qe|sJt/fdd|DrĐqe|d}|t,j0ks|t,j.krt,j1}||||||j| qe|j}tj+|s|j}|j||||d}jt|g}|2d|}|S)Ncsg|]}t|qSr()rg).0itemrhr(r) sz.handle_partial_with_callee..rhr5TFrN)startcs|Sr.r()i) actual_typesr(r)r2sz,handle_partial_with_callee..) actual_kinds actual_names formal_kinds formal_namesactual_arg_typecSsh|]}|jqSr()idrjtvr(r(r) sz-handle_partial_with_callee..cSs0g|]}|tjkr tjn |tjkrtjn|qSr()rrARG_OPT ARG_NAMED ARG_NAMED_OPT)rjr/r(r(r)rls  cs$g|]}|jvst|ts|qSr()rur?rrv) can_infer_idsr(r)rls )rPr> variablesr)rarKrP arg_namesanalyzedrarKrPr~context)r>c3s$|] }|tjtjfvVqdSr.)rrARG_STAR)rjj)actual_arg_kindsr(r) s"z-handle_partial_with_callee..)rQrPr~r>__mypy_partial)3r?r8rbrcrdrerr make_unionitemsextract_callable_typewith_unpacked_kwargswith_normalized_var_args type_context is_type_objnamed_generic_typerr>set enumeraterKaddappendrPr~rQrupdater copy_modifiedr}lineset_liner rr expr_checker check_callrrr#fullnamerfis_valid_inferred_typerrranyrzr{copy_with_extra_attr)r*ra callee_properfn_type last_contextr>wrapped_return actual_argsactual_arg_names seen_argsrnparamraformal_to_actualrW defaulted call_expr_bound partial_kinds partial_types partial_namesactualskindpartially_appliedretr()rror|r*r)rgs                          rgmypy.plugin.MethodContextc Cst|jtjjr t|jtr |jjjtks |jj r d|jj j vr#|j S|jj j d}t |j dkr4|j Sg}g}g}t}t|jD]0\}}t|D]'\}} | |vrSqJ|| || ||j||||j||qJqB|jjj|||||jd} | dS)z@Infer a more precise return type for functools.partial.__call__.rrr)r?r8rbrcrdr#rrr extra_attrsattrsrerfrQrrrKrrrPr~rrr) r* partial_typerrrrrnrrrresultr(r(r)partial_call_callback2sB   r)F)r*r+r,r!r-r!)rMr r-r)r*r+r-rX)r*r_r-r)r*r_rarr-r)r*rr-r).__doc__ __future__rtypingrr mypy.checkerrb mypy.plugin mypy.semanal mypy.argmapr mypy.nodesrrrr r r r mypy.plugins.commonr mypy.typeopsr mypy.typesrrrrrrrrrrrr'rrr rLr<r7rirgrr(r(r(r)s.  $  0   '   (