a C gvA@sddlmZddlZddlZddlmZddlmZmZddl m Z ddl m Z ddl mZddlmZdd lmZdd lmZmZd d Zd dZdddZddZddZGdddejZdS))unicode_literalsN)FieldDoesNotExist)models transaction)ForeignObjectRel)is_protected_type)DjangoJSONEncoder)settings)timezone) ParentalKeyParentalManyToManyFieldcCs|jdur|j||jdud}t|tjrjtjrjt|rZt }t || tjj }nt |tjj }t|rv|S||Snt||SdS)N)add) remote_fieldZpre_savepk isinstancedatetimer USE_TZr Zis_naiveget_default_timezone make_aware astimezoneutc localtimerZvalue_to_stringgetattrZ get_attname)fieldmodelvaluedefault_timezonerR/var/www/lab.imftr.de/x/nb_venv/lib/python3.9/site-packages/modelcluster/models.pyget_field_values   rcCsZ|jj}|jr$|jjr$|jjjj}qdt||i}|jjD]}|jr:t||||j<q:|S)z Return a serialised version of the model's fields which exist as local database columns (i.e. excluding m2m and incoming foreign key relations) r) _metarr parent_linkrrfields serializename)rpk_fieldobjrrrr get_serializable_data_for_fields's r'TFc CsB|jj}i}|jr6|jjr6|d||j<|jjjj}q |d||j<|D]\}}z|j|}Wnty|YqLYn0t |t rqL|jrt |jt j r|jjj j|d} t| ||j<qL|jrt |jt jr|durd||j<n|jjj|jj|} | ||j<|rz"|jjj jfi|jj| iWnz|jjjy|jjt jkr`nN|jjt jkr|r~YdSd||j<n$|jjt jkrd||j<ntdYn0qL||} t |t jrtjr| durt} t| rt | | } n t!| | } | ||j"<qL|fi|} |ddur>d| j#_$| S)Nr)Zpk__inzRcan't currently handle on_delete types other than CASCADE, SET_NULL and DO_NOTHINGF)%r rrr!attnameritems get_fieldrrrrZ ManyToManyRelZ_default_managerfilterlistZ ManyToOneRel field_nameZ to_pythongetZ DoesNotExistZ on_deleteZ DO_NOTHINGZCASCADEZSET_NULL ExceptionZ DateTimeFieldr rr rZis_awarerrr$_stateZadding)rdata check_fks strict_fksr%kwargsr-Z field_valuerZrelated_objectsZ clean_valuerrr&rrrmodel_from_serializable_data:sZ     "       r5cCsdd|jDS)z Return a list of RelatedObject records for child relations of the given model, including ones attached to ancestors of the model cSsg|]}t|jtr|qSr)rrr .0rrrr s z+get_all_child_relations..r get_fieldsrrrrget_all_child_relationssr<cCsdd|jDS)z} Return a list of ParentalManyToManyFields on the given model, including ones attached to ancestors of the model cSsg|]}t|tr|qSr)rr r6rrrr8s z/get_all_child_m2m_relations..r9r;rrrget_all_child_m2m_relationssr=cseZdZfddZfddZddZddZedd d Zedd dZ e j dddZ dddZ dddZGdddZZS)ClusterableModelc sddt|Dddt|D}tfdd|Dr}i}|D]}|vrJ||||<qJtj|i||D]\}}t|||qntj|idS)zv Extend the standard model constructor to allow child object lists to be passed in via kwargs cSsg|] }|qSrget_accessor_namer7relrrrr8z-ClusterableModel.__init__..cSsg|] }|jqSrr$r6rrrr8rCc3s|]}|vVqdSNr)r7r$r4rr rCz,ClusterableModel.__init__..N) r<r=anycopypopsuper__init__r)setattr) selfargsr4child_relation_namesZkwargs_for_superZrelation_assignmentsrel_namer-Zrelated_instances __class__rFrrLszClusterableModel.__init__c  sddt|D}ddt|D}|dd}|durFd}|}|}nHg}g}g}|D]6}||vrn||qV||vr||qV||qVtjfd|i||D]} t|| q|D]}t||qdS)z@ Save the model and commit all child relations. cSsg|] }|qSrr?rArrrr8rCz)ClusterableModel.save..cSsg|] }|jqSrrDr6rrrr8rC update_fieldsN)r<r=rJappendrKsavercommit) rNr4rPZchild_m2m_field_namesrTZreal_update_fieldsZrelations_to_commitZm2m_fields_to_commitrZrelationrRrrrVs*    zClusterableModel.savecCst|}t|D]L}|}t||}t|jdrJdd|D||<qdd|D||<qt|D].}|jrft||j }dd|D||j <qf|S)Nserializable_datacSsg|] }|qSr)rXr7childrrrr8rCz6ClusterableModel.serializable_data..cSsg|] }t|qSr)r'rYrrrr8rCcSsg|] }|jqSr)rrYrrrr8rC) r'r<r@rallhasattr related_modelr=r#r$)rNr&rBrQchildrenrrrrrXs   z"ClusterableModel.serializable_datacCstj|tdS)N)cls)jsondumpsrXr)rNrrrto_jsonszClusterableModel.to_jsonTFc st|||d}|durdSt|}|D]}|}z ||}WntyVYq(Yn0|jtdr~fdd|D} nfdd|D} tdd| } t||| q(|S) a Build an instance of this model from the JSON-like structure passed in, recursing into related objects as required. If check_fks is true, it will check whether referenced foreign keys still exist in the database. - dangling foreign keys on related objects are dealt with by either nullifying the key or dropping the related object, according to the 'on_delete' setting. - dangling foreign keys on the base object will be nullified, unless strict_fks is true, in which case any dangling foreign keys with on_delete=CASCADE will cause None to be returned for the entire object. r2r3Nfrom_serializable_datacsg|]}j|ddqSTrc)rdr7Z child_datar2r]rrr8sz;ClusterableModel.from_serializable_data..csg|]}t|ddqSre)r5rfrgrrr8 scSs|duSrEr)rZrrrrCz9ClusterableModel.from_serializable_data..)r5r<r@KeyErrorr]r\r+rM) r_r1r2r3r&Zchild_relationsrBrQZchild_data_listr^rrgrrds*       z'ClusterableModel.from_serializable_datacCs|jt|||dS)Nrc)rdr`loads)r_Z json_datar2r3rrr from_jsonszClusterableModel.from_jsonc Csi}t|tr|j|}t|jts.td|jj}t || }t || }|s^| | dD]p} | j} | du} d| _t| ||j|| | dur| ||| f<ql|df|vrg||df<||df| ql|r||S)a[ Copies all of the objects in the accessor_name to the target object. For example, say we have an event with speakers (my_event) and we need to copy these to another event (my_other_event): my_event.copy_child_relation('speakers', my_other_event) By default, this copies the child objects without saving them. Set the commit paremter to True to save the objects but note that this would cause an exception if the target object is not saved. This will overwrite the child relation on the target object. This is to avoid any issues with unique keys and/or sort_order. If you want it to append. set the `append` parameter to True. This method returns a dictionary mapping the child relation/primary key on the source object to the new object created for the target object. zQcopy_child_relation can only be used for relationships defined with a ParentalKeyrN)rstrr r*rr LookupErrorrr(rr@clearr[Zorder_byrrMidr rUrW) rNchild_relationtargetrWrUchild_object_mapZparental_key_nameZsource_managerZtarget_managerZ child_objectZold_pkZis_savedrrrcopy_child_relations0      z$ClusterableModel.copy_child_relationNc CsD|pg}i}t|D]*}||vr&q||j||||dq|S)a Copies all of the objects in all child relations to the target object. This will overwrite all of the child relations on the target object. Set exclude to a list of child relation accessor names that shouldn't be copied. This method returns a dictionary mapping the child_relation/primary key on the source object to the new object created for the target object. )rWrU)r<r@updaters)rNrqexcluderWrUrrrprrrcopy_all_child_relationsSs   z)ClusterableModel.copy_all_child_relationscCs|pg}i}|jD]}|j|vr&q|jr.q|jrlt|trt||j}t|dr| }|r|||j<qt|t j r|j j rqt|t jrd||j<t||j||j<qt||j||j<q|jfi|}|j||d}||fS)a Makes a copy of this object and all child relations. Includes all field data including child relations and parental many to many fields. Doesn't include non-parental many to many. The result of this method is unsaved. r[N)ru)r r:r$Z auto_createdZ many_to_manyrr rr\r[rZ OneToOneFieldrr!Z ForeignKeyr(rSrv)rNZexclude_fieldsZ data_dictrZparental_fieldvaluesrIrrrrr copy_clusteris0        zClusterableModel.copy_clusterc@seZdZdZdS)zClusterableModel.MetaTN)__name__ __module__ __qualname__ZabstractrrrrMetasr|)TF)TF)FF)NFF)N)ryrzr{rLrVrXrb classmethodrdrkrZatomicrsrvrxr| __classcell__rrrRrr>s   +  :  ;r>)TF) __future__rr`rZdjango.core.exceptionsrZ django.dbrrZdjango.db.models.fields.relatedrZdjango.utils.encodingrZdjango.core.serializers.jsonrZ django.confr Z django.utilsr Zmodelcluster.fieldsr r rr'r5r<r=ZModelr>rrrrs        H