U 5iu/@sdZddlZddlmZddlmZddlmZeZ GdddZ Gdd d e Z Gd d d e ZGd d d e ZGddde ZGdddZGdddZGdddZGdddZddZddZddZdS)z; Module for abstract serializer/unserializer base classes. N)StringIO)ObjectDoesNotExist)modelsc@s*eZdZdZd ddZddZddZdS) PickleSerializerzd Simple wrapper around pickle to be used in signing.dumps()/loads() and cache backends. NcCs|dkrtjn||_dSN)pickleHIGHEST_PROTOCOLprotocol)selfr r @/tmp/pip-unpacked-wheel-gcyiijx6/django/core/serializers/base.py__init__szPickleSerializer.__init__cCst||jSr)rdumpsr r objr r r rszPickleSerializer.dumpscCs t|Sr)rloads)r datar r r rszPickleSerializer.loads)N)__name__ __module__ __qualname____doc__r rrr r r r r s rc@seZdZdZdS)SerializerDoesNotExistz'The requested serializer was not found.Nrrrrr r r r rsrc@seZdZdZdS)SerializationErrorz,Something bad happened during serialization.Nrr r r r r!src@seZdZdZeddZdS)DeserializationErrorz.Something bad happened during deserialization.cCs|d||||fS)zs Factory method for creating a deserialization error which has a more explanatory message. z#%s: (%s:pk=%s) field_value was '%s'r )cls original_excmodelZfk field_valuer r r WithData)szDeserializationError.WithDataN)rrrr classmethodrr r r r r&src@seZdZdZddZdS)M2MDeserializationErrorzCSomething bad happened during deserialization of a ManyToManyField.cCs||_||_dSr)rpk)r rr"r r r r 4sz M2MDeserializationError.__init__N)rrrrr r r r r r!2sr!c@s eZdZdZddZddZdS) ProgressBarKcCs||_||_d|_dS)Nr)output total_count prev_done)r r%r&r r r r <szProgressBar.__init__cCs|js dS|d|j}||jd}|j|kr4dS||_|jdkrHdnd}|j|dd|d|j|d||jkr|jd |jdS) Nd [. ] )r%r&progress_widthr'writeflush)r countpercdoneZcrr r r updateAs *  zProgressBar.updateN)rrrr1r r7r r r r r#9sr#c@steZdZdZdZeZeZdddddddddZ dd Z d d Z d d Z ddZ ddZddZddZddZdS) Serializerz) Abstract serializer base class. FNr)streamfieldsuse_natural_foreign_keysuse_natural_primary_keysprogress_output object_countcKs||_|dk r|n||_||_||_||_|||} |d|_t |ddD]\} } | | | j j } |jr| j j } | jr| jjr| nd}nd}| j jD]l}|js||kr|jdkr|jdks|j|jkr|| |q|jdks|jdd|jkr|| |q| j jD]4}|jr|jdks>|j|jkr|| |q|| | | |jold|_qT||S)z' Serialize a queryset. NTr))startF)options stream_classr9Zselected_fieldsr;r<progress_classstart_serializationfirst enumerate start_object_metaconcrete_modelr" remote_fieldZ parent_linkZ local_fields serializeattname handle_fieldhandle_fk_fieldZlocal_many_to_manyhandle_m2m_field end_objectr7end_serializationgetvalue)r Zquerysetr9r:r;r<r=r>rA progress_barr4rrIr"Z pk_parentfieldr r r rK[s>        zSerializer.serializecCs tddS)zA Called when serializing of the queryset starts. zDsubclasses of Serializer must provide a start_serialization() methodNNotImplementedErrorr r r r rDszSerializer.start_serializationcCsdS)z? Called when serializing of the queryset ends. Nr rWr r r rQszSerializer.end_serializationcCs tddS)z> Called when serializing of an object starts. z=subclasses of Serializer must provide a start_object() methodNrUrr r r rGszSerializer.start_objectcCsdS)z< Called when serializing of an object ends. Nr rr r r rPszSerializer.end_objectcCs tddS)zW Called to handle each individual (non-relational) field on an object. z=subclasses of Serializer must provide a handle_field() methodNrUr rrTr r r rMszSerializer.handle_fieldcCs tddS)z6 Called to handle a ForeignKey field. z@subclasses of Serializer must provide a handle_fk_field() methodNrUrXr r r rNszSerializer.handle_fk_fieldcCs tddS)z5 Called to handle a ManyToManyField. zAsubclasses of Serializer must provide a handle_m2m_field() methodNrUrXr r r rOszSerializer.handle_m2m_fieldcCs tt|jddr|jSdS)zn Return the fully serialized queryset (or None if the output stream is not seekable). rRN)callablegetattrr9rRrWr r r rRszSerializer.getvalue)rrrrZinternal_use_onlyr#rCrrBrKrDrQrGrPrMrNrOrRr r r r r8Ps" .r8c@s(eZdZdZddZddZddZdS) Deserializerz+ Abstract base deserializer class. cKs&||_t|trt||_n||_dS)zA Init this serializer given a stream or a string N)rA isinstancestrrr9)r Zstream_or_stringrAr r r r s  zDeserializer.__init__cCs|Srr rWr r r __iter__szDeserializer.__iter__cCs tddS)z9Iteration interface -- return the next item in the streamz;subclasses of Deserializer must provide a __next__() methodNrUrWr r r __next__szDeserializer.__next__N)rrrrr r^r_r r r r r[s r[c@s6eZdZdZd ddZddZd dd Zdd d ZdS)DeserializedObjectae A deserialized model. Basically a container for holding the pre-saved deserialized data along with the many-to-many data saved with the object. Call ``save()`` to save the object (with the many-to-many data) to the database; call ``save(save_m2m=False)`` to save just the object fields (and not touch the many-to-many stuff.) NcCs||_||_||_dSr)objectm2m_datadeferred_fields)r rrbrcr r r r szDeserializedObject.__init__cCsd|jj|jjj|jjfS)Nz<%s: %s(pk=%s)>) __class__rrarHlabelr"rWr r r __repr__s zDeserializedObject.__repr__TcKsVtjj|jf|dd||jrL|rL|jD]\}}t|j||q0d|_dS)NT)usingraw)rModelZ save_baserarbitemsrZset)r Zsave_m2mrgkwargsZ accessor_nameZ object_listr r r saves  zDeserializedObject.savec Csi|_|jD]\}}|jj}|jd|j}t|jt j rzt |||dd}Wn:t k r}zt |j||jj|jW5d}~XYnX||j|j<qt|jt jrzt|||dd}Wn6tk r}zt |||jj|W5d}~XYnXt|j|j|q|dS)Nr-F)handle_forward_references)rbrcrjrarHZ app_labelZ model_namer\rJrZ ManyToManyReldeserialize_m2m_valuesr!rrrr"nameZ ManyToOneReldeserialize_fk_value ExceptionsetattrrLrm) r rgrTroptsrevaluesevaluer r r save_deferred_fieldss"*&z'DeserializedObject.save_deferred_fields)NN)TN)N)rrrrr rfrmrxr r r r r`s   r`cCs|jj}||jjj}|dkrt|drt|dr|f|}z*|jj||j |j||jjj<Wn|j k rYnX|f|S)z Build a model instance. If the model instance doesn't have a primary key and the model supports natural keys, try to retrieve it from the database. Nget_by_natural_key natural_key) rHdefault_managergetr"rLhasattrrz to_python db_managerryZ DoesNotExist)rirdbr{r"rzr r r build_instance src s|jjtjdr$fdd}n fdd}z t|}Wn,tk rh}zt||W5d}~XYnXz"g}|D]}|||qt|WStk r}z(t |t r|rt WYSt||W5d}~XYnXdS)Nrycs:t|dr(t|ts(jj|jSjj|SdS)Nr^) r}r\r]_default_managerrryr"rHr~)rwrrgr r m2m_convert$sz+deserialize_m2m_values..m2m_convertcsjj|Sr)rHr"r~)v)rr r r*s) rJrr}riter TypeErrorr!appendrrr\r DEFER_FIELD) rTrrgrnrZpks_iterrvrur"r rr ro!s"    roc Cs|dkr dS|jj}|j}|jj}t|drt|drt|tsz||j|}Wn"t k rv|rpt YSYnXt ||}|j j jr|j }|S|j ||S)Nryr^)rJrr field_namer}r\r]rryrrrZrHr" get_fieldr~) rTrrgrnrr{rrrwr r r rq=s*   rq)rriorZdjango.core.exceptionsrZ django.dbrrarrKeyErrorrrrrrr!r#r8r[r`rrorqr r r r s"    l9