baseballcommand 0.0.3__tar.gz → 0.0.5__tar.gz

{baseballcommand-0.0.3 → baseballcommand-0.0.5}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: baseballcommand
-Version: 0.0.3
+Version: 0.0.5
 Summary: This is a tool that will retrieve Major League Baseball scores and stats in your command line.
 License-Expression: MIT
 Author: Owen Powell

{baseballcommand-0.0.3 → baseballcommand-0.0.5}/pyproject.toml

@@ -4,7 +4,7 @@ build-backend = "poetry.core.masonry.api"
 
 [project]
 name = "baseballcommand"
-version = "0.0.3"
+version = "0.0.5"
 description = "This is a tool that will retrieve Major League Baseball scores and stats in your command line."
 authors = [
     {name = "Owen Powell",email = "opowell2899@gmail.com"}
@@ -21,5 +21,6 @@ base = "baseballcommand:cli" # CLI entry point
 
 [dependency-groups]
 dev = [
-    "ruff (>=0.16.5,<0.17.0)"
+    "ruff (>=0.16.5,<0.17.0)",
+    "pytest (>=9.2.0,<10.0.0)"
 ]

baseballcommand-0.0.5/src/baseballcommand/commands/players.py

@@ -0,0 +1,136 @@
+import difflib
+import re
+import unicodedata
+from datetime import datetime
+from zoneinfo import ZoneInfo
+
+import requests
+
+BASE_URL = 'https://statsapi.mlb.com/api/v1'
+
+HITTING_COLUMNS = [
+    ('G', 'gamesPlayed'), ('AB', 'atBats'), ('R', 'runs'), ('H', 'hits'), ('2B', 'doubles'),
+    ('3B', 'triples'), ('HR', 'homeRuns'), ('RBI', 'rbi'), ('SB', 'stolenBases'), ('BB', 'baseOnBalls'),
+    ('SO', 'strikeOuts'), ('AVG', 'avg'), ('OBP', 'obp'), ('SLG', 'slg'), ('OPS', 'ops'),
+]
+PITCHING_COLUMNS = [
+    ('G', 'gamesPitched'), ('GS', 'gamesStarted'), ('W', 'wins'), ('L', 'losses'), ('SV', 'saves'),
+    ('IP', 'inningsPitched'), ('H', 'hits'), ('ER', 'earnedRuns'), ('BB', 'baseOnBalls'),
+    ('SO', 'strikeOuts'), ('HR', 'homeRuns'), ('ERA', 'era'), ('WHIP', 'whip'),
+]
+
+
+def normalize_name(name):
+    # Strip accents (Acuña -> acuna), punctuation, and suffixes so user input matches API names
+    name = unicodedata.normalize('NFKD', name).encode('ascii', 'ignore').decode()
+    name = re.sub(r'[^a-z ]', '', name.lower().replace('-', ' '))
+    words = [word for word in name.split() if word not in ('jr', 'sr', 'ii', 'iii', 'iv')]
+    return ' '.join(words)
+
+
+def get_active_players(season):
+    response = requests.get(f'{BASE_URL}/sports/1/players?season={season}')
+    response.raise_for_status()
+    return response.json().get('people', [])
+
+
+def player_names(player):
+    # Both the full name and the name the player goes by (e.g. "Mike" vs "Michael")
+    return {normalize_name(player['fullName']),
+            normalize_name(f'{player.get("useName", "")} {player.get("lastName", "")}')}
+
+
+def find_players(name, players):
+    target = normalize_name(name)
+    names = {}
+    for player in players:
+        for key in player_names(player):
+            names.setdefault(key, []).append(player)
+
+    if target in names:
+        return names[target]
+
+    close = difflib.get_close_matches(target, names.keys(), n=5, cutoff=0.75)
+    matches = []
+    for key in close:
+        for player in names[key]:
+            if player not in matches:
+                matches.append(player)
+    return matches
+
+
+def choose_player(matches):
+    print('Multiple players found:')
+    for i, player in enumerate(matches, start=1):
+        print(f'  {i}. {player["fullName"]} ({player["primaryPosition"]["abbreviation"]})')
+
+    try:
+        choice = int(input('Select a player: '))
+    except (ValueError, EOFError):
+        return None
+    return matches[choice - 1] if 1 <= choice <= len(matches) else None
+
+
+def get_season_stats(player_id, group, season):
+    response = requests.get(f'{BASE_URL}/people/{player_id}/stats?stats=season&group={group}&season={season}')
+    response.raise_for_status()
+    stats = response.json().get('stats', [])
+    if not stats or not stats[0].get('splits'):
+        return None
+    # Players traded mid-season have one split per team plus a combined split without a team
+    splits = stats[0]['splits']
+    combined = [split for split in splits if 'team' not in split]
+    return (combined or splits)[-1]['stat']
+
+
+def print_stats(stat, columns):
+    widths = [max(len(label), len(str(stat.get(key, '-')))) for label, key in columns]
+    print(' | '.join(label.rjust(width) for (label, _), width in zip(columns, widths)))
+    print('-+-'.join('-' * width for width in widths))
+    print(' | '.join(str(stat.get(key, '-')).rjust(width) for (_, key), width in zip(columns, widths)))
+
+
+def player_search(first_name, last_name):
+    season = datetime.now(ZoneInfo('America/New_York')).year
+    name = f'{first_name} {last_name}'
+
+    try:
+        players = get_active_players(season)
+        matches = find_players(name, players)
+    except requests.RequestException as e:
+        print(f'Error searching for player: {e}')
+        return
+
+    if not matches:
+        print(f'No active MLB player found matching "{name}".')
+        return
+
+    player = matches[0] if len(matches) == 1 else choose_player(matches)
+    if player is None:
+        print('No player selected.')
+        return
+
+    if normalize_name(name) not in player_names(player):
+        print(f'Showing results for {player["fullName"]}')
+
+    position = player['primaryPosition']
+    if position['abbreviation'] == 'TWP':
+        groups = ['hitting', 'pitching']
+    elif position['type'] == 'Pitcher':
+        groups = ['pitching']
+    else:
+        groups = ['hitting']
+
+    print(f'\n{player["fullName"]} ({position["abbreviation"]}) - {season} Season')
+    for group in groups:
+        try:
+            stat = get_season_stats(player['id'], group, season)
+        except requests.RequestException as e:
+            print(f'Error retrieving stats: {e}')
+            return
+
+        print(f'\n{group.capitalize()}')
+        if stat is None:
+            print(f'No {group} stats for {season}.')
+        else:
+            print_stats(stat, HITTING_COLUMNS if group == 'hitting' else PITCHING_COLUMNS)

baseballcommand-0.0.3/src/baseballcommand/commands/players.py

@@ -1,41 +0,0 @@
-
-import requests
-
-
-# TODO: improve name matching. Example: searching for Pete Crowe-Armstrong returns Pete Alonso
-# solution: check name match on JSON return?
-def player_search(first_name, last_name):
-    url = (f'https://statsapi.mlb.com/api/v1/people/search?names={first_name.lower()}%20'
-           f'{last_name.lower()}'
-           f'&active=True'
-           f'&sportIDs=1'
-           f'&limit=1')
-
-    response = requests.get(url)
-
-    if response.status_code == 200:
-        data = response.json()
-
-        # print(json.dumps(data, sort_keys=True, indent=4))
-
-        player = data["people"][0]
-        player_id = player["id"]
-
-        # TODO: change URL if player is a pitcher
-        stats_url = f'https://statsapi.mlb.com/api/v1/people/{player_id}/stats?stats=season&group=hitting'
-
-        stats_response = requests.get(stats_url)
-
-        if stats_response.status_code == 200:
-            # stats_data = stats_response.json()
-
-            # print(json.dumps(stats_data, sort_keys=True, indent=4))
-
-            # stat = stats_data["stats"][0]["splits"][0]["stat"]
-            print(f'{"Games Played | ":14}{"At Bats | ":9}{"Hits | ":6}{"Runs | ":6}{"RBI | ":5}{"Home Runs | ":11}'
-                  f'{"Doubles | ":9}{"Triples | ":9}{"AVG | ":5}{"OBP | ":5}{"OPS | ":5}{"SLG | ":5}')
-            print('-' * 100)
-
-    else:
-        print(f'Response: {response.status_code}')
-        return