U m f3@sdZddlZddlZddlmZddlmZddlmZddl m Z e Z GdddZ Gd d d eZGd d d eZGd ddeZGdddeZGdddZGdddZGdddZGdddZddZddZddZdS)z; Module for abstract serializer/unserializer base classes. N)StringIO)ObjectDoesNotExist)models)RemovedInDjango50Warningc@s*eZdZdZd ddZddZddZdS) PickleSerializerzd Simple wrapper around pickle to be used in signing.dumps()/loads() and cache backends. NcCs$tdt|dkrtjn||_dS)NzTPickleSerializer is deprecated due to its security risk. Use JSONSerializer instead.)warningswarnrpickleHIGHEST_PROTOCOLprotocol)selfr r @/tmp/pip-unpacked-wheel-siwebuq3/django/core/serializers/base.py__init__s zPickleSerializer.__init__cCst||jSN)r dumpsr r objr r rrszPickleSerializer.dumpscCs t|Sr)r loads)r datar r rr szPickleSerializer.loads)N)__name__ __module__ __qualname____doc__rrrr r r rrs rc@seZdZdZdS)SerializerDoesNotExistz'The requested serializer was not found.Nrrrrr r r rr$src@seZdZdZdS)SerializationErrorz,Something bad happened during serialization.Nrr r r rr*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 rWithData3s  zDeserializationError.WithDataN)rrrr classmethodr"r r r rr0src@seZdZdZddZdS)M2MDeserializationErrorzCSomething bad happened during deserialization of a ManyToManyField.cCs||_||_dSr)rpk)r rr%r r rrBsz M2MDeserializationError.__init__N)rrrrrr r r rr$?sr$c@s eZdZdZddZddZdS) ProgressBarKcCs||_||_d|_dS)Nr)output total_count prev_done)r r(r)r r rrJszProgressBar.__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 rupdateOs    zProgressBar.updateN)rrrr4rr:r r r rr&Gsr&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_classr<Zselected_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_objectr:end_serializationgetvalue)r Zquerysetr<r=r>r?r@rArD progress_barr7rrLr%Z pk_parentfieldr r rrNksR         zSerializer.serializecCs tddS)zA Called when serializing of the queryset starts. zDsubclasses of Serializer must provide a start_serialization() methodNNotImplementedErrorr r r rrGszSerializer.start_serializationcCsdS)z? Called when serializing of the queryset ends. Nr rZr r rrTszSerializer.end_serializationcCs tddS)z> Called when serializing of an object starts. z=subclasses of Serializer must provide a start_object() methodNrXrr r rrJszSerializer.start_objectcCsdS)z< Called when serializing of an object ends. Nr rr r rrSszSerializer.end_objectcCs tddS)zW Called to handle each individual (non-relational) field on an object. z=subclasses of Serializer must provide a handle_field() methodNrXr rrWr r rrPszSerializer.handle_fieldcCs tddS)z6 Called to handle a ForeignKey field. z@subclasses of Serializer must provide a handle_fk_field() methodNrXr[r r rrQszSerializer.handle_fk_fieldcCs tddS)z5 Called to handle a ManyToManyField. zAsubclasses of Serializer must provide a handle_m2m_field() methodNrXr[r r rrRszSerializer.handle_m2m_fieldcCs tt|jddr|jSdS)zn Return the fully serialized queryset (or None if the output stream is not seekable). rUN)callablegetattrr<rUrZr r rrUszSerializer.getvalue)rrrrZinternal_use_onlyr&rFrrErNrGrTrJrSrPrQrRrUr r r rr;`s& Cr;c@s(eZdZdZddZddZddZdS) Deserializerz+ Abstract base deserializer class. cKs&||_t|trt||_n||_dS)zA Init this serializer given a stream or a string N)rD isinstancestrrr<)r Zstream_or_stringrDr r rrs  zDeserializer.__init__cCs|Srr rZr r r__iter__szDeserializer.__iter__cCs tddS)z9Iteration interface -- return the next item in the streamz;subclasses of Deserializer must provide a __next__() methodNrXrZr r r__next__szDeserializer.__next__N)rrrrrrarbr r r rr^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 rrerfr r rrszDeserializedObject.__init__cCsd|jj|jjj|jjfS)Nz<%s: %s(pk=%s)>) __class__rrdrKlabelr%rZr r r__repr__s zDeserializedObject.__repr__TcKsVtjj|jf|dd||jrL|rL|jD]\}}t|j||q0d|_dS)NT)usingraw)rModelZ save_baserdreitemsr]set)r Zsave_m2mrjkwargsZ accessor_nameZ object_listr r rsaves  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)Nr0F)handle_forward_references)rerfrmrdrKZ app_labelZ model_namer_rMrZ ManyToManyReldeserialize_m2m_valuesr$rr"rr%nameZ ManyToOneReldeserialize_fk_value ExceptionsetattrrOrp) r rjrWr!optsrhvaluesevaluer r rsave_deferred_fields)sJ  z'DeserializedObject.save_deferred_fields)NN)TN)N)rrrrrrirpr{r r r rrcs   rccCs|jj}||jjj}|dkrt|drt|dr|f|}||j_|}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) rKdefault_managergetr%rOhasattr_statedbr} to_python db_managerr|Z DoesNotExist)rlrrr~r%rr}r r rbuild_instanceEs$ rc 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)Nr|cs:t|dr(t|ts(jj|jSjj|SdS)Nra) rr_r`_default_managerrr|r%rKr)rzr rjr r m2m_convertcs  z+deserialize_m2m_values..m2m_convertcsjj|Sr)rKr%r)v)r r rros) rMr rriter TypeErrorr$appendrur_r DEFER_FIELD) rWr!rjrqrZpks_iterryrxr%r rrrr_s"    rrc Cs|dkr dS|jj}|j}|jj}t|drt|drt|tsz||j|}Wn"t k rv|rpt YSYnXt ||}|j j jr|j }|S|j ||S)Nr|ra)rMr r field_namerr_r`rr|rrr]rKr% get_fieldr) rWr!rjrqr r~rrrzr r rrts,  rt)rr riorZdjango.core.exceptionsrZ django.dbrZdjango.utils.deprecationrrdrrKeyErrorrrurrr$r&r;r^rcrrrrtr r r rs(     A#