From 050bd99488ac66f9e490548ec9fa8e3fde47cbc4 Mon Sep 17 00:00:00 2001 From: Jylpah Date: Fri, 10 Sep 2021 21:03:41 +0300 Subject: [PATCH] get_filter_categories(): fix, add room_type=5: Quick Tournament(?) --- analyze_wotb_replays.py | 5 ++++- 1 file changed, 4 insertions(+), 1 deletion(-) diff --git a/analyze_wotb_replays.py b/analyze_wotb_replays.py index 9fae5b5..04f71c1 100755 --- a/analyze_wotb_replays.py +++ b/analyze_wotb_replays.py @@ -167,6 +167,7 @@ class BattleCategorizationList(): 'Regular' : 1, 'Training Room' : 2, 'Tournament' : 4, + 'Quick Tournament' : 5, # maybe? 'Rating' : 7, 'Mad Games' : 8, 'Realistic Battles' : 22, @@ -416,7 +417,7 @@ class BattleCategorizationList(): def get_filter_categories(self, cat: str, filters: list) -> dict: try: - return self.categorizations[cat].get_filter_categories(filters) + return self.categorizations[str(cat)].get_filter_categories(filters) except Exception as err: bu.error(exception=err) return None @@ -1577,6 +1578,8 @@ async def main(argv): players = get_players(results) replays = filter_replays(replays, results) + + (player_stats, stat_id_map) = await process_player_stats(players, OPT_WORKERS_N, args, db) bu.debug('Number of player stats: ' + str(len(player_stats))) teamresults = calc_team_stats(results, player_stats, stat_id_map, args)