hockey-blast-common-lib 0.1.50__tar.gz → 0.1.51__tar.gz

{hockey_blast_common_lib-0.1.50 → hockey_blast_common_lib-0.1.51}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: hockey-blast-common-lib
-Version: 0.1.50
+Version: 0.1.51
 Summary: Common library for shared functionality and DB models
 Author: Pavel Kletskov
 Author-email: kletskov@gmail.com

hockey_blast_common_lib-0.1.51/hockey_blast_common_lib/aggregate_h2h_stats.py

@@ -0,0 +1,230 @@
+import sys, os
+from datetime import datetime
+
+# Add the package directory to the Python path
+sys.path.append(os.path.dirname(os.path.dirname(os.path.abspath(__file__))))
+
+from hockey_blast_common_lib.models import Game, GameRoster, Goal, Penalty
+from hockey_blast_common_lib.h2h_models import H2HStats, H2HStatsMeta
+from hockey_blast_common_lib.db_connection import create_session
+from sqlalchemy.sql import func
+from sqlalchemy import types
+
+# Max games to process (set to None to process all)
+MAX_GAMES_TO_PROCESS = None  # Set to None to process all games
+
+def aggregate_h2h_stats():
+    session = create_session("boss")
+    meta = None #session.query(H2HStatsMeta).order_by(H2HStatsMeta.id.desc()).first()
+    h2h_stats_dict = {}  # (h1, h2) -> H2HStats instance
+    if meta is None or meta.last_run_timestamp is None or meta.last_processed_game_id is None:
+        # Full run: delete all existing stats and process all games
+        session.query(H2HStats).delete()
+        session.commit()
+        games_query = session.query(Game).order_by(Game.date, Game.time, Game.id)
+        print("No previous run found, deleted all existing H2H stats, processing all games...")
+    else:
+        # Incremental: only process games after last processed
+        # Load all existing stats into memory
+        for stat in session.query(H2HStats).all():
+            h2h_stats_dict[(stat.human1_id, stat.human2_id)] = stat
+        last_game = session.query(Game).filter(Game.id == meta.last_processed_game_id).first()
+        if last_game:
+            last_dt = datetime.combine(last_game.date, last_game.time)
+            games_query = session.query(Game).filter(
+                func.cast(func.concat(Game.date, ' ', Game.time), types.TIMESTAMP()) > last_dt
+            ).order_by(Game.date, Game.time, Game.id)
+            print(f"Resuming from game after id {meta.last_processed_game_id} ({last_dt})...")
+        else:
+            games_query = session.query(Game).order_by(Game.date, Game.time, Game.id)
+            print("Previous game id not found, processing all games...")
+
+    total_games = games_query.count()
+    print(f"Total games to process: {total_games}")
+    processed = 0
+    latest_game_id = None
+    for game in games_query:
+        if MAX_GAMES_TO_PROCESS is not None and processed >= MAX_GAMES_TO_PROCESS:
+            break
+        # --- Gather all relevant data for this game ---
+        # Get all GameRoster entries for this game
+        rosters = session.query(GameRoster).filter(GameRoster.game_id == game.id).all()
+        # Map: team_id -> set of human_ids (players)
+        team_to_players = {}
+        human_roles = {}  # human_id -> set of roles in this game
+        for roster in rosters:
+            team_to_players.setdefault(roster.team_id, set()).add(roster.human_id)
+            human_roles.setdefault(roster.human_id, set()).add(roster.role)
+        # Get all human_ids in this game
+        all_humans = set(human_roles.keys())
+        # Add goalies from Game table (home/visitor)
+        if game.home_goalie_id:
+            all_humans.add(game.home_goalie_id)
+            human_roles.setdefault(game.home_goalie_id, set()).add('G')
+        if game.visitor_goalie_id:
+            all_humans.add(game.visitor_goalie_id)
+            human_roles.setdefault(game.visitor_goalie_id, set()).add('G')
+        # Add referees from Game table (NOT from roster!)
+        if game.referee_1_id:
+            all_humans.add(game.referee_1_id)
+            human_roles.setdefault(game.referee_1_id, set()).add('R')
+        if game.referee_2_id:
+            all_humans.add(game.referee_2_id)
+            human_roles.setdefault(game.referee_2_id, set()).add('R')
+        # --- Build all pairs of humans in this game ---
+        all_humans = list(all_humans)
+        for i in range(len(all_humans)):
+            for j in range(i+1, len(all_humans)):
+                h1, h2 = sorted([all_humans[i], all_humans[j]])
+                key = (h1, h2)
+                h2h = h2h_stats_dict.get(key)
+                if not h2h:
+                    h2h = H2HStats(
+                        human1_id=h1,
+                        human2_id=h2,
+                        first_game_id=game.id,
+                        last_game_id=game.id,
+                        games_together=0,
+                        games_against=0,
+                        games_tied_together=0,
+                        games_tied_against=0,
+                        wins_together=0,
+                        losses_together=0,
+                        h1_wins_vs_h2=0,
+                        h2_wins_vs_h1=0,
+                        games_h1_goalie=0,
+                        games_h2_goalie=0,
+                        games_h1_ref=0,
+                        games_h2_ref=0,
+                        games_both_referees=0,
+                        goals_h1_when_together=0,
+                        goals_h2_when_together=0,
+                        assists_h1_when_together=0,
+                        assists_h2_when_together=0,
+                        penalties_h1_when_together=0,
+                        penalties_h2_when_together=0,
+                        gm_penalties_h1_when_together=0,
+                        gm_penalties_h2_when_together=0,
+                        h1_goalie_h2_scorer_goals=0,
+                        h2_goalie_h1_scorer_goals=0,
+                        shots_faced_h1_goalie_vs_h2=0,
+                        shots_faced_h2_goalie_vs_h1=0,
+                        goals_allowed_h1_goalie_vs_h2=0,
+                        goals_allowed_h2_goalie_vs_h1=0,
+                        save_percentage_h1_goalie_vs_h2=0.0,
+                        save_percentage_h2_goalie_vs_h1=0.0,
+                        h1_ref_h2_player_games=0,
+                        h2_ref_h1_player_games=0,
+                        h1_ref_penalties_on_h2=0,
+                        h2_ref_penalties_on_h1=0,
+                        h1_ref_gm_penalties_on_h2=0,
+                        h2_ref_gm_penalties_on_h1=0,
+                        penalties_given_both_refs=0,
+                        gm_penalties_given_both_refs=0,
+                        h1_shootout_attempts_vs_h2_goalie=0,
+                        h1_shootout_goals_vs_h2_goalie=0,
+                        h2_shootout_attempts_vs_h1_goalie=0,
+                        h2_shootout_goals_vs_h1_goalie=0
+                    )
+                    h2h_stats_dict[key] = h2h
+                # Update first/last game ids
+                if game.id < h2h.first_game_id:
+                    h2h.first_game_id = game.id
+                if game.id > h2h.last_game_id:
+                    h2h.last_game_id = game.id
+                # --- Determine roles and teams ---
+                h1_roles = human_roles.get(h1, set())
+                h2_roles = human_roles.get(h2, set())
+                h1_team = None
+                h2_team = None
+                for team_id, players in team_to_players.items():
+                    if h1 in players:
+                        h1_team = team_id
+                    if h2 in players:
+                        h2_team = team_id
+                # --- General stats ---
+                h2h.games_together += 1  # Both present in this game
+                if h1_team and h2_team:
+                    if h1_team == h2_team:
+                        h2h.wins_together += int(_is_win(game, h1_team))
+                        h2h.losses_together += int(_is_loss(game, h1_team))
+                        if _is_tie(game):
+                            h2h.games_tied_together += 1
+                    else:
+                        h2h.games_against += 1
+                        if _is_win(game, h1_team):
+                            h2h.h1_wins_vs_h2 += 1
+                        if _is_win(game, h2_team):
+                            h2h.h2_wins_vs_h1 += 1
+                        if _is_tie(game):
+                            h2h.games_tied_against += 1
+                # --- Role-specific stats ---
+                if 'G' in h1_roles:
+                    h2h.games_h1_goalie += 1
+                if 'G' in h2_roles:
+                    h2h.games_h2_goalie += 1
+                if 'R' in h1_roles:
+                    h2h.games_h1_ref += 1
+                if 'R' in h2_roles:
+                    h2h.games_h2_ref += 1
+                if 'R' in h1_roles and 'R' in h2_roles:
+                    h2h.games_both_referees += 1
+                # --- Goals, assists, penalties ---
+                # Goals
+                goals = session.query(Goal).filter(Goal.game_id == game.id).all()
+                for goal in goals:
+                    if goal.goal_scorer_id == h1 and (goal.assist_1_id == h2 or goal.assist_2_id == h2):
+                        h2h.goals_h1_when_together += 1
+                    if goal.goal_scorer_id == h2 and (goal.assist_1_id == h1 or goal.assist_2_id == h1):
+                        h2h.goals_h2_when_together += 1
+                # Penalties
+                penalties = session.query(Penalty).filter(Penalty.game_id == game.id).all()
+                for pen in penalties:
+                    if pen.penalized_player_id == h1:
+                        h2h.penalties_h1_when_together += 1
+                        if pen.penalty_minutes and 'GM' in pen.penalty_minutes:
+                            h2h.gm_penalties_h1_when_together += 1
+                    if pen.penalized_player_id == h2:
+                        h2h.penalties_h2_when_together += 1
+                        if pen.penalty_minutes and 'GM' in pen.penalty_minutes:
+                            h2h.gm_penalties_h2_when_together += 1
+                # --- TODO: Add more detailed logic for goalie/skater, referee/player, shootouts, etc. ---
+        latest_game_id = game.id
+        processed += 1
+        if processed % 10 == 0 or processed == total_games:
+            print(f"\rProcessed {processed}/{total_games} games ({(processed/total_games)*100:.2f}%)", end="")
+            sys.stdout.flush()
+    # Commit all stats at once
+    session.query(H2HStats).delete()
+    session.add_all(list(h2h_stats_dict.values()))
+    session.commit()
+    print(f"\rProcessed {processed}/{total_games} games (100.00%)")
+    # Save/update meta
+    meta = H2HStatsMeta(
+        last_run_timestamp=datetime.utcnow(),
+        last_processed_game_id=latest_game_id
+    )
+    session.add(meta)
+    session.commit()
+    print("H2H aggregation complete.")
+
+# --- Helper functions for win/loss/tie ---
+def _is_win(game, team_id):
+    if team_id == game.home_team_id:
+        return (game.home_final_score or 0) > (game.visitor_final_score or 0)
+    if team_id == game.visitor_team_id:
+        return (game.visitor_final_score or 0) > (game.home_final_score or 0)
+    return False
+
+def _is_loss(game, team_id):
+    if team_id == game.home_team_id:
+        return (game.home_final_score or 0) < (game.visitor_final_score or 0)
+    if team_id == game.visitor_team_id:
+        return (game.visitor_final_score or 0) < (game.home_final_score or 0)
+    return False
+
+def _is_tie(game):
+    return (game.home_final_score is not None and game.visitor_final_score is not None and game.home_final_score == game.visitor_final_score)
+
+if __name__ == "__main__":
+    aggregate_h2h_stats()

hockey_blast_common_lib-0.1.51/hockey_blast_common_lib/h2h_models.py

@@ -0,0 +1,77 @@
+from hockey_blast_common_lib.models import db
+
+class H2HStats(db.Model):
+    __tablename__ = 'h2h_stats'
+    id = db.Column(db.Integer, primary_key=True)
+    human1_id = db.Column(db.Integer, db.ForeignKey('humans.id'), nullable=False)
+    human2_id = db.Column(db.Integer, db.ForeignKey('humans.id'), nullable=False)
+    # Always store with human1_id < human2_id for uniqueness
+    __table_args__ = (
+        db.UniqueConstraint('human1_id', 'human2_id', name='_h2h_human_pair_uc'),
+        db.Index('ix_h2h_human_pair', 'human1_id', 'human2_id'),  # Composite index for fast lookup
+    )
+
+    # General
+    games_together = db.Column(db.Integer, default=0, nullable=False)  # Games where both played (any role, any team)
+    games_against = db.Column(db.Integer, default=0, nullable=False)   # Games where both played on opposing teams
+    games_tied_together = db.Column(db.Integer, default=0, nullable=False)
+    games_tied_against = db.Column(db.Integer, default=0, nullable=False)   # Games against each other that ended in a tie
+    wins_together = db.Column(db.Integer, default=0, nullable=False)   # Games both played on same team and won
+    losses_together = db.Column(db.Integer, default=0, nullable=False) # Games both played on same team and lost
+    h1_wins_vs_h2 = db.Column(db.Integer, default=0, nullable=False)   # Games h1's team won vs h2's team
+    h2_wins_vs_h1 = db.Column(db.Integer, default=0, nullable=False)   # Games h2's team won vs h1's team
+
+    # Role-specific counts
+    games_h1_goalie = db.Column(db.Integer, default=0, nullable=False)  # Games where h1 was a goalie and h2 played
+    games_h2_goalie = db.Column(db.Integer, default=0, nullable=False)  # Games where h2 was a goalie and h1 played
+    games_h1_ref = db.Column(db.Integer, default=0, nullable=False)     # Games where h1 was a referee and h2 played
+    games_h2_ref = db.Column(db.Integer, default=0, nullable=False)     # Games where h2 was a referee and h1 played
+    games_both_referees = db.Column(db.Integer, default=0, nullable=False) # Games where both were referees
+
+    # Goals/Assists/Penalties (when both played)
+    goals_h1_when_together = db.Column(db.Integer, default=0, nullable=False)    # Goals by h1 when both played
+    goals_h2_when_together = db.Column(db.Integer, default=0, nullable=False)    # Goals by h2 when both played
+    assists_h1_when_together = db.Column(db.Integer, default=0, nullable=False)  # Assists by h1 when both played
+    assists_h2_when_together = db.Column(db.Integer, default=0, nullable=False)  # Assists by h2 when both played
+    penalties_h1_when_together = db.Column(db.Integer, default=0, nullable=False) # Penalties on h1 when both played
+    penalties_h2_when_together = db.Column(db.Integer, default=0, nullable=False) # Penalties on h2 when both played
+    gm_penalties_h1_when_together = db.Column(db.Integer, default=0, nullable=False) # GM penalties on h1 when both played
+    gm_penalties_h2_when_together = db.Column(db.Integer, default=0, nullable=False) # GM penalties on h2 when both played
+
+    # Goalie/Skater head-to-head (when one is goalie, other is skater on opposing team)
+    h1_goalie_h2_scorer_goals = db.Column(db.Integer, default=0, nullable=False) # Goals scored by h2 against h1 as goalie
+    h2_goalie_h1_scorer_goals = db.Column(db.Integer, default=0, nullable=False) # Goals scored by h1 against h2 as goalie
+    shots_faced_h1_goalie_vs_h2 = db.Column(db.Integer, default=0, nullable=False) # Shots faced by h1 as goalie vs h2 as skater
+    shots_faced_h2_goalie_vs_h1 = db.Column(db.Integer, default=0, nullable=False) # Shots faced by h2 as goalie vs h1 as skater
+    goals_allowed_h1_goalie_vs_h2 = db.Column(db.Integer, default=0, nullable=False) # Goals allowed by h1 as goalie vs h2 as skater
+    goals_allowed_h2_goalie_vs_h1 = db.Column(db.Integer, default=0, nullable=False) # Goals allowed by h2 as goalie vs h1 as skater
+    save_percentage_h1_goalie_vs_h2 = db.Column(db.Float, default=0.0, nullable=False) # Save % by h1 as goalie vs h2 as skater
+    save_percentage_h2_goalie_vs_h1 = db.Column(db.Float, default=0.0, nullable=False) # Save % by h2 as goalie vs h1 as skater
+
+    # Referee/Player
+    h1_ref_h2_player_games = db.Column(db.Integer, default=0, nullable=False)    # Games h1 was referee, h2 was player
+    h2_ref_h1_player_games = db.Column(db.Integer, default=0, nullable=False)    # Games h2 was referee, h1 was player
+    h1_ref_penalties_on_h2 = db.Column(db.Integer, default=0, nullable=False)    # Penalties given by h1 (as ref) to h2
+    h2_ref_penalties_on_h1 = db.Column(db.Integer, default=0, nullable=False)    # Penalties given by h2 (as ref) to h1
+    h1_ref_gm_penalties_on_h2 = db.Column(db.Integer, default=0, nullable=False) # GM penalties given by h1 (as ref) to h2
+    h2_ref_gm_penalties_on_h1 = db.Column(db.Integer, default=0, nullable=False) # GM penalties given by h2 (as ref) to h1
+
+    # Both referees (when both are referees in the same game)
+    penalties_given_both_refs = db.Column(db.Integer, default=0, nullable=False)     # Total penalties given by both
+    gm_penalties_given_both_refs = db.Column(db.Integer, default=0, nullable=False)  # Total GM penalties given by both
+
+    # Shootouts
+    h1_shootout_attempts_vs_h2_goalie = db.Column(db.Integer, default=0, nullable=False) # h1 shootout attempts vs h2 as goalie
+    h1_shootout_goals_vs_h2_goalie = db.Column(db.Integer, default=0, nullable=False)    # h1 shootout goals vs h2 as goalie
+    h2_shootout_attempts_vs_h1_goalie = db.Column(db.Integer, default=0, nullable=False) # h2 shootout attempts vs h1 as goalie
+    h2_shootout_goals_vs_h1_goalie = db.Column(db.Integer, default=0, nullable=False)    # h2 shootout goals vs h1 as goalie
+
+    # First and last game IDs where both were present
+    first_game_id = db.Column(db.Integer, nullable=False)  # Game.id of the first game where both were present
+    last_game_id = db.Column(db.Integer, nullable=False)   # Game.id of the most recent game where both were present
+
+class H2HStatsMeta(db.Model):
+    __tablename__ = 'h2h_stats_meta'
+    id = db.Column(db.Integer, primary_key=True)
+    last_run_timestamp = db.Column(db.DateTime, nullable=True)  # When the h2h stats were last updated
+    last_processed_game_id = db.Column(db.Integer, nullable=True)  # Game.id of the latest processed game

{hockey_blast_common_lib-0.1.50 → hockey_blast_common_lib-0.1.51}/hockey_blast_common_lib/wsgi.py

@@ -6,6 +6,7 @@ sys.path.append(os.path.dirname(os.path.dirname(os.path.abspath(__file__))))
 from flask_migrate import Migrate
 from flask import Flask
 from hockey_blast_common_lib.models import *
+from hockey_blast_common_lib.h2h_models import *
 from hockey_blast_common_lib.stats_models import *
 from hockey_blast_common_lib.stats_models import db
 from hockey_blast_common_lib.db_connection import get_db_params

{hockey_blast_common_lib-0.1.50 → hockey_blast_common_lib-0.1.51}/hockey_blast_common_lib.egg-info/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: hockey-blast-common-lib
-Version: 0.1.50
+Version: 0.1.51
 Summary: Common library for shared functionality and DB models
 Author: Pavel Kletskov
 Author-email: kletskov@gmail.com

{hockey_blast_common_lib-0.1.50 → hockey_blast_common_lib-0.1.51}/hockey_blast_common_lib.egg-info/SOURCES.txt

@@ -4,12 +4,14 @@ setup.py
 hockey_blast_common_lib/__init__.py
 hockey_blast_common_lib/aggregate_all_stats.py
 hockey_blast_common_lib/aggregate_goalie_stats.py
+hockey_blast_common_lib/aggregate_h2h_stats.py
 hockey_blast_common_lib/aggregate_human_stats.py
 hockey_blast_common_lib/aggregate_referee_stats.py
 hockey_blast_common_lib/aggregate_skater_stats.py
 hockey_blast_common_lib/assign_skater_skill.py
 hockey_blast_common_lib/db_connection.py
 hockey_blast_common_lib/dump_sample_db.sh
+hockey_blast_common_lib/h2h_models.py
 hockey_blast_common_lib/hockey_blast_sample_backup.sql.gz
 hockey_blast_common_lib/models.py
 hockey_blast_common_lib/options.py

{hockey_blast_common_lib-0.1.50 → hockey_blast_common_lib-0.1.51}/setup.py

@@ -2,7 +2,7 @@ from setuptools import setup, find_packages
 
 setup(
     name='hockey-blast-common-lib',  # The name of your package
-    version='0.1.50',
+    version='0.1.51',
     description='Common library for shared functionality and DB models',
     author='Pavel Kletskov',
     author_email='kletskov@gmail.com',