from django.contrib.auth.models import User, Permission
from rest_framework import viewsets, permissions, filters
import django_filters
from core.models import Client, Project, Entry, Task
from .serializers import (
UserSerializer,
ClientSerializer,
PermissionSerializer,
ProjectSerializer,
EntrySerializer,
TaskSerializer,
)
from .pagination import LimitOffsetPaginationWithTotals
class UserViewSet(viewsets.ModelViewSet):
queryset = User.objects.all()
serializer_class = UserSerializer
pagination_class = None
def get_permissions(self):
# Prevent rest_framework from checking for the "view" perm.
return (permissions.IsAuthenticated(),)
def get_queryset(self):
if self.request.user.is_superuser:
queryset = User.objects.all()
else:
queryset = User.objects.filter(id=self.request.user.id)
return queryset
class PermissionViewSet(viewsets.ModelViewSet):
queryset = Permission.objects.all()
serializer_class = PermissionSerializer
pagination_class = None
def get_permissions(self):
# Prevent rest_framework from checking for the "view" perm.
return (permissions.IsAuthenticated(),)
def get_queryset(self):
if self.request.user.is_superuser:
queryset = Permission.objects.all()
else:
queryset = Permission.objects.filter(user=self.request.user)
return queryset
class ClientViewSet(viewsets.ModelViewSet):
queryset = Client.on_site.filter(archive=False)
serializer_class = ClientSerializer
pagination_class = None
filter_fields = ("id",)
def get_queryset(self):
return Client.on_site.all()
class ProjectViewSet(viewsets.ModelViewSet):
queryset = Project.objects.filter(archive=False)
serializer_class = ProjectSerializer
pagination_class = None
filter_fields = ("id", "client")
class EntryFilter(django_filters.rest_framework.FilterSet):
min_date = django_filters.DateFilter(field_name="date", lookup_expr="gte")
max_date = django_filters.DateFilter(field_name="date", lookup_expr="lte")
class Meta:
model = Entry
fields = ("id", "date", "user", "task", "project", "project__client")
class EntryViewSet(viewsets.ModelViewSet):
queryset = Entry.objects.all()
serializer_class = EntrySerializer
pagination_class = LimitOffsetPaginationWithTotals
filter_class = EntryFilter
filter_backends = (
filters.SearchFilter,
django_filters.rest_framework.DjangoFilterBackend,
filters.OrderingFilter,
)
ordering_fields = (
"date",
"user__username",
"task__name",
"project__name",
"project__client__name",
)
ordering = ("-date", "-id")
search_fields = (
"id",
"date",
"note",
"user__username",
"task__name",
"project__name",
"project__client__name",
)
def get_queryset(self):
return Entry.on_site.all()
class TaskViewSet(viewsets.ModelViewSet):
queryset = Task.objects.all()
serializer_class = TaskSerializer
pagination_class = None
def get_queryset(self):
return Task.on_site.all()
