o KDi@sdZddlmZddlZddlmZmZmZddl m Z m Z m Z m Z mZmZddlZddlmZddlmZdd lmZe rDdd lmZeeee feeee ffZeeee feeee ffZe d eeZe d ed ZddZGdddZ dS)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) TYPE_CHECKINGAnyGenericTypeVarUnioncast)DeclarativeMeta)ObjectDeletedError)get_primary_keys)Session _LoadDataT _ModelType)boundcCs,ttjdddkrtd|Std|S)N marshmallowr _LoadDataV4 _LoadDataV3)int importlibmetadataversionr datarc/var/www/Datamplify/venv/lib/python3.10/site-packages/marshmallow_sqlalchemy/load_instance_mixin.py _cast_datas  r c@s.eZdZGdddZGdddeeZdS)LoadInstanceMixincs>eZdZUded<ded<ded<ded<fdd ZZS) LoadInstanceMixin.Optsztype[DeclarativeMeta] | NonemodelSession | None sqla_sessionbool load_instance transientcsVtj|g|Ri|t|dd|_t|dd|_t|dd|_t|dd|_dS)Nr#r%r'Fr()super__init__getattrr#r%r'r()selfmetaargskwargs __class__rrr*+s zLoadInstanceMixin.Opts.__init__)__name__ __module__ __qualname____annotations__r* __classcell__rrr0rOpts%s r7cseZdZUded<ded<ded<ded<d ed <ed+d d Zejd,dd Zed-ddZejd.ddZfddZd/ddZ e j d0ddZ ddddd1fd#d$Z dd%d2fd'd(Zd)d*ZZS)3zLoadInstanceMixin.Schemar"opts_ModelType | Noneinstancer$_sessionz bool | None _transientr&_load_instancereturncCs|jp|jjS)z+The SQLAlchemy session used to load models.)r;r8r%r,rrrsession9sz LoadInstanceMixin.Schema.sessionr@rNonecC ||_dSN)r;)r,r@rrrr@> cCs|jdur|jS|jjS)z8Whether model instances are loaded in a transient state.N)r<r8r(r?rrrr(Bs z"LoadInstanceMixin.Schema.transientr(cCrBrC)r<)r,r(rrrr(IrDcsR|dd|_|dd|_|dd|_|d|jj|_tj|i|dS)Nr@r:r(r') popr;r:r<r8r'r=r)r*)r,r.r/r0rrr*Ms z!LoadInstanceMixin.Schema.__init__csn|jrdStd|jj}t|}fdd|D}d|vr5z td|j||WSty4YdSwdS)zRetrieve an existing record by primary key(s). If the schema instance is transient, return `None`. :param data: Serialized data to inform lookup. Nztype[_ModelType]csi|] }|j|jqSr)keyget).0proprrr ^sz9LoadInstanceMixin.Schema.get_instance..r) r(r r8r#rvaluesr@rGr )r,rr#propsfiltersrrr get_instanceTs  z%LoadInstanceMixin.Schema.get_instancerc Ks|js|S|jp ||}|dur"|D] \}}t|||q|S||\}}td|jj}|di|}|D] \}}t|||q;|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. Nr r) r=r:rNitemssetattr_split_model_kwargs_associationr r8r#) r,rr/r:rFvalueassociation_attrs ModelClassattrrrr make_instancefs z&LoadInstanceMixin.Schema.make_instanceNF)r@r:r(rrrc sn|p|j|_|p |j|_|jr|js|jstd|p|j|_ztdtj t |fi|Wd|_Sd|_w)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 session ma.SchemaN) r;r<r=r(r@ ValueErrorr:r r)loadr )r,rr@r:r(r/r0rrrY}s   zLoadInstanceMixin.Schema.load)r@dict[str, list[str]]c s>|p|j|_|js|jstdtdtjt|fi|S)zGSame as `marshmallow.Schema.validate` but allows passing a ``session``.zValidation requires a sessionrW)r;r(r@rXr r)validater )r,rr@r/r0rrr[s  z!LoadInstanceMixin.Schema.validatecs6fdd|Dfdd|D}|fS)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. cs,i|]\}}ttjj|ddr||qS)N remote_attr)hasattrr+r8r#rHrFrRr?rrrJs zLLoadInstanceMixin.Schema._split_model_kwargs_association..cs,i|]\}}tjj|r|vr||qSr)r]r8r#r^rSr,rrrJs )rO)r,rr/rr_rrQs  z8LoadInstanceMixin.Schema._split_model_kwargs_association)r>r$)r@rr>rA)r>r&)r(r&r>rA)r>r9)r>r) rrr@r$r:r9r(r&r>r)rrr@r$r>rZ)r2r3r4r5propertyr@setterr(r*rNma post_loadrVrYr[rQr6rrr0rSchema2s2          rdN)r2r3r4r7rrrdrrrrr!$s r!)!__doc__ __future__rimportlib.metadatarcollections.abcrrrtypingrrrr r r rrbsqlalchemy.ext.declarativer sqlalchemy.orm.excr fieldsrsqlalchemy.ormrstrrrrrr r!rrrrs"