add savedpoint views

This commit is contained in:
ThePetrovich 2025-07-01 22:07:12 +08:00
parent 741aefca31
commit c446ac5a1b
4 changed files with 59 additions and 5 deletions

View file

@ -1,9 +1,10 @@
from rest_framework import status, generics, permissions
from rest_framework.response import Response
from rest_framework.views import APIView
from rest_framework.viewsets import ModelViewSet
from django.utils import timezone
from .models import Prediction, User
from .serializers import PredictionSerializer, PredictionRequestSerializer, PredictionListSerializer, PredictionDetailSerializer
from .models import Prediction, User, Satellite, SavedPoint, SavedRateProfile, PreditctionTemplate
from .serializers import PredictionSerializer, PredictionRequestSerializer, PredictionListSerializer, PredictionDetailSerializer, SavedPointSerializer, SavedRateProfileSerializer, PreditctionTemplateSerializer
from rest_framework.permissions import IsAuthenticated
import requests
from django.views.decorators.csrf import csrf_exempt
@ -13,7 +14,7 @@ from .services.tawhiri import TawhiriClient
from django.contrib.auth import get_user_model
from .models import Satellite, TelemetryPacket
from .serializers import TelemetryPacketSerializer
from .permissions import ReadOnlyOrAuthenticated
from .permissions import ReadOnlyOrAuthenticated, IsOwner
import time
from django.http import JsonResponse
from rest_framework.authentication import SessionAuthentication, BasicAuthentication
@ -222,6 +223,17 @@ def logout_view(request):
return JsonResponse({'detail': 'Successfully logged out.'})
class SavedPointViewset(ModelViewSet):
authentication_classes = [SessionAuthentication, BasicAuthentication]
permission_classes = [IsOwner]
serializer_class = SavedPointSerializer
pagination_class = None
def get_queryset(self):
return SavedPoint.objects.filter(user=self.request.user)
def perform_create(self, serializer):
serializer.save(user=self.request.user)
#class PredictionCreateView(APIView):
#permission_classes = [IsAuthenticated]