o tBh^@sbUdZddlmZddlmZddlmZmZmZm Z m Z m Z m Z m Z mZmZmZmZmZmZmZmZmZmZmZmZmZmZddlmZmZddlm Z m!Z!m"Z"m#Z#ddl$m%Z%ddl&m'Z'dd l(m)Z)dd l*m+Z+m,Z,m-Z-m.Z.m/Z/m0Z0m1Z1m2Z2m3Z3m4Z4d d hZ5d e6d<dhZ7d e6d<dZ8d e6d<GdddZ9GdddZ:d&ddZ;d'ddZr?) selfr3r5r7r8r9r;r<r>r?rEo/var/www/html/riverr-enterprise-integrations-main/venv/lib/python3.10/site-packages/mypy/plugins/dataclasses.py__init__@s  zDataclassAttribute.__init__r cCsNt}|jr |jr t}n|jr|jst}n|js|jrt}t||jd|dS)N)variabletype_annotation initializerkind) rr?r8rrrr to_varr<)rDarg_kindrErErF to_argumentVs   zDataclassAttribute.to_argumentrcCst|j|jSrB)rr3r<rDrErErFrLbszDataclassAttribute.to_varrc Cs4|jsJ|j|j|j|j|j|j|j|jdS)N)r3r5r7r8r9r;r<r?) r<r3r5r7r8r9r; serializer?rOrErErFrPes zDataclassAttribute.serializedataapircCsB|}|ddurd|d<t|d|}|d||d|S)Nr?Fr<)r<r>rE)copygetrpop)clsr>rQrRtyprErErF deserializers zDataclassAttribute.deserializesub_typecCs$|jdurt|j||j|_dSdS)zoExpands type vars in the context of a subtype when an attribute is inherited from a generic super type.N)r<r"r>)rDrYrErErFexpand_typevar_from_subtype|s z.DataclassAttribute.expand_typevar_from_subtypeN)r3r4r5r6r7r6r8r6r9r:r;r:r<r=r>rr?r6r@rA)r@r )r@r)r@r)r>rrQrrRrr@r2)rYrr@rA) __name__ __module__ __qualname__rGrNrLrP classmethodrXrZrErErErFr2?s     r2c@sneZdZdZd$ddZd%d d Zd&ddZd'ddZd(ddZd)ddZ d*d+ddZ d,dd Z d-d!d"Z d#S).DataclassTransformeraImplement the behavior of @dataclass. Note that this may be executed multiple times on the same class, so everything here must be idempotent. This runs after the main semantic analysis pass, so you can assume that there are no placeholders. ctxrr@rAcCs ||_dSrB)_ctx)rDr`rErErFrGs zDataclassTransformer.__init__r6c sFj}jjj}}|durdS|D] }|jdurdSqtjddtjddtjddtjddtjddtjd dd }jjjj}|drd |j vs[|j d j r|rfd d |D}|j r|D] }|j t krut|_ qktd} t| ttjdtg|t| ttjdt}t|d |td|dr|ddus|dr|jd} tt|jdtg| } tt| |j t<|dr'|ds|jd|jdD]P} |jd} t t|jdtdg| } |jd}ttd| | dt g}|| }|dur|j s|j!sJ|jd| d|j!t|| ||| | dq|dr9j"|dd#|n"||drMj$|||dkd%|||d rd|j vse|j dj r|r|dkr|jdfd d |D}t&||jd!}t'|j|jd|(d"d |D|dd#|j)d$<dS)%zApply all the necessary transformations to the underlying dataclass so as to ensure it is fully type checked according to the rules in PEP 557. NFinitTeqorderfrozenslots match_args)rbrcrdrerfrgrGcs&g|]}|jr|js|qSrE)r5_is_kw_only_typer<rN.0attrrOrErF s z2DataclassTransformer.transform..)args return_type__eq__zbuiltins.object.z eq must be True if order is True)__lt____gt____le____ge__z builtins.boolotherzYou may not have a custom z method when order=True)rnro self_typetvar_def)settable) )correct_version__match_args__ builtins.strcsg|] }|jrt|jqSrE)r5r&r3ri)str_typerErFrl s  zbuiltins.tuplecSsg|]}|qSrE)rPrirErErFrls) attributesrer-)*rarVr>collect_attributesr<rrRoptionspython_versionnamesplugin_generatedfallback_to_anyrKrrrr r#r*explicitrr rr'rT named_typerr1fullnamerr failr+node_propertize_callables_freeze add_slotsreset_init_only_varsr(r%_add_dataclass_fields_magic_attributemetadata)rDr`r>rrkdecorator_arguments py_versionrnarg nameless_varobj_typeself_tvar_expr method_nameorder_tvar_deforder_return_type order_argsexisting_methodliteralsmatch_args_typerE)rDrrF transforms                            zDataclassTransformer.transformr>rrlist[DataclassAttribute]r}cCsx|s|jjd|jjdSdd|D}|jdur|j|ks%|jdr7|jjd|jjj |jjdS||_dS)NzPKeyword argument "slots" for "dataclass" is only valid in Python 3.10 and highercSsh|]}|jqSrE)r3rirErErF 'sz1DataclassTransformer.add_slots.. __slots__z;"{}" both defines "__slots__" and is used with "slots=True") rarRrreasonrfrrTformatrVr3)rDr>rr}generated_slotsrErErFrs$ zDataclassTransformer.add_slotscCs||D]9}|jr;|j|jvr|j|j=n|jsJ|jjjD]}t|tr:|jr:|j d}t|t r:|j|jkr:d|_ qqdS)zERemove init-only vars from the class and reset init var declarations.rN) r7r3rdefndefsbody isinstancer unanalyzed_typelvaluesrr)rDr>rrkstmtlvaluerErErFr9s    z)DataclassTransformer.reset_init_only_varslist[DataclassAttribute] | NonecCs|j}|jj}g}t}t|dd}|jjD]}t|tr |js!q|j d}t|t s,q|j j |j}|dur9q|j} t| trCJt| trP|jd| qt| tsWJ| jr[qd} t| j} t| tru| jjdkrud} | jd| _|| r|d}t|j|\} } | d}|durd}nt|j|}d}| rd | vpd | v}nt|jt sd}|sd|_!|}| d}|durt|j|}|"|j|#t$|j|| ||j%|j&|j|j |d q|'}|j j(d d D]}d|j)vrd|j)vrdSd|j)vrqg}|j*t+|j|j)ddD][}|d}||vrTt$,|||j}t-.|jj/j0|1|jj Wdn 1sCwY|"||#|q|ro|D]}|j|krm|2||#|nqYq||}|j3dddqd}d}|D]]}|r|j4r|j5s|j6s||vrt7|j%|j&dn|j}|jd||p|j5o|j4}|r||jr||vrt7|j%|j&dn|j}|jd||p||j}q|S)a@Collect all attributes declared in the dataclass and its parents. All assignments of the form a: SomeType b: SomeOtherType = ... are collected. Return None if some dataclass base class hasn't been processed yet and thus we'll need to ask for another pass. r?FrNzFType aliases inside dataclass definitions are not supported at runtimezdataclasses.InitVarTrbdefaultdefault_factoryrCrv dataclass_tagr-rr3cSs|jSrB)r?)arErErFsz9DataclassTransformer.collect_attributes..)key)r9r;z>Attributes without a default cannot follow attributes with onez:There may not be more than one field with the KW_ONLY type)8rarVsetrrrrr new_syntaxrrr>rrTr3rrrrRrr is_classvarr,r<r%rrnrh_collect_field_argsrvaluer6 parse_boolrimplicitaddappendr2r9r;rSmroradd_plugin_dependencyr rXr!strict_optional_setrstrict_optionalrZremovesortr5r8r?r)rDr`rVattrs known_attrsr?rlhssymrr7 node_typehas_field_call field_argsis_in_init_paramr5r8 is_kw_onlyfield_kw_only_param all_attrsr> super_attrsrQr3rk found_defaultfound_kw_sentinelcontextrErErFrJs                     z'DataclassTransformer.collect_attributescCs|jjj}|D]7}|j|j}|dur"|j}t|tsJd|_ q| }||_d|_ |j d|j|_ t t||j|j<qdS)zaConverts all attributes to @property methods in order to emulate frozen classes. NTrq)rarVr>rrTr3rrr is_propertyrLr _fullnamerr )rDrr>rksym_nodevarrErErFrs zDataclassTransformer._freezeTrzcCsf|jjj}|D])}tt|jtr0|}||_d|_||_ |j d|j |_ t t||j|j <qdS)aConverts all attributes with callable types to @property methods. This avoids the typechecker getting confused and thinking that `my_dataclass_instance.callable_attr(foo)` is going to receive a `self` argument (it is not). TrqN)rarVr>rr,r<r$rLris_settable_propertyrr3rrr r)rDrrzr>rkrrErErFrs z*DataclassTransformer._propertize_callablesrr=cCs.|durdSt|}t|tsdS|jjdkS)z=Checks if the type of the node is the KW_ONLY sentinel value.NFzdataclasses.KW_ONLY)r,rr%r<r)rDrrrErErFrh#s   z%DataclassTransformer._is_kw_only_typecCsd}ttj}|jjd|gp|}|jjd|jjd|g}t||d}|jjj |_ |jjj j d||_ t t |dd|jjj j|<dS) N__dataclass_fields__zdataclasses.Fieldz builtins.dictr)r3r<rqT)rKrr)r#r*rrarRnamed_type_or_nonerrrVr>rrrr r)rD attr_nameany_type field_type attr_typerrErErFr,s   z:DataclassTransformer._add_dataclass_fields_magic_attributeNr`rr@rA)r@r6)r>rrrr}r6r@rA)r>rrrr@rA)r@r)rrr@rA)T)rrrzr6r@rA)rr=r@r6)r@rA) r[r\r]__doc__rGrrrrrrrhrrErErErFr_s    4  r_r`rr@rAcCsi|jjjd<dS)zRecord that we have a dataclass in the main semantic analysis pass. The later pass implemented by DataclassTransformer will use this to detect dataclasses in base classes. rN)rVr>r)r`rErErFdataclass_tag_callback;srr6cCst|}|S)zIHooks into the class typechecking process to add support for dataclasses.)r_r)r` transformerrErErFdataclass_class_maker_callbackEsrexprr"tuple[bool, dict[str, Expression]]cCst|trLt|jtrL|jjtvrLi}t|j|j|j D]+\}}}| s=|j ddr.d}nd}|j ||difS|dusCJ|||<qd|fSdifS)zReturns a tuple where the first value represents whether or not the expression is a call to dataclass.field and the second is a dictionary of the keyword arguments that field() was called with. T)starz0Unpacking **kwargs in "field()" is not supportedz."field()" does not accept positional argumentsNF) rr calleerrr/zip arg_namesrn arg_kindsis_namedrRr)rr`rnr3rrKmessagerErErFrKs"      rNr)r`rr@r6)rrr`rr@r)>r __future__rtyping_extensionsr mypy.nodesrrrrrr r r r r rrrrrrrrrrrr mypy.pluginrrmypy.plugins.commonrrrrmypy.server.triggerr mypy.stater! mypy.typeopsr" mypy.typesr#r$r%r&r'r(r)r*r+r,r.__annotations__r/r1r2r_rrrrErErErFs*  `   0 D ;