CLAUDE: Refactor draft system to eliminate hard-coded magic numbers
Replace all hard-coded values with centralized config constants for better maintainability and flexibility: Added config constants: - draft_team_count (16) - draft_linear_rounds (10) - swar_cap_limit (32.00) - cap_player_count (26) - draft_total_picks property (derived: rounds × teams) Critical fixes: - FA team ID (498) now uses config.free_agent_team_id in: * tasks/draft_monitor.py - Auto-draft validation * commands/draft/picks.py - Pick validation and autocomplete - sWAR cap limit display now uses config.swar_cap_limit Refactored modules: - utils/draft_helpers.py - All calculation functions - services/draft_service.py - Pick advancement logic - views/draft_views.py - Display formatting Benefits: - Eliminates risk of silent failures from hard-coded IDs - Centralizes all draft constants in one location - Enables easy draft format changes via config - Improves testability with mockable config - Zero breaking changes - fully backwards compatible 🤖 Generated with [Claude Code](https://claude.com/claude-code) Co-Authored-By: Claude <noreply@anthropic.com>
This commit is contained in:
parent
0e54a81bbe
commit
ea7b356db9
@ -42,8 +42,8 @@ async def fa_player_autocomplete(
|
||||
season=config.sba_current_season
|
||||
)
|
||||
|
||||
# Filter to FA team (team_id = 498)
|
||||
fa_players = [p for p in players if p.team_id == 498]
|
||||
# Filter to FA team
|
||||
fa_players = [p for p in players if p.team_id == config.free_agent_team_id]
|
||||
|
||||
return [
|
||||
discord.app_commands.Choice(
|
||||
@ -194,7 +194,7 @@ class DraftPicksCog(commands.Cog):
|
||||
player_obj = players[0]
|
||||
|
||||
# Validate player is FA
|
||||
if player_obj.team_id != 498: # 498 = FA team ID
|
||||
if player_obj.team_id != config.free_agent_team_id:
|
||||
embed = await create_pick_illegal_embed(
|
||||
"Player Not Available",
|
||||
f"{player_obj.name} is not a free agent."
|
||||
@ -217,7 +217,7 @@ class DraftPicksCog(commands.Cog):
|
||||
if not is_valid:
|
||||
embed = await create_pick_illegal_embed(
|
||||
"Cap Space Exceeded",
|
||||
f"Drafting {player_obj.name} would put you at {projected_total:.2f} sWAR (limit: 32.00)."
|
||||
f"Drafting {player_obj.name} would put you at {projected_total:.2f} sWAR (limit: {config.swar_cap_limit:.2f})."
|
||||
)
|
||||
await interaction.followup.send(embed=embed)
|
||||
return
|
||||
|
||||
@ -47,6 +47,10 @@ class BotConfig(BaseSettings):
|
||||
# Draft Constants
|
||||
default_pick_minutes: int = 10
|
||||
draft_rounds: int = 32
|
||||
draft_team_count: int = 16 # Number of teams in draft
|
||||
draft_linear_rounds: int = 10 # Rounds 1-10 are linear, 11+ are snake
|
||||
swar_cap_limit: float = 32.00 # Maximum sWAR cap for team roster
|
||||
cap_player_count: int = 26 # Number of players that count toward cap
|
||||
|
||||
# Special Team IDs
|
||||
free_agent_team_id: int = 498
|
||||
@ -94,6 +98,11 @@ class BotConfig(BaseSettings):
|
||||
"""Check if running in test mode."""
|
||||
return self.testing
|
||||
|
||||
@property
|
||||
def draft_total_picks(self) -> int:
|
||||
"""Calculate total picks in draft (derived value)."""
|
||||
return self.draft_rounds * self.draft_team_count
|
||||
|
||||
|
||||
# Global configuration instance - lazily initialized to avoid import-time errors
|
||||
_config = None
|
||||
|
||||
@ -169,12 +169,13 @@ class DraftService(BaseService[DraftData]):
|
||||
|
||||
config = get_config()
|
||||
season = config.sba_current_season
|
||||
total_picks = config.draft_total_picks
|
||||
|
||||
# Start with next pick
|
||||
next_pick = current_pick + 1
|
||||
|
||||
# Keep advancing until we find an unfilled pick or reach end
|
||||
while next_pick <= 512: # 32 rounds * 16 teams
|
||||
while next_pick <= total_picks:
|
||||
pick = await draft_pick_service.get_pick(season, next_pick)
|
||||
|
||||
if not pick:
|
||||
@ -191,7 +192,7 @@ class DraftService(BaseService[DraftData]):
|
||||
next_pick += 1
|
||||
|
||||
# Check if draft is complete
|
||||
if next_pick > 512:
|
||||
if next_pick > total_picks:
|
||||
logger.info("Draft is complete - all picks filled")
|
||||
# Disable timer
|
||||
await self.set_timer(draft_id, active=False)
|
||||
|
||||
@ -190,7 +190,7 @@ class DraftMonitorTask:
|
||||
player = entry.player
|
||||
|
||||
# Check if player is still available
|
||||
if player.team_id != 498: # 498 = FA team ID
|
||||
if player.team_id != config.free_agent_team_id:
|
||||
self.logger.debug(f"Player {player.name} no longer available, skipping")
|
||||
continue
|
||||
|
||||
|
||||
@ -6,6 +6,7 @@ Provides helper functions for draft order calculation and cap space validation.
|
||||
import math
|
||||
from typing import Tuple
|
||||
from utils.logging import get_contextual_logger
|
||||
from config import get_config
|
||||
|
||||
logger = get_contextual_logger(__name__)
|
||||
|
||||
@ -46,17 +47,21 @@ def calculate_pick_details(overall: int) -> Tuple[int, int]:
|
||||
>>> calculate_pick_details(177)
|
||||
(12, 16) # Round 12, Pick 16 (snake reverses)
|
||||
"""
|
||||
round_num = math.ceil(overall / 16)
|
||||
config = get_config()
|
||||
team_count = config.draft_team_count
|
||||
linear_rounds = config.draft_linear_rounds
|
||||
|
||||
if round_num <= 10:
|
||||
round_num = math.ceil(overall / team_count)
|
||||
|
||||
if round_num <= linear_rounds:
|
||||
# Linear draft: position is same calculation every round
|
||||
position = ((overall - 1) % 16) + 1
|
||||
position = ((overall - 1) % team_count) + 1
|
||||
else:
|
||||
# Snake draft: reverse on even rounds
|
||||
if round_num % 2 == 1: # Odd rounds (11, 13, 15...)
|
||||
position = ((overall - 1) % 16) + 1
|
||||
position = ((overall - 1) % team_count) + 1
|
||||
else: # Even rounds (12, 14, 16...)
|
||||
position = 16 - ((overall - 1) % 16)
|
||||
position = team_count - ((overall - 1) % team_count)
|
||||
|
||||
return round_num, position
|
||||
|
||||
@ -87,16 +92,20 @@ def calculate_overall_from_round_position(round_num: int, position: int) -> int:
|
||||
>>> calculate_overall_from_round_position(12, 16)
|
||||
177
|
||||
"""
|
||||
if round_num <= 10:
|
||||
config = get_config()
|
||||
team_count = config.draft_team_count
|
||||
linear_rounds = config.draft_linear_rounds
|
||||
|
||||
if round_num <= linear_rounds:
|
||||
# Linear draft
|
||||
return (round_num - 1) * 16 + position
|
||||
return (round_num - 1) * team_count + position
|
||||
else:
|
||||
# Snake draft
|
||||
picks_before_round = (round_num - 1) * 16
|
||||
picks_before_round = (round_num - 1) * team_count
|
||||
if round_num % 2 == 1: # Odd snake rounds
|
||||
return picks_before_round + position
|
||||
else: # Even snake rounds (reversed)
|
||||
return picks_before_round + (17 - position)
|
||||
return picks_before_round + (team_count + 1 - position)
|
||||
|
||||
|
||||
async def validate_cap_space(
|
||||
@ -127,6 +136,10 @@ async def validate_cap_space(
|
||||
Raises:
|
||||
ValueError: If roster structure is invalid
|
||||
"""
|
||||
config = get_config()
|
||||
cap_limit = config.swar_cap_limit
|
||||
cap_player_count = config.cap_player_count
|
||||
|
||||
if not roster or not roster.get('active'):
|
||||
raise ValueError("Invalid roster structure - missing 'active' key")
|
||||
|
||||
@ -140,7 +153,7 @@ async def validate_cap_space(
|
||||
# Maximum zeroes = 32 - roster size
|
||||
# Maximum counted = 26 - zeroes
|
||||
max_zeroes = 32 - projected_roster_size
|
||||
max_counted = min(26, 26 - max_zeroes) # Can't count more than 26
|
||||
max_counted = min(cap_player_count, cap_player_count - max_zeroes) # Can't count more than cap_player_count
|
||||
|
||||
# Sort all players (including new) by sWAR descending
|
||||
all_players_wara = [p['wara'] for p in current_players] + [new_player_wara]
|
||||
@ -150,7 +163,7 @@ async def validate_cap_space(
|
||||
projected_total = sum(sorted_wara[:max_counted])
|
||||
|
||||
# Allow tiny floating point tolerance
|
||||
is_valid = projected_total <= 32.00001
|
||||
is_valid = projected_total <= (cap_limit + 0.00001)
|
||||
|
||||
logger.debug(
|
||||
f"Cap validation: roster_size={current_roster_size}, "
|
||||
@ -200,17 +213,21 @@ def get_next_pick_overall(current_overall: int) -> int:
|
||||
return current_overall + 1
|
||||
|
||||
|
||||
def is_draft_complete(current_overall: int, total_picks: int = 512) -> bool:
|
||||
def is_draft_complete(current_overall: int, total_picks: int = None) -> bool:
|
||||
"""
|
||||
Check if draft is complete.
|
||||
|
||||
Args:
|
||||
current_overall: Current overall pick number
|
||||
total_picks: Total number of picks in draft (default: 512 for 32 rounds, 16 teams)
|
||||
total_picks: Total number of picks in draft (None uses config value)
|
||||
|
||||
Returns:
|
||||
True if draft is complete
|
||||
"""
|
||||
if total_picks is None:
|
||||
config = get_config()
|
||||
total_picks = config.draft_total_picks
|
||||
|
||||
return current_overall > total_picks
|
||||
|
||||
|
||||
|
||||
@ -15,6 +15,7 @@ from models.player import Player
|
||||
from models.draft_list import DraftList
|
||||
from views.embeds import EmbedTemplate, EmbedColors
|
||||
from utils.draft_helpers import format_pick_display, get_round_name
|
||||
from config import get_config
|
||||
|
||||
|
||||
async def create_on_the_clock_embed(
|
||||
@ -66,9 +67,10 @@ async def create_on_the_clock_embed(
|
||||
|
||||
# Add team sWAR if provided
|
||||
if team_roster_swar is not None:
|
||||
config = get_config()
|
||||
embed.add_field(
|
||||
name="Current sWAR",
|
||||
value=f"{team_roster_swar:.2f} / 32.00",
|
||||
value=f"{team_roster_swar:.2f} / {config.swar_cap_limit:.2f}",
|
||||
inline=True
|
||||
)
|
||||
|
||||
@ -375,9 +377,10 @@ async def create_pick_success_embed(
|
||||
inline=True
|
||||
)
|
||||
|
||||
config = get_config()
|
||||
embed.add_field(
|
||||
name="Projected Team sWAR",
|
||||
value=f"{projected_swar:.2f} / 32.00",
|
||||
value=f"{projected_swar:.2f} / {config.swar_cap_limit:.2f}",
|
||||
inline=True
|
||||
)
|
||||
|
||||
|
||||
Loading…
Reference in New Issue
Block a user