53 lines
1.5 KiB
Python
53 lines
1.5 KiB
Python
# -*- coding: utf-8 -*-
|
|
|
|
from django.db.models import Q
|
|
|
|
from rest_framework import filters
|
|
|
|
|
|
class QueryParamsFilterMixin(object):
|
|
_special_values_dict = {
|
|
'true': True,
|
|
'false': False,
|
|
'null': None,
|
|
}
|
|
|
|
def filter_queryset(self, request, queryset, view):
|
|
query_params = {}
|
|
|
|
if not hasattr(view, "filter_fields"):
|
|
return queryset
|
|
|
|
for field_name in view.filter_fields:
|
|
if field_name in request.QUERY_PARAMS:
|
|
field_data = request.QUERY_PARAMS[field_name]
|
|
if field_data in self._special_values_dict:
|
|
query_params[field_name] = self._special_values_dict[field_data]
|
|
else:
|
|
query_params[field_name] = field_data
|
|
|
|
if query_params:
|
|
queryset = queryset.filter(**query_params)
|
|
|
|
return queryset
|
|
|
|
|
|
class FilterBackend(QueryParamsFilterMixin,
|
|
filters.BaseFilterBackend):
|
|
"""
|
|
Default filter backend.
|
|
"""
|
|
pass
|
|
|
|
|
|
class IsProjectMemberFilterBackend(FilterBackend):
|
|
def filter_queryset(self, request, queryset, view):
|
|
queryset = super(IsProjectMemberFilterBackend, self).filter_queryset(
|
|
request, queryset, view)
|
|
user = request.user
|
|
|
|
if user.is_authenticated():
|
|
queryset = queryset.filter(Q(project__members=request.user) |
|
|
Q(project__owner=request.user))
|
|
return queryset.distinct()
|