Update elo.py

elo.py

@@ -1,11 +1,14 @@
 import math
 
-def update_elo_ratings(ratings_dict, winner, loser):
-    # Extract old ratings and games played
-    winner_old_rating = ratings_dict.get(winner, {}).get('elo_rating', 1200)
-    loser_old_rating = ratings_dict.get(loser, {}).get('elo_rating', 1200)
-    winner_games_played = ratings_dict.get(winner, {}).get('games_played', 0)
-    loser_games_played = ratings_dict.get(loser, {}).get('games_played', 0)
+def update_elo_ratings(ratings_dict, winner, loser, category):
+    # Extract old ratings and games played for the specific category
+    winner_category_data = ratings_dict.get(winner, {}).get(category, {})
+    loser_category_data = ratings_dict.get(loser, {}).get(category, {})
+
+    winner_old_rating = winner_category_data.get('elo_rating', 1200)
+    loser_old_rating = loser_category_data.get('elo_rating', 1200)
+    winner_games_played = winner_category_data.get('games_played', 0)
+    loser_games_played = loser_category_data.get('games_played', 0)
 
     # Function to determine the K-factor based on games played
     def determine_k_factor(games_played):
@@ -35,8 +38,19 @@ def update_elo_ratings(ratings_dict, winner, loser):
     # Calculate new ratings
     winner_new_rating, loser_new_rating = elo(winner_old_rating, loser_old_rating, k_factor_winner=winner_k_factor, k_factor_loser=loser_k_factor)
 
-    # Update ratings and games played in the dictionary
-    ratings_dict[winner] = {'elo_rating': winner_new_rating, 'games_played': winner_games_played + 1}
-    ratings_dict[loser] = {'elo_rating': loser_new_rating, 'games_played': loser_games_played + 1}
+    # Update ratings and games played in the dictionary for the specific category
+    if winner not in ratings_dict:
+        ratings_dict[winner] = {}
+    if category not in ratings_dict[winner]:
+        ratings_dict[winner][category] = {}
+    ratings_dict[winner][category]['elo_rating'] = winner_new_rating
+    ratings_dict[winner][category]['games_played'] = winner_games_played + 1
+
+    if loser not in ratings_dict:
+        ratings_dict[loser] = {}
+    if category not in ratings_dict[loser]:
+        ratings_dict[loser][category] = {}
+    ratings_dict[loser][category]['elo_rating'] = loser_new_rating
+    ratings_dict[loser][category]['games_played'] = loser_games_played + 1
 
     return ratings_dict