from DB import UseDB


def recalculation_events_func(difficulty, well):
    difficulties = {
        "Простая": [5, 5],
        "Средняя": [5, 2],
        "Сложная": [5, 1],
        "Проигрывать весело": [10, 1]
    }
    db = UseDB("events")
    not_well_events = db.find_document({"well": False})
    well_events = db.find_document({"well": True})
    if well:
        for i in not_well_events:
            db.update_document({"_id": i["_id"]},
                               {"probability": i["probability"] + difficulties[difficulty][0]})
        ratio = len(not_well_events) * difficulties[difficulty][0] / len(well_events)
        for i in well_events:
            db.update_document({"_id": i["_id"]}, {"probability": i["probability"] - ratio})
    else:
        for i in well_events:
            db.update_document({"_id": i["_id"]},
                               {"probability": i["probability"] + difficulties[difficulty][1]})
        ratio = len(well_events) * difficulties[difficulty][1] / len(not_well_events)
        for i in not_well_events:
            db.update_document({"_id": i["_id"]}, {"probability": i["probability"] - ratio})