62 lines
2.7 KiB
Python
62 lines
2.7 KiB
Python
from modules.DB import UseDB
|
|
from loguru import logger
|
|
import random
|
|
import configparser
|
|
|
|
|
|
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):
|
|
difficulties = {
|
|
"Простая": [5, 5],
|
|
"Средняя": [5, 2],
|
|
"Сложная": [5, 1],
|
|
"Проигрывать весело": [10, 1]
|
|
}
|
|
config = configparser.ConfigParser()
|
|
config.read("conf/config.ini")
|
|
difficulty = config['Settings']['difficulty']
|
|
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})
|
|
|
|
|
|
def start_calculating():
|
|
logger.debug("Валидация началась")
|
|
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.debug("Валидация игры успешно прошла")
|