D6idZddlmZddlZddlmZmZmZddl m Z m Z m Z m Z mZddlZddlmZddlmZddlmZd d lmZe eee feeee ffZe eee feeee ffZe d eeZe d e ZdZGddZy)zMixin that adds model instance loading behavior. .. warning:: This module is treated as private API. Users should not need to use this module directly. ) annotationsN)IterableMappingSequence)AnyGenericTypeVarUnioncast)DeclarativeMeta)Session)ObjectDeletedError)get_primary_keys _LoadDataT _ModelType)boundcttjjdddk\rt t |St t |S)N marshmallowr)int importlibmetadataversionr _LoadDataV4 _LoadDataV3)datas m/home/azureuser/techstart-app/venv/lib/python3.12/site-packages/marshmallow_sqlalchemy/load_instance_mixin.py _cast_datars? 9   % %m 4Q 78A=K&&  T ""c<eZdZGddZGddeeZy)LoadInstanceMixincHeZdZUded<ded<ded<ded<fdZxZS) LoadInstanceMixin.Optsztype[DeclarativeMeta] | NonemodelSession | None sqla_sessionbool load_instance transientct||g|i|t|dd|_t|dd|_t|dd|_t|dd|_y)Nr%r'r)Fr*)super__init__getattrr%r'r)r*)selfmetaargskwargs __class__s rr-zLoadInstanceMixin.Opts.__init__)s\ G T 3D 3F 3 w5DJ 'nd CD !(!FD $T;>DNr )__name__ __module__ __qualname____annotations__r- __classcell__r3s@rOptsr$#s$++$$ ? ?r r:c:eZdZUded<ded<ded<ded<d ed <edd Zej dd Zedd Zej ddZfdZddZ e jddZ dddd dfdZ dd d fdZdZxZS)!LoadInstanceMixin.Schemar$opts_ModelType | Noneinstancer&_sessionz bool | None _transientr(_load_instancecJ|jxs|jjS)z+The SQLAlchemy session used to load models.)r@r=r'r/s rsessionz LoadInstanceMixin.Schema.session7s==:DII$:$: :r c||_yN)r@)r/rEs rrEz LoadInstanceMixin.Schema.session<s #DMr c^|j |jS|jjS)z8Whether model instances are loaded in a transient state.)rAr=r*rDs rr*z"LoadInstanceMixin.Schema.transient@s(*&99&& &r c||_yrG)rA)r/r*s rr*z"LoadInstanceMixin.Schema.transientGs 'DOr c|jdd|_|jdd|_|jdd|_|jd|jj |_t| |i|y)NrEr?r*r)) popr@r?rAr=r)rBr,r-)r/r1r2r3s rr-z!LoadInstanceMixin.Schema.__init__Ksh"JJy$7DM"JJz48DM$jjd;DO"(**_dii>U>U"VD  G d -f -r c|jryttt|jj }t |}|Dcic](}|j|j|j*}}d|jvr+ tt|jj||Sycc}w#t$rYywxYw)zRetrieve an existing record by primary key(s). If the schema instance is transient, return `None`. :param data: Serialized data to inform lookup. N) r*r typerr=r%rkeygetvaluesr rEr)r/rr%propspropfilterss r get_instancez%LoadInstanceMixin.Schema.get_instanceRs ~~j)499??;E$U+E@EFtxx$((!33FGF7>>++ 6::5'JJ G*  s-B5 )B:: CCc ||js|S|jxs|j|}|'|jD]\}}t ||||S|j |\}}t t|jj}|di|}|jD]\}}t ||||S)aDeserialize data to an instance of the model if self.load_instance is True. Update an existing row if specified in `self.instance` or loaded by primary key(s) in the data; else create a new row. :param data: Data to deserialize. ) rBr?rTitemssetattr_split_model_kwargs_associationr r r=r%) r/rr2r?rNvalueassociation_attrs ModelClassattrs r make_instancez&LoadInstanceMixin.Schema.make_instanceds&& }}?(9(9$(?H#"&**,2JCHc512(,(L(LT(R %F%otyy?J!+F+H0668 / e$. /Or NF)rEr?r*c |xs |j|_|xs |j|_|jr#|js|js t d|xs |j |_ ttjtjt|fi|d|_S#d|_wxYw)aDeserialize data. If ``load_instance`` is set to `True` in the schema meta options, load the data as model instance(s). :param data: The data to deserialize. :param session: SQLAlchemy `session `. :param instance: Existing model instance to modify. :param transient: If `True`, load transient model instance(s). :param kwargs: Same keyword arguments as `marshmallow.Schema.load`. z"Deserialization requires a sessionN) r@rArBr*rE ValueErrorr?r maSchemar,loadr)r/rrEr?r*r2r3s rrczLoadInstanceMixin.Schema.load{s$$4t}}DM':4??DO""DNNdll !EFF$5 DM %4tBIIuw/44Z5EPP $  s 1;B44 B=)rEc |xs |j|_|js|js tdt t j tjt|fi|S)zGSame as `marshmallow.Schema.validate` but allows passing a ``session``.zValidation requires a session) r@r*rEr`r rarbr,validater)r/rrEr2r3s rrez!LoadInstanceMixin.Schema.validatesX$4t}}DMNNdll !@AA44 57+44Z5EPP Pr c L|jDcic]3\}}tt|jj|ddr||5}}}|jDcic],\}}t|jj|r||vr||.}}}||fScc}}wcc}}w)a$Split serialized attrs to ensure association proxies are passed separately. This is necessary for Python < 3.6.0, as the order in which kwargs are passed is non-deterministic, and associations must be parsed by sqlalchemy after their intermediate relationship, unless their `creator` has been set. Ignore invalid keys at this point - behaviour for unknowns should be handled elsewhere. :param data: serialized dictionary of attrs to split on association_proxy. N remote_attr)rWhasattrr.r=r%)r/rrNrZr[r2s rrYz8LoadInstanceMixin.Schema._split_model_kwargs_associations#'**,!C7499??C> NU ! !#'**,CDIIOOS1cAR6RU F ,, ,! s 8B"1B )returnr&)rEr riNone)rir()r*r(rirj)rir>)rir) rrrEr&r?r>r*r(rir)rrrEr&rizdict[str, list[str]])r4r5r6r7propertyrEsetterr*r-rTra post_loadr^rcrerYr8r9s@rrbr<0s $$##    ;  ;  $  $  '  '    (  ( . $    4'+*.#  % %$  % (  %   % %@'+ Q Q$ Q " Q -r rbN)r4r5r6r:rrrbrVr rr"r""s ? ?K-$K-r r") __doc__ __future__rimportlib.metadatarcollections.abcrrrtypingrrr r r rrasqlalchemy.ext.declarativer sqlalchemy.ormr sqlalchemy.orm.excrfieldsrstrrrrrrr"rVr rrxs#77556"1$GCH%xS0A'BBC GCH%xS0A'BBC \; < \ 9 # Y-Y-r