This commit is contained in:
Ross
2022-04-02 22:34:14 +01:00
parent 8498b17678
commit 1646bac2cb
5 changed files with 87 additions and 8 deletions
+31
View File
@@ -46,6 +46,37 @@ def get_series(request):
return queryset
class CaseCollectionFilter(django_filters.FilterSet):
author = django_filters.ModelMultipleChoiceFilter(
queryset=get_authors, null_label="No author"
)
cases = django_filters.ModelMultipleChoiceFilter(
queryset=get_cases, null_label="No cases"
)
class Meta:
model = Case
fields = (
"archive",
"author",
) # "site",
def __init__(
self,
data=None,
queryset=None,
prefix=None,
strict=None,
user=None,
request=None,
):
if not request.user.groups.filter(name="atlas_editor").exists():
queryset = queryset.prefetch_related("author").filter(author__id=request.user.id)
super(CaseCollectionFilter, self).__init__(
data=data, queryset=queryset, prefix=prefix, request=request
)
pass
class CaseFilter(django_filters.FilterSet):
author = django_filters.ModelMultipleChoiceFilter(
queryset=get_authors, null_label="No author"