from datetime import datetime, timedelta

from django.db.models import Min
from django.contrib.auth import get_user_model
from django.views.generic import TemplateView
from django.utils.timezone import now

from apps.course.models import Course
from apps.school.models import SchoolSchedule
from apps.payment.models import SchoolPayment

User = get_user_model()


class IndexView(TemplateView):
    template_name = 'templates/lilcity/main.html'

    def get_context_data(self):
        context = super().get_context_data()

        now_time = now()
        try:
            school_schedule = SchoolSchedule.objects.get(weekday=now_time.isoweekday())
        except SchoolSchedule.DoesNotExist:
            online = False
        else:
            end_at = datetime.combine(now_time.today(), school_schedule.start_at)
            online = (
                school_schedule.start_at <= now_time.time() and
                (end_at + timedelta(hours=2)).time() >= now_time.time() and
                school_schedule.current_live_lesson()
            )
        date_now = now_time.date()
        if self.request.user.is_authenticated:
            school_payment = SchoolPayment.objects.filter(
                user=self.request.user,
                date_start__lte=date_now,
                date_end__gte=date_now
            )
            school_payment_exists = school_payment.exists()
            school_schedules_purchased = school_payment.values_list('weekdays', flat=True)
            school_schedules_purchased = school_schedules_purchased[0] if school_schedules_purchased else []
        else:
            school_payment_exists = False
            school_schedules_purchased = []

        context.update({
            'online': online,
            'course_items': Course.objects.filter(status=Course.PUBLISHED)[:6],
            'is_purchased': school_payment_exists,
            'min_school_price': SchoolSchedule.objects.aggregate(Min('month_price'))['month_price__min'],
            'school_schedules': SchoolSchedule.objects.all(),
            'school_schedules_purchased': school_schedules_purchased,
            'teachers': User.objects.filter(role=User.TEACHER_ROLE, show_in_mainpage=True),
            'subscription_ends': school_payment.filter(add_days=False).first().date_end if school_payment_exists else None,
            'subscription_ends_humanize': school_payment.filter(add_days=False).first().date_end_humanize if school_payment_exists else None,
        })
        return context


class SchoolSchedulesView(TemplateView):
    template_name = 'templates/lilcity/school_schedules.html'

    def get_context_data(self):
        context = super().get_context_data()
        context['school_schedules'] = SchoolSchedule.objects.all()
        return context