import threading
import time
import datetime
from .models import DailyRewardsList, DailyReward, User

def calculate_reward(day):
    if day == 1:
        return 20
    elif day == 2:
        return 50
    elif day == 3:
        return 100
    elif day == 4:
        return 150
    elif day == 5:
        return 200
    elif day == 6:
        return 250
    elif day == 7:
        return 300
    elif day == 8:
        return 350
    elif day == 9:
        return 400
    elif day == 10:
        return 500
    else:
        return 500 + (day - 10) * 100

def my_scheduled_job():
    # Удаляем старые записи списка наград
    DailyRewardsList.objects.all().delete()

    # Определяем максимальное количество дней в текущем месяце
    now = datetime.datetime.now()
    next_month = now.month + 1 if now.month < 12 else 1
    next_year = now.year if now.month < 12 else now.year + 1
    max_day = (datetime.date(next_year, next_month, 1) - datetime.date(now.year, now.month, 1)).days

    # Создаем новый DailyRewardsList
    month_max_day = f'До {max_day}'
    new_daily_rewards_list = DailyRewardsList.objects.create(active_day=1, month_max_day=month_max_day)

    # Очищаем предыдущие награды
    DailyReward.objects.all().delete()

    # Создаем новые ежедневные награды для каждого пользователя
    users = User.objects.all()
    for user in users:
        for day in range(1, max_day + 1):
            reward_amount = calculate_reward(day)
            DailyReward.objects.create(user=user, day=day, reward_amount=reward_amount)

def run_scheduler():
    while True:
        now = datetime.datetime.now()
        # Проверяем, что текущее время 00:05
        
        if now.day == 1 and now.hour == 21 and now.minute == 5:
            my_scheduled_job()
            # Ждем 60 секунд, чтобы избежать повторного выполнения задачи в эту же минуту
            time.sleep(60)

def start_scheduler():
    thread = threading.Thread(target=run_scheduler, daemon=True)
    thread.start()