from rest_framework.views import APIView
from rest_framework.generics import ListAPIView
from rest_framework.response import Response
from rest_framework import status
from rest_framework import generics
from .models import ObjektSigurnosti
from .serializers import ObjektSigurnostiSerializer
from django.shortcuts import get_object_or_404
from rest_framework.pagination import LimitOffsetPagination



class CustomObjektSigurnostiPagination(LimitOffsetPagination):
    default_limit = 20
    

class ObjektSigurnostiList(generics.ListCreateAPIView):
    queryset = ObjektSigurnosti.objects.all().order_by("naziv")
    serializer_class = ObjektSigurnostiSerializer
    pagination_class = CustomObjektSigurnostiPagination
    permission_classes = []


def get_queryset(self):
    user = self.request.user
    return user.accounts.all()
    
def perform_create(self, serializer):       
    serializer.save(operater=self.request.user)

def get_serializer_class(self):
    if self.request.method == "GET":
      return ObjektSigurnostiSerializer
    return self.serializer_class()
    

class ObjektSigurnostiDetail(generics.RetrieveUpdateDestroyAPIView):     
    queryset = ObjektSigurnosti.objects.all()
    serializer_class = ObjektSigurnostiSerializer
    permission_classes = []