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>
59 lines
1.8 KiB
Python
59 lines
1.8 KiB
Python
"""
|
|
Helper functions for Discord Bot v2.0
|
|
|
|
Contains utility functions for salary cap calculations and other common operations.
|
|
"""
|
|
from config import get_config
|
|
|
|
# Get default values from config
|
|
_config = get_config()
|
|
|
|
# Salary cap constants - default from config, tolerance for float comparisons
|
|
DEFAULT_SALARY_CAP = _config.swar_cap_limit # 32.0
|
|
SALARY_CAP_TOLERANCE = 0.001 # Small tolerance for floating point comparisons
|
|
|
|
|
|
def get_team_salary_cap(team) -> float:
|
|
"""
|
|
Get the salary cap for a team, falling back to the default if not set.
|
|
|
|
Args:
|
|
team: Team data - can be a dict or Pydantic model with 'salary_cap' attribute.
|
|
|
|
Returns:
|
|
float: The team's salary cap, or DEFAULT_SALARY_CAP (32.0) if not set.
|
|
|
|
Why: Teams may have custom salary caps (e.g., for expansion teams or penalties).
|
|
This centralizes the fallback logic so all cap checks use the same source of truth.
|
|
"""
|
|
if team is None:
|
|
return DEFAULT_SALARY_CAP
|
|
|
|
# Handle both dict and Pydantic model (or any object with salary_cap attribute)
|
|
if isinstance(team, dict):
|
|
salary_cap = team.get('salary_cap')
|
|
else:
|
|
salary_cap = getattr(team, 'salary_cap', None)
|
|
|
|
if salary_cap is not None:
|
|
return salary_cap
|
|
return DEFAULT_SALARY_CAP
|
|
|
|
|
|
def exceeds_salary_cap(wara: float, team) -> bool:
|
|
"""
|
|
Check if a WAR total exceeds the team's salary cap.
|
|
|
|
Args:
|
|
wara: The total WAR value to check
|
|
team: Team data - can be a dict or Pydantic model
|
|
|
|
Returns:
|
|
bool: True if wara exceeds the team's salary cap (with tolerance)
|
|
|
|
Why: Centralizes the salary cap comparison logic with proper floating point
|
|
tolerance handling. All cap validation should use this function.
|
|
"""
|
|
cap = get_team_salary_cap(team)
|
|
return wara > (cap + SALARY_CAP_TOLERANCE)
|