#!/usr/bin/python3.3
import json, yaml, schedule
import schedules as bot
from datetime import datetime

path_users = "json/users.json"  # Нужный путь до json файлов
path_schedule = "yaml_groups/{}.yaml"
path_students = "json/students.json"
path_groups = "json/groups.json"

allow_update = True


def save_user(infos):
    for i in range(len(infos)):
        if type(infos[i]) == type(None):
            infos[i] = ""
        infos[i] = str(infos[i])

    infos = {
        "id": infos[0],
        "first_name": infos[1],
        "last_name": infos[2],
        "username": infos[3],
        "group": infos[4],
    }

    data = {}
    with open(path_users, "r", encoding="utf-8") as json_file:
        data = json.load(json_file)

    data[infos["id"]] = {
        "username": infos["username"],
        "first_name": infos["first_name"],
        "last_name": infos["last_name"],
        "group": infos["group"],
        "timeout": "10",
        "allow_message": "yes",
        "thread": "General",
    }

    with open(path_users, "w", encoding="utf-8") as json_file:
        json.dump(data, json_file, ensure_ascii=False, indent=4)

    create_schedule_tasks()

    return


def change_user_param(id, key, value):
    id = str(id)
    data = {}
    with open(path_users, "r", encoding="utf-8") as json_file:
        data = json.load(json_file)

    data[id][key] = value

    with open(path_users, "w", encoding="utf-8") as json_file:
        json.dump(data, json_file, ensure_ascii=False, indent=4)
    create_schedule_tasks()

    # bot.send_admin_message(
    #     "Изменения:\
    #                       \n<i>user:</i> @{}\
    #                       \n<i>id:</i> <code>{}</code>\
    #                       \n<i>key:</i> <code>{}</code>\
    #                       \n<i>value:</i> <code>{}</code>".format(
    #         data[id]["username"], id, key, value
    #     )
    # )

    return


def parse_lesson(lesson: dict[str, str]) -> str:
    text = ""
    start, end = lesson["begin"], lesson["end"]

    if not lesson["name"]:
        return ""

    if len(lesson["teacher.room"]) == 1:
        text = "{}-{} | {}\
                \n<b>{}</b>\
                \n<i>{}</i>".format(
            start,
            end,
            lesson["teacher.room"][0]["r"],
            lesson["name"],
            lesson["teacher.room"][0]["t"],
        )
    else:
        text = "{}-{}\
                \n<b>{}</b>".format(
            start,
            end,
            lesson["name"],
        )

        for tr in lesson["teacher.room"]:
            if not tr["t"] and not tr["r"]:
                continue
            text += "\n({}) <i>{}</i>".format(tr["r"], tr["t"])

    return text


def create_schedule_tasks(manual=False):
    global allow_update

    if not manual:
        if not allow_update:
            return

        # bot.send_admin_message("Произошёл auto-update")

    allow_update = True

    schedule.clear()
    users = {}
    schdl = {}
    with open(path_users, "r", encoding="utf-8") as json_file:
        users = json.load(json_file)
    for id, params in users.items():
        if params["allow_message"] != "yes":
            continue
        group = params["group"]
        if group == "other":
            continue

        with open(path_schedule.format(group), "r", encoding="utf-8") as json_file:
            # schdl = json.load(json_file)["schedule"]
            schdl = yaml.safe_load(json_file)["schedule"]

        for day in schdl:
            for lesson in schdl[day]:
                if not lesson["name"]:  # Неправильный формат schedule
                    continue

                start = lesson["begin"]
                text = parse_lesson(lesson)

                if not text:
                    continue

                thread_id = params["thread"]

                delta = "00:" + params["timeout"]
                format = "%H:%M"
                start_task = datetime.strptime(start, format) - datetime.strptime(
                    delta, format
                )
                tmp = ""
                for i in str(start_task).split(":"):
                    tmp += (("0" + i) if len(i) < 2 else i) + ":"
                start_task = tmp[:-1]

                if day == "mon":
                    schedule.every().monday.at(start_task).do(
                        bot.send_message, id, text, thread_id, lesson["parity"]
                    )
                elif day == "tue":
                    schedule.every().tuesday.at(start_task).do(
                        bot.send_message, id, text, thread_id, lesson["parity"]
                    )
                elif day == "wed":
                    schedule.every().wednesday.at(start_task).do(
                        bot.send_message, id, text, thread_id, lesson["parity"]
                    )
                elif day == "thu":
                    schedule.every().thursday.at(start_task).do(
                        bot.send_message, id, text, thread_id, lesson["parity"]
                    )
                elif day == "fri":
                    schedule.every().friday.at(start_task).do(
                        bot.send_message, id, text, thread_id, lesson["parity"]
                    )
                elif day == "sat":
                    schedule.every().saturday.at(start_task).do(
                        bot.send_message, id, text, thread_id, lesson["parity"]
                    )

    return


def groups_in_json() -> list[str]:
    groups = {}
    with open(path_groups, "r", encoding="utf-8") as json_file:
        groups = json.load(json_file)

    return groups


def students_in_group(group) -> list[str]:
    group = str(group)

    data = {}
    with open(path_users, "r", encoding="utf-8") as json_file:
        data = json.load(json_file)

    students = []

    for i in data.keys():
        if data[i]["group"] == group:
            students.append(i)

    return students


def students_in_json(id: str, key: str = "") -> list[str] | None:
    id = str(id)

    group = {}
    with open(path_users, "r", encoding="utf-8") as json_file:
        group = json.load(json_file)
    group = group[id]["group"]

    stud = {}
    with open(path_students, "r", encoding="utf-8") as json_file:
        stud = json.load(json_file)
    stud = stud.get(group)

    if stud == None:
        return None

    if key == "":
        return list(stud.keys())

    return stud[key]


def return_infos(id) -> dict[str, str] | None:
    id = str(id)
    data = {}
    with open(path_users, "r", encoding="utf-8") as json_file:
        data = json.load(json_file)
    data = data.get(id)

    return data


def pause_bot():
    global allow_update

    if allow_update:
        schedule.clear()
        allow_update = False
        bot.send_admin_message("Бот больше не отправляет расписание")
    else:
        create_schedule_tasks(True)
        allow_update = True
        bot.send_admin_message("Бот возобновил рассылку!")

    return


def get_schedule(id: str, day: str) -> str:
    id = str(id)

    if day == "sun":
        day = "mon"

    days = ["mon", "tue", "wed", "thu", "fri", "sat"]
    russian_days = ["понедельник", "вторник", "среду", "четверг", "пятницу", "субботу"]

    text = "<u>Расписание на {}</u>:\n\n".format(russian_days[days.index(day)])

    schdl_today = {}

    group = ""
    with open(path_users, "r", encoding="utf-8") as user_file:
        group = json.load(user_file)[id]["group"]

    if group == "other":
        return "У тебя не выбрана группа для рассылки сообщений"

    try:
        with open(path_schedule.format(group), "r", encoding="utf-8") as schedule_file:
            try:
                # schdl_today = json.load(schedule_file)["schedule"][day]
                schdl_today = yaml.safe_load(schedule_file)["schedule"][day]
            except:
                return "Расписания твоей группы на {} нет".format(
                    russian_days[days.index(day)]
                )

            for lesson in schdl_today:
                str_lesson = parse_lesson(lesson)
                if str_lesson:
                    text += "•" + str_lesson + "\n\n"
    except:
        return "Расписания твоей группы нет"

    return text


if __name__ == "__main__":
    print(groups_in_json())

    pass