Add franchise logic

This commit is contained in:
Cal Corum 2023-10-30 21:18:57 -05:00
parent 3e8a73ba92
commit 46ec97e501
2 changed files with 41 additions and 3 deletions

View File

@ -431,6 +431,44 @@ CLUB_LIST = {
'KCR': 'Kansas City Royals', 'KCR': 'Kansas City Royals',
'LAA': 'Los Angeles Angels', 'LAA': 'Los Angeles Angels',
'LAD': 'Los Angeles Dodgers', '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', 'MIA': 'Miami Marlins',
'MIL': 'Milwaukee Brewers', 'MIL': 'Milwaukee Brewers',
'MIN': 'Minnesota Twins', 'MIN': 'Minnesota Twins',

View File

@ -18,7 +18,7 @@ import sys
from creation_helpers import pd_players_df, get_batting_stats, pd_battingcards_df, pd_battingcardratings_df, \ 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_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 db_calls import db_get, db_put, db_post, db_patch, DB_URL
from typing import Literal from typing import Literal
from bs4 import BeautifulSoup from bs4 import BeautifulSoup
@ -168,7 +168,7 @@ async def main(args):
'image': f'{CARD_BASE_URL}/{df_data["player_id"]}/battingcard' 'image': f'{CARD_BASE_URL}/{df_data["player_id"]}/battingcard'
f'{urllib.parse.quote("?d=")}{release_directory}', f'{urllib.parse.quote("?d=")}{release_directory}',
'mlbclub': CLUB_LIST[df_data['Tm_vL']], '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'], 'cardset_id': cardset['id'],
'set_num': int(float(df_data['key_fangraphs'])), 'set_num': int(float(df_data['key_fangraphs'])),
'rarity_id': 99, 'rarity_id': 99,
@ -606,7 +606,7 @@ async def main(args):
'image': f'{CARD_BASE_URL}/{df_data["player_id"]}/' 'image': f'{CARD_BASE_URL}/{df_data["player_id"]}/'
f'pitchingcard{urllib.parse.quote("?d=")}{release_directory}', f'pitchingcard{urllib.parse.quote("?d=")}{release_directory}',
'mlbclub': CLUB_LIST[df_data['Tm_vL']], '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'], 'cardset_id': cardset['id'],
'set_num': int(float(df_data['key_fangraphs'])), 'set_num': int(float(df_data['key_fangraphs'])),
'rarity_id': 99, 'rarity_id': 99,