o Ti5@sddlmZddlmZmZmZddlmZddlm Z m Z ddl m Z ddlmZmZmZmZmZddlmZmZeZGdd d ejZGd d d ejZGd d d ejZddZddZejj eejj eddZ!GdddejZ"GdddejZ#GdddejZ$dS))ImproperlyConfigured)modelsrouter transaction)Now) post_savepre_save) gettext_lazy)AutoCreatedFieldAutoLastModifiedField MonitorField StatusField UUIDField) QueryManagerSoftDeletableManagercsFeZdZdZeedZeedZfddZ GdddZ Z S)TimeStampedModelzl An abstract base class model that provides self-updating ``created`` and ``modified`` fields. createdmodifiedcs:|dd}|rt|dh|d<tj|i|dS)z Overriding the save method in order to make sure that modified field is updated even if it is not given as a parameter to the update field argument. update_fieldsNrgetsetunionsupersaveselfargskwargsr __class__R/var/www/html/evchargy.com/venv/lib/python3.10/site-packages/model_utils/models.pyrs zTimeStampedModel.savec@eZdZdZdS)zTimeStampedModel.MetaTN__name__ __module__ __qualname__abstractr!r!r!r"Meta(r)) r%r&r'__doc__r _rr rrr) __classcell__r!r!rr"rs     rc@sFeZdZdZejeddddZejeddddZGdddZ dS) TimeFramedModelzm An abstract base class model that provides ``start`` and ``end`` fields to record a timeframe. startT)nullblankendc@r#)zTimeFramedModel.MetaTNr$r!r!r!r"r)5r*r)N) r%r&r'r+rZ DateTimeFieldr,r/r2r)r!r!r!r"r.,s r.csJeZdZdZeedZeedddZfddZ GdddZ Z S) StatusModelaD An abstract base class model with a ``status`` field that automatically uses a ``STATUS`` class attribute of choices, a ``status_changed`` date-time field that records when ``status`` was last modified, and an automatically-added manager for each status that returns objects with that status only. statuszstatus changed)ZmonitorcsB|dd}|rd|vrt|dh|d<tj|i|dS)z Overriding the save method in order to make sure that status_changed field is updated even if it is not given as a parameter to the update field argument. rNr4status_changedrrrr!r"rEs  zStatusModel.savec@r#)zStatusModel.MetaTNr$r!r!r!r"r)Qr*r)) r%r&r'r+r r,r4r r5rr)r-r!r!rr"r39s    r3cKsht|tsdS|jj}t|ddD]\}}t||r#td|j|f||t |dq|j |j_ dS)z? Add a Querymanager for each status item dynamically. NZSTATUSr!z^StatusModel: Model '%s' has a field named '%s' which conflicts with a status of the same name.)r4) issubclassr3_metadefault_managergetattr _field_existsrr% add_to_classrnameZdefault_manager_name)senderrr8valuedisplayr!r!r"add_status_query_managersUs  r@c Ksft|tsdSt|drtd|j|dttjt dtjddBtjt dtjddB@dS) z7 Add a QueryManager for a specific timeframe. NZ timeframedz[Model '%s' has a field named 'timeframed' which conflicts with the TimeFramedModel manager.)Z start__lteT)Z start__isnull)Zend__gte)Z end__isnull) r6r.r:rr%r;rrQnow)r=rr!r!r"add_timeframed_query_managerks   rCcCs|dd|jjDvS)NcSsg|]}|jqSr!)Zattname).0fr!r!r" sz!_field_exists..)r7Z local_fields)Z model_class field_namer!r!r"r:sr:csTeZdZdZejddZGdddZeddZ eZ e Z d fd d Z ZS) SoftDeletableModelz An abstract base class model with a ``is_removed`` field that marks entries that are not going to be used anymore, but are kept in db for any reason. Default manager returns only not-removed entries. F)defaultc@r#)zSoftDeletableModel.MetaTNr$r!r!r!r"r)r*r)T)Z_emit_deprecation_warningsNcs0|r d|_|j|ddStj|d|i|S)z Soft delete object (set its ``is_removed`` field to True). Actually delete object if setting ``soft`` to False. T)usingrJN) is_removedrrdelete)rrJZsoftrrrr!r"rLszSoftDeletableModel.delete)NT)r%r&r'r+rZ BooleanFieldrKr)robjectsZavailable_objectsManagerZ all_objectsrLr-r!r!rr"rHs  rHc@s,eZdZdZeddddZGdddZdS) UUIDModelzz This abstract base class provides id field on any model that inherits from it which will be the primary key. TF)Z primary_keyversioneditablec@r#)zUUIDModel.MetaTNr$r!r!r!r"r)r*r)N)r%r&r'r+ridr)r!r!r!r"rOsrOcs>eZdZdZGdddZd fdd Z  d dd ZZS) SaveSignalHandlingModelz An abstract base class model to pass a parameter ``signals_to_disable`` to ``save`` method in order to disable signals c@r#)zSaveSignalHandlingModel.MetaTNr$r!r!r!r"r)r*r)Ncs |pg|_tj|i|dS)zp Add an extra parameters to hold which signals to disable If empty, nothing will change N)signals_to_disablerr)rrUrrrr!r"rs zSaveSignalHandlingModel.saveFc Cs|p tj|j|d}|r|s|rJ|dust|dksJ|j}}|jjr+|jj}|j}|js@d|jvr@t j |||||dt j |dd|sQ| ||||||||||} Wdn1sewY||j_d|j_|jsd|jvrtj ||| |||d g|_dS) z Copied from base class for a minor change. This is an ugly overwriting but since Django's ``save_base`` method does not differ between versions 1.8 and 1.10, that way of implementing wouldn't harm the flow )instanceNrr)r=rVrawrJrF)rJZ savepointr)r=rVrrrWrJ)rZ db_for_writer lenr7proxyZconcrete_modelZ auto_createdrUrsendrZatomicZ _save_parentsZ _save_table_statedbZaddingr) rrWZ force_insertZ force_updaterJrclsoriginmetaupdatedr!r!r" save_bases2  z!SaveSignalHandlingModel.save_base)N)FFFNN)r%r&r'r+r)rrar-r!r!rr"rTs rTN)%Zdjango.core.exceptionsrZ django.dbrrrZdjango.db.models.functionsrZdjango.db.models.signalsrrZdjango.utils.translationr r,Zmodel_utils.fieldsr r r r rZmodel_utils.managersrrrBZModelrr.r3r@rCZsignalsZclass_preparedconnectr:rHrOrTr!r!r!r"s&