blob: e4712cd71246f506816b6a7a6de388781587f97f (
plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
|
from django.db import models
class TrashableQuerySet(models.query.QuerySet):
"""Special QuerySet that explicitly calls delete() method on each model.
If we don't do this, QuerySet delete() will bypass our TrashableMixin
delete().
This is only used for the NonTrashedManager. TrashedManager returns
trashed objects, which should be truly deleted when delete() is called.
Logic is the same for TrashableMixin at second delete().
@author: Charl P. Botha <cpbotha@timescapers.com>
"""
def delete(self, trash=True):
"""Overrides normal QuerySet delete to call explicitly object's
delete() method.
"""
if trash:
for m in self:
m.delete()
else:
super(TrashableQuerySet, self).delete()
class NonTrashedManager(models.Manager):
''' Query only objects which have not been trashed. '''
def get_query_set(self):
# class Manager instantiates QuerySet() at every call of
# get_query_set() also. see django/db/models/manager.py
# we use special TrashableQuerySet which makes sure that
# model delete()s get called explicitly.
query_set = TrashableQuerySet(self.model, using=self._db)
return query_set.filter(trashed_at__isnull=True)
class TrashedManager(models.Manager):
''' Query only objects which have been trashed. '''
def get_query_set(self):
query_set = super(TrashedManager, self).get_query_set()
return query_set.filter(trashed_at__isnull=False)
|
