2i=ddlmZddlmZmZmZmZmZddlm Z ddl m Z m Z dZ dZdZdZd Z dd Z dd Zd Zy )) 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/html/acx/venv/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** ;;>>  r2Nc x|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 :return: List of objs with IDs TF savepoint) batch_sizeignore_conflictsr)r= default_userdefault_change_reason default_dateNc|dduS)Nr )xs r%z*bulk_create_with_history..ws!A$d"2r2)excludethenr )r get_fieldsr8rr9r_default_managerratomic bulk_createrbulk_history_create enumeratedictrr itemsrappendrlistrr)objsrr=r>r?r@rAr!exclude_fieldshistory_manager model_managersecond_transaction_required objs_with_idcumulative_filter obj_when_listiobj attributesqobj_lists r%bulk_create_with_historyr_<s<[[++-  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 ;# H y  # # s#F1AF$CF0$F-0F9c t|}|xs |j} | j|ur tdt j d5| j |||} |j||d|||ddd| S#1swY SxYw)ag 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 :return: The number of model rows updated, not including any history objects z/The given manager does not belong to the model.Fr;)r=T)r=updater?r@rAN)rrIrr rrJ bulk_updaterL) rRrrr=r?r@rArrTrU rows_updateds r%bulk_update_with_historyrds64E:O5u55M%'%&WXX   e ,  $00v*0U ++ !%"7% ,       s ,BB c4t|dr t|dSy)N_change_reason)hasattrr)r[s r%get_change_reason_from_objectrhss$%s,-- r2)NFNNN)NNNNN) django.dbrdjango.db.modelsrrrrrdjango.forms.modelsr simple_history.exceptionsr r r&rr1r5r-r_rdrhr r2r%rmsa!GG-V*(6 X~ *Zr2