21 lines
623 B
Python
21 lines
623 B
Python
from typing import Any, List, Optional
|
|
|
|
from django.db.models import Manager, QuerySet
|
|
|
|
|
|
class BaseManager(Manager):
|
|
"""
|
|
A base manager for all REST-models.
|
|
Provides a base implementation for `get_prefetched_queryset` and
|
|
allows filtering of the queryset by ids.
|
|
"""
|
|
|
|
def get_queryset(self, ids: Optional[List[int]] = None) -> QuerySet:
|
|
queryset = super().get_queryset()
|
|
if ids:
|
|
queryset = queryset.filter(pk__in=ids)
|
|
return queryset
|
|
|
|
def get_prefetched_queryset(self, *args: Any, **kwargs: Any) -> QuerySet:
|
|
return self.get_queryset(*args, **kwargs)
|