28 lines
1.2 KiB
Python
28 lines
1.2 KiB
Python
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})
|