from django.shortcuts import render
import requests
import json
from django.http import Http404
from rest_framework.views import APIView
from rest_framework.response import Response
from rest_framework.status import (
    HTTP_400_BAD_REQUEST,
    HTTP_404_NOT_FOUND,
    HTTP_200_OK,
    HTTP_500_INTERNAL_SERVER_ERROR
)
from .serializers import FlightInventorySerializer, AirLineInventorySerializer
from django.core.exceptions import ObjectDoesNotExist
from .models import FlightInventory, AirLineInventory


class FlightInventoryRequest(APIView):

    def get(self, request, format=None):
        try:
            data = FlightInventory.objects.all().order_by('-id')
            serializer = FlightInventorySerializer(data, many=True)
            return Response(serializer.data, status=HTTP_200_OK)
        except ObjectDoesNotExist as e:
            return Response({'error': str(e), "status": False}, status=HTTP_404_NOT_FOUND)
        except Exception as e:
            return Response({'error': str(e), "status": False}, status=HTTP_500_INTERNAL_SERVER_ERROR)

    def post(self, request):
        serializer = FlightInventorySerializer(data=request.data)
        if serializer.is_valid():
            serializer.save()
            return Response(serializer.data, status=HTTP_200_OK)
        return Response(serializer.errors, status=HTTP_400_BAD_REQUEST)


class UpdateFlightInventory(APIView):

    def get_object(self, pk):
        try:
            return FlightInventory.objects.get(pk=pk)
        except FlightInventory.DoesNotExist:
            raise Http404

    def put(self, request, pk, format=None):
        FlightInventory = self.get_object(pk)
        serializer = FlightInventorySerializer(FlightInventory, data=request.data)
        if serializer.is_valid():
            serializer.save()
            return Response(serializer.data, status=HTTP_200_OK)
        return Response(serializer.errors, status=HTTP_400_BAD_REQUEST)


class GetFlightInventory(APIView):

    def get(self, request, pk, format=None):
        try:
            data = FlightInventory.objects.get(pk=pk)
            serializer = FlightInventorySerializer(data, many=False)
            return Response(serializer.data, status=HTTP_200_OK)
        except ObjectDoesNotExist as e:
            return Response({'error': str(e), "status": False}, status=HTTP_404_NOT_FOUND)
        except Exception as e:
            return Response({'error': str(e), "status": False}, status=HTTP_500_INTERNAL_SERVER_ERROR)


class GetCustomerFlightInventory(APIView):

    def get(self, request, **kwargs):
        try:
            fromAirportCode = kwargs['fromAirportCode']
            toAirportCode = kwargs['toAirportCode']
            travelDate = kwargs['travelDate']
            print(toAirportCode)
            print(travelDate)
            print(fromAirportCode)
            data = FlightInventory.objects.all().order_by('-id')
            serializer = FlightInventorySerializer(data, many=True)
            return Response(serializer.data, status=HTTP_200_OK)
        except ObjectDoesNotExist as e:
            return Response({'error': str(e), "status": False}, status=HTTP_404_NOT_FOUND)
        except Exception as e:
            return Response({'error': str(e), "status": False}, status=HTTP_500_INTERNAL_SERVER_ERROR)