from modules.DB import UseDB
from loguru import logger
import random


def get_random_event():
    db = UseDB("events")
    response = db.find_document({})
    random_event = random.choices([i["_id"] for i in response], weights=[i["probability"] for i in response])
    return db.find_document({"_id": random_event[0]})[0]


def recalculation_events(well):
    if well:
        db = UseDB("events")
        not_well_events = db.find_document({"well": False})
        for i in not_well_events:
            db.update_document({"_id": i["_id"]}, {"probability": i["probability"] + 5})
        well_events = db.find_document({"well": True})
        cooficent = len(not_well_events) * 5 / len(well_events)
        for i in well_events:
            db.update_document({"_id": i["_id"]}, {"probability": i["probability"] - cooficent})
    else:
        db = UseDB("events")
        well_events = db.find_document({"well": True})
        for i in well_events:
            db.update_document({"_id": i["_id"]}, {"probability": i["probability"] + 5})
        not_well_events = db.find_document({"well": False})
        cooficent = len(well_events) * 5 / len(not_well_events)
        for i in not_well_events:
            db.update_document({"_id": i["_id"]}, {"probability": i["probability"] - cooficent})


def start_calculating():
    logger.remove()
    logger.add("conf/log.log", level="INFO")
    logger.info("Валидация началась")
    db = UseDB("events")
    response = db.find_document({})
    sum_percent_probability = sum([i["probability"] for i in response])
    sum_percent_probability_in_events = sum([sum(i["probability_consequence"]) for i in response])
    if sum_percent_probability != 100:
        logger.critical("Сумма вероятностей всех ивентов не равна 100, это плохо, не надо так")
        quit(0)
    if sum_percent_probability_in_events != len(response) * 100:
        logger.critical("Проблема в каком-то ивенте, сумма вероятностей не правильная")
        quit(0)
    for i in response:
        if not (len(i["variants"]) == len(i["consequence"]) == len(i["probability_consequence"])):
            logger.critical(f"Ошибка в ивенте {i['name']}. Разница в длинах variants, consequence "
                            f"и probability_consequence")
            quit(0)
    logger.info("Валидация игры успешно прошла")