hockey-blast-common-lib 0.1.63__py3-none-any.whl → 0.1.65__py3-none-any.whl

hockey_blast_common_lib/options.py

@@ -1,4 +1,5 @@
 import argparse
+
 from .models import Human
 
 MAX_HUMAN_SEARCH_RESULTS = 25
@@ -7,38 +8,52 @@ MIN_GAMES_FOR_ORG_STATS = 1
 MIN_GAMES_FOR_DIVISION_STATS = 1
 MIN_GAMES_FOR_LEVEL_STATS = 1
 
-orgs = {'caha', 'sharksice', 'tvice'}
+orgs = {"caha", "sharksice", "tvice"}
+
 
 def parse_args():
-    parser = argparse.ArgumentParser(description="Process data for a specific organization.")
-    parser.add_argument("org", choices=orgs, help="The organization to process (e.g., 'caha', 'sharksice', 'tvice').")
-    parser.add_argument("--reprocess", action="store_true", help="Reprocess existing data.")
-    parser.add_argument("--pre_process", action="store_true", help="Pre-Process existing data.")
+    parser = argparse.ArgumentParser(
+        description="Process data for a specific organization."
+    )
+    parser.add_argument(
+        "org",
+        choices=orgs,
+        help="The organization to process (e.g., 'caha', 'sharksice', 'tvice').",
+    )
+    parser.add_argument(
+        "--reprocess", action="store_true", help="Reprocess existing data."
+    )
+    parser.add_argument(
+        "--pre_process", action="store_true", help="Pre-Process existing data."
+    )
     return parser.parse_args()
 
+
 def get_or_create_empty_net_human(session, game_date):
     """Get or create the special 'Empty Net' human record for tracking pulled goalies.
-    
+
     Args:
         session: Database session
         game_date: Date of the game (used for first_date/last_date if creating)
-    
+
     Returns:
         int: The human_id of the Empty Net special record
     """
-    empty_net_human = session.query(Human).filter_by(
-        first_name="Empty", middle_name="", last_name="Net"
-    ).first()
-    
+    empty_net_human = (
+        session.query(Human)
+        .filter_by(first_name="Empty", middle_name="", last_name="Net")
+        .first()
+    )
+
     if not empty_net_human:
         empty_net_human = Human(
-            first_name="Empty", 
-            middle_name="", 
+            first_name="Empty",
+            middle_name="",
             last_name="Net",
             first_date=game_date,
-            last_date=game_date
+            last_date=game_date,
         )
         session.add(empty_net_human)
         session.commit()
-    
+
     return empty_net_human.id

hockey_blast_common_lib/progress_utils.py

@@ -1,5 +1,5 @@
 import time
-from datetime import datetime, timedelta
+
 
 class ProgressTracker:
     """
@@ -14,7 +14,7 @@ class ProgressTracker:
         self.processed_items = 0
         self.last_update_time = self.start_time
         self.custom_counters = custom_counters if custom_counters else {}
-        
+
     def update(self, processed_count=None, **kwargs):
         """
         Update progress. If processed_count is None, increment by 1.
@@ -33,23 +33,26 @@ class ProgressTracker:
         current_time = time.time()
 
         # Only update display if at least 0.1 seconds have passed (to avoid spamming)
-        if current_time - self.last_update_time >= 0.1 or self.processed_items == self.total_items:
+        if (
+            current_time - self.last_update_time >= 0.1
+            or self.processed_items == self.total_items
+        ):
             self._display_progress()
             self.last_update_time = current_time
-            
+
     def _display_progress(self):
         """
         Display progress with percentage, ETA, and elapsed time.
         """
         current_time = time.time()
         elapsed_time = current_time - self.start_time
-        
+
         if self.processed_items == 0:
             percentage = 0.0
             eta_str = "calculating..."
         else:
             percentage = (self.processed_items / self.total_items) * 100
-            
+
             # Calculate ETA
             if self.processed_items == self.total_items:
                 eta_str = "completed!"
@@ -58,7 +61,7 @@ class ProgressTracker:
                 remaining_items = self.total_items - self.processed_items
                 eta_seconds = avg_time_per_item * remaining_items
                 eta_str = self._format_time(eta_seconds)
-        
+
         elapsed_str = self._format_time(elapsed_time)
 
         progress_msg = f"\r{self.description}: {self.processed_items}/{self.total_items} ({percentage:.1f}%) | "
@@ -66,15 +69,17 @@ class ProgressTracker:
 
         # Add custom counters to the display
         if self.custom_counters:
-            counter_parts = [f"{key}: {value}" for key, value in self.custom_counters.items()]
+            counter_parts = [
+                f"{key}: {value}" for key, value in self.custom_counters.items()
+            ]
             progress_msg += " | " + ", ".join(counter_parts)
 
         print(progress_msg, end="", flush=True)
-        
+
         # Add newline when complete
         if self.processed_items == self.total_items:
             print()  # Newline to finish the progress line
-    
+
     def _format_time(self, seconds):
         """
         Format seconds into a human-readable string.
@@ -89,7 +94,7 @@ class ProgressTracker:
             hours = int(seconds // 3600)
             minutes = int((seconds % 3600) // 60)
             return f"{hours}h {minutes}m"
-    
+
     def finish(self):
         """
         Mark progress as complete and add final newline.
@@ -97,9 +102,12 @@ class ProgressTracker:
         self.processed_items = self.total_items
         self._display_progress()
 
-def create_progress_tracker(total_items, description="Processing", custom_counters=None):
+
+def create_progress_tracker(
+    total_items, description="Processing", custom_counters=None
+):
     """
     Factory function to create a progress tracker.
     custom_counters: dict of counter_name: initial_value for additional metrics to track
     """
-    return ProgressTracker(total_items, description, custom_counters)
+    return ProgressTracker(total_items, description, custom_counters)

hockey_blast_common_lib/skills_in_divisions.py

@@ -1,20 +1,31 @@
-import sys
 import os
+import sys
 from collections import defaultdict
 
 # Add the project root 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, Division, Level, Season, League
 from hockey_blast_common_lib.db_connection import create_session
+from hockey_blast_common_lib.models import Division, Game, League, Level, Season
 from hockey_blast_common_lib.utils import get_fake_level
 
+
 def analyze_levels(org):
     session = create_session(org)
 
     # Query to get games and their divisions for season_number 33 and 35 with league_number 1
-    games_season_33 = session.query(Game.home_team_id, Game.visitor_team_id, Division.level).join(Division, Game.division_id == Division.id).filter(Division.season_number == 33, Division.league_number == 1).all()
-    games_season_35 = session.query(Game.home_team_id, Game.visitor_team_id, Division.level).join(Division, Game.division_id == Division.id).filter(Division.season_number == 35, Division.league_number == 1).all()
+    games_season_33 = (
+        session.query(Game.home_team_id, Game.visitor_team_id, Division.level)
+        .join(Division, Game.division_id == Division.id)
+        .filter(Division.season_number == 33, Division.league_number == 1)
+        .all()
+    )
+    games_season_35 = (
+        session.query(Game.home_team_id, Game.visitor_team_id, Division.level)
+        .join(Division, Game.division_id == Division.id)
+        .filter(Division.season_number == 35, Division.league_number == 1)
+        .all()
+    )
 
     # Dictionary to store levels for each team by season
     team_levels_season_33 = defaultdict(set)
@@ -43,32 +54,131 @@ def analyze_levels(org):
     for new_level in sorted(level_connections.keys()):
         connections = level_connections[new_level]
         connections_list = sorted(connections.items(), key=lambda x: x[0])
-        connections_str = ", ".join([f"{old_level}: {count}" for old_level, count in connections_list])
+        connections_str = ", ".join(
+            [f"{old_level}: {count}" for old_level, count in connections_list]
+        )
         print(f"{new_level}: {connections_str}")
 
     session.close()
 
+
 def fill_seed_levels():
     session = create_session("boss")
 
     # List of Skill objects based on the provided comments
     levels = [
-        Level(is_seed=True, org_id=1, skill_value=10.0, level_name='Adult Division 1', level_alternative_name='Senior A'),
-        Level(is_seed=True, org_id=1, skill_value=20.0, level_name='Adult Division 2', level_alternative_name='Senior B'),
-        Level(is_seed=True, org_id=1, skill_value=30.0, level_name='Adult Division 3A', level_alternative_name='Senior BB'),
-        Level(is_seed=True, org_id=1, skill_value=35.0, level_name='Adult Division 3B', level_alternative_name='Senior C'),
-        Level(is_seed=True, org_id=1, skill_value=40.0, level_name='Adult Division 4A', level_alternative_name='Senior CC'),
-        Level(is_seed=True, org_id=1, skill_value=45.0, level_name='Adult Division 4B', level_alternative_name='Senior CCC,Senior CCCC'),
-        Level(is_seed=True, org_id=1, skill_value=50.0, level_name='Adult Division 5A', level_alternative_name='Senior D,Senior DD'),
-        Level(is_seed=True, org_id=1, skill_value=55.0, level_name='Adult Division 5B', level_alternative_name='Senior DDD'),
-        Level(is_seed=True, org_id=1, skill_value=60.0, level_name='Adult Division 6A', level_alternative_name='Senior DDDD'),
-        Level(is_seed=True, org_id=1, skill_value=65.0, level_name='Adult Division 6B', level_alternative_name='Senior DDDDD'),
-        Level(is_seed=True, org_id=1, skill_value=70.0, level_name='Adult Division 7A', level_alternative_name='Senior E'),
-        Level(is_seed=True, org_id=1, skill_value=75.0, level_name='Adult Division 7B', level_alternative_name='Senior EE'),
-        Level(is_seed=True, org_id=1, skill_value=80.0, level_name='Adult Division 8', level_alternative_name='Senior EEE'),
-        Level(is_seed=True, org_id=1, skill_value=80.0, level_name='Adult Division 8A', level_alternative_name='Senior EEE'),
-        Level(is_seed=True, org_id=1, skill_value=85.0, level_name='Adult Division 8B', level_alternative_name='Senior EEEE'),
-        Level(is_seed=True, org_id=1, skill_value=90.0, level_name='Adult Division 9', level_alternative_name='Senior EEEEE')
+        Level(
+            is_seed=True,
+            org_id=1,
+            skill_value=10.0,
+            level_name="Adult Division 1",
+            level_alternative_name="Senior A",
+        ),
+        Level(
+            is_seed=True,
+            org_id=1,
+            skill_value=20.0,
+            level_name="Adult Division 2",
+            level_alternative_name="Senior B",
+        ),
+        Level(
+            is_seed=True,
+            org_id=1,
+            skill_value=30.0,
+            level_name="Adult Division 3A",
+            level_alternative_name="Senior BB",
+        ),
+        Level(
+            is_seed=True,
+            org_id=1,
+            skill_value=35.0,
+            level_name="Adult Division 3B",
+            level_alternative_name="Senior C",
+        ),
+        Level(
+            is_seed=True,
+            org_id=1,
+            skill_value=40.0,
+            level_name="Adult Division 4A",
+            level_alternative_name="Senior CC",
+        ),
+        Level(
+            is_seed=True,
+            org_id=1,
+            skill_value=45.0,
+            level_name="Adult Division 4B",
+            level_alternative_name="Senior CCC,Senior CCCC",
+        ),
+        Level(
+            is_seed=True,
+            org_id=1,
+            skill_value=50.0,
+            level_name="Adult Division 5A",
+            level_alternative_name="Senior D,Senior DD",
+        ),
+        Level(
+            is_seed=True,
+            org_id=1,
+            skill_value=55.0,
+            level_name="Adult Division 5B",
+            level_alternative_name="Senior DDD",
+        ),
+        Level(
+            is_seed=True,
+            org_id=1,
+            skill_value=60.0,
+            level_name="Adult Division 6A",
+            level_alternative_name="Senior DDDD",
+        ),
+        Level(
+            is_seed=True,
+            org_id=1,
+            skill_value=65.0,
+            level_name="Adult Division 6B",
+            level_alternative_name="Senior DDDDD",
+        ),
+        Level(
+            is_seed=True,
+            org_id=1,
+            skill_value=70.0,
+            level_name="Adult Division 7A",
+            level_alternative_name="Senior E",
+        ),
+        Level(
+            is_seed=True,
+            org_id=1,
+            skill_value=75.0,
+            level_name="Adult Division 7B",
+            level_alternative_name="Senior EE",
+        ),
+        Level(
+            is_seed=True,
+            org_id=1,
+            skill_value=80.0,
+            level_name="Adult Division 8",
+            level_alternative_name="Senior EEE",
+        ),
+        Level(
+            is_seed=True,
+            org_id=1,
+            skill_value=80.0,
+            level_name="Adult Division 8A",
+            level_alternative_name="Senior EEE",
+        ),
+        Level(
+            is_seed=True,
+            org_id=1,
+            skill_value=85.0,
+            level_name="Adult Division 8B",
+            level_alternative_name="Senior EEEE",
+        ),
+        Level(
+            is_seed=True,
+            org_id=1,
+            skill_value=90.0,
+            level_name="Adult Division 9",
+            level_alternative_name="Senior EEEEE",
+        ),
     ]
 
     for skill in levels:
@@ -77,6 +187,7 @@ def fill_seed_levels():
 
     print("Seed skills have been populated into the database.")
 
+
 def assign_fake_level_to_divisions(session, fake_level):
     # Assign the special fake Skill to every existing Division
     divisions = session.query(Division).all()
@@ -85,45 +196,71 @@ def assign_fake_level_to_divisions(session, fake_level):
     session.commit()
     print("Assigned special fake Skill to all Division records.")
 
+
 def delete_all_levels():
     session = create_session("boss")
     fake_level = get_fake_level(session)
     assign_fake_level_to_divisions(session, fake_level)
     # Delete all Skill records except the fake skill
-    session.query(Level).filter(Level.id != fake_level.id).delete(synchronize_session=False)
+    session.query(Level).filter(Level.id != fake_level.id).delete(
+        synchronize_session=False
+    )
     session.commit()
     print("All Skill records except the fake skill have been deleted.")
 
+
 def populate_season_ids():
     session = create_session("boss")
     divisions = session.query(Division).all()
     for division in divisions:
         # Find the Season record that matches the season_number
-        season = session.query(Season).filter_by(season_number=division.season_number, org_id=division.org_id, league_number=division.league_number).first()
+        season = (
+            session.query(Season)
+            .filter_by(
+                season_number=division.season_number,
+                org_id=division.org_id,
+                league_number=division.league_number,
+            )
+            .first()
+        )
         if season:
             division.season_id = season.id
-            print(f"Assigned season_id {season.id} for Division with season_number {division.season_number}")
+            print(
+                f"Assigned season_id {season.id} for Division with season_number {division.season_number}"
+            )
         else:
-            print(f"Season not found for Division with season_number {division.season_number}")
+            print(
+                f"Season not found for Division with season_number {division.season_number}"
+            )
     session.commit()
     print("Season IDs have been populated into the Division table.")
 
+
 def populate_league_ids():
     session = create_session("boss")
     seasons = session.query(Season).all()
     for season in seasons:
         # Find the League record that matches the league_number and org_id
-        league = session.query(League).filter_by(league_number=season.league_number, org_id=season.org_id).first()
+        league = (
+            session.query(League)
+            .filter_by(league_number=season.league_number, org_id=season.org_id)
+            .first()
+        )
         if league:
             season.league_id = league.id
-            print(f"Assigned league_id {league.id} for Season with league_number {season.league_number}")
+            print(
+                f"Assigned league_id {league.id} for Season with league_number {season.league_number}"
+            )
         else:
-            print(f"League not found for Season with league_number {season.league_number}")
+            print(
+                f"League not found for Season with league_number {season.league_number}"
+            )
     session.commit()
     print("League IDs have been populated into the Season table.")
 
-#if __name__ == "__main__":
-    # delete_all_levels()
-    #fill_seed_levels()
-    #populate_season_ids()  # Call the function to populate season_ids
-    #populate_league_ids()  # Call the new function to populate league_ids
+
+# if __name__ == "__main__":
+# delete_all_levels()
+# fill_seed_levels()
+# populate_season_ids()  # Call the function to populate season_ids
+# populate_league_ids()  # Call the new function to populate league_ids