2i;ddlmZmZmZddlmZddlmZddlm Z m Z m Z m Z m Z mZddlmZGddej"ZGd d eZGd d eZy ))OptionalTupleType)settings)models)DELETED_INVISIBLEDELETED_ONLY_VISIBLEDELETED_VISIBLE FIELD_NAME SOFT_DELETESOFT_DELETE_CASCADE)SafeDeleteQuerysetceZdZUdZeZeed<dZe ed<e Z dde e e ffd ZdZdej"fd Zdej"fd Zdej"fd Zddeej,efffd Zed ZxZS)SafeDeleteManageraDefault manager for the SafeDeleteModel. If _safedelete_visibility == DELETED_VISIBLE_BY_PK, the manager can returns deleted objects if they are accessed by primary key. :attribute _safedelete_visibility: define what happens when you query masked objects. It can be one of ``DELETED_INVISIBLE`` and ``DELETED_VISIBLE_BY_PK``. Defaults to ``DELETED_INVISIBLE``. >>> from safedelete.models import SafeDeleteModel >>> from safedelete.managers import SafeDeleteManager >>> class MyModelManager(SafeDeleteManager): ... _safedelete_visibility = DELETED_VISIBLE_BY_PK ... >>> class MyModel(SafeDeleteModel): ... _safedelete_policy = SOFT_DELETE ... my_field = models.TextField() ... objects = MyModelManager() ... >>> :attribute _queryset_class: define which class for queryset should be used This attribute allows to add custom filters for both deleted and not deleted objects. It is ``SafeDeleteQueryset`` by default. Custom queryset classes should be inherited from ``SafeDeleteQueryset``. _safedelete_visibilitypk_safedelete_visibility_fieldqueryset_classc>tt| |r||_yy)a%Hook for setting custom ``_queryset_class``. Example: class CustomQueryset(models.QuerySet): pass class MyModel(models.Model): my_field = models.TextField() objects = SafeDeleteManager(CustomQuerySet) N)superr__init___queryset_class)selfr __class__s J/var/www/html/acx/venv/lib/python3.12/site-packages/safedelete/managers.pyrzSafeDeleteManager.__init__1s# /1 #1D  c|j|j|j}|j|j_|j |j_|S)N)using)rmodel_dbrqueryr)rquerysets r get_querysetzSafeDeleteManager.get_querysetBsH'' $(('C040K0K-6:6W6W3rreturnc.|jtS)zShow all models including the soft deleted models. .. note:: This is useful for related managers as those don't have access to ``all_objects``. force_visibility)allr rs rall_with_deletedz"SafeDeleteManager.all_with_deletedIsxx,  rc.|jtS)zOnly show the soft deleted models. .. note:: This is useful for related managers as those don't have access to ``deleted_objects``. r')r)r r*s r deleted_onlyzSafeDeleteManager.deleted_onlyTsxx1  rc p|jdd}|j}|||j_|S)aFPass kwargs to ``SafeDeleteQuerySet.all()``. Args: force_visibility: Show deleted models. (default: {None}) .. note:: The ``force_visibility`` argument is meant for related managers when no other managers like ``all_objects`` or ``deleted_objects`` are available. r(N)popr$r"_safedelete_force_visibility)rkwargsr(qss rr)zSafeDeleteManager.all_s="::&8$?     '4DBHH 1 rc d}|jjs0d|vs,|jjjj|vr|j j di|jditdij}|r?|j|jvr#t|td|jd}tt|B|fi|\}}|rt#t$ddrd}||fS)a:See :func:`~django.db.models.Query.update_or_create.`. Change to regular djangoesk function: Regular update_or_create() fails on soft-deleted, existing record with unique constraint on non-id field If object is soft-deleted we don't update-or-create it but reset the deleted field to None. So the object is visible again like a create in any other case. Attention: If the object is "revived" from a soft-deleted state the created return value will still be false because the object is technically not created unless you set SAFE_DELETE_INTERPRET_UNDELETED_OBJECTS_AS_CREATED = True in the django settings. Args: defaults: Dict with defaults to update/create model instance with kwargs: Attributes to lookup model instance with FrNT2SAFE_DELETE_INTERPRET_UNDELETED_OBJECTS_AS_CREATED)r has_unique_fields_metarnamer+filterexcluder first_safedelete_policyget_soft_delete_policiessetattrsaverrupdate_or_creategetattrr)rdefaultsr1revived_soft_deleted_objectdeleted_objectobjcreatedrs rr@z"SafeDeleteManager.update_or_createqs$',# :: ' ' )TV^tzz?O?O?R?R?W?W[a?aM;T224;;EfEMMcQ[]aPbciikN."C"CtGdGdGf"f D9##%.2+.FxZSYZ W '"VX]^GG|rcttgS)z;Returns all states which stand for some kind of soft-delete)r rr5rrr=z*SafeDeleteManager.get_soft_delete_policiess011r)N)__name__ __module__ __qualname____doc__r rint__annotations__rstrrrrrrr$rQuerySetr+r-r)rModelboolr@ staticmethodr= __classcell__)rs@rrrs6#4C3(, #,(O2x5G0H'I2"  &//    foo  v$%5tAS;T%N22rrceZdZdZeZy)SafeDeleteAllManagerzSafeDeleteManager with ``_safedelete_visibility`` set to ``DELETED_VISIBLE``. .. note:: This is used in :py:attr:`safedelete.models.SafeDeleteModel.all_objects`. N)rHrIrJrKr rr5rrrUrUs -rrUceZdZdZeZy)SafeDeleteDeletedManagerzSafeDeleteManager with ``_safedelete_visibility`` set to ``DELETED_ONLY_VISIBLE``. .. note:: This is used in :py:attr:`safedelete.models.SafeDeleteModel.deleted_objects`. N)rHrIrJrKr rr5rrrWrWs 2rrWN)typingrrr django.confr django.dbrconfigr r r r r rr#rManagerrrUrWr5rrr]sM(( )J2J2Z-,-202r