ii4ddlmZddlZddlmZddlmZddlmZm Z m Z m Z ddl m Z mZddlmZddlmZmZdd lmZmZmZmZmZmZdd lmZerdd lmZd%dZedd&dZ d%dZ!d'd(dZ"d)dZ#d'd*dZ$d'd+dZ%d'd(dZ&d,dZ'd-d!Z(d.d$Z)dS)/) annotationsN)deque) is_dataclass) TYPE_CHECKINGAnyMappingSequence)get_args get_origin) TYPE_MAPPING)check_for_deprecated_parameters deprecated) is_annotated is_new_type is_optionalis_safe_subclass is_type_aliasis_unionNoneType)Random annotationrreturncPt|r|j}t||S)zReturn base type if given annotation is a type derived with NewType, otherwise annotation. :param annotation: A type annotation, possibly one created using 'types.NewType' :returns: The unwrapped annotation. )r __supertype__rs q/var/www/development/aibuddy-work/election-extract/venv/lib/python3.11/site-packages/polyfactory/utils/helpers.pyunwrap_new_typers6 j ! !.-  j ! !. zv2.21.0randomrct|r@tt|}||}t|@|S)zUnwraps union types recursively and picks a random type from each union. :param annotation: A type annotation, possibly a type union. :param random: An instance of random.Random. :returns: A type annotation )rlistr choicerr argss r unwrap_unionr&'sT :  )HZ(())]]4((  :  ) rct|r5tdt|D}t|5|S)zUnwraps optional union types - recursively. :param annotation: A type annotation, possibly an optional union. :returns: A type annotation c30K|]}|tdfv |VdSNr.0args r z"unwrap_optional..=s2]]#XW[L\A\A\#A\A\A\A\]]r)rnextr rs runwrap_optionalr/5sR j ! !^]]*)=)=]]]]]  j ! !^ r Random | Nonectdd|fft|s-t|st|st |rt|rt |}nZt|rt |}n;t|rt|d}nt |r|j}t|t|t|t ||S)zUnwraps an annotation. :param annotation: A type annotation. :param random: An instance of random.Random. :returns: The unwrapped annotation. 2.21.0r  parametersr) r rrrrrr/unwrap_annotated __value__rr s runwrap_annotationr8As!$H8V:L9NOOOO j ! !.[%<%<. Z@X@X.\ijt\u\u. z " " .(44JJ  $ $ .(44JJ * % % .)*55a8JJ : & & .#-J j ! !.[%<%<. Z@X@X.\ijt\u\u. r list[Any]cLg}t|r0|tt|nt |r5t |D]$}|t|%nt |r6|tt |dnYt|r5t |D]$}|t|%n|||S)uDFlattens an annotation into an array of possible types. For example: * Union[str, int] → [str, int] * Optional[str] → [str, None] * Union[str, Optional[int]] → [str, int, None] * NewType('UserId', int) → [int] :param annotation: A type annotation. :returns: The flattened annotations. r) rextendflatten_annotationrrr rrappend)rflatas rr<r<Xs' D:   &z'B'BCCDDDD Z  *%% / /A KK*1-- . . . . / j ! !  &x ';';A'>??@@@@ *   *%% / /A KK*1-- . . . . / J Krtuple[Any, ...]cdtdd|fftt|S)zUnwrap the annotation and return any type args. :param annotation: A type annotation :param random: An instance of random.Random. :returns: A tuple of type args. r2r r3rr r r8r7s r unwrap_argsrCus9$H8V:L9NOOOO %<<< = ==rtuple[Any, list[Any]]ctdd|ffdt|D}t|d|ddfS)zUnwrap an annotated type and return a tuple of type args and optional metadata. :param annotation: An annotated type annotation :param random: An instance of random.Random. :returns: A tuple of type args. r2r r3cg|]}||Sr)r*s r z$unwrap_annotated..s C C CC3?C???rrNrBr$s rr5r5sY$H8V:L9NOOOO C C8J// C C CD T!W % %tABBx //rctdd|fft|rt|}t|rt |d}t jdkr|St|p|}|tvr t|}t|x}r,td|D}|tur||n|S|S)zvNormalize an annotation. :param annotation: A type annotation. :returns: A normalized type annotation. r2r r3r) c34K|]}t|VdSr))normalize_annotationr*s rr-z'normalize_annotation..s+??3)#..??????r) r rrrr5sys version_infor r r tupletype)rr originr%s rrNrNs$H8V:L9NOOOO:1$Z00 J5%j11!4  &    # # 1zF f% ###tB??$?????%T11vd||zA Mr Sequence[Any]c0t|ddS)zpGet the metadata in the annotation. :param annotation: A type annotation. :returns: The metadata. rIN)r rs rget_annotation_metadatarVs J   ##r3type[list | tuple | set | frozenset | dict | deque]ct|trtSt|trtSt|trtSt|t rt St|t rt St|trtSt|trtSd|}t|)zGet the collection type from the annotation. :param annotation: A type annotation. :returns: The collection type. zUnknown collection type - ) rr"rdictrQset frozensetrr ValueError)rmsgs rget_collection_typer^s D))  G,,  E**  C((  I.. E**  H-- 3z 3 3C S//robjboolcLt|ot|t Sr))r isinstancerR)r_s ris_dataclass_instancercs"    :ZT%:%:!::r)rrrr)rrr rrrr))rrr r0rr)rrrr9)rrr r0rr@)rrr r0rrD)rrrrT)rrrrW)r_rrr`)* __future__rrO collectionsr dataclassesrtypingrrrr typing_extensionsr r polyfactory.constantsr polyfactory.utils.deprecationr rpolyfactory.utils.predicatesrrrrrrpolyfactory.utils.typesrr rrr&r/r8r<rCr5rNrVr^rcrGrrrms:"""""" $$$$$$88888888888822222222......UUUUUUUU-,,,,,     I        .: > > > > > 0 0 0 0 0@$$$$6;;;;;;r