o 7f@sUdZddlmZddlmZmZddlZddlm Z m Z m Z m Z m Z ddlmZddlmZmZmZmZmZmZdhZd ed <hd Zd ed <Gd ddeZ ddddZd ddZd!ddZdS)"z__ge____gt____le____lt___ORDERING_METHODSc@seZdZUded<ded<dS) _MethodInfobool is_staticr typeN)__name__ __module__ __qualname____annotations__rrO/home/ertert/spirit/venv/lib/python3.10/site-packages/mypy/plugins/functools.pyrs  rFctxmypy.plugin.ClassDefContextauto_attribs_defaultrreturnc 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|fS)Nr)kcomparison_methodsrr #sz9functools_total_ordering_maker_callback..)keyz builtins.bool.rotherN)_analyze_classapifailreasonmax_find_other_type named_typerret_typer isinstancernamesplitr rimplementation_artifactrgetrr rr cls) r!r#root root_method other_type bool_typer4proper_ret_type additional_opargsrr&r 'functools_total_ordering_maker_callbacks4    rBmethodr cCsz|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) rzipr arg_kinds arg_types is_positionalrr rr8)rC first_arg_pos cur_pos_arg other_argarg_kindarg_typerrr r2=s   r2dict[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.Nr+)r:infomrornamesnoder5rrr rrr ris_staticmethod)r!r'r:r6rR proper_typerrr r-Ss"    r-)F)r!r"r#rr$r)rCrr$r )r!r"r$rN)__doc__ __future__rtypingrr mypy.pluginmypy mypy.nodesrrrrr mypy.plugins.commonr mypy.typesr r r rrrrrrrrBr2r-rrrr s    '