def add_new_event():
    name = input("Введите название ивента\n")
    description = input("Введите описание ивента\n")
    try:
        well = input("Этот ивент хороший?\n")
        if well == "True":
            well = True
        else:
            well = False
    except Exception as e:
        return e
    variants = input("Введите варианты развития ивентов, разделённые */!/*\n").split("*/!/*")
    consequence = input("Введите функции которые выполнятся при варианте ивента, "
                        "разделённые */!/*\n").split("*/!/*")
    probability_consequence = [int(i) for i in input("Введите вероятности вариантов ивентов,"
                                                     " разделённые */!/*\n").split("*/!/*")]
    if len(variants) == len(consequence) == len(probability_consequence):
        response = {
            "name": name,
            "description": description,
            "variants": variants,
            "consequence": consequence,
            "probability_consequence": probability_consequence,
            "probability": 50,
            "well": well
        }
        from modules.DB import UseDB
        db = UseDB("events")
        db.insert_document(response)
        return response
    else:
        return "Длины variants, consequence и probability_consequence не равны"


def main():
    while True:
        answer = input("Что сегодня сделаем?\n")
        if answer == "добавить эвент":
            print(add_new_event())
        else:
            print("Ничего не понятно")
            quit(0)


if __name__ == '__main__':
    main()