Update elo.py

elo.py

@@ -1,14 +1,10 @@
-import math
+def update_elo_ratings(ratings_dict, winner, loser):
 
-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)
+    # 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)
 
     # Function to determine the K-factor based on games played
     def determine_k_factor(games_played):
@@ -38,19 +34,8 @@ def update_elo_ratings(ratings_dict, winner, loser, category):
     # 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 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
+    # 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}
+
+    return ratings_dict