Ross
32 lines
1.4 KiB
Python
Executable File
32 lines
1.4 KiB
Python
Executable File
import django_filters
|
|
|
|
from .models import Rapid, CidUserAnswer
|
|
|
|
|
|
class RapidFilter(django_filters.FilterSet):
|
|
class Meta:
|
|
model = Rapid
|
|
fields = ("normal", "abnormality", "exams", "region", "examination",
|
|
"laterality",
|
|
#"site",
|
|
"created_date", "open_access", "author")
|
|
|
|
def __init__(self, data=None, queryset=None, prefix=None, strict=None, user=None, request=None):
|
|
if not request.user.groups.filter(name="rapid_checker").exists():
|
|
queryset = queryset.filter(open_access=True) | queryset.filter(author__id=request.user.id)
|
|
super(RapidFilter, self).__init__(data=data, queryset=queryset, prefix=prefix, request=request)
|
|
pass
|
|
|
|
class RapidQuestionFilter(django_filters.FilterSet):
|
|
class Meta:
|
|
model = RapidQuestion
|
|
fields = ("question_type", "exams",
|
|
"examination",
|
|
"modality", "region",
|
|
"normal", "created_date", "open_access", "author")
|
|
|
|
def __init__(self, data=None, queryset=None, prefix=None, strict=None, user=None, request=None):
|
|
if not request.user.groups.filter(name="rapid_checker").exists():
|
|
queryset = queryset.filter(open_access=True) | queryset.filter(author__id=request.user.id)
|
|
super(RapidQuestionFilter, self).__init__(data=data, queryset=queryset, prefix=prefix, request=request)
|
|
pass |