2026_MLB_Model / visualization /logo_utils.py
Syntrex's picture
Update visualization/logo_utils.py
eb7fd81 verified
raw
history blame
3.53 kB
from __future__ import annotations
TEAM_FLAG_URL_MAP = {
"Australia": "https://flagcdn.com/w40/au.png",
"Brazil": "https://flagcdn.com/w40/br.png",
"Canada": "https://flagcdn.com/w40/ca.png",
"China": "https://flagcdn.com/w40/cn.png",
"Chinese Taipei": "https://flagcdn.com/w40/tw.png",
"Colombia": "https://flagcdn.com/w40/co.png",
"Cuba": "https://flagcdn.com/w40/cu.png",
"Czechia": "https://flagcdn.com/w40/cz.png",
"Dominican Republic": "https://flagcdn.com/w40/do.png",
"Great Britain": "https://flagcdn.com/w40/gb.png",
"Israel": "https://flagcdn.com/w40/il.png",
"Italy": "https://flagcdn.com/w40/it.png",
"Japan": "https://flagcdn.com/w40/jp.png",
"Korea": "https://flagcdn.com/w40/kr.png",
"Mexico": "https://flagcdn.com/w40/mx.png",
"Netherlands": "https://flagcdn.com/w40/nl.png",
"Nicaragua": "https://flagcdn.com/w40/ni.png",
"Panama": "https://flagcdn.com/w40/pa.png",
"Puerto Rico": "https://flagcdn.com/w40/pr.png",
"United States": "https://flagcdn.com/w40/us.png",
"Venezuela": "https://flagcdn.com/w40/ve.png",
}
MLB_TEAM_LOGOS = {
"angels": "https://a.espncdn.com/i/teamlogos/mlb/500/ana.png",
"astros": "https://a.espncdn.com/i/teamlogos/mlb/500/hou.png",
"athletics": "https://a.espncdn.com/i/teamlogos/mlb/500/oak.png",
"blue jays": "https://a.espncdn.com/i/teamlogos/mlb/500/tor.png",
"braves": "https://a.espncdn.com/i/teamlogos/mlb/500/atl.png",
"brewers": "https://a.espncdn.com/i/teamlogos/mlb/500/mil.png",
"cardinals": "https://a.espncdn.com/i/teamlogos/mlb/500/stl.png",
"cubs": "https://a.espncdn.com/i/teamlogos/mlb/500/chc.png",
"diamondbacks": "https://a.espncdn.com/i/teamlogos/mlb/500/ari.png",
"dodgers": "https://a.espncdn.com/i/teamlogos/mlb/500/lad.png",
"giants": "https://a.espncdn.com/i/teamlogos/mlb/500/sf.png",
"guardians": "https://a.espncdn.com/i/teamlogos/mlb/500/cle.png",
"mariners": "https://a.espncdn.com/i/teamlogos/mlb/500/sea.png",
"marlins": "https://a.espncdn.com/i/teamlogos/mlb/500/mia.png",
"mets": "https://a.espncdn.com/i/teamlogos/mlb/500/nym.png",
"nationals": "https://a.espncdn.com/i/teamlogos/mlb/500/wsh.png",
"orioles": "https://a.espncdn.com/i/teamlogos/mlb/500/bal.png",
"padres": "https://a.espncdn.com/i/teamlogos/mlb/500/sd.png",
"phillies": "https://a.espncdn.com/i/teamlogos/mlb/500/phi.png",
"pirates": "https://a.espncdn.com/i/teamlogos/mlb/500/pit.png",
"rangers": "https://a.espncdn.com/i/teamlogos/mlb/500/tex.png",
"rays": "https://a.espncdn.com/i/teamlogos/mlb/500/tb.png",
"red sox": "https://a.espncdn.com/i/teamlogos/mlb/500/bos.png",
"reds": "https://a.espncdn.com/i/teamlogos/mlb/500/cin.png",
"rockies": "https://a.espncdn.com/i/teamlogos/mlb/500/col.png",
"royals": "https://a.espncdn.com/i/teamlogos/mlb/500/kc.png",
"tigers": "https://a.espncdn.com/i/teamlogos/mlb/500/det.png",
"twins": "https://a.espncdn.com/i/teamlogos/mlb/500/min.png",
"white sox": "https://a.espncdn.com/i/teamlogos/mlb/500/cws.png",
"yankees": "https://a.espncdn.com/i/teamlogos/mlb/500/nyy.png",
}
def get_team_flag_url(team_name: str) -> str:
return TEAM_FLAG_URL_MAP.get(str(team_name).strip(), "")
def get_team_logo_url(team_name: str) -> str:
team_norm = str(team_name or "").strip().lower()
for key, url in MLB_TEAM_LOGOS.items():
if team_norm == key or team_norm.endswith(f" {key}"):
return url
return ""