o "i' @sddlmZddlmZmZmZddlmZmZddl m Z ddl m Z m Z mZmZddlmZgdZd ed efd d Zed efdede deegefdedeeefd df ddZ ddede ded dfddZdS)) defaultdict)AnyCallableUnion)NOTHING NothingType) BaseConverter)get_newtype_base is_literal is_subclass is_union_type) is_type_alias)configure_tagged_unionconfigure_union_passthroughdefault_tag_generatortypreturncCs|jS)zReturn the class name.)__name__)rrY/var/www/html/premium_crap/venv/lib/python3.10/site-packages/cattrs/strategies/_unions.pyrsr_typeunion converter tag_generatortag_namedefaultNcst|r|j}|j}i}i}|D]/}|} ||} ||} || fdtd|fdd} | fd|dtfdd} | || <| ||<qfdd|D}turn|}|fdtfd d tfd d |}tfd d |}|||fd|dtfdd}turt|ddr||fdtd|fdd}n0||fdtd|fdd}n#t|ddr|||fdtd|fdd}n|||fdtd|fdd}| ||| ||dS)a Configure the converter so that `union` (which should be a union, or a type alias of one) is un/structured with the help of an additional piece of data in the unstructured payload, the tag. :param converter: The converter to apply the strategy to. :param tag_generator: A `tag_generator` function is used to map each member of the union to a tag, which is then included in the unstructured payload. The default tag generator returns the name of the class. :param tag_name: The key under which the tag will be set in the unstructured payload. By default, `'_type'`. :param default: An optional class to be used if the tag information is not present when structuring. The tagged union strategy currently only works with the dict un/structuring base strategy. .. versionadded:: 23.1.0 .. versionchanged:: 25.1 Type aliases of unions are now also supported. valrcS |||SNrr_cl_hrrrstructure_union_member> z6configure_tagged_union..structure_union_membercSs||Srr)rr"rrrunstructure_union_memberAsz8configure_tagged_union..unstructure_union_membercsi|]}||qSrr).0cl)rrr Gsz*configure_tagged_union..cSrrrr rrrstructure_defaultLr$z1configure_tagged_union..structure_defaultcSrrr)r)rrOz(configure_tagged_union..cr*rrr)rrrr+Pr,cSs ||j|}||j||<|Sr) __class__)r_exact_cl_unstruct_hooks _cl_to_tag _tag_nameresrrrunstructure_tagged_unionRsz8configure_tagged_union..unstructure_tagged_unionforbid_extra_keysFcSs|}||||Srcopypopr_ _tag_to_clr0rrrstructure_tagged_union_sz6configure_tagged_union..structure_tagged_unioncSs||||Srrr7rrrr:gscSs,||vr|}||||S|||Srr4rr8 _tag_to_hookr0_dh_defaultrrrr:os cSs"||vr ||||S|||Srrr;rrrr:~s N) r __value____args__get_structure_hookget_unstructure_hookdictrrgetattrregister_unstructure_hookregister_structure_hook)rrrrrargs tag_to_hookexact_cl_unstruct_hooksr'tagstruct_handlerunstruct_handlerr#r% cl_to_tagdefault_handlerr2r:r)rr)rrrs|           rTaccept_ints_as_floatscsJt|jdtdtffdd }dtdtffdd }||dS)ae Configure the converter to support validating and passing through unions of the provided types and their subsets. For example, all mature JSON libraries natively support producing unions of ints, floats, Nones, and strings. Using this strategy, a converter can be configured to efficiently validate and pass through unions containing these types. The most important point is that another library (in this example the JSON library) handles producing the union, and the converter is configured to just validate it. Literals of provided types are also supported, and are checked by value. NewTypes of provided types are also supported. The strategy is designed to be O(1) in execution time, and independent of the ordering of types in the union. If the union contains a class and one or more of its subclasses, the subclasses will also be included when validating the superclass. :param accept_ints_as_floats: When set (the default), if the provided union contains both ints and floats, actual unions containing only floats will also accept ints. See https://typing.python.org/en/latest/spec/special-types.html#special-cases-for-float-and-complex for more information. .. versionadded:: 23.2.0 .. versionchanged:: 25.2.0 Introduced the `accept_ints_as_floats` parameter. exact_typercsdd|jD}dd|jDfdd|jDfddDOfdd|jD}rFtvrFtvrFtvrFtvrFt|rnt|dkrTtt|ntt|}||fdt d t d |ffd d }|S|fdt d t d |ffd d }|S)NcSs$h|]}t|r|jD]}|q qSr)r r@r&tvrrr szSconfigure_union_passthrough..make_structure_native_union..cS&h|]}t|r|jD]}|jq qSrr r@r-rQrrrrTcs0h|]}t|st|p |vrt|p|qSrr r r&rRrGrrrTs cs&h|]tfddDrqS)c3s|]}t|VqdSr)r )r&carr sz]configure_union_passthrough..make_structure_native_union...)any)r&non_literal_classesr\rrTs cs(h|]}t|p |vrt|s|qSr)r r )r&r]r`rrrTsrr8rcs0|jvr ||vr |S|j|vr|S|||Sr)r- structure)rr8classesvalsr spilloverliteral_classesrrstructure_native_unions   z`configure_union_passthrough..make_structure_native_union..structure_native_unioncs>|jvr ||vr |S|j|vr|St|d|jd|)Nz (z) not part of )r- TypeError)rr8rdrergrrris  ) r@intfloataddlenrtuplenextiterr)rPliteral_valuesrfspillover_typerirOrGr)rhrarmake_structure_native_unions\     " z@configure_union_passthrough..make_structure_native_unioncsZt|r+t|j}t|dkrtd|vrdSdd|D}dd|D}||B@SdS)zCan we handle this type?rNFcSrUrrV)r&rRlit_argrrrrT rWzMconfigure_union_passthrough..contains_native_union..cSs h|] }t|st|p |qSrrXrYrrrrTs  )r setr@rntype)rP type_argsrhnon_literal_typesrZrrcontains_native_unions  z:configure_union_passthrough..contains_native_unionN)rwr@rrboolregister_structure_hook_factory)rrrOrur{rrtrrs "Qr)T) collectionsrtypingrrrattrsrrr _compatr r r r typealiasesr__all__rxstrrrr|rrrrrsD      y