Nwg%ddlmZddlmZmZmZmZmZddlm Z ddl m Z m Z dZ dZdZdZd Zd ed efd Zd ed efd Z ddZ ddZdZy)) transaction)Case ForeignKeyManyToManyFieldQWhen) model_to_dict)AlternativeManagerErrorNotHistoricalModelErrorc8i}t|}|j|n|}t|}|jjj Dcgc]}|j }}|jj D]W}|j |vrt||j }|jdur|9|||j <I|||j <Y|jdi|jdj} || _ | jycc}w)NTz -history_date)typepkget_history_manager_for_modelmodel_metafieldsattnamegetattr primary_keyfilterorder_byfirsthistory_change_reasonsave) instancereasonattrsrmanagerhistoryfieldhistory_fieldsvaluerecords K/var/www/horilla/myenv/lib/python3.12/site-packages/simple_history/utils.pyupdate_change_reasonr&s E NE"++1huG+G4G181D1D1K1KLemmLNL&&) == . %--0    $ ',emm$#(E%-- )W^^ $e $ - -o > D D FF#)F  KKMMs Dc |jj}t ||S#t$rtd|dwxYw)z1Return the history manager for a given app model.z#Cannot find a historical model for .)r simple_history_manager_attributeAttributeErrorr r)r manager_names r%rrsOV{{CC  5, '' V%(KE7RS&TUUVs$=c t|j}t|jjdi|t ||iS)zL Return the history manager, based on an existing history instance. r )get_app_model_primary_key_name instance_typerrr)history_instancekey_names r% get_history_manager_from_historyr1&sL..>.L.LMH O ()9)G)G H O O  W-x8 9 c,t|jS)z/Return the history model for a given app model.)rrrs r%get_history_model_for_modelr50s ( / 5 55r2ct|jjtr#|jjjdzS|jjjS)z2Return the primary key name for a given app model._id) isinstancerrrnamer4s r%r-r-5sA%++..*-{{~~""U** ;;>>  r2 m2m_fieldreturnc"|jS)a1 Returns the field name of an M2M field's through model that corresponds to the model the M2M field is defined on. E.g. for a ``votes`` M2M field on a ``Poll`` model that references a ``Vote`` model (and with a default-generated through model), this function would return ``"poll"``. )m2m_field_namer:s r%get_m2m_field_namer?<s  # # %%r2c"|jS)a. Returns the field name of an M2M field's through model that corresponds to the model the M2M field references. E.g. for a ``votes`` M2M field on a ``Poll`` model that references a ``Vote`` model (and with a default-generated through model), this function would return ``"vote"``. )m2m_reverse_field_namer>s r%get_m2m_reverse_field_namerBHs  + + --r2Nc ||jjDcgc]}t|tr |j } }t |} |j } d} tjd5| j|||} | r*| djr|sd} | j| |||||ddd| rtjd5d}g}t D]f\}}ttdt|| j!}t#d i|}|r||zn|}|j%t'd i|d |ih| r/t)| j|j+t-|ng}| j||||||ddd}  Scc}w#1swYxYw#1swYxYw) 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 :param custom_historical_attrs: Optional dict of field `name`:`value` to specify values for custom fields :return: List of objs with IDs TF savepoint) batch_sizeignore_conflictsr)rF default_userdefault_change_reason default_datecustom_historical_attrsNc|dduS)Nr )xs r%z*bulk_create_with_history..s!A$d"2r2)excludethenr )r get_fieldsr8rr9r_default_managerratomic bulk_createrbulk_history_create enumeratedictrr itemsrappendrlistrr)objsrrFrGrHrIrJrKr!exclude_fieldshistory_manager model_managersecond_transaction_required objs_with_idcumulative_filter obj_when_listiobj attributesqobj_lists r%bulk_create_with_historyriTsB[[++-  e_ -  N 4E:O**M"&   e , $00 Z:J1  LO..7G*/ '  / /%)&;)(? 0  #   % 0$ !% M#L1 A3!2%c>BHHJ O O?P%6%:VW! $$T%?J%?Q%?@ A* !(():;DDm,    / /%)&;)(? 0 ;$ J }  $ $ s#F!1AF&CF2&F/2F;c (t|} |xs |j} | j|ur tdt j d5|sd} n| j |||} | j||d||||ddd| S#1swY SxYw) 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. If empty, no model objects will be changed, but history records will still be created. :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 :param custom_historical_attrs: Optional dict of field `name`:`value` to specify values for custom fields :return: The number of model rows updated, not including any history objects z/The given manager does not belong to the model.FrDr)rFT)rFupdaterHrIrJrKN)rrSrr rrT bulk_updaterV) r\rrrFrHrIrJrrKr^r_ rows_updateds r%bulk_update_with_historyrns>4E:O5u55M%'%&WXX   e , L(44f5L ++ !%"7%$; ,  & ' & s 2BBc4t|dr t|dSy)N_change_reason)hasattrr)res r%get_change_reason_from_objectrrss$%s,-- r2)NFNNNN)NNNNNN) django.dbrdjango.db.modelsrrrrrdjango.forms.modelsr simple_history.exceptionsr r r&rr1r5r-strr?rBrirnrrr r2r%rxs!GG-V*(6  &/ &c & ./ .c . ]H  7tr2