From 46ec97e501d147cfdbadb97ac98acc7e71d2c077 Mon Sep 17 00:00:00 2001
From: Cal Corum <calcorum@users.noreply.github.com>
Date: Mon, 30 Oct 2023 21:18:57 -0500
Subject: [PATCH] Add franchise logic

---
 creation_helpers.py   | 38 ++++++++++++++++++++++++++++++++++++++
 live_series_update.py |  6 +++---
 2 files changed, 41 insertions(+), 3 deletions(-)

diff --git a/creation_helpers.py b/creation_helpers.py
index 1356b10..0bb31c3 100644
--- a/creation_helpers.py
+++ b/creation_helpers.py
@@ -431,6 +431,44 @@ CLUB_LIST = {
     'KCR': 'Kansas City Royals',
     'LAA': 'Los Angeles Angels',
     'LAD': 'Los Angeles Dodgers',
+    'FLA': 'Florida Marlins',
+    'MIA': 'Miami Marlins',
+    'MIL': 'Milwaukee Brewers',
+    'MIN': 'Minnesota Twins',
+    'NYM': 'New York Mets',
+    'NYY': 'New York Yankees',
+    'OAK': 'Oakland Athletics',
+    'PHI': 'Philadelphia Phillies',
+    'PIT': 'Pittsburgh Pirates',
+    'SDP': 'San Diego Padres',
+    'SEA': 'Seattle Mariners',
+    'SFG': 'San Francisco Giants',
+    'STL': 'St Louis Cardinals',
+    'TBR': 'Tampa Bay Rays',
+    'TEX': 'Texas Rangers',
+    'TOR': 'Toronto Blue Jays',
+    'WSN': 'Washington Nationals',
+    'TOT': 'None',
+    '2 Tms': 'None',
+    '3 Tms': 'None',
+    '4 Tms': 'None'
+}
+FRANCHISE_LIST = {
+    'ARI': 'Arizona Diamondbacks',
+    'ATL': 'Atlanta Braves',
+    'BAL': 'Baltimore Orioles',
+    'BOS': 'Boston Red Sox',
+    'CHC': 'Chicago Cubs',
+    'CHW': 'Chicago White Sox',
+    'CIN': 'Cincinnati Reds',
+    'CLE': 'Cleveland Guardians',
+    'COL': 'Colorado Rockies',
+    'DET': 'Detroit Tigers',
+    'HOU': 'Houston Astros',
+    'KCR': 'Kansas City Royals',
+    'LAA': 'Los Angeles Angels',
+    'LAD': 'Los Angeles Dodgers',
+    'FLA': 'Miami Marlins',
     'MIA': 'Miami Marlins',
     'MIL': 'Milwaukee Brewers',
     'MIN': 'Minnesota Twins',
diff --git a/live_series_update.py b/live_series_update.py
index c8fd76e..c331ce9 100644
--- a/live_series_update.py
+++ b/live_series_update.py
@@ -18,7 +18,7 @@ import sys
 
 from creation_helpers import pd_players_df, get_batting_stats, pd_battingcards_df, pd_battingcardratings_df, \
     get_pitching_stats, get_all_pybaseball_ids, pd_pitchingcards_df, pd_pitchingcardratings_df, pd_positions_df, \
-    get_args, mlbteam_and_franchise, CLUB_LIST
+    get_args, mlbteam_and_franchise, CLUB_LIST, FRANCHISE_LIST
 from db_calls import db_get, db_put, db_post, db_patch, DB_URL
 from typing import Literal
 from bs4 import BeautifulSoup
@@ -168,7 +168,7 @@ async def main(args):
             'image': f'{CARD_BASE_URL}/{df_data["player_id"]}/battingcard'
                      f'{urllib.parse.quote("?d=")}{release_directory}',
             'mlbclub': CLUB_LIST[df_data['Tm_vL']],
-            'franchise': CLUB_LIST[df_data['Tm_vL']],
+            'franchise': FRANCHISE_LIST[df_data['Tm_vL']],
             'cardset_id': cardset['id'],
             'set_num': int(float(df_data['key_fangraphs'])),
             'rarity_id': 99,
@@ -606,7 +606,7 @@ async def main(args):
             'image': f'{CARD_BASE_URL}/{df_data["player_id"]}/'
                      f'pitchingcard{urllib.parse.quote("?d=")}{release_directory}',
             'mlbclub': CLUB_LIST[df_data['Tm_vL']],
-            'franchise': CLUB_LIST[df_data['Tm_vL']],
+            'franchise': FRANCHISE_LIST[df_data['Tm_vL']],
             'cardset_id': cardset['id'],
             'set_num': int(float(df_data['key_fangraphs'])),
             'rarity_id': 99,