f64fee8d2e
Ran `ruff check --select F401 --fix` to auto-remove 221 unused imports, manually removed 4 unused `import discord` from package __init__.py files, and fixed test import for DISAPPOINTMENT_TIERS to reference canonical location. Co-Authored-By: Claude Opus 4.6 <noreply@anthropic.com>
71 lines
2.5 KiB
Python
71 lines
2.5 KiB
Python
"""
|
|
Injury model for tracking player injuries
|
|
|
|
Represents an injury record with game timeline and status information.
|
|
"""
|
|
from typing import Any
|
|
from pydantic import Field, model_validator
|
|
|
|
from models.base import SBABaseModel
|
|
|
|
|
|
class Injury(SBABaseModel):
|
|
"""Injury model representing a player injury."""
|
|
|
|
# Override base model to make id required for database entities
|
|
id: int = Field(..., description="Injury ID from database")
|
|
|
|
season: int = Field(..., description="Season number")
|
|
player_id: int = Field(..., description="Player ID who is injured")
|
|
total_games: int = Field(..., description="Total games player will be out")
|
|
|
|
@model_validator(mode='before')
|
|
@classmethod
|
|
def extract_player_id(cls, data: Any) -> Any:
|
|
"""
|
|
Extract player_id from nested player object if present.
|
|
|
|
The API returns injuries with a nested 'player' object:
|
|
{'id': 123, 'player': {'id': 456, ...}, ...}
|
|
|
|
This validator extracts the player ID before validation:
|
|
{'id': 123, 'player_id': 456, ...}
|
|
"""
|
|
if isinstance(data, dict):
|
|
# If player_id is missing but player object exists, extract it
|
|
if 'player_id' not in data and 'player' in data:
|
|
if isinstance(data['player'], dict) and 'id' in data['player']:
|
|
data['player_id'] = data['player']['id']
|
|
|
|
return data
|
|
|
|
# Injury timeline
|
|
start_week: int = Field(..., description="Week injury started")
|
|
start_game: int = Field(..., description="Game number injury started (1-4)")
|
|
end_week: int = Field(..., description="Week player returns")
|
|
end_game: int = Field(..., description="Game number player returns (1-4)")
|
|
|
|
# Status
|
|
is_active: bool = Field(True, description="Whether injury is currently active")
|
|
|
|
@property
|
|
def return_date(self) -> str:
|
|
"""Format return date as 'w##g#' string."""
|
|
return f'w{self.end_week:02d}g{self.end_game}'
|
|
|
|
@property
|
|
def start_date(self) -> str:
|
|
"""Format start date as 'w##g#' string."""
|
|
return f'w{self.start_week:02d}g{self.start_game}'
|
|
|
|
@property
|
|
def duration_display(self) -> str:
|
|
"""Return a human-readable duration string."""
|
|
if self.total_games == 1:
|
|
return "1 game"
|
|
return f"{self.total_games} games"
|
|
|
|
def __str__(self):
|
|
status = "Active" if self.is_active else "Cleared"
|
|
return f"Injury (Season {self.season}, {self.duration_display}, {status})"
|