diff --git a/helpers/utils.py b/helpers/utils.py index d956435..9fc70e3 100644 --- a/helpers/utils.py +++ b/helpers/utils.py @@ -77,8 +77,14 @@ def get_roster_sheet_legacy(team): def get_roster_sheet(team): - """Get roster sheet URL for a team.""" - return f'https://docs.google.com/spreadsheets/d/{team["gsheet"]}/edit' + """ + Get roster sheet URL for a team. + + Handles both dict and Team object formats. + """ + # Handle both dict (team["gsheet"]) and object (team.gsheet) formats + gsheet = team.get("gsheet") if isinstance(team, dict) else getattr(team, "gsheet", None) + return f'https://docs.google.com/spreadsheets/d/{gsheet}/edit' def get_player_url(team, player) -> str: diff --git a/utils.py b/utils.py index bb71a96..e8e5f5d 100644 --- a/utils.py +++ b/utils.py @@ -74,8 +74,14 @@ def get_roster_sheet_legacy(team): def get_roster_sheet(team): - """Get roster sheet URL for a team.""" - return f'https://docs.google.com/spreadsheets/d/{team["gsheet"]}/edit' + """ + Get roster sheet URL for a team. + + Handles both dict and Team object formats. + """ + # Handle both dict (team["gsheet"]) and object (team.gsheet) formats + gsheet = team.get("gsheet") if isinstance(team, dict) else getattr(team, "gsheet", None) + return f'https://docs.google.com/spreadsheets/d/{gsheet}/edit' def get_player_url(team, player) -> str: