diff --git a/backend/time_tracking/urls.py b/backend/time_tracking/urls.py index 83e7ac4..8bde29b 100644 --- a/backend/time_tracking/urls.py +++ b/backend/time_tracking/urls.py @@ -7,7 +7,7 @@ from .views import ( TimeEntryUpdateView, WeeklySummaryView, ) - +from .views import DebugUsersView urlpatterns = [ path('entries/', TimeEntryListView.as_view(), name='time-entry-list'), path('entries/create/', TimeEntryCreateView.as_view(), name='time-entry-create'), diff --git a/backend/time_tracking/views.py b/backend/time_tracking/views.py index b23d3b6..e4fec6c 100644 --- a/backend/time_tracking/views.py +++ b/backend/time_tracking/views.py @@ -2,7 +2,7 @@ from rest_framework import generics, status from rest_framework.response import Response from rest_framework.permissions import IsAuthenticated from rest_framework.views import APIView -from django.contrib.auth.models import User +from django.contrib.auth import get_user_model from django.db import transaction from django.utils import timezone from datetime import timedelta @@ -16,6 +16,8 @@ from .serializers import ( WeeklySummarySerializer ) +User = get_user_model() + class TimeEntryListView(generics.ListAPIView): serializer_class = TimeEntrySerializer permission_classes = [IsAuthenticated]