from datetime import datetime, date, time
from driver.models import Driver
from earning.models import Plan
from tariff.models import City, CitySubZone
from settlements.models import LoginIncentiveSchem

def get_login_incentive_obj(city, plan, slot_start_time, slot_end_time, slot_name, per_minute_pay):
    login_incentive = LoginIncentiveScheme()
    login_incentive.city = city
    login_incentive.slot_name = slot_name
    login_incentive.minimum_checkin_minutes = 0
    login_incentive.per_minute_pay = per_minute_pay
    login_incentive.slot_start_time = slot_start_time
    login_incentive.slot_end_time = slot_end_time
    login_incentive.driver_plan = driver_plan
    login_incentive.incentive_type = "checkin"
    login_incentive.is_active = True
    login_incentive.valid_from = date(2017, 4, 25)
    login_incentive.valid_till = date(2017, 5, 31)
    login_incentive.save()
    return login_incentive

def create_checkin_incentives():
    city = City.objects.get(id=3)
    plan = Plan.objects.get(type="flexible", city=city)
    #Centeral South Mumbai 00:00am to 02:00am
    login_incentive = get_login_incentive_obj(city, plan, time(0, 0, 1), time(1, 59, 59), "CS Mumbai 00:00 hrs to 02:00 hrs", 90)
    zones_list = ["Centeral Mumbai", "South Mumbai"]
    zones_id_list = list(CitySubZone.objects.filter(zone_name__in=zones_list).values_list('id', flat=True))
    login_incentive.zone.add(*zones_id_list)
    #West Mumbai 00:00am to 02:00am
    login_incentive = get_login_incentive_obj(city, plan, time(0, 0, 1), time(1, 59, 59), "West Mumbai 00:00 hrs to 02:00 hrs", 30)
    zones_list = ["West Mumbai"]
    zones_id_list = list(CitySubZone.objects.filter(zone_name__in=zones_list).values_list('id', flat=True))
    login_incentive.zone.add(*zones_id_list)
    #06:00am to 08:00am
    login_incentive = get_login_incentive_obj(city, plan, time(6, 0, 1), time(7, 59, 59), "CSWHL Mumbai 06:00 hrs to 08:00 hrs", 60)
    zones_list = ["Centeral Mumbai", "South Mumbai", "West Mumbai", "Harbour Line Mumbai"]
    zones_id_list = list(CitySubZone.objects.filter(zone_name__in=zones_list).values_list('id', flat=True))
    login_incentive.zone.add(*zones_id_list)
    #08:00am to 11:00am
    login_incentive = get_login_incentive_obj(city, plan, time(8, 0, 1), time(10, 59, 59), "CS Mumbai 08:00 hrs to 11:00 hrs", 30)
    zones_list = ["Centeral Mumbai", "South Mumbai"]
    zones_id_list = list(CitySubZone.objects.filter(zone_name__in=zones_list).values_list('id', flat=True))
    login_incentive.zone.add(*zones_id_list)
    #11:00am to 08:00pm
    login_incentive = get_login_incentive_obj(city, plan, time(11, 0, 1), time(19, 59, 59), "CS Mumbai 11:00 hrs to 20:00 hrs", 30)
    zones_list = ["Centeral Mumbai", "South Mumbai"]
    zones_id_list = list(CitySubZone.objects.filter(zone_name__in=zones_list).values_list('id', flat=True))
    login_incentive.zone.add(*zones_id_list)
    #08:00pm to 00:00am
    login_incentive = get_login_incentive_obj(city, plan, time(20, 0, 1), time(23, 59, 59), "CS Mumbai 20:00 hrs to 00:00 hrs", 60)
    zones_list = ["Centeral Mumbai", "South Mumbai"]
    zones_id_list = list(CitySubZone.objects.filter(zone_name__in=zones_list).values_list('id', flat=True))
    login_incentive.zone.add(*zones_id_list)
    #08:00pm to 00:00am
    login_incentive = get_login_incentive_obj(city, plan, time(20, 0, 1), time(23, 59, 59), "West Mumbai 20:00 hrs to 00:00 hrs", 30)
    zones_list = ["West Mumbai"]
    zones_id_list = list(CitySubZone.objects.filter(zone_name__in=zones_list).values_list('id', flat=True))
    login_incentive.zone.add(*zones_id_list)