elo.py

import math

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):
        # Define K-factor based on number of games played. Adjust these thresholds as needed.
        if games_played < 30:
            return 40
        elif games_played < 100:
            return 20
        else:
            return 10

    # Determine K-factors
    winner_k_factor = determine_k_factor(winner_games_played)
    loser_k_factor = determine_k_factor(loser_games_played)

    def elo(winner_rating, loser_rating, k_factor_winner=32, k_factor_loser=32):
        # Calculate the expected scores for each player
        winner_expected = 1 / (1 + 10 ** ((loser_rating - winner_rating) / 400))
        loser_expected = 1 / (1 + 10 ** ((winner_rating - loser_rating) / 400))

        # Calculate the new ratings for each player
        winner_new_rating = winner_rating + k_factor_winner * (1 - winner_expected)
        loser_new_rating = loser_rating + k_factor_loser * (0 - loser_expected)

        return winner_new_rating, loser_new_rating

    # 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