correctly reset per-role ratings when recalculating

Author: spookybear0

handlers/admin/index.py

@@ -27,7 +27,7 @@ async def admin_index(request: Request) -> str:
 @admin_only
 async def admin_recalculate_ratings(request: Request) -> str:
     request.app.ctx.banner["text"] = "Rating recalculation in progress, stats may be inaccurate and the site may be slow"
-    request.app.create_task(recalculate_ratings(), name="Recalculate Ratings").add_done_callback(lambda _: reset_banner())
+    request.app.add_task(recalculate_ratings(), name="Recalculate Ratings").add_done_callback(lambda _: reset_banner())
 
     return response.json({"status": "ok"})
 
@@ -36,7 +36,7 @@ async def admin_recalculate_ratings(request: Request) -> str:
 @admin_only
 async def admin_recalculate_sm5_ratings(request: Request) -> str:
     request.app.ctx.banner["text"] = "Rating recalculation in progress, stats may be inaccurate and the site may be slow"
-    request.app.create_task(recalculate_sm5_ratings(), name="Recalculate SM5 Ratings").add_done_callback(lambda _: reset_banner())
+    request.app.add_task(recalculate_sm5_ratings(), name="Recalculate SM5 Ratings").add_done_callback(lambda _: reset_banner())
 
     return response.json({"status": "ok"})
 
@@ -45,7 +45,7 @@ async def admin_recalculate_sm5_ratings(request: Request) -> str:
 @admin_only
 async def admin_recalculate_lb_ratings(request: Request) -> str:
     request.app.ctx.banner["text"] = "Rating recalculation in progress, stats may be inaccurate and the site may be slow"
-    request.app.create_task(recalculate_laserball_ratings(), name="Recalculate Laserball Ratings").add_done_callback(lambda _: reset_banner())
+    request.app.add_task(recalculate_laserball_ratings(), name="Recalculate Laserball Ratings").add_done_callback(lambda _: reset_banner())
 
     return response.json({"status": "ok"})
 

helpers/ratinghelper.py

@@ -610,8 +610,18 @@ async def recalculate_sm5_ratings(*, _sample_size: int=99999) -> None:
 
     # reset sm5 ratings
 
+    logger.info("Resetting sm5 general ratings")
+
     await Player.all().update(sm5_mu=MU, sm5_sigma=SIGMA)
 
+    # reset per-role ratings
+
+    for role in IntRole:
+        if role == IntRole.OTHER:
+            continue
+        logger.info(f"Resetting {str(role).lower()} ratings")
+        await Player.all().update(**{f"{str(role).lower()}_mu": MU, f"{str(role).lower()}_sigma": SIGMA})
+
     # get all games and recalculate ratings
 
     sm5_games = await SM5Game.all().order_by("start_time").limit(_sample_size)