django-model-utils/tests/managers.py

14 lines
375 B
Python
Raw Normal View History

2020-11-29 20:58:00 +00:00
from model_utils.managers import SoftDeletableManager, SoftDeletableQuerySet
class CustomSoftDeleteQuerySet(SoftDeletableQuerySet):
def only_read(self):
return self.filter(is_read=True)
class CustomSoftDeleteManager(SoftDeletableManager):
_queryset_class = CustomSoftDeleteQuerySet
def only_read(self):
return self.get_queryset().only_read()