CLAUDE: Add test game creation utilities
- test_game_data.py: Static lineup data based on real SBA Game 2519 - West Virginia Black Bears vs Columbus Hydra - Complete lineup data with player info - create_test_game.py: Script to end active games and create fresh test game - Usage: uv run python scripts/create_test_game.py 🤖 Generated with [Claude Code](https://claude.com/claude-code) Co-Authored-By: Claude <noreply@anthropic.com>
This commit is contained in:
parent
1a562a75d2
commit
6f0fe24701
100
backend/app/config/test_game_data.py
Normal file
100
backend/app/config/test_game_data.py
Normal file
@ -0,0 +1,100 @@
|
|||||||
|
"""
|
||||||
|
Test Game Data - Static lineup data for test game creation.
|
||||||
|
|
||||||
|
Based on SBA Game 2519 (Season 12, Week 1):
|
||||||
|
- West Virginia Black Bears (499) vs Columbus Hydra (544)
|
||||||
|
- Final Score: WV 4 - CLS 3
|
||||||
|
|
||||||
|
Author: Claude
|
||||||
|
Date: 2025-01-10
|
||||||
|
"""
|
||||||
|
|
||||||
|
# Team Information
|
||||||
|
AWAY_TEAM = {
|
||||||
|
"id": 499,
|
||||||
|
"abbrev": "WV",
|
||||||
|
"short_name": "Black Bears",
|
||||||
|
"long_name": "West Virginia Black Bears",
|
||||||
|
"manager_id": 3,
|
||||||
|
"manager_name": "Cal",
|
||||||
|
}
|
||||||
|
|
||||||
|
HOME_TEAM = {
|
||||||
|
"id": 544,
|
||||||
|
"abbrev": "CLS",
|
||||||
|
"short_name": "Hydra",
|
||||||
|
"long_name": "Columbus Hydra",
|
||||||
|
"manager_id": 14,
|
||||||
|
"manager_name": "Mason",
|
||||||
|
}
|
||||||
|
|
||||||
|
# Away Team Lineup - West Virginia Black Bears (Team 499)
|
||||||
|
# Pitcher: Zac Gallen
|
||||||
|
AWAY_LINEUP = [
|
||||||
|
# Batters (batting_order 1-9)
|
||||||
|
{"player_id": 12288, "name": "Ronald Acuna Jr", "position": "RF", "batting_order": 1},
|
||||||
|
{"player_id": 12395, "name": "Trea Turner", "position": "SS", "batting_order": 2},
|
||||||
|
{"player_id": 11483, "name": "Alec Burleson", "position": "1B", "batting_order": 3},
|
||||||
|
{"player_id": 11487, "name": "Alex Bregman", "position": "3B", "batting_order": 4},
|
||||||
|
{"player_id": 12262, "name": "Ramon Urias", "position": "2B", "batting_order": 5},
|
||||||
|
{"player_id": 12356, "name": "Steven Kwan", "position": "LF", "batting_order": 6},
|
||||||
|
{"player_id": 12339, "name": "Shea Langeliers", "position": "C", "batting_order": 7},
|
||||||
|
{"player_id": 11706, "name": "David Fry", "position": "DH", "batting_order": 8},
|
||||||
|
{"player_id": 11545, "name": "Blake Perkins", "position": "CF", "batting_order": 9},
|
||||||
|
# Starting Pitcher
|
||||||
|
{"player_id": 12479, "name": "Zac Gallen", "position": "P", "batting_order": None},
|
||||||
|
]
|
||||||
|
|
||||||
|
# Home Team Lineup - Columbus Hydra (Team 544)
|
||||||
|
# Pitcher: Tanner Houck
|
||||||
|
HOME_LINEUP = [
|
||||||
|
# Batters (batting_order 1-9)
|
||||||
|
{"player_id": 12154, "name": "Michael Busch", "position": "2B", "batting_order": 1},
|
||||||
|
{"player_id": 12455, "name": "Willy Adames", "position": "SS", "batting_order": 2},
|
||||||
|
{"player_id": 12110, "name": "Luke Raley", "position": "LF", "batting_order": 3},
|
||||||
|
{"player_id": 12135, "name": "Matt Chapman", "position": "3B", "batting_order": 4},
|
||||||
|
{"player_id": 11768, "name": "Eric Haase", "position": "C", "batting_order": 5},
|
||||||
|
{"player_id": 11523, "name": "Anthony Santander", "position": "RF", "batting_order": 6},
|
||||||
|
{"player_id": 11611, "name": "Cal Raleigh", "position": "DH", "batting_order": 7},
|
||||||
|
{"player_id": 12250, "name": "Pete Alonso", "position": "1B", "batting_order": 8},
|
||||||
|
{"player_id": 11927, "name": "JJ Bleday", "position": "CF", "batting_order": 9},
|
||||||
|
# Starting Pitcher
|
||||||
|
{"player_id": 12366, "name": "Tanner Houck", "position": "P", "batting_order": None},
|
||||||
|
]
|
||||||
|
|
||||||
|
|
||||||
|
def get_test_game_config() -> dict:
|
||||||
|
"""
|
||||||
|
Get complete test game configuration.
|
||||||
|
|
||||||
|
Returns:
|
||||||
|
Dictionary with away_team, home_team, away_lineup, home_lineup
|
||||||
|
"""
|
||||||
|
return {
|
||||||
|
"away_team": AWAY_TEAM,
|
||||||
|
"home_team": HOME_TEAM,
|
||||||
|
"away_lineup": AWAY_LINEUP,
|
||||||
|
"home_lineup": HOME_LINEUP,
|
||||||
|
"league_id": "sba",
|
||||||
|
}
|
||||||
|
|
||||||
|
|
||||||
|
def get_lineup_for_team(team_id: int) -> list[dict]:
|
||||||
|
"""
|
||||||
|
Get lineup for a specific team.
|
||||||
|
|
||||||
|
Args:
|
||||||
|
team_id: Team identifier (499 or 544)
|
||||||
|
|
||||||
|
Returns:
|
||||||
|
List of player lineup entries
|
||||||
|
|
||||||
|
Raises:
|
||||||
|
ValueError: If team_id is not recognized
|
||||||
|
"""
|
||||||
|
if team_id == 499:
|
||||||
|
return AWAY_LINEUP
|
||||||
|
elif team_id == 544:
|
||||||
|
return HOME_LINEUP
|
||||||
|
else:
|
||||||
|
raise ValueError(f"Unknown team_id: {team_id}. Use 499 (WV) or 544 (CLS)")
|
||||||
142
backend/scripts/create_test_game.py
Normal file
142
backend/scripts/create_test_game.py
Normal file
@ -0,0 +1,142 @@
|
|||||||
|
"""
|
||||||
|
Create Test Game Script
|
||||||
|
|
||||||
|
Ends all active games and creates a new test game using real SBA lineup data.
|
||||||
|
|
||||||
|
Usage:
|
||||||
|
uv run python scripts/create_test_game.py
|
||||||
|
"""
|
||||||
|
|
||||||
|
import asyncio
|
||||||
|
import sys
|
||||||
|
from uuid import uuid4, UUID
|
||||||
|
from pathlib import Path
|
||||||
|
|
||||||
|
# Add backend to path
|
||||||
|
sys.path.insert(0, str(Path(__file__).parent.parent))
|
||||||
|
|
||||||
|
from app.database.session import AsyncSessionLocal
|
||||||
|
from app.database.operations import DatabaseOperations
|
||||||
|
from app.models.db_models import Game
|
||||||
|
from app.config.test_game_data import get_test_game_config
|
||||||
|
from app.services.lineup_service import lineup_service
|
||||||
|
from sqlalchemy import select, update
|
||||||
|
|
||||||
|
|
||||||
|
async def end_all_active_games(db_ops: DatabaseOperations) -> int:
|
||||||
|
"""End all active games by setting status to 'completed'."""
|
||||||
|
async with AsyncSessionLocal() as session:
|
||||||
|
# Find all active games
|
||||||
|
result = await session.execute(
|
||||||
|
select(Game).where(Game.status == 'active')
|
||||||
|
)
|
||||||
|
active_games = result.scalars().all()
|
||||||
|
|
||||||
|
count = len(active_games)
|
||||||
|
|
||||||
|
if count > 0:
|
||||||
|
# Update all to completed
|
||||||
|
await session.execute(
|
||||||
|
update(Game).where(Game.status == 'active').values(status='completed')
|
||||||
|
)
|
||||||
|
await session.commit()
|
||||||
|
print(f"Ended {count} active game(s)")
|
||||||
|
else:
|
||||||
|
print("No active games to end")
|
||||||
|
|
||||||
|
return count
|
||||||
|
|
||||||
|
|
||||||
|
async def create_test_game(db_ops: DatabaseOperations) -> UUID:
|
||||||
|
"""Create a new test game with real SBA lineup data."""
|
||||||
|
|
||||||
|
# Get test game configuration
|
||||||
|
config = get_test_game_config()
|
||||||
|
away_team = config["away_team"]
|
||||||
|
home_team = config["home_team"]
|
||||||
|
away_lineup = config["away_lineup"]
|
||||||
|
home_lineup = config["home_lineup"]
|
||||||
|
|
||||||
|
# Create game ID
|
||||||
|
game_id = uuid4()
|
||||||
|
|
||||||
|
print(f"\nCreating game: {game_id}")
|
||||||
|
print(f"Away: {away_team['long_name']} ({away_team['id']})")
|
||||||
|
print(f"Home: {home_team['long_name']} ({home_team['id']})")
|
||||||
|
|
||||||
|
# Create game in database
|
||||||
|
await db_ops.create_game(
|
||||||
|
game_id=game_id,
|
||||||
|
league_id="sba",
|
||||||
|
home_team_id=home_team["id"],
|
||||||
|
away_team_id=away_team["id"],
|
||||||
|
game_mode="friendly",
|
||||||
|
visibility="public"
|
||||||
|
)
|
||||||
|
|
||||||
|
# Add away team lineup (using lineup_service to fetch player data)
|
||||||
|
print(f"\nAway lineup ({away_team['abbrev']}):")
|
||||||
|
for player in away_lineup:
|
||||||
|
entry = await lineup_service.add_sba_player_to_lineup(
|
||||||
|
game_id=game_id,
|
||||||
|
team_id=away_team["id"],
|
||||||
|
player_id=player["player_id"],
|
||||||
|
position=player["position"],
|
||||||
|
batting_order=player["batting_order"],
|
||||||
|
is_starter=True
|
||||||
|
)
|
||||||
|
order = player["batting_order"] if player["batting_order"] else "P"
|
||||||
|
print(f" {order}. {entry.player_name} ({entry.position})")
|
||||||
|
|
||||||
|
# Add home team lineup (using lineup_service to fetch player data)
|
||||||
|
print(f"\nHome lineup ({home_team['abbrev']}):")
|
||||||
|
for player in home_lineup:
|
||||||
|
entry = await lineup_service.add_sba_player_to_lineup(
|
||||||
|
game_id=game_id,
|
||||||
|
team_id=home_team["id"],
|
||||||
|
player_id=player["player_id"],
|
||||||
|
position=player["position"],
|
||||||
|
batting_order=player["batting_order"],
|
||||||
|
is_starter=True
|
||||||
|
)
|
||||||
|
order = player["batting_order"] if player["batting_order"] else "P"
|
||||||
|
print(f" {order}. {entry.player_name} ({entry.position})")
|
||||||
|
|
||||||
|
# Update game status to active
|
||||||
|
async with AsyncSessionLocal() as session:
|
||||||
|
await session.execute(
|
||||||
|
update(Game).where(Game.id == game_id).values(
|
||||||
|
status="active",
|
||||||
|
current_inning=1,
|
||||||
|
current_half="top"
|
||||||
|
)
|
||||||
|
)
|
||||||
|
await session.commit()
|
||||||
|
|
||||||
|
print(f"\nGame created and activated: {game_id}")
|
||||||
|
return game_id
|
||||||
|
|
||||||
|
|
||||||
|
async def main():
|
||||||
|
"""Main function to end active games and create a new test game."""
|
||||||
|
print("=" * 60)
|
||||||
|
print("Test Game Creator")
|
||||||
|
print("=" * 60)
|
||||||
|
|
||||||
|
db_ops = DatabaseOperations()
|
||||||
|
|
||||||
|
# End all active games
|
||||||
|
await end_all_active_games(db_ops)
|
||||||
|
|
||||||
|
# Create new test game
|
||||||
|
game_id = await create_test_game(db_ops)
|
||||||
|
|
||||||
|
print("\n" + "=" * 60)
|
||||||
|
print(f"SUCCESS! Test game ready: {game_id}")
|
||||||
|
print("=" * 60)
|
||||||
|
|
||||||
|
return game_id
|
||||||
|
|
||||||
|
|
||||||
|
if __name__ == "__main__":
|
||||||
|
game_id = asyncio.run(main())
|
||||||
Loading…
Reference in New Issue
Block a user