a xd@s|ddlZddlZddlmZddlmZddlmZddlm Z m Z ddZ dd Z d d Z dd dZdddZddZdS)N) transaction)ManyToManyField) model_to_dict)AlternativeManagerErrorNotHistoricalModelErrorc Csi}t|}|jdur|n|}t|}dd|jjjD}|jjD]D}|j|vrRqBt||j}|jdur||dur|||j<qB|||j<qB|j fi| d } || _ | dS)NcSsg|] }|jqS)attname.0fieldrrL/var/www/html/Ranjet/env/lib/python3.9/site-packages/simple_history/utils.py z(update_change_reason..Tz -history_date)typeidget_history_manager_for_modelmodel_metafieldsrgetattrZ primary_keyfilterZorder_byfirstZhistory_change_reasonsave) instancereasonattrsrmanagerhistoryZhistory_fieldsr valuerecordrrr update_change_reason s       r cCs:z |jj}Wn"ty.tdj|dYn0t||S)z1Return the history manager for a given app model.z+Cannot find a historical model for {model}.r)rZ simple_history_manager_attributeAttributeErrorrformatr)rZ manager_namerrr r s   rcCs t|jS)z/Return the history model for a given app model.)rrr!rrr get_history_model_for_model+sr$Fc Csdd|jD}t|}|j} d} tjddJ| j|||d} | rn| djrn|snd} |j| ||||dWd n1s0Y| rg} tjdd`| D]6} t t d d t | |d  }| | j fi|7} q|j| ||||dWd n1s 0Y| } | S) a Bulk create the objects specified by objs while also bulk creating their history (all in one transaction). Because of not providing primary key attribute after bulk_create on any DB except Postgres (https://docs.djangoproject.com/en/2.2/ref/models/querysets/#bulk-create) Divide this process on two transactions for other DB's :param objs: List of objs (not yet saved to the db) of type model :param model: Model class that should be created :param batch_size: Number of objects that should be created in each batch :param default_user: Optional user to specify as the history_user in each historical record :param default_change_reason: Optional change reason to specify as the change_reason in each historical record :param default_date: Optional date to specify as the history_date in each historical record :return: List of objs with IDs cSsg|]}t|tr|jqSr) isinstancernamer rrr r Ls z,bulk_create_with_history..TFZ savepoint) batch_sizeignore_conflictsr)r( default_userdefault_change_reason default_dateNcSs |dduS)Nr)xrrr hrz*bulk_create_with_history..)exclude) rZ get_fieldsr_default_managerratomicZ bulk_createpkbulk_history_createdictrritems)objsrr(r)r*r+r,Zexclude_fieldshistory_manager model_managerZsecond_transaction_requiredZ objs_with_idZobj_listobj attributesrrr bulk_create_with_history0sN$&r<c Cszt|}|p|j} | j|ur$tdtjdd6| j|||d|j||d|||dWdn1sl0YdS)a Bulk update the objects specified by objs while also bulk creating their history (all in one transaction). :param objs: List of objs of type model to be updated :param model: Model class that should be updated :param fields: The fields that are updated :param batch_size: Number of objects that should be updated in each batch :param default_user: Optional user to specify as the history_user in each historical record :param default_change_reason: Optional change reason to specify as the change_reason in each historical record :param default_date: Optional date to specify as the history_date in each historical record :param manager: Optional model manager to use for the model instead of the default manager z/The given manager does not belong to the model.Fr')r(T)r(updater*r+r,N)rr1rrrr2Z bulk_updater4) r7rrr(r*r+r,rr8r9rrr bulk_update_with_historyxs  r>cCst|drt|dSdS)NZ_change_reason)hasattrr)r:rrr get_change_reason_from_objects  r@)NFNNN)NNNNN)warningsZdjangoZ django.dbrZdjango.db.modelsrZdjango.forms.modelsrZsimple_history.exceptionsrrr rr$r<r>r@rrrr s*     L +