!%i4BddlmZmZddlmZmZddlmZddlm Z ddl m Z ddl m Z mZmZddlmZmZddlmZdd lmZdd lmZdd lmZGd d ej8ZGddeZGddeZGddej8Z y)) OrderedDict defaultdict)ListTuple)GenericRelation) ContentType)FieldDoesNotExist)ProtectedErrorSET_NULL SET_DEFAULT)ForeignObjectRel ManyToManyRel) gettext_lazy) serializers)ValidationError)UniqueValidatorcZeZdZdZdZdZdZdZdZdZ dZ d Z fd Z d Z xZS) BaseNestedModelSerializerct}t}|jjD]6\}}|jr |j |\}}t |tjrjt |jtjrF|j|vrv|j|j||j|jf||<t |tjs|j|vr|j|j|r|j|j|r||jf||<%|||jf||<9||fS#t $rYJwxYwN)rfieldsitems read_only_get_related_fieldr isinstancerListSerializerchildModelSerializersourcepopget)selfvalidated_datareverse_relations relations field_namefield related_fielddirects t/var/www/html/navyabakers_fullstack/navyabakers_prod/venv/lib/python3.12/site-packages/drf_writable_nested/mixins.py_extract_relationsz,BaseNestedModelSerializer._extract_relationssd'MM "&!2!2!4% < J (,(?(?(F% v%!;!;<u{{K,G,GH<<~5""5<<0"5;; 1>!*-%! ?AA 6"&&u'E'EFJ(:''1&1 1"5??,V, ,c|jtjj||j|j iSr)content_type_field_namerobjects get_for_modelobject_id_field_namepk)r"r7r(s r*_get_generic_lookupz-BaseNestedModelSerializer._get_generic_lookupcs8  1 1##11(;  . .   rDc|jdxs/|j|jjj}|r t |Sy)NrJ)r!r0rJattnamestr)r"r;r4rJs r*_get_related_pkz)BaseNestedModelSerializer._get_related_pkjs> XXd^ Etxx (9(9(<(<(D(DE r7NrDc|jj}g}td|D](}|j||}|s|j |*|Sr)r.r/filterrOappend)r"r' related_datar4pk_listdrJs r*_extract_related_pksz.BaseNestedModelSerializer._extract_related_pksrsTjj&& l+ #A%%a5Br" # rDc|jj}|j||}|jj |Dcic]}t |j |}}|Scc}w)Npk__in)r.r/rVrGrQrNrJ)r"r'rSr4rTrelated_instance instancess r*_prefetch_related_instancesz5BaseNestedModelSerializer._prefetch_related_instances|szjj&& ++E<@%0$7$7$>$>%?%    ## $&6 6   sA(c|jD]\}\}}}|jj|d}|.|jrc|jj j jj}||vrd|vrd}||vrt||d} | r| j||<|g}|j||} |j|} t|tr"| j|j||n|j s|| |j"<g} g} |D]}| j|j%||jj }|j'|||} |j)d|j*di| } | j|d<| j-| | j-it3| r)|jrt/|| dit/|| i|j st||}|j4| y#t.$r&}| j-|j0Yd}~/d}~wwxYw)NrJr7r;Traise_exceptionrr<)r get_initialr! one_to_oner.r/r0rJrMgetattrr\_get_save_kwargsrrr=rK many_to_manynamerOrCis_validsaverRrdetailanyadd)r"r7r$r&r(r' field_sourcerSpk_namerZr[ save_kwargsnew_related_instanceserrorsr;objrBexc m2m_managers r*"update_or_create_reverse_relationsz??))%h =  !67? 8f'.MM#**--.s1AH)) I2IIc|jD]\}\}}d}|j|}|jj}|j ||} | r*|j j | j}|j|||} | jd| jdi|j|||<y#t$r} t|| jid} ~ wwxYw)N)rJr^Tr_r<)rrar.r/rOrGrQfirstrCrgrhrdrri) r"attrsr%r&r'rlrqr;r4rJrBrrs r*!update_or_create_direct_relationsz;BaseNestedModelSerializer.update_or_create_direct_relationss1:1B @ -J- C##%j1D****K%%dK8B!))001%'778J  @##D#9&5joo'++J7'l## @(# @%z3::&>?? @s6C C6C11C6c Ttt||_tt|di|S)Nr<)rdict _save_kwargssuperrrh)r"rAr@s r*rhzBaseNestedModelSerializer.saves('f5.:DVDDrDcl|j|}t|tstt d|S)Nz6Arguments to nested serializer's `save` must be dict's)r{rrz TypeError_)r"r&rns r*rdz*BaseNestedModelSerializer._get_save_kwargss:'' 3 +t,JK rD)__name__ __module__ __qualname__r+rrCrKrOrVr\rtrxrhrd __classcell__r@s@r*rrs?,,\#&-"  B8H@0E rDrc"eZdZdZfdZxZS)NestedCreateMixinz$ Adds nested create feature c|j|\}}|j||tt||}|j |||Sr)r+rxr|rcreatert)r"r#r%r$r7r@s r*rzNestedCreateMixin.createsY'+'>'>~'N$ $ ..   *D8H //:KLrD)rrr__doc__rrrs@r*rrsrDrcBeZdZdZdediZfdZdZdZxZ S)NestedUpdateMixinz$ Adds update nested feature cannot_delete_protectedz;Cannot delete {instances} because protected relation existsc|j|\}}|j||tt|||}|j |||j |||j|Sr)r+rxr|rr=rt delete_reverse_relations_if_needrefresh_from_db)r"r7r#r%r$r@s r*r=zNestedUpdateMixin.updates~'+'>'>~'N$ $ ..   *D8    //:KL --h8IJ  "rDcj|jrt||}|j|y|jj |}|j j }|ttfvr;|tk(r|j} nd} |jdi|j| iy|jy)NrXr<) rercremoverGrQ remote_field on_deleter r get_defaultr=rfdelete) r" pks_to_deleter4r7r(rlrsqsrdefaults r*perform_nested_delete_or_updatez1NestedUpdateMixin.perform_nested_delete_or_update&s  % %!(L9K K   .$$++=+AB%22< ?? ?s%(AE F)*F$:F  F$$F)) rrrrrdefault_error_messagesr=rrrrs@r*rr s0 "1 ($) &$2?rDrc@eZdZdZgZfdZdZfdZfdZxZ S)UniqueFieldsMixinah Moves `UniqueValidator`'s from the validation stage to the save stage. It solves the problem with nested validation for unique fields on update. If you want more details, you can read related issues and articles: https://github.com/beda-software/drf-writable-nested/issues/1 http://www.django-rest-framework.org/api-guide/validators/#updating-nested-serializers Example of usage: ``` class Child(models.Model): field = models.CharField(unique=True) class Parent(models.Model): child = models.ForeignKey('Child') class ChildSerializer(UniqueFieldsMixin, serializers.ModelSerializer): class Meta: model = Child class ParentSerializer(NestedUpdateMixin, serializers.ModelSerializer): child = ChildSerializer() class Meta: model = Parent ``` Note: `UniqueFieldsMixin` must be applied only on the serializer which has unique fields. Note: When you are using both mixins (`UniqueFieldsMixin` and `NestedCreateMixin` or `NestedUpdateMixin`) you should put `UniqueFieldsMixin` ahead. ctg|_tt|}|j D]\}}|j Dcgc]}t |tr|}}|s2|jj||df|j Dcgc]}t |ts|c}|_|Scc}wcc}w)Nr) _unique_fieldsr|r get_fieldsr validatorsrrrR)r"rr&r' validatorunique_validatorsr@s r*rzUniqueFieldsMixin.get_fieldss ($:<!' C J272B2B!L%.$.y/$J"+!L !L!##**J8I!8L+MN/4/?/?$C"+%iA$C  C !L $Cs B0 B5cB|jD]h}|\}}|jr||vr t|dr*|j|j||||n||||j|jy#t $r}t ||j id}~wwxYw)N set_context)rr9r>rrrri)r"r# unique_fieldr&unique_validatorrrs r*_validate_unique_fieldsz)UniqueFieldsMixin._validate_unique_fieldss // @L+7 (J(|| . @ @+];$00Z1HI$^J%?@$^J%?%)[[%<> @# @%z3::&>?? @sAA99 BBBcL|j|tt||Sr)rr|rr)r"r#r@s r*rzUniqueFieldsMixin.creates$ $$^4&4^DDrDcN|j|tt|||Sr)rr|rr=)r"r7r#r@s r*r=zUniqueFieldsMixin.updates& $$^4&4X~NNrD) rrrrrrrrr=rrs@r*rrms-$JN"@ EOOrDrN)! collectionsrrtypingrr"django.contrib.contenttypes.fieldsr"django.contrib.contenttypes.modelsrdjango.core.exceptionsr django.db.modelsr r r django.db.models.fields.relatedr rdjango.utils.translationrrrest_frameworkrrest_framework.exceptionsrrest_framework.validatorsrrrrrrr<rDr*rsq0>:4BBK6&55a ; ;aH1*a?1a?HOO 33OOrD