Added pre-commit-hook

Run ruff lint and formatter pre-commit.

Author: Kjeldgaard

.pre-commit-config.yaml

@@ -0,0 +1,10 @@
+repos:
+- repo: https://github.com/astral-sh/ruff-pre-commit
+  # Ruff version.
+  rev: v0.7.1
+  hooks:
+    # Run the linter.
+    - id: ruff
+      args: [ --fix ]
+    # Run the formatter.
+    - id: ruff-format

fantasy_stats.py

@@ -35,17 +35,19 @@ def __init__(
         self.logger = logger
         self.year = year
 
-        self.logger.info(f"Getting League data: Started")
+        self.logger.info("Getting League data: Started")
         self.league = League(league_id=league_id, year=year, espn_s2=espn_s2, swid=swid)
-        self.logger.info(f"Getting League data: Done")
+        self.logger.info("Getting League data: Done")
 
         self.team_map = self._get_team_map()
 
         self.finished_weeks = self._get_finished_weeks()
 
         self.logger.info(f"Finished week: {self.finished_weeks}")
         self.logger.info(f"Current week: {self.league.current_week}")
-        self.logger.info(f"Regular season weeks: {self.league.settings.reg_season_count}")
+        self.logger.info(
+            f"Regular season weeks: {self.league.settings.reg_season_count}"
+        )
 
         self.player_ids = self._get_player_ids()
 
@@ -57,28 +59,28 @@ def __init__(
 
         self.players = self._get_all_player_scoring()
 
-        self.logger.info(f"Fantasy stats init: Done")
+        self.logger.info("Fantasy stats init: Done")
 
     def _get_team_owner(self):
         self._team_owner_map = {}
         for member in self.league.members:
-            id = member.get('id')
+            id = member.get("id")
             name = f"{member.get('firstName')} {member.get('lastName')}"
             self._team_owner_map[id] = name
 
     def _get_team_map(self):
-        self.logger.info(f"Generating team id mapping: Started")
+        self.logger.info("Generating team id mapping: Started")
         team_map = {team.team_id: team.team_name for team in self.league.teams}
-        self.logger.info(f"Generating team id mapping: Done")
+        self.logger.info("Generating team id mapping: Done")
         return team_map
 
     def _get_player_ids(self):
-        self.logger.info(f"Getting player ids: Started")
+        self.logger.info("Getting player ids: Started")
         pro_players = self.league.espn_request.get_pro_players()
         player_ids = []
         for player in pro_players:
             player_ids.append(player.get("id"))
-        self.logger.info(f"Getting player ids: Done")
+        self.logger.info("Getting player ids: Done")
         return player_ids
 
     def print_player_injuries(self):
@@ -106,7 +108,7 @@ def print_team_scoring(self):
         return points_for, points_against
 
     def _get_games(self):
-        self.logger.info(f"Getting games: Started")
+        self.logger.info("Getting games: Started")
         games = []
         for week in range(
             1, min(self.league.settings.reg_season_count + 1, self.league.current_week)
@@ -154,7 +156,7 @@ def _get_games(self):
                 "Score diff",
             ],
         )
-        self.logger.info(f"Getting games: Done")
+        self.logger.info("Getting games: Done")
         return df
 
     def _get_games_played(self, player) -> int:
@@ -169,10 +171,10 @@ def _get_games_played(self, player) -> int:
         return games_played
 
     def _had_bye_week(self, player) -> bool:
-        return (self.finished_weeks > len(player.stats.items()) - 1)
+        return self.finished_weeks > len(player.stats.items()) - 1
 
     def _get_draft_class(self):
-        self.logger.info(f"Getting draft class: Started")
+        self.logger.info("Getting draft class: Started")
         draft_ids = []
         for pick in self.league.draft:
             draft_ids.append(pick.playerId)
@@ -208,12 +210,12 @@ def _get_draft_class(self):
                 "Games Missed",
             ],
         )
-        self.logger.info(f"Getting draft class: Done")
+        self.logger.info("Getting draft class: Done")
         return df
 
     def _get_player_scoring(self, player_ids: list):
         players = self.league.player_info(playerId=player_ids)
-        if players == None:
+        if players is None:
             return []
 
         players_stats = []
@@ -237,7 +239,7 @@ def _get_player_scoring(self, player_ids: list):
         return players_stats
 
     def _get_all_player_scoring(self):
-        self.logger.info(f"Getting player score data: Started")
+        self.logger.info("Getting player score data: Started")
         players_stats = []
         num_lists = int(
             (len(self.player_ids) + self.players_per_call - 1) / self.players_per_call
@@ -258,7 +260,7 @@ def _get_all_player_scoring(self):
                 "Diff",
             ],
         )
-        self.logger.info(f"Getting player score data: Done")
+        self.logger.info("Getting player score data: Done")
         return df
 
     def _get_player_score(self, player_stats) -> int:
@@ -449,7 +451,6 @@ def get_perfect_record(self) -> str:
                     losers_actual.append(game.home_team.team_name)
                     winners_actual.append(game.away_team.team_name)
 
-
         standings = []
         for team in self.league.teams:
             wins_perfect = winners_perfect.count(team.team_name)
@@ -485,7 +486,7 @@ def get_league_overview(self):
             team_info.append(team.points_for)
             team_info.append(team.points_against)
             team_info.append(team.acquisitions)
-            owners = [self._team_owner_map[owner.get('id')] for owner in team.owners]
+            owners = [self._team_owner_map[owner.get("id")] for owner in team.owners]
             team_info.append(", ".join(owners))
             league_overview.extend([team_info])
 

main.py

@@ -1,6 +1,5 @@
 #!/usr/bin/env python3
 
-from email.policy import default
 import sys
 import argparse
 from pathlib import Path
@@ -9,7 +8,6 @@
 from datetime import datetime
 import logging
 import json
-from fantasy_stats import FantasyStats
 
 
 def setup_logger():
@@ -127,9 +125,9 @@ def main(
 
     with open(outputfile, "w") as f:
         f.write(output)
-        logger.info(f"Write to output file")
+        logger.info("Write to output file")
 
-    logger.info(f"Script completed")
+    logger.info("Script completed")
 
 
 if __name__ == "__main__":

pyproject.toml

@@ -10,4 +10,5 @@ dependencies = [
     "natsort>=8.4.0",
     "numpy>=2.1.1",
     "pandas>=2.2.3",
+    "pre-commit>=4.0.1",
 ]