Я хотел бы расширить метод _filter_or_exclude в своем менеджере, но по какой-то причине этот метод никогда не вызывается.
Чего я хочу добиться, так это неявно фильтровать модели, в которых поле не удалено = 0 (со всеми get()
и filter()
и all()
и exclude()
и, возможно, другими методами)
Мои модели
from django.db.models import Model, Manager, Q, QuerySet
class BaseManager(Manager):
def _filter_or_exclude(self, *args, get_deleted=False, **kwargs):
clone = super()._filter_or_exclude(*args, **kwargs)
if not get_deleted:
clone.query.add(Q(deleted=0))
return clone
# Just for sure, I extended the method on a querySet as well
class BaseQuerySet(QuerySet):
def _filter_or_exclude(self, *args, get_deleted=False, **kwargs):
clone = super()._filter_or_exclude(*args, **kwargs)
if not get_deleted:
clone.query.add(Q(deleted=0))
return clone
class BaseModel(Model):
id = models.UUIDField(primary_key=True, default=uuid.uuid4, editable=False)
deleted = models.IntegerField(default=0)
...
objects = BaseManager.from_queryset(BaseQuerySet)
class Meta:
abstract = True
def delete(self, from_db=False, using=None, keep_parents=False):
if from_db:
super().delete(using, keep_parents)
return
self.deleted = 1
self.save()
class Operation(BaseModel):
...
Теперь, когда я бегу
Operation.objects.get(id = "...")
Расширенный метод _filter_or_exclude
вызывается не в кастомном менеджере, а в дефолтном.
Что я делаю неправильно?
Вы можете переопределить метод get_queryset
для менеджера. И чтобы добавить все экземпляры, включая удаленные, вы можете добавить еще один метод. так.
class BaseManager(Manager):
def get_queryset(self):
qs = super().get_queryset()
qs = qs.filter(deleted=1) # or whatever your condition is
return qs
def all_objects(self):
return super().get_queryset()
И использовать это так.
Model.objects.get() # filtered results
Model.objects.all_objects.get() # include deleted ones.
Это не отвечает на мой вопрос, но делает именно то, что мне нужно! Спасибо, сэр!