o "i&@sdZddlmZddlZddlmZddlmZmZmZm Z ddl m Z ddl m Z mZmZdd lmZdd lmZd/ddZd0ddZd1ddZede dZ   d2d3d%d&Zd4d)d*Zd5d-d.ZdS)6z.Strategies for customizing subclass behaviors.) annotationsN)collect)AnyCallableTypeVarUnion) BaseConverter)AttributeOverridemake_dict_structure_fnmake_dict_unstructure_fn)already_generating) subclassescltypereturn list[type]cCs&t|p|}|gddt|DS)NcSsg|] }t|D]}|qqS)_make_subclasses_tree).0sclssclrr]/var/www/html/premium_crap/venv/lib/python3.10/site-packages/cattrs/strategies/_subclasses.py s z)_make_subclasses_tree..)typing get_originr)r cls_originrrrrs rgiven_subclassestuple[type, ...]boolcCs.t|p|}tt|}t|}t||@S)z?Whether the given class has subclasses from `given_subclasses`.)rrsetrr)rrractualgivenrrr_has_subclassess  r#given_subclasses_tree tuple[type] type | NonecCs8tt|}tt|t|@}t|dkrt|SdS)Nr)tuplerr lenr)rr$actual_subclass_tree class_treerrr_get_union_types r+C)bound converterrtuple[type, ...] | Noneunion_strategyCallable[[Any, C], Any] | None overrides#dict[str, AttributeOverride] | NoneNonecCsTt|dur|g|R}ntt|}|dur!t||||dSt||||dS)a` Configure the converter so that the attrs/dataclass `cl` is un/structured as if it was a union of itself and all its subclasses that are defined at the time when this strategy is applied. :param cl: A base `attrs` or `dataclass` class. :param converter: The `Converter` on which this strategy is applied. Do note that the strategy does not work for a :class:`cattrs.BaseConverter`. :param subclasses: A tuple of sublcasses whose ancestor is `cl`. If left as `None`, subclasses are detected using recursively the `__subclasses__` method of `cl` and its descendents. :param union_strategy: A callable of two arguments passed by position (`subclass_union`, `converter`) that defines the union strategy to use to disambiguate the subclasses union. If `None` (the default), the automatic unique field disambiguation is used which means that every single subclass participating in the union must have an attribute name that does not exist in any other sibling class. :param overrides: a mapping of `cl` attribute names to overrides (instantiated with :func:`cattrs.gen.override`) to customize un/structuring. .. versionadded:: 23.1.0 .. versionchanged:: 24.1.0 When overrides are not provided, hooks for individual classes are retrieved from the converter instead of generated with no overrides, using converter defaults. .. versionchanged:: 25.2.0 Slotted dataclasses work on Python 3.14 via :func:`cattrs.subclasses.subclasses`, which filters out duplicate classes caused by slotting. N)rr'r*_include_subclasses_without_union_strategy'_include_subclasses_with_union_strategy)rr.rr0r2parent_subclass_treerrrinclude_subclasses(s$ r8r r7c Cs|D]a}t||}|fdd}|dur&t||fi|}t||fi|}n ||}||}|dur=||fddd}n|j||d } |||| fdd d}|||fdd d } |||||| qdS)NcS||uSNrcls_clrrr cls_is_clmz=_include_subclasses_without_union_strategy..cls_is_clvaldictrrcS |||Sr:r)r@_r= _base_hookrrr struct_hooky z?_include_subclasses_without_union_strategy..struct_hook)r2cSs&||}||ur |||S|||S)z If val is disambiguated to the class `cl`, use its base hook. If val is disambiguated to a subclass, dispatch on its exact runtime type. ) structure)r@rC_cr=rD_dis_fndis_clrrrrEs  parent_subclass_tree[0]cSs"|j|ur ||S|j||jdS)z If val is an instance of the class `cl`, use the hook. If val is an instance of a subclass, dispatch on its exact runtime type. )unstructure_as) __class__ unstructure)r@rHr=rDrrr unstruct_hooks zA_include_subclasses_without_union_strategy..unstruct_hook)r@rArr)r@rKrrA)r+r r get_structure_hookget_unstructure_hook _get_dis_funcregister_structure_hook_funcregister_unstructure_hook_func) rr.r7r2subclass_unionr>base_struct_hookbase_unstruct_hookrEdis_fnrOrrrr5\s.      r5 union_classesCallable[[Any, C], Any]c sfddD}|s dSi}i}D]Ctht_z+|dur5t|fi|}t|fi|}n|jdd}|jdd}Wtt_ntt_w||<||<qt} |D]\} fdd} | | | q_|D]\} fdd} | | | qt|| ||| }|| }D];fd d} | | |t fd dD} t | d krt| } || ||| }| |fddd}| | |qdS)a^ This function is tricky because we're dealing with what is essentially a circular reference. We need to generate a structure hook for a class that is both: * specific for that particular class and its own fields * but should handle specific functions for all its descendants too Hence the dance with registering below. csg|] }t|r|qSr)r#)rr)rYrrrsz;_include_subclasses_with_union_strategy..NF) cache_resultcSr9r:rr;rrrr>r?z:_include_subclasses_with_union_strategy..cls_is_clcSr9r:rr;rrrr>r?cSr9r:rr;rrrr>r?cs.g|]}tt|p |tpr|qSr) issubclassrr)rc)rrrrs payloadrArrcSrBr:r)r_rC_u_srrrshrFz3_include_subclasses_with_union_strategy..sh)r_rArr) r r working_setr r rQrPritemsrTrSr'r()r.rYr0r2parent_classesoriginal_unstruct_hooksoriginal_struct_hooksrOrE final_unionhookr>rurbr)rrYrr6sV             r6)rrrr)rrrrrr)rrr$r%rr&)NNN) rrr.r,rr/r0r1r2r3rr4)r.r r7rr2r3)r.r,rYrr0rZr2r3)__doc__ __future__rrgcrrrrr convertersr genr r r gen._constsr rrr#r+r,r8r5r6rrrrs&          4K