9 fhddlZddlZddlZddlmZmZmZejdZ Gddej Z Gdde Z Gd d e Z Gd d e ZGd de ZGddZefdZGdde ZGdde ZGdde ZGddeZGdde ZGdde ZGddZGd d!e ZGd"d#eZGd$d%eZGd&d'ZeZGd(d)e ZGd*d+ej ZGd,d-eZ Gd.d/eZ!Gd0d1ej"Z#Gd2d3e Z$Gd4d5e$Z%Gd6d7e$Z&Gd8d9e&Z'Gd:d;Z(Gd<d=e$Z)dS)>N)enumserrorsutilszfactory.generatecZeZdZdZejjZdZdZ fdZ dZ dZ dZ dZxZS) BaseDeclarationzA factory declaration. Declarations mark an attribute as needing lazy evaluation. This allows them to refer to attributes defined by other BaseDeclarations in the same factory. FTc Zt|pi|_dSN)super__init__ _defaults)selfdefaults __class__s J/var/www/equiseq/venv/lib/python3.11/site-packages/factory/declarations.pyr zBaseDeclaration.__init__ s( !RcBt}||j|||js|St d|Ds|Sddl}|jj}| |||j S)Nc3>K|]}tj|VdSr )rget_builder_phase).0vs r z1BaseDeclaration.unroll_context..+s-MM!5*1--MMMMMMrrforce_sequence) dictupdater UNROLL_CONTEXT_BEFORE_EVALUATIONanyvalues factory.basebase DictFactoryrecursesequence)rinstancestepcontext full_contextfactory subfactorys runroll_contextzBaseDeclaration.unroll_context$svv DN+++G$$$4  MM|7J7J7L7LMMMMM  \- ||J T]|SSSrc`t|tr||||S|S)zEvaluate a wrapped pre-declaration. This is especially useful for declarations wrapping another one, e.g. Maybe or Transformer. r%r& overrides) isinstancer evaluate_pre)rwrappedr%r&r.s r_unwrap_evaluate_prez$BaseDeclaration._unwrap_evaluate_pre3sC g / / ''!#(  rc^||||}||||Sr )r+evaluate)rr%r&r.r's rr0zBaseDeclaration.evaluate_preAs/%%hi@@}}XtW555rc td)a%Evaluate this declaration. Args: instance (builder.Resolver): The object holding currently computed attributes step: a factory.builder.BuildStep extra (dict): additional, call-time added kwargs for the step. zThis is an abstract methodNotImplementedErrorrr%r&extras rr4zBaseDeclaration.evaluateEs"">???r)__name__ __module__ __qualname____doc__r BuilderPhaseATTRIBUTE_RESOLUTIONFACTORY_BUILDER_PHASECAPTURE_OVERRIDESrr r+r2r0r4 __classcell__rs@rrr s".C (,$((((( T T T   666 @ @ @ @ @ @ @rrceZdZdZdS)OrderedDeclaration CompatibilityN)r:r;r<r=rrrErERsrrEc(eZdZdZfdZdZxZS) LazyFunctionzSimplest BaseDeclaration computed by calling the given function. Attributes: function (function): a function without arguments and returning the computed value. cVt||_dSr r r functionrrLrs rr zLazyFunction.__init__`$   rcltd|j||S)Nz!LazyFunction: Evaluating %r on %rloggerdebugrLr8s rr4zLazyFunction.evaluateds) 8$-NNN}}rr:r;r<r=r r4rBrCs@rrIrIXsQ!!!!!rrIc(eZdZdZfdZdZxZS) LazyAttributezSpecific BaseDeclaration computed using a lambda. Attributes: function (function): a function, expecting the current LazyStub and returning the computed value. cVt||_dSr rKrMs rr zLazyAttribute.__init__qrNrcntd|j|||S)Nz"LazyAttribute: Evaluating %r on %rrPr8s rr4zLazyAttribute.evaluateus- 94=(SSS}}X&&&rrSrCs@rrUrUisQ!!!!!'''''''rrUcFeZdZdZdZGddZfdZdZxZS) TransformerTFceZdZdZdZdZdS)Transformer.Forcez Bypass a transformer's transformation. The forced value can be any declaration, and will be evaluated as if it had been passed instead of the Transformer declaration. c||_dSr ) forced_value)rr]s rr zTransformer.Force.__init__s ,D   rc2dt|jdS)NzTransformer.Force())reprr]rs r__repr__zTransformer.Force.__repr__sBT->(?(?BBB BrN)r:r;r<r=r rbrGrrForcer[~sA   - - - C C C C Crrccdt||_||_dSr )r r default transform)rrerfrs rr zTransformer.__init__s+  "rc|d|j}t||jr d}|j}nd}|||||}|r|S||S)NTFr-)poprer/rcr]r2rf)rr%r&r.value_or_declarationbypass_transformvalues rr0zTransformer.evaluate_pres(}}R>> *DJ 7 7 %# #7#D $ ))  *    L~~e$$$r) r:r;r<rArrcr r0rBrCs@rrYrYzs',$ C C C C C C C C##### %%%%%%%rrYceZdZdS) _UNSPECIFIEDNr:r;r<rGrrrnrnDrrnc d|vr8|dd\}}tt||||St||S#t$r|tur|cYSwxYw)aTry to retrieve the given attribute of an object, digging on '.'. This is an extended getattr, digging deeper if '.' is found. Args: obj (object): the object of which an attribute should be read name (str): the name of an attribute to look up. default (object): the default value to use if the attribute wasn't found Returns: the attribute pointed to by 'name', splitting on '.'. Raises: AttributeError: if obj has no 'name' attribute. .r)split deepgetattrgetattrAttributeErrorrn)objnamereattrsubnames rrtrts  $;; JJsA..MD'wsD117GDD D3%% %  l " " NNN s;AAA'&A'c2eZdZdZeffd ZdZdZxZS) SelfAttributeaSpecific BaseDeclaration copying values from other fields. If the field name starts with two dots or more, the lookup will be anchored in the related 'parent'. Attributes: depth (int): the number of steps to go up in the containers chain attribute_name (str): the name of the attribute to copy. default (object): the default value to use if the attribute doesn't exist. ctt|t|dz }||d}||_||_||_dS)Nrr)r r lenlstripdepthattribute_namere)rrrerrs rr zSelfAttribute.__init__si N##c.*?*?*D*D&E&EE'/ , rc|jdkr|j|jdz }n|}td|j|t ||j|jS)Nrz)SelfAttribute: Picking attribute %r on %r)rchainrQrRrrtre)rr%r&r9targets rr4zSelfAttribute.evaluatesX :>>Z Q/FFF @$BUW]^^^64#6 EEErcBd|jjd|jd|jdS)N<(z , default=z)>)rr:rreras rrbzSelfAttribute.__repr__s1 N # # #     LLL  r) r:r;r<r=rnr r4rbrBrCs@rr|r|sm  0<FFF       rr|c0eZdZdZdfd ZdZdZxZS)IteratoraFill this value using the values returned by an iterator. Warning: the iterator should not end ! Attributes: iterator (iterable): the iterator whose value should be used. getter (callable or None): a function to parse returned values TNct||_d|_|r fd|_dSfd|_dS)NcPtjtjSr )rResetableIterator itertoolscycleiteratorsrz#Iterator.__init__..sE,CIOT\D]D],^,^rc,tjSr )rrrsrrz#Iterator.__init__..sE,CH,M,Mr)r r getterriterator_builder)rrrrrs ` rr zIterator.__init__s_    N$^$^$^$^D ! ! !$M$M$M$MD ! ! !rc|j||_td|jt t |j}|j|S||S)Nz%Iterator: Fetching next value from %r)rrrQrRnextiterr)rr%r&r9rls rr4zIterator.evaluatesl =  1133DM Composite of a LazyAttribute and a Sequence. Attributes: function (function): A function, expecting the current LazyStub and the current sequence counter. type (function): A function converting an integer into the expected kind of counter for the 'function' attribute. ctd|j|j|||t |jS)NzDLazyAttributeSequence: Computing next value of %r for seq=%s, obj=%rrr8s rr4zLazyAttributeSequence.evaluate/sF R M4=( 4 4 4}}Xs4='9'9:::rN)r:r;r<r=r4rGrrrr&s-;;;;;rrc*eZdZdZdfd ZdZxZS)ContainerAttributeaVariant of LazyAttribute, also receives the containers of the object. Attributes: function (function): A function, expecting the current LazyStub and the (optional) object having a subfactory containing this attribute. strict (bool): Whether evaluating should fail when the containers are not passed in (i.e used outside a SubFactory). Tcdt||_||_dSr )r r rLstrict)rrLrrs rr zContainerAttribute.__init__?s+    rc||jdd}|jr|std|||S)aYEvaluate the current ContainerAttribute. Args: obj (LazyStub): a lazy stub of the object being constructed, if needed. containers (list of LazyStub): a list of lazy stubs of factories being evaluated in a chain, each item being a future field of next one. rNzKA ContainerAttribute in 'strict' mode can only be used within a SubFactory.)rr TypeErrorrL)rr%r&r9rs rr4zContainerAttribute.evaluateDsQ 122 ; (u ('(( (}}Xu---r)TrSrCs@rrr6sV .......rrceZdZdZdZdZdS)ParameteredAttributezBase class for attributes expecting parameters. Attributes: defaults (dict): Default values for the parameters. May be overridden by call-time parameters. c.|||S)a Evaluate the current definition and fill its attributes. Uses attributes definition in the following order: - values defined when defining the ParameteredAttribute - additional values defined when instantiating the containing factory Args: instance (builder.Resolver): The object holding currently computed attributes step: a factory.builder.BuildStep extra (dict): additional, call-time added kwargs for the step. )generater8s rr4zParameteredAttribute.evaluate`s}}T5)))rct)aActually generate the related attribute. Args: sequence (int): the current sequence number obj (LazyStub): the object being constructed create (bool): whether the calling factory was in 'create' or 'build' mode params (dict): parameters inherited from init and evaluation-time overrides. Returns: Computed value for the current declaration. r6)rr&paramss rrzParameteredAttribute.generateps"###rN)r:r;r<r=r4rrGrrrrXs<*** $$$$$rrc$eZdZdZdZdZdZdS)_FactoryWrapperzHandle a 'factory' arg. Such args can be either a Factory subclass, or a fully qualified import path for that subclass (e.g 'myapp.factories.MyFactory'). cd|_dx|_|_t|tr ||_dSt|t rd|vst d|z|dd\|_|_dS)NrhrrzrA factory= argument must receive either a class or the fully qualified path to a Factory subclass; got %r instead.r)r)modulerxr/typestr ValueErrorrsplit)rfactory_or_paths rr z_FactoryWrapper.__init__s "$$ di ot , , D*DLLL44 59O9O "$34555&5%;%;C%C%C "DKrcf|j$tj|j|j|_|jSr )r)r import_objectrrxras rgetz_FactoryWrapper.gets2 <  .  DL|rcV|jd|jd|jdSd|jjdS)Nz<_FactoryImport: rr>)r)rrxrras rrbz_FactoryWrapper.__repr__s= < At{AATYAAA A@t|'=@@@ @rN)r:r;r<r=r rrbrGrrrrsS D D DAAAAArrc6eZdZdZdZdZfdZdZdZxZ S) SubFactoryzBase class for attributes based upon a sub-factory. Attributes: defaults (dict): Overrides to the defaults defined in the wrapped factory factory (base.Factory): the wrapped factory Fc btjdi|t||_dS)NrG)r r rfactory_wrapper)rr)kwargsrs rr zSubFactory.__init__s6""6""".w77rc4|jSz.Retrieve the wrapped factory.Factory subclass.rrras r get_factoryzSubFactory.get_factory#'')))rc|}td|j|jt j|||jr|jnd}| |||S)zEvaluate the current definition and fill its attributes. Args: step: a factory.builder.BuildStep params (dict): additional, call-time added kwargs for the step. z.SubFactory: Instantiating %s.%s(%s), create=%r)rNr) rrQrRr;r:r log_pprintFORCE_SEQUENCEr$r#)rr%r&r9r*rs rr4zSubFactory.evaluatesz%%''  <  !:#6  E * * *     +/*=G4||Jn|MMMr) r:r;r<r=rrr rr4rBrCs@rrrstN',$88888***NNNNNNNrrc(eZdZdZdZdfd ZxZS)Dictz$Fill a dict with usual declarations.Tfactory.DictFactorycVtj|fit|dSr )r r r)rr dict_factoryrs rr z Dict.__init__s.66f66666r)rr:r;r<r=rr rBrCs@rrrsH..N7777777777rrc(eZdZdZdZdfd ZxZS)Listz'Fill a list with standard declarations.Tfactory.ListFactorycndt|D}tj|fi|dS)Nc4i|]\}}t||SrG)r)rirs r z!List.__init__..s$:::1#a&&!:::r) enumerater r )rr list_factoryrs rr z List.__init__sC:: &(9(9:::0000000r)rrrCs@rrrsH11N1111111111rrceZdZdZdS)SkipcdS)NFrGras r__bool__z Skip.__bool__surN)r:r;r<rrGrrrrs#rrc6eZdZeeffd ZdZdZdZxZS)Maybecttj|t |d}||_||_||_tj|tj|d}d|D}t|dkrtd|d||r| ntj j |_dS)Nre)yes_declarationno_declarationch|]}||Sr rG)rphases r z!Maybe.__init__..sOOOU=Nu=N=N=NrrzInconsistent phases for z: )r r rrr|decideryesnorr~rrir>r?r@)rrrrphases used_phasesrs rr zMaybe.__init__s   "7 + + 3#GT:::G "  %6GG#5nEE  PO&--//OOO {  a  KtKKKKLL L:E%r[__%6%6%65K]Kr"""rctj|j}|tjjkr#|j|j||}n4|tjjksJ|j||i}|r|j n|j }tj|tjjkr||||S|S)z#Handle post-generation declarationsr-) rrrr>r?r0stubPOST_INSTANTIATION evaluate_postrr)rr%r&r. decider_phasechoicers rrzMaybe.evaluate_posts/ == E.C C C\../DDFF!E$6$IIIII\//!0<s rN)r:r;r<r=rrrGrrrr*s< $ $ $rrc:eZdZfdZdZedZxZS)SimpleParametercVt||_dSr )r r rl)rrlrs rr zSimpleParameter.__init__Ds$  rc||jiSr )rlrs rrzSimpleParameter.as_declarationsHs    rcnt|ts ||S||Sr )r/rtouch_creation_counter)clsrls rwrapzSimpleParameter.wrapMs9%++ 3u::  $$&&& r)r:r;r<r r classmethodrrBrCs@rrrCsb   [rrc4eZdZdZfdZdZdZdZxZS)Traitz_The simplest complex parameter, it enables a bunch of new declarations based on a boolean flag.c Vt||_dSr )r r r.)rr.rs rr zTrait.__init__Ws$ "rc  i}|jD]g\}}ttd|t jzd|d|||t||<h|S)NrrFr)rrr) r.itemsrr|countrSPLITTERrr)rrrr. maybe_field new_values rrzTrait.as_declarations[s &*n&:&:&<&<   "K%*%k//?????" " !*+// TBB & & &Ik " "rc fd|DS)z'This might alter fields it's injecting.c&g|] }|jv |SrG)r.)rparamrs r z%Trait.get_revdeps..ms%III%$.1H1H1H1H1HrrGrs` rrzTrait.get_revdepsksIIII:IIIIrc|jjddd|jDdS)Nrz, c3 K|] }d|zV dS)z%s=%rNrG)rts rrz!Trait.__repr__..rs&BBagkBBBBBBrr_)rr:joinr.r ras rrbzTrait.__repr__osL N # # # IIBB4>+?+?+A+ABBB B B B B  r) r:r;r<r=r rrrbrBrCs@rr r Usnii##### JJJ       rr c\eZdZUeed<ejed<ejeejfed<dS)PostGenerationContextvalue_providedrlr9N) r:r;r<bool__annotations__TAnyrrrGrrrrzsB 5LLL 6#qu* rrc6eZdZdZejjZdZdZ dS)PostGenerationDeclarationzCDeclarations to be called once the model object has been generated.c||||}ttd|v|dd|D}||||S)Nrhc&i|]\}}|dk ||SrhrG)rkrs rrz;PostGenerationDeclaration.evaluate_post..s#???DAqqBww1awwwr)rrlr9)r+rrrr call)rr%r&r.r'postgen_contexts rrz'PostGenerationDeclaration.evaluate_postsz%%hi@@/g ..++b//??GMMOO???    yy4999rct)aNCall this hook; no return value is expected. Args: instance (object): the newly generated object step (bool): whether the object was 'built' or 'created' context: a builder.PostGenerationContext containing values extracted from the containing factory's declaration r6)rr%r&r's rr&zPostGenerationDeclaration.calls"###rN) r:r;r<r=rr>rr@rr&rGrrr!r!sDMM!.A::: $ $ $ $ $rr!c(eZdZdZfdZdZxZS)PostGenerationz:Calls a given function once the object has been generated.cVt||_dSr rKrMs rr zPostGeneration.__init__rNrc td|jj|jjt j||f||jj tj k}|j|||j fi|j S)Nz!PostGeneration: Calling %s.%s(%s))rQrRrLr;r:rr_asdictbuilderstrategyrCREATE_STRATEGYrlr9)rr%r&r'creates rr&zPostGeneration.calls / M $ M "  4 !!      &%*??t} fgm>>/6}>> >rr:r;r<r=r r&rBrCs@rr*r*sMDD!!!!! > > > > > > >rr*c4eZdZdZdZdfd ZdZdZxZS)RelatedFactorya6Calls a factory once the object has been generated. Attributes: factory (Factory): the factory to call defaults (dict): extra declarations for calling the related factory name (str): the name to use to refer to the generated object when calling the related factory Frhc t||_||_t ||_dSr )r r rxrrr)rr)factory_related_namerrs rr zRelatedFactory.__init__s= (   .w77rc4|jSrrras rrzRelatedFactory.get_factoryrrc |}|jr3td|j|j|j|jSt|j}| |j |j r |||j <td|j|jtj |f||||S)Nz>RelatedFactory: Using provided %r instead of generating %s.%s.z$RelatedFactory: Generating %s.%s(%s))rrrQrRrlr;r:rrrr9rxrrr#)rr%r&r'r) passed_kwargss rr&zRelatedFactory.calls""$$  ! ! LLP "G$4    = T]++ W]+++ 9 0'/M$) $ 2      dWm 4 4    ||G]333rr$) r:r;r<r=rr rr&rBrCs@rr4r4sm(-$888888***4444444rr4c.eZdZdZdfd ZfdZxZS)RelatedFactoryListaCalls a factory 'size' times once the object has been generated. Attributes: factory (Factory): the factory to call "size-times" defaults (dict): extra declarations for calling the related factory factory_related_name (str): the name to use to refer to the generated object when calling the related factory size (int|lambda): the number of times 'factory' is called, ultimately returning a list of 'factory' objects w/ size 'size'. rhc L||_tj||fi|dSr )sizer r )rr)r6r>rrs rr zRelatedFactoryList.__init__s2 "6CC(CCCCCrctfdtt|jtr|jn|DS)Nc>g|]}SrG)r&)rrr'r%parentr&s rrz+RelatedFactoryList.call..s9    KK$ 0 0   r)r ranger/r>r)rr%r&r'rArs ```@rr&zRelatedFactoryList.callsq        49c(B(BS499 TT    r)rhr<r2rCs@rr;r;sf  DDDDDD         rr;ceZdZdS) NotProvidedNrorGrrrDrDrprrDc(eZdZdZfdZdZxZS)PostGenerationMethodCallaCalls a method of the generated object. Attributes: method_name (str): the method to call method_args (list): arguments to pass to the method method_kwargs (dict): keyword arguments to pass to the method Example: class UserFactory(factory.Factory): ... password = factory.PostGenerationMethodCall('set_pass', password='') ctt|dkrtjd||_|r|dnt |_||_dS)Nrz|A PostGenerationMethodCall can only handle 1 positional argument; please provide other parameters through keyword arguments.r) r r r~rInvalidDeclarationError method_namerD method_arg method_kwargs)rrIargsrrs rr z!PostGenerationMethodCall.__init__ so  t99q==0M '%):$q''{#rc T|js|jturd}n|jf}n|jf}t |j}||jt||j }t d||j tj ||||i|S)NrGz+PostGenerationMethodCall: Calling %r.%s(%s))rrJrDrlrrKrr9rurIrQrRrr)rr%r&r'rLrmethods rr&zPostGenerationMethodCall.calls% $+--)M#Dd()) gm$$$4#344 9     T6 * *    vt&v&&&rr2rCs@rrFrFsQ   $ $ $ $ $'''''''rrF)*rloggingtypingrrhrrr getLoggerrQ OrderedBaserrErIrUrYrnrtr|rrrrrrrrrrrrrrr  NamedTuplerr!r*r4r;rDrFrGrrrTsW""""""""""  - . .B@B@B@B@B@e'B@B@B@J ?"'''''O'''"(%(%(%(%(%/(%(%(%V        $0:% % % % % O% % % P#"#"#"#"#"#"#"#"L11111111$ ; ; ; ; ;H ; ; ; ........D&$&$&$&$&$?&$&$&$RAAAAAAAAD&N&N&N&N&N&N&N&NR77777:77711111:111  tvv<L<L<L<L<LO<L<L<L~!2i$     I   JAL $$$$$$$$4>>>>>.>>>*.4.4.4.4.4..4.4.4b        0        *'*'*'*'*'8*'*'*'*'*'r