import pytz
from datetime import datetime, time
from django.db.models import Sum, Max
from rest_framework.generics import (
    ListAPIView,
    RetrieveAPIView,
    CreateAPIView
)
from rest_framework.views import APIView
from rest_framework.response import Response

from rest_framework.permissions import IsAuthenticated
from rest_framework.filters import SearchFilter, OrderingFilter
from mosquito.api.pagination import PostLimitOffsetPagination, PostPageNumberPagination
from .serializers import (
    DeviceCountSerializer,
    DeviceInfoSerializer,
)
from ..models import DeviceCount, DeviceInfo


class DeviceLogListAPIView(ListAPIView):
    serializer_class = DeviceCountSerializer
    permission_classes = [IsAuthenticated]
    filter_backends = [SearchFilter, OrderingFilter]
    pagination_class = PostPageNumberPagination
    search_fields = ['device_id']
    queryset = DeviceCount.objects.get_queryset().order_by('-data_time')


class DeviceInfoListAPIView(ListAPIView):
    serializer_class = DeviceInfoSerializer
    permission_classes = [IsAuthenticated]
    filter_backends = [SearchFilter, OrderingFilter]
    pagination_class = PostPageNumberPagination
    search_fields = ['device_id', 'device_name', 'chip_type', 'chip_id']
    queryset = DeviceInfo.objects.all()


class DeviceLogStatisticAPIView(APIView):
    permission_classes = [IsAuthenticated]

    def get(self, request, *args, **kwargs):
        tz = pytz.timezone("UTC")
        today = datetime.now(tz).date()
        midnight = tz.localize(datetime.combine(today, time(0, 0)), is_dst=None)

        # calc total count
        total_count_queryset = [DeviceCount.objects.get(device_id=x['device_id'], data_time=x['max_time'])
                                for x in DeviceCount.objects.values('device_id').annotate(max_time=Max('data_time'))]
        if total_count_queryset:
            total_count = sum(map(lambda x: int(x.count), total_count_queryset))
        else:
            total_count = 0

        # calc daily count
        one_day_ago_queryset = [DeviceCount.objects.get(device_id=x['device_id'], data_time=x['max_time'])
                                for x in DeviceCount.objects.filter(
                                        data_time__lt=midnight).values('device_id').annotate(max_time=Max('data_time'))]
        if one_day_ago_queryset:
            one_day_ago_count = sum(map(lambda x: int(x.count), one_day_ago_queryset))
            daily_count = total_count - one_day_ago_count
        else:
            daily_count = 0
        data = {'total_count': total_count, 'daily_count': daily_count}
        return Response(data)


class DeviceInfoStatisticAPIView(APIView):
    permission_classes = [IsAuthenticated]

    def get(self, request, *args, **kwargs):
        online_count = DeviceInfo.objects.filter(online=1).count()
        offline_count = DeviceInfo.objects.filter(online=0).count()
        data = {'online_count': online_count, 'offline_count': offline_count}
        return Response(data)