U Ha-@sdZddlmZddlmZddlmZeZGddde Z Gddde Z Gd d d e Z Gd d d e ZGd ddZGdddZGdddZGdddZddZddZddZdS)z; Module for abstract serializer/unserializer base classes. )StringIO)ObjectDoesNotExist)modelsc@seZdZdZdS)SerializerDoesNotExistz'The requested serializer was not found.N__name__ __module__ __qualname____doc__r r @/tmp/pip-unpacked-wheel-3jxiddxt/django/core/serializers/base.pyr src@seZdZdZdS)SerializationErrorz,Something bad happened during serialization.Nrr r r r r sr c@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 WithDataszDeserializationError.WithDataN)rrr r classmethodrr r r r rsrc@seZdZdZddZdS)M2MDeserializationErrorzCSomething bad happened during deserialization of a ManyToManyField.cCs||_||_dSN)rpk)selfrrr r r __init__$sz M2MDeserializationError.__init__N)rrr r rr r r r r"src@s eZdZdZddZddZdS) ProgressBarKcCs||_||_d|_dS)Nr)output total_count prev_done)rrrr r r r,szProgressBar.__init__cCs|js dS|d|j}||jd}|j|kr4dS||_|jdkrHdnd}|j|dd|d|j|d||jkr|jd |jdS) Nd [. ] )rrprogress_widthrwriteflush)rcountpercdoneZcrr r r update1s *  zProgressBar.updateN)rrr r(rr.r r r r r)src@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_classr0Zselected_fieldsr2r3progress_classstart_serializationfirst enumerate start_object_metaconcrete_modelr remote_fieldZ parent_linkZ local_fields serializeattname handle_fieldhandle_fk_fieldZlocal_many_to_manyhandle_m2m_field end_objectr.end_serializationgetvalue)rZquerysetr0r1r2r3r4r5r8 progress_barr+objr@rZ pk_parentfieldr r r rBKs>        zSerializer.serializecCs tddS)zA Called when serializing of the queryset starts. zDsubclasses of Serializer must provide a start_serialization() methodNNotImplementedErrorrr r r r;yszSerializer.start_serializationcCsdS)z? Called when serializing of the queryset ends. Nr rOr r r rHszSerializer.end_serializationcCs tddS)z> Called when serializing of an object starts. z=subclasses of Serializer must provide a start_object() methodNrMrrKr r r r>szSerializer.start_objectcCsdS)z< Called when serializing of an object ends. Nr rPr r r rGszSerializer.end_objectcCs tddS)zW Called to handle each individual (non-relational) field on an object. z=subclasses of Serializer must provide a handle_field() methodNrMrrKrLr r r rDszSerializer.handle_fieldcCs tddS)z6 Called to handle a ForeignKey field. z@subclasses of Serializer must provide a handle_fk_field() methodNrMrQr r r rEszSerializer.handle_fk_fieldcCs tddS)z5 Called to handle a ManyToManyField. zAsubclasses of Serializer must provide a handle_m2m_field() methodNrMrQr r r rFszSerializer.handle_m2m_fieldcCs tt|jddr|jSdS)zn Return the fully serialized queryset (or None if the output stream is not seekable). rIN)callablegetattrr0rIrOr r r rIszSerializer.getvalue)rrr r Zinternal_use_onlyrr:rr9rBr;rHr>rGrDrErFrIr r r r r/@s" .r/c@s(eZdZdZddZddZddZdS) Deserializerz+ Abstract base deserializer class. cKs&||_t|trt||_n||_dS)zA Init this serializer given a stream or a string N)r8 isinstancestrrr0)rZstream_or_stringr8r r r rs  zDeserializer.__init__cCs|Srr rOr r r __iter__szDeserializer.__iter__cCs tddS)z9Iteration interface -- return the next item in the streamz;subclasses of Deserializer must provide a __next__() methodNrMrOr r r __next__szDeserializer.__next__N)rrr r rrWrXr r r r rTs rTc@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)rrKr[r\r r r rszDeserializedObject.__init__cCsd|jj|jjj|jjfS)Nz<%s: %s(pk=%s)>) __class__rrZr?labelrrOr r r __repr__s zDeserializedObject.__repr__TcKsVtjj|jf|dd||jrL|rL|jD]\}}t|j||q0d|_dS)NT)usingraw)rModelZ save_baserZr[itemsrSset)rZsave_m2mr`kwargsZ 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)r[r\rcrZr?Z app_labelZ model_namerUrArZ ManyToManyReldeserialize_m2m_valuesrrrrrnameZ ManyToOneReldeserialize_fk_value ExceptionsetattrrCrf) rr`rLroptsr^valuesevaluer r r save_deferred_fieldss"*&z'DeserializedObject.save_deferred_fields)NN)TN)N)rrr r rr_rfrqr r r r rYs   rYcCs|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) r?default_managergetrrChasattrrs to_python db_managerrrZ DoesNotExist)rbdatadbrtrrsr r r build_instancesr{c 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)Nrrcs:t|dr(t|ts(jj|jSjj|SdS)NrW) rvrUrV_default_managerrxrrrr?rw)rprr`r r m2m_convertsz+deserialize_m2m_values..m2m_convertcsjj|Sr)r?rrw)v)rr r r~s) rArrvr|iter TypeErrorrappendrkrUr DEFER_FIELD) rLrr`rgr~Zpks_iterrornrr r}r rhs"    rhc Cs|dkr dS|jj}|j}|jj}t|drt|drt|tsz||j|}Wn"t k rv|rpt YSYnXt ||}|j j jr|j }|S|j ||S)NrrrW)rArr| field_namervrUrVrxrrrrrSr?r get_fieldrw) rLrr`rgrrtrrKrpr r r rj-s*   rjN)r iorZdjango.core.exceptionsrZ django.dbrrZrKeyErrorrrkr rrrr/rTrYr{rhrjr r r r s    l9