1330 lines
54 KiB
Python
1330 lines
54 KiB
Python
import datetime
|
|
import logging
|
|
from typing import Literal
|
|
|
|
import discord
|
|
import pydantic
|
|
|
|
from pydantic import field_validator
|
|
from sqlmodel import Session, SQLModel, UniqueConstraint, create_engine, select, or_, Field, Relationship, text
|
|
from sqlalchemy import func, desc
|
|
|
|
from exceptions import *
|
|
from in_game.managerai_responses import DefenseResponse, JumpResponse, TagResponse, ThrowResponse, UncappedRunResponse
|
|
|
|
|
|
logger = logging.getLogger('discord_app')
|
|
sqlite_url = 'sqlite:///storage/gameplay.db'
|
|
connect_args = {"check_same_thread": False}
|
|
engine = create_engine(sqlite_url, echo=False, connect_args=connect_args)
|
|
CACHE_LIMIT = 1209600 # in seconds
|
|
SBA_COLOR = 'a6ce39'
|
|
SBA_LOGO = 'https://sombaseball.ddns.net/static/images/sba-logo.png'
|
|
|
|
|
|
class ManagerAiBase(SQLModel):
|
|
id: int | None = Field(primary_key=True)
|
|
name: str = Field(index=True)
|
|
steal: int | None = Field(default=5)
|
|
running: int | None = Field(default=5)
|
|
hold: int | None = Field(default=5)
|
|
catcher_throw: int | None = Field(default=5)
|
|
uncapped_home: int | None = Field(default=5)
|
|
uncapped_third: int | None = Field(default=5)
|
|
uncapped_trail: int | None = Field(default=5)
|
|
bullpen_matchup: int | None = Field(default=5)
|
|
behind_aggression: int | None = Field(default=5)
|
|
ahead_aggression: int | None = Field(default=5)
|
|
decide_throw: int | None = Field(default=5)
|
|
|
|
|
|
class GameCardsetLink(SQLModel, table=True):
|
|
game_id: int | None = Field(default=None, foreign_key='game.id', primary_key=True)
|
|
cardset_id: int | None = Field(default=None, foreign_key='cardset.id', primary_key=True)
|
|
priority: int | None = Field(default=1, index=True)
|
|
|
|
game: 'Game' = Relationship(back_populates='cardset_links')
|
|
cardset: 'Cardset' = Relationship(back_populates='game_links')
|
|
|
|
|
|
class TeamBase(SQLModel):
|
|
id: int = Field(primary_key=True)
|
|
abbrev: str = Field(index=True)
|
|
sname: str
|
|
lname: str
|
|
gmid: int = Field(index=True)
|
|
gmname: str
|
|
gsheet: str
|
|
wallet: int
|
|
team_value: int
|
|
collection_value: int
|
|
logo: str | None = Field(default=None)
|
|
color: str
|
|
season: int
|
|
career: int
|
|
ranking: int
|
|
has_guide: bool
|
|
is_ai: bool
|
|
created: datetime.datetime = Field(default_factory=datetime.datetime.now, nullable=True)
|
|
|
|
@property
|
|
def description(self) -> str:
|
|
return f'{self.id}. {self.abbrev} {self.lname}, {"is_ai" if self.is_ai else "human"}'
|
|
|
|
|
|
class Team(TeamBase, table=True):
|
|
cards: list['Card'] = Relationship(back_populates='team', cascade_delete=True)
|
|
lineups: list['Lineup'] = Relationship(back_populates='team', cascade_delete=True)
|
|
# away_games: list['Game'] = Relationship(back_populates='away_team')
|
|
# home_games: list['Game'] = Relationship(back_populates='home_team')
|
|
|
|
@property
|
|
def embed(self) -> discord.Embed:
|
|
embed = discord.Embed(
|
|
title=f'{self.lname}',
|
|
color=int(self.color, 16) if self.color else int(SBA_COLOR, 16)
|
|
)
|
|
embed.set_footer(text=f'Paper Dynasty Season {self.season}', icon_url=SBA_LOGO)
|
|
embed.set_thumbnail(url=self.logo if self.logo else SBA_LOGO)
|
|
return embed
|
|
|
|
|
|
class Game(SQLModel, table=True):
|
|
id: int | None = Field(default=None, primary_key=True)
|
|
away_team_id: int = Field(foreign_key='team.id')
|
|
home_team_id: int = Field(foreign_key='team.id')
|
|
channel_id: int = Field(index=True)
|
|
season: int
|
|
active: bool | None = Field(default=True)
|
|
is_pd: bool | None = Field(default=True)
|
|
ranked: bool | None = Field(default=False)
|
|
short_game: bool | None = Field(default=False)
|
|
week: int | None = Field(default=None)
|
|
game_num: int | None = Field(default=None)
|
|
away_roster_id: int | None = Field(default=None)
|
|
home_roster_id: int | None = Field(default=None)
|
|
first_message: str | None = Field(default=None)
|
|
ai_team: str | None = Field(default=None)
|
|
game_type: str
|
|
roll_buttons: bool | None = Field(default=False)
|
|
auto_roll: bool | None = Field(default=True)
|
|
|
|
cardset_links: list[GameCardsetLink] = Relationship(back_populates='game', cascade_delete=True)
|
|
away_team: Team = Relationship(
|
|
# back_populates='away_games',
|
|
# sa_relationship_kwargs={
|
|
# 'primaryjoin': 'Game.away_team_id==Team.id',
|
|
# 'foreign_keys': '[Game.away_team.id]',
|
|
# 'lazy': 'joined'
|
|
# }
|
|
sa_relationship_kwargs=dict(foreign_keys="[Game.away_team_id]")
|
|
)
|
|
home_team: Team = Relationship(
|
|
# back_populates='home_games',
|
|
# sa_relationship_kwargs={
|
|
# 'primaryjoin': 'Game.home_team_id==Team.id',
|
|
# 'foreign_keys': '[Game.home_team.id]',
|
|
# 'lazy': 'joined'
|
|
# }
|
|
sa_relationship_kwargs=dict(foreign_keys="[Game.home_team_id]")
|
|
)
|
|
lineups: list['Lineup'] = Relationship(back_populates='game', cascade_delete=True)
|
|
plays: list['Play'] = Relationship(back_populates='game', cascade_delete=True)
|
|
|
|
@field_validator('ai_team', 'game_type')
|
|
def lowercase_strings(cls, value: str) -> str:
|
|
return value.lower()
|
|
|
|
@property
|
|
def cardset_param_string(self) -> str:
|
|
pri_cardsets = ''
|
|
back_cardsets = ''
|
|
for link in self.cardset_links:
|
|
if link.priority == 1:
|
|
pri_cardsets += f'&cardset_id={link.cardset_id}'
|
|
else:
|
|
back_cardsets += f'&backup_cardset_id={link.cardset_id}'
|
|
return f'{pri_cardsets}{back_cardsets}'
|
|
|
|
def current_play_or_none(self, session: Session):
|
|
this_play = session.exec(select(Play).where(Play.game == self, Play.complete == False).order_by(Play.id.desc()).limit(1)).all()
|
|
if len(this_play) == 1:
|
|
return this_play[0]
|
|
else:
|
|
return None
|
|
|
|
# async def get_scorebug_embed(self, session: Session, full_length: bool = True, classic: bool = True) -> discord.Embed:
|
|
# gt_string = ' - Unlimited'
|
|
# if self.game_type == 'minor-league':
|
|
# gt_string = ' - Minor League'
|
|
# elif self.game_type == 'major-league':
|
|
# gt_string = ' - Major League'
|
|
# elif self.game_type == 'hall-of-fame':
|
|
# gt_string = ' - Hall of Fame'
|
|
# elif 'gauntlet' in self.game_type:
|
|
# gt_string = ' - Gauntlet'
|
|
# elif 'flashback' in self.game_type:
|
|
# gt_string = ' - Flashback'
|
|
# elif 'exhibition' in self.game_type:
|
|
# gt_string = ' - Exhibition'
|
|
# logger.info(f'gameplay_models - Game.get_scorebug_embed - this_game: {self} / gt_string: {gt_string}')
|
|
|
|
# embed = discord.Embed(
|
|
# title=f'{self.away_team.sname} @ {self.home_team.sname}{gt_string}',
|
|
# color=int(SBA_COLOR, 16)
|
|
# )
|
|
|
|
# curr_play = self.current_play_or_none(session)
|
|
|
|
# if curr_play is None:
|
|
# try:
|
|
# curr_play = self.initialize_play(session)
|
|
# except LineupsMissingException as e:
|
|
# logger.debug(f'gameplay_models - Game.get_scorebug_embed - Could not initialize play')
|
|
|
|
# if curr_play is not None:
|
|
# embed.add_field(
|
|
# name='Game State',
|
|
# value=curr_play.scorebug_ascii,
|
|
# inline=False
|
|
# )
|
|
|
|
# def steal_string(batting_card: BattingCard) -> str:
|
|
# steal_string = '-/- (---)'
|
|
# if batting_card.steal_jump > 0:
|
|
# jump_chances = round(batting_card.steal_jump * 36)
|
|
|
|
# if jump_chances == 6:
|
|
# good_jump = 7
|
|
# elif jump_chances == 5:
|
|
# good_jump = 6
|
|
# elif jump_chances == 4:
|
|
# good_jump = 5
|
|
# elif jump_chances == 3:
|
|
# good_jump = 4
|
|
# elif jump_chances == 2:
|
|
# good_jump = 3
|
|
# elif jump_chances == 1:
|
|
# good_jump = 2
|
|
# elif jump_chances == 7:
|
|
# good_jump = '4,5'
|
|
# elif jump_chances == 8:
|
|
# good_jump = '4,6'
|
|
# elif jump_chances == 9:
|
|
# good_jump = '3-5'
|
|
# elif jump_chances == 10:
|
|
# good_jump = '2-5'
|
|
# elif jump_chances == 11:
|
|
# good_jump = '6,7'
|
|
# elif jump_chances == 12:
|
|
# good_jump = '4-6'
|
|
# elif jump_chances == 13:
|
|
# good_jump = '2,4-6'
|
|
# elif jump_chances == 14:
|
|
# good_jump = '3-6'
|
|
# elif jump_chances == 15:
|
|
# good_jump = '2-6'
|
|
# elif jump_chances == 16:
|
|
# good_jump = '2,5-6'
|
|
# elif jump_chances == 17:
|
|
# good_jump = '3,5-6'
|
|
# elif jump_chances == 18:
|
|
# good_jump = '4-6'
|
|
# elif jump_chances == 19:
|
|
# good_jump = '2,4-7'
|
|
# elif jump_chances == 20:
|
|
# good_jump = '3-7'
|
|
# elif jump_chances == 21:
|
|
# good_jump = '2-7'
|
|
# elif jump_chances == 22:
|
|
# good_jump = '2-7,12'
|
|
# elif jump_chances == 23:
|
|
# good_jump = '2-7,11'
|
|
# elif jump_chances == 24:
|
|
# good_jump = '2,4-8'
|
|
# elif jump_chances == 25:
|
|
# good_jump = '3-8'
|
|
# elif jump_chances == 26:
|
|
# good_jump = '2-8'
|
|
# elif jump_chances == 27:
|
|
# good_jump = '2-8,12'
|
|
# elif jump_chances == 28:
|
|
# good_jump = '2-8,11'
|
|
# elif jump_chances == 29:
|
|
# good_jump = '3-9'
|
|
# elif jump_chances == 30:
|
|
# good_jump = '2-9'
|
|
# elif jump_chances == 31:
|
|
# good_jump = '2-9,12'
|
|
# elif jump_chances == 32:
|
|
# good_jump = '2-9,11'
|
|
# elif jump_chances == 33:
|
|
# good_jump = '2-10'
|
|
# elif jump_chances == 34:
|
|
# good_jump = '3-11'
|
|
# elif jump_chances == 35:
|
|
# good_jump = '2-11'
|
|
# else:
|
|
# good_jump = '2-12'
|
|
# steal_string = f'{"*" if batting_card.steal_auto else ""}{good_jump}/- ({batting_card.steal_high}-{batting_card.steal_low})'
|
|
# return steal_string
|
|
|
|
# baserunner_string = ''
|
|
# if curr_play.on_first is not None:
|
|
# runcard = curr_play.on_first.card.batterscouting.battingcard
|
|
# baserunner_string += f'On First: {curr_play.on_first.player.name_card_link('batting')}\nSteal: {steal_string(runcard)}, Run: {runcard.running}'
|
|
# if curr_play.on_second is not None:
|
|
# baserunner_string += f'On Second: {curr_play.on_second.player.name_card_link('batting')}\nSteal: {steal_string(runcard)}, Run: {runcard.running}'
|
|
# if curr_play.on_third is not None:
|
|
# baserunner_string += f'On Third: {curr_play.on_third.player.name_card_link('batting')}\nSteal: {steal_string(runcard)}, Run: {runcard.running}'
|
|
# logger.info(f'gameplay_models - Game.get_scorebug_embed - baserunner_string: {baserunner_string}')
|
|
|
|
# pit_string = f'{curr_play.pitcher.player.name_card_link('pitching')}'
|
|
# bat_string = f'{curr_play.batter.player.name_card_link('batting')}'
|
|
# if len(baserunner_string) > 0:
|
|
# pitchingcard = curr_play.pitcher.card.pitcherscouting.pitchingcard
|
|
# pit_string += f'\nHold: {pitchingcard.hold}, WP: {pitchingcard.wild_pitch}, Bk: {pitchingcard.balk}'
|
|
|
|
# # battingcard = curr_play.batter.card.batterscouting.battingcard
|
|
# # bat_string += f'\nBunt: {battingcard.bunting}, HnR: {battingcard.hit_and_run}'
|
|
|
|
# embed.add_field(
|
|
# name='Pitcher',
|
|
# value=pit_string
|
|
# )
|
|
# embed.add_field(
|
|
# name='Batter',
|
|
# value=bat_string
|
|
# )
|
|
|
|
# if len(baserunner_string) > 0:
|
|
# embed.add_field(name=' ', value=' ', inline=False)
|
|
# embed.add_field(name='Baserunners', value=baserunner_string)
|
|
|
|
# c_query = session.exec(select(PositionRating).where(PositionRating.player_id == curr_play.catcher.card.player.id, PositionRating.position == 'C', PositionRating.variant == curr_play.catcher.card.variant)).all()
|
|
# if len(c_query) > 0:
|
|
# catcher_rating = c_query[0]
|
|
# else:
|
|
# log_exception(PositionNotFoundException, f'No catcher rating found for {curr_play.catcher.card.player.name}')
|
|
|
|
# cat_string = f'{curr_play.catcher.player.name_card_link('batter')}\nArm: {catcher_rating.arm}'
|
|
# embed.add_field(name='Catcher', value=cat_string)
|
|
|
|
# ai_note = curr_play.ai_note
|
|
# logger.info(f'gameplay_models - Game.get_scorebug_embed - ai_note: {ai_note}')
|
|
# if len(ai_note) > 0:
|
|
# gm_name = self.home_team.gmname if self.ai_team == 'home' else self.away_team.gmname
|
|
# embed.add_field(name=f'{gm_name} will...', value=ai_note, inline=False)
|
|
# else:
|
|
# embed.add_field(name=' ', value=' ', inline=False)
|
|
|
|
# if full_length:
|
|
# embed.add_field(
|
|
# name=f'{self.away_team.abbrev} Lineup',
|
|
# value=self.team_lineup(session, self.away_team)
|
|
# )
|
|
# embed.add_field(
|
|
# name=f'{self.home_team.abbrev} Lineup',
|
|
# value=self.team_lineup(session, self.home_team)
|
|
# )
|
|
# else:
|
|
# embed.add_field(
|
|
# name=f'{self.away_team.abbrev} Lineup',
|
|
# value=self.team_lineup(session, self.away_team)
|
|
# )
|
|
# embed.add_field(
|
|
# name=f'{self.home_team.abbrev} Lineup',
|
|
# value=self.team_lineup(session, self.home_team)
|
|
# )
|
|
|
|
# return embed
|
|
|
|
def initialize_play(self, session: Session):
|
|
"""
|
|
Commits new_play
|
|
"""
|
|
existing_play = self.current_play_or_none(session)
|
|
if existing_play is not None:
|
|
return existing_play
|
|
|
|
all_plays = session.exec(select(func.count(Play.id)).where(Play.game == self)).one()
|
|
if all_plays > 0:
|
|
raise PlayInitException(f'{all_plays} plays for game {self.id} already exist, but all are complete.')
|
|
|
|
leadoff_batter, home_pitcher, home_catcher = None, None, None
|
|
home_positions, away_positions = [], []
|
|
for line in [x for x in self.lineups if x.active]:
|
|
if line.team == self.away_team:
|
|
if line.position not in away_positions:
|
|
away_positions.append(line.position)
|
|
if line.batting_order == 1:
|
|
leadoff_batter = line
|
|
else:
|
|
if line.position not in home_positions:
|
|
home_positions.append(line.position)
|
|
if line.position == 'P':
|
|
home_pitcher = line
|
|
elif line.position == 'C':
|
|
home_catcher = line
|
|
|
|
if len(home_positions) != 10:
|
|
e_msg = f'Only {len(home_positions)} players found on home team'
|
|
log_exception(LineupsMissingException, e_msg)
|
|
if len(away_positions) != 10:
|
|
e_msg = f'Only {len(away_positions)} players found on away team'
|
|
log_exception(LineupsMissingException, e_msg)
|
|
if None in [leadoff_batter, home_pitcher, home_catcher]:
|
|
e_msg = f'Could not set the initial pitcher, catcher, and batter'
|
|
log_exception(LineupsMissingException, e_msg)
|
|
|
|
manager_ai_id = ((datetime.datetime.now().day * (self.away_team_id if self.ai_team == 'away' else self.home_team_id)) % 3) + 1
|
|
if manager_ai_id > 3 or manager_ai_id < 1:
|
|
manager_ai_id = 1
|
|
|
|
new_play = Play(
|
|
game=self,
|
|
play_num=1,
|
|
batter=leadoff_batter,
|
|
pitcher=home_pitcher,
|
|
batter_pos=leadoff_batter.position,
|
|
catcher=home_catcher,
|
|
is_tied=True,
|
|
is_new_inning=True,
|
|
managerai_id=manager_ai_id
|
|
)
|
|
session.add(new_play)
|
|
session.commit()
|
|
session.refresh(new_play)
|
|
|
|
new_play.init_ai(session)
|
|
|
|
return new_play
|
|
|
|
def team_lineup(self, session: Session, team: Team) -> str:
|
|
all_lineups = session.exec(select(Lineup).where(Lineup.team == team, Lineup.game == self).order_by(Lineup.batting_order)).all()
|
|
|
|
lineup_val = ''
|
|
for line in all_lineups:
|
|
lineup_val += f'{line.batting_order}. {line.player.name_card_link("batting" if line.position != "P" else "pitching")} {line.position}\n'
|
|
|
|
return lineup_val
|
|
|
|
|
|
class ManagerAi(ManagerAiBase, table=True):
|
|
plays: list['Play'] = Relationship(back_populates='managerai')
|
|
def create_ai(session: Session = None):
|
|
def get_new_ai(this_session: Session):
|
|
all_ai = this_session.exec(select(ManagerAi.id)).all()
|
|
if len(all_ai) == 0:
|
|
logger.info(f'Creating ManagerAI records')
|
|
new_ai = [
|
|
ManagerAi(
|
|
name='Balanced'
|
|
),
|
|
ManagerAi(
|
|
name='Yolo',
|
|
steal=10,
|
|
running=10,
|
|
hold=5,
|
|
catcher_throw=10,
|
|
uncapped_home=10,
|
|
uncapped_third=10,
|
|
uncapped_trail=10,
|
|
bullpen_matchup=3,
|
|
behind_aggression=10,
|
|
ahead_aggression=10,
|
|
decide_throw=10
|
|
),
|
|
ManagerAi(
|
|
name='Safe',
|
|
steal=3,
|
|
running=3,
|
|
hold=8,
|
|
catcher_throw=5,
|
|
uncapped_home=5,
|
|
uncapped_third=3,
|
|
uncapped_trail=5,
|
|
bullpen_matchup=8,
|
|
behind_aggression=5,
|
|
ahead_aggression=1,
|
|
decide_throw=1
|
|
)
|
|
]
|
|
for x in new_ai:
|
|
session.add(x)
|
|
session.commit()
|
|
|
|
if session is None:
|
|
with Session(engine) as session:
|
|
get_new_ai(session)
|
|
else:
|
|
get_new_ai(session)
|
|
|
|
return True
|
|
|
|
def check_jump(self, session: Session, this_game: Game, to_base: Literal[2, 3, 4]) -> JumpResponse | None:
|
|
this_resp = JumpResponse()
|
|
this_play = this_game.current_play_or_none(session)
|
|
if this_play is None:
|
|
raise GameException(f'No game found while checking for jump')
|
|
|
|
num_outs = this_play.starting_outs
|
|
run_diff = this_play.away_score - this_play.home_score
|
|
if this_game.ai_team == 'home':
|
|
run_diff = run_diff * -1
|
|
|
|
if to_base == 2:
|
|
match self.steal:
|
|
case 10:
|
|
this_resp.min_safe = 12 + num_outs
|
|
case self.steal if self.steal > 8 and run_diff <= 5:
|
|
this_resp.min_safe = 13 + num_outs
|
|
case self.steal if self.steal > 6 and run_diff <= 5:
|
|
this_resp.min_safe = 14 + num_outs
|
|
case self.steal if self.steal > 4 and num_outs < 2 and run_diff <= 5:
|
|
this_resp.min_safe = 15 + num_outs
|
|
case self.steal if self.steal > 2 and num_outs < 2 and run_diff <= 5:
|
|
this_resp.min_safe = 16 + num_outs
|
|
case _:
|
|
this_resp = 17 + num_outs
|
|
|
|
if self.steal > 7 and num_outs < 2 and run_diff <= 5:
|
|
this_resp.run_if_auto_jump = True
|
|
elif self.steal < 5:
|
|
this_resp.must_auto_jump = True
|
|
|
|
elif to_base == 3:
|
|
match self.steal:
|
|
case 10:
|
|
this_resp.min_safe = 12 + num_outs
|
|
case self.steal if self.steal > 6 and num_outs < 2 and run_diff <= 5:
|
|
this_resp.min_safe = 15 + num_outs
|
|
case _:
|
|
this_resp.min_safe = None
|
|
|
|
if self.steal == 10 and num_outs < 2 and run_diff <= 5:
|
|
this_resp.run_if_auto_jump = True
|
|
elif self.steal <= 5:
|
|
this_resp.must_auto_jump = True
|
|
|
|
elif run_diff in [-1, 0]:
|
|
if self.steal == 10:
|
|
this_resp.min_safe = 5
|
|
elif self.steal > 5:
|
|
this_resp.min_safe = 7
|
|
elif this_play.inning_num > 7 and self.steal >= 5:
|
|
this_resp.min_safe = 6
|
|
|
|
return this_resp
|
|
|
|
def tag_from_second(self, session: Session, this_game: Game) -> TagResponse:
|
|
this_resp = TagResponse()
|
|
this_play = this_game.current_play_or_none(session)
|
|
if this_play is None:
|
|
raise GameException(f'No game found while checking tag_from_second')
|
|
|
|
ai_rd = this_play.ai_run_diff()
|
|
aggression_mod = abs(self.ahead_aggression - 5 if ai_rd > 0 else self.behind_aggression - 5)
|
|
adjusted_running = self.running + aggression_mod
|
|
|
|
if adjusted_running >= 8:
|
|
this_resp.min_safe = 4
|
|
elif adjusted_running >= 5:
|
|
this_resp.min_safe = 7
|
|
else:
|
|
this_resp.min_safe = 10
|
|
|
|
if this_play.starting_outs == 1:
|
|
this_resp.min_safe -= 2
|
|
else:
|
|
this_resp.min_safe += 2
|
|
|
|
return this_resp
|
|
|
|
def throw_at_uncapped(self, session: Session, this_game: Game) -> ThrowResponse:
|
|
this_resp = ThrowResponse()
|
|
this_play = this_game.current_play_or_none(session)
|
|
if this_play is None:
|
|
raise GameException(f'No game found while checking throw_at_uncapped')
|
|
|
|
ai_rd = this_play.ai_run_diff()
|
|
aggression = self.ahead_aggression if ai_rd > 0 else self.behind_aggression
|
|
current_outs = this_play.starting_outs + this_play.outs
|
|
|
|
if ai_rd > 5:
|
|
if self.ahead_aggression > 5:
|
|
this_resp.at_trail_runner = True
|
|
this_resp.trail_max_safe_delta = -4 + current_outs
|
|
else:
|
|
this_resp.cutoff = True
|
|
elif ai_rd > 2:
|
|
if self.ahead_aggression > 8:
|
|
this_resp.at_trail_runner = True
|
|
this_resp.trail_max_safe_delta = -4 + current_outs
|
|
elif ai_rd > 0:
|
|
if self.ahead_aggression > 8:
|
|
this_resp.at_trail_runner = True
|
|
this_resp.trail_max_safe_delta = -6 + current_outs
|
|
elif ai_rd > -3:
|
|
if self.behind_aggression < 5:
|
|
this_resp.at_trail_runner = True
|
|
this_resp.trail_max_safe_delta = -6 + current_outs
|
|
elif ai_rd > -6:
|
|
if self.behind_aggression < 5:
|
|
this_resp.at_trail_runner = True
|
|
this_resp.trail_max_safe_delta = -4 + current_outs
|
|
else:
|
|
if self.behind_aggression < 5:
|
|
this_resp.at_trail_runner = True
|
|
this_resp.trail_max_safe_delta = -4
|
|
|
|
return this_resp
|
|
|
|
def uncapped_advance(self, session: Session, this_game: Game, lead_base: int, trail_base: int) -> UncappedRunResponse:
|
|
this_resp = UncappedRunResponse()
|
|
this_play = this_game.current_play_or_none(session)
|
|
if this_play is None:
|
|
raise GameException(f'No game found while checking uncapped_advance_lead')
|
|
|
|
ai_rd = this_play.ai_run_diff()
|
|
aggression = self.ahead_aggression - 5 if ai_rd > 0 else self.behind_aggression - 5
|
|
|
|
if ai_rd > 4:
|
|
if lead_base == 4:
|
|
this_resp.min_safe = 16 - this_play.starting_outs - aggression
|
|
this_resp.send_trail = True
|
|
this_resp.trail_min_safe = 10 - aggression - this_play.starting_outs - this_play.outs
|
|
elif lead_base == 3:
|
|
this_resp.min_safe = 14 + (this_play.starting_outs * 2) - aggression
|
|
if this_play.starting_outs + this_play.outs >= 2:
|
|
this_resp.send_trail = False
|
|
elif ai_rd > 1 or ai_rd < -2:
|
|
if lead_base == 4:
|
|
this_resp.min_safe = 12 - this_play.starting_outs - aggression
|
|
this_resp.send_trail = True
|
|
this_resp.trail_min_safe = 10 - aggression - this_play.starting_outs - this_play.outs
|
|
elif lead_base == 3:
|
|
this_resp.min_safe = 12 + (this_play.starting_outs * 2) - (aggression * 2)
|
|
if this_play.starting_outs + this_play.outs >= 2:
|
|
this_resp.send_trail = False
|
|
else:
|
|
if lead_base == 4:
|
|
this_resp.min_safe = 10 - this_play.starting_outs - aggression
|
|
this_resp.send_trail = True
|
|
this_resp.trail_min_safe = 2
|
|
elif lead_base == 3:
|
|
this_resp.min_safe = 14 + (this_play.starting_outs * 2) - aggression
|
|
if this_play.starting_outs + this_play.outs >= 2:
|
|
this_resp.send_trail = False
|
|
|
|
if this_resp.min_safe > 20:
|
|
this_resp.min_safe = 20
|
|
if this_resp.min_safe < 1:
|
|
this_resp.min_safe = 1
|
|
if this_resp.trail_min_safe > 20:
|
|
this_resp.min_safe = 20
|
|
if this_resp.trail_min_safe < 1:
|
|
this_resp.min_safe = 1
|
|
|
|
return this_resp
|
|
|
|
def defense_alignment(self, session: Session, this_game: Game) -> DefenseResponse:
|
|
this_resp = DefenseResponse()
|
|
this_play = this_game.current_play_or_none(session)
|
|
if this_play is None:
|
|
raise GameException(f'No game found while checking uncapped_advance_lead')
|
|
|
|
ai_rd = this_play.ai_run_diff()
|
|
aggression = self.ahead_aggression - 5 if ai_rd > 0 else self.behind_aggression - 5
|
|
|
|
if self.starting_outs == 2 and self.on_base_code > 0:
|
|
if self.on_base_code == 1:
|
|
this_resp.hold_first = True
|
|
elif self.on_base_code == 2:
|
|
this_resp.hold_second = True
|
|
elif self.on_base_code in [4, 5, 7]:
|
|
this_resp.hold_first = True
|
|
this_resp.hold_second = True
|
|
# elif self.on_base_code == 5:
|
|
# ai_note += f'- hold the runner on first\n'
|
|
elif self.on_base_code == 6:
|
|
ai_note += f'- hold {self.on_second.player.name} on 2nd\n'
|
|
elif self.on_base_code in [1, 5]:
|
|
runner = self.on_first.player
|
|
if self.on_first.card.batterscouting.battingcard.steal_auto:
|
|
ai_note += f'- hold {runner.name} on 1st\n'
|
|
elif self.on_base_code in [2, 4]:
|
|
if self.on_second.card.batterscouting.battingcard.steal_low + max(self.pitcher.card.pitcherscouting.pitchingcard.hold, 5) >= 14:
|
|
ai_note += f'- hold {self.on_second.player.name} on 2nd\n'
|
|
|
|
# Defensive Alignment
|
|
if self.on_third and self.starting_outs < 2:
|
|
if self.could_walkoff:
|
|
ai_note += f'- play the outfield and infield in'
|
|
elif abs(self.away_score - self.home_score) <= 3:
|
|
ai_note += f'- play the whole infield in\n'
|
|
else:
|
|
ai_note += f'- play the corners in\n'
|
|
|
|
if len(ai_note) == 0 and self.on_base_code > 0:
|
|
ai_note += f'- play straight up\n'
|
|
|
|
|
|
class CardsetBase(SQLModel):
|
|
id: int | None = Field(default=None, primary_key=True)
|
|
name: str
|
|
ranked_legal: bool | None = Field(default=False)
|
|
|
|
|
|
class Cardset(CardsetBase, table=True):
|
|
game_links: list[GameCardsetLink] = Relationship(back_populates='cardset', cascade_delete=True)
|
|
players: list['Player'] = Relationship(back_populates='cardset')
|
|
|
|
|
|
class PlayerBase(SQLModel):
|
|
id: int | None = Field(primary_key=True)
|
|
name: str
|
|
cost: int
|
|
image: str
|
|
mlbclub: str
|
|
franchise: str
|
|
cardset_id: int | None = Field(default=None, foreign_key='cardset.id')
|
|
set_num: int
|
|
rarity_id: int | None = Field(default=None)
|
|
pos_1: str
|
|
description: str
|
|
quantity: int | None = Field(default=999)
|
|
image2: str | None = Field(default=None)
|
|
pos_2: str | None = Field(default=None)
|
|
pos_3: str | None = Field(default=None)
|
|
pos_4: str | None = Field(default=None)
|
|
pos_5: str | None = Field(default=None)
|
|
pos_6: str | None = Field(default=None)
|
|
pos_7: str | None = Field(default=None)
|
|
pos_8: str | None = Field(default=None)
|
|
headshot: str | None = Field(default=None)
|
|
vanity_card: str | None = Field(default=None)
|
|
strat_code: str | None = Field(default=None)
|
|
bbref_id: str | None = Field(default=None)
|
|
fangr_id: str | None = Field(default=None)
|
|
mlbplayer_id: int | None = Field(default=None)
|
|
created: datetime.datetime = Field(default_factory=datetime.datetime.now, nullable=True)
|
|
|
|
@field_validator('pos_1', 'pos_2', 'pos_3', 'pos_4', 'pos_5', 'pos_6', 'pos_7', 'pos_8')
|
|
def uppercase_strings(cls, value: str) -> str:
|
|
if value is not None:
|
|
return value.upper()
|
|
else:
|
|
return value
|
|
|
|
@property
|
|
def p_card_url(self):
|
|
if 'pitching' in self.image:
|
|
return self.image
|
|
elif self.image2 is not None and 'pitching' in self.image2:
|
|
return self.image2
|
|
else:
|
|
logger.error(f'gameplay_models - PlayerBase - pitching card url not found for {self.id}. {self.description} {self.name}')
|
|
return self.image
|
|
|
|
@property
|
|
def b_card_url(self):
|
|
if 'batting' in self.image:
|
|
return self.image
|
|
elif self.image2 is not None and 'batting' in self.image2:
|
|
return self.image2
|
|
else:
|
|
logger.error(f'gameplay_models - PlayerBase - batting card url not found for {self.id}. {self.description} {self.name}')
|
|
return self.image
|
|
|
|
def name_card_link(self, which: Literal['pitching', 'batting']):
|
|
if which == 'pitching':
|
|
return f'[{self.name}]({self.p_card_url})'
|
|
else:
|
|
return f'[{self.name}]({self.b_card_url})'
|
|
|
|
|
|
class Player(PlayerBase, table=True):
|
|
cardset: Cardset = Relationship(back_populates='players')
|
|
cards: list['Card'] = Relationship(back_populates='player', cascade_delete=True)
|
|
lineups: list['Lineup'] = Relationship(back_populates='player', cascade_delete=True)
|
|
positions: list['PositionRating'] = Relationship(back_populates='player', cascade_delete=True)
|
|
|
|
@property
|
|
def name_with_desc(self):
|
|
return f'{self.description} {self.name}'
|
|
|
|
@property
|
|
def batter_card_url(self):
|
|
if 'batting' in self.image:
|
|
return self.image
|
|
elif 'batting' in self.image2:
|
|
return self.image2
|
|
else:
|
|
return None
|
|
|
|
@property
|
|
def pitcher_card_url(self):
|
|
if 'pitching' in self.image:
|
|
return self.image
|
|
elif 'pitching' in self.image2:
|
|
return self.image2
|
|
else:
|
|
return None
|
|
|
|
|
|
def player_description(player: Player = None, player_dict: dict = None) -> str:
|
|
if player is None and player_dict is None:
|
|
err = 'One of "player" or "player_dict" must be included to get full description'
|
|
logger.error(f'gameplay_models - player_description - {err}')
|
|
raise TypeError(err)
|
|
|
|
if player is not None:
|
|
return f'{player.description} {player.name}'
|
|
|
|
r_val = f'{player_dict['description']}'
|
|
if 'name' in player_dict:
|
|
r_val += f' {player_dict["name"]}'
|
|
elif 'p_name' in player_dict:
|
|
r_val += f' {player_dict["p_name"]}'
|
|
return r_val
|
|
|
|
|
|
class BattingCardBase(SQLModel):
|
|
id: int | None = Field(default=None, primary_key=True)
|
|
variant: int | None = Field(default=0)
|
|
steal_low: int = Field(default=0, ge=0, le=20)
|
|
steal_high: int = Field(default=0, ge=0, le=20)
|
|
steal_auto: bool = Field(default=False)
|
|
steal_jump: float = Field(default=0.0, ge=0.0, le=1.0)
|
|
bunting: str = Field(default='C')
|
|
hit_and_run: str = Field(default='C')
|
|
running: int = Field(default=10, ge=1, le=20)
|
|
offense_col: int = Field(ge=1, le=3)
|
|
hand: str
|
|
created: datetime.datetime = Field(default_factory=datetime.datetime.now, nullable=True)
|
|
# created: datetime.datetime | None = Field(sa_column_kwargs={"server_default": text("CURRENT_TIMESTAMP"),})
|
|
|
|
@field_validator('hand')
|
|
def lowercase_hand(cls, value: str) -> str:
|
|
return value.lower()
|
|
|
|
|
|
class BattingCard(BattingCardBase, table=True):
|
|
pass
|
|
|
|
|
|
class BattingRatingsBase(SQLModel):
|
|
id: int | None = Field(default=None, primary_key=True)
|
|
homerun: float = Field(default=0.0, ge=0.0, le=108.0)
|
|
bp_homerun: float = Field(default=0.0, ge=0.0, le=108.0)
|
|
triple: float = Field(default=0.0, ge=0.0, le=108.0)
|
|
double_three: float = Field(default=0.0, ge=0.0, le=108.0)
|
|
double_two: float = Field(default=0.0, ge=0.0, le=108.0)
|
|
double_pull: float = Field(default=0.0, ge=0.0, le=108.0)
|
|
single_two: float = Field(default=0.0, ge=0.0, le=108.0)
|
|
single_one: float = Field(default=0.0, ge=0.0, le=108.0)
|
|
single_center: float = Field(default=0.0, ge=0.0, le=108.0)
|
|
bp_single: float = Field(default=0.0, ge=0.0, le=10.0)
|
|
hbp: float = Field(default=0.0, ge=0.0, le=108.0)
|
|
walk: float = Field(default=0.0, ge=0.0, le=108.0)
|
|
strikeout: float = Field(default=0.0, ge=0.0, le=108.0)
|
|
lineout: float = Field(default=0.0, ge=0.0, le=108.0)
|
|
popout: float = Field(default=0.0, ge=0.0, le=108.0)
|
|
flyout_a: float = Field(default=0.0, ge=0.0, le=108.0)
|
|
flyout_bq: float = Field(default=0.0, ge=0.0, le=108.0)
|
|
flyout_lf_b: float = Field(default=0.0, ge=0.0, le=108.0)
|
|
flyout_rf_b: float = Field(default=0.0, ge=0.0, le=108.0)
|
|
groundout_a: float = Field(default=0.0, ge=0.0, le=108.0)
|
|
groundout_b: float = Field(default=0.0, ge=0.0, le=108.0)
|
|
groundout_c: float = Field(default=0.0, ge=0.0, le=108.0)
|
|
avg: float = Field(default=0.0, ge=0.0, le=1.0)
|
|
obp: float = Field(default=0.0, ge=0.0, le=1.0)
|
|
slg: float = Field(default=0.0, ge=0.0, le=4.0)
|
|
pull_rate: float = Field(default=0.0, ge=0.0, le=1.0)
|
|
center_rate: float = Field(default=0.0, ge=0.0, le=1.0)
|
|
slap_rate: float = Field(default=0.0, ge=0.0, le=1.0)
|
|
created: datetime.datetime = Field(default_factory=datetime.datetime.now, nullable=True)
|
|
|
|
|
|
class BattingRatings(BattingRatingsBase, table=True):
|
|
pass
|
|
|
|
|
|
class BatterScoutingBase(SQLModel):
|
|
id: int | None = Field(default=None, primary_key=True)
|
|
battingcard_id: int | None = Field(default=None, foreign_key='battingcard.id')
|
|
ratings_vl_id: int | None = Field(default=None, foreign_key='battingratings.id')
|
|
ratings_vr_id: int | None = Field(default=None, foreign_key='battingratings.id')
|
|
created: datetime.datetime = Field(default_factory=datetime.datetime.now, nullable=True)
|
|
|
|
|
|
class BatterScouting(BatterScoutingBase, table=True):
|
|
battingcard: BattingCard = Relationship() #back_populates='batterscouting')
|
|
ratings_vl: BattingRatings = Relationship(
|
|
sa_relationship_kwargs=dict(foreign_keys="[BatterScouting.ratings_vl_id]")
|
|
)
|
|
ratings_vr: BattingRatings = Relationship(
|
|
sa_relationship_kwargs=dict(foreign_keys="[BatterScouting.ratings_vr_id]")
|
|
)
|
|
cards: list['Card'] = Relationship(back_populates='batterscouting', cascade_delete=True)
|
|
|
|
|
|
class PitchingCardBase(SQLModel):
|
|
id: int | None = Field(default=None, primary_key=True)
|
|
variant: int | None = Field(default=0)
|
|
balk: int = Field(default=0, ge=0, le=20)
|
|
wild_pitch: int = Field(default=0, ge=0, le=20)
|
|
hold: int = Field(default=0, ge=-9, le=9)
|
|
starter_rating: int = Field(default=1, ge=1, le=10)
|
|
relief_rating: int = Field(default=1, ge=1, le=10)
|
|
closer_rating: int | None = Field(default=None, ge=0, le=9)
|
|
offense_col: int = Field(ge=1, le=3)
|
|
batting: str = Field(default='#1WR-C')
|
|
hand: str
|
|
created: datetime.datetime = Field(default_factory=datetime.datetime.now, nullable=True)
|
|
|
|
@field_validator('hand')
|
|
def lowercase_hand(cls, value: str) -> str:
|
|
return value.lower()
|
|
|
|
|
|
class PitchingCard(PitchingCardBase, table=True):
|
|
pass
|
|
|
|
|
|
class PitchingRatingsBase(SQLModel):
|
|
id: int | None = Field(default=None, primary_key=True)
|
|
homerun: float = Field(default=0.0, ge=0.0, le=108.0)
|
|
bp_homerun: float = Field(default=0.0, ge=0.0, le=108.0)
|
|
triple: float = Field(default=0.0, ge=0.0, le=108.0)
|
|
double_three: float = Field(default=0.0, ge=0.0, le=108.0)
|
|
double_two: float = Field(default=0.0, ge=0.0, le=108.0)
|
|
double_cf: float = Field(default=0.0, ge=0.0, le=108.0)
|
|
single_two: float = Field(default=0.0, ge=0.0, le=108.0)
|
|
single_one: float = Field(default=0.0, ge=0.0, le=108.0)
|
|
single_center: float = Field(default=0.0, ge=0.0, le=108.0)
|
|
bp_single: float = Field(default=0.0, ge=0.0, le=108.0)
|
|
hbp: float = Field(default=0.0, ge=0.0, le=108.0)
|
|
walk: float = Field(default=0.0, ge=0.0, le=108.0)
|
|
strikeout: float = Field(default=0.0, ge=0.0, le=108.0)
|
|
flyout_lf_b: float = Field(default=0.0, ge=0.0, le=108.0)
|
|
flyout_cf_b: float = Field(default=0.0, ge=0.0, le=108.0)
|
|
flyout_rf_b: float = Field(default=0.0, ge=0.0, le=108.0)
|
|
groundout_a: float = Field(default=0.0, ge=0.0, le=108.0)
|
|
groundout_b: float = Field(default=0.0, ge=0.0, le=108.0)
|
|
xcheck_p: float = Field(default=0.0, ge=0.0, le=108.0)
|
|
xcheck_c: float = Field(default=0.0, ge=0.0, le=108.0)
|
|
xcheck_1b: float = Field(default=0.0, ge=0.0, le=108.0)
|
|
xcheck_2b: float = Field(default=0.0, ge=0.0, le=108.0)
|
|
xcheck_3b: float = Field(default=0.0, ge=0.0, le=108.0)
|
|
xcheck_ss: float = Field(default=0.0, ge=0.0, le=108.0)
|
|
xcheck_lf: float = Field(default=0.0, ge=0.0, le=108.0)
|
|
xcheck_cf: float = Field(default=0.0, ge=0.0, le=108.0)
|
|
xcheck_rf: float = Field(default=0.0, ge=0.0, le=108.0)
|
|
avg: float = Field(default=0.0, ge=0.0, le=1.0)
|
|
obp: float = Field(default=0.0, ge=0.0, le=1.0)
|
|
slg: float = Field(default=0.0, ge=0.0, le=4.0)
|
|
created: datetime.datetime = Field(default_factory=datetime.datetime.now, nullable=True)
|
|
|
|
|
|
class PitchingRatings(PitchingRatingsBase, table=True):
|
|
pass
|
|
|
|
|
|
class PitcherScoutingBase(SQLModel):
|
|
id: int | None = Field(default=None, primary_key=True)
|
|
pitchingcard_id: int | None = Field(default=None, foreign_key='pitchingcard.id',)
|
|
ratings_vl_id: int | None = Field(default=None, foreign_key='pitchingratings.id')
|
|
ratings_vr_id: int | None = Field(default=None, foreign_key='pitchingratings.id')
|
|
created: datetime.datetime = Field(default_factory=datetime.datetime.now, nullable=True)
|
|
|
|
|
|
class PitcherScouting(PitcherScoutingBase, table=True):
|
|
pitchingcard: PitchingCard = Relationship()
|
|
ratings_vl: PitchingRatings = Relationship(
|
|
sa_relationship_kwargs=dict(foreign_keys="[PitcherScouting.ratings_vl_id]")
|
|
)
|
|
ratings_vr: PitchingRatings = Relationship(
|
|
sa_relationship_kwargs=dict(foreign_keys="[PitcherScouting.ratings_vr_id]")
|
|
)
|
|
cards: list['Card'] = Relationship(back_populates='pitcherscouting', cascade_delete=True)
|
|
|
|
|
|
class CardBase(SQLModel):
|
|
id: int | None = Field(default=None, primary_key=True)
|
|
player_id: int = Field(foreign_key='player.id', index=True, ondelete='CASCADE')
|
|
team_id: int = Field(foreign_key='team.id', index=True, ondelete='CASCADE')
|
|
batterscouting_id: int | None = Field(default=None, foreign_key='batterscouting.id', ondelete='CASCADE')
|
|
pitcherscouting_id: int | None = Field(default=None, foreign_key='pitcherscouting.id', ondelete='CASCADE')
|
|
variant: int | None = Field(default=0)
|
|
created: datetime.datetime = Field(default_factory=datetime.datetime.now, nullable=True)
|
|
|
|
|
|
class Card(CardBase, table=True):
|
|
player: Player = Relationship(back_populates='cards')
|
|
team: Team = Relationship(back_populates='cards')
|
|
lineups: list['Lineup'] = Relationship(back_populates='card', cascade_delete=True)
|
|
batterscouting: BatterScouting = Relationship(back_populates='cards')
|
|
pitcherscouting: PitcherScouting = Relationship(back_populates='cards')
|
|
|
|
|
|
class PositionRatingBase(SQLModel):
|
|
__table_args__ = (UniqueConstraint("player_id", "variant", "position"),)
|
|
id: int | None = Field(default=None, primary_key=True)
|
|
player_id: int = Field(foreign_key='player.id', index=True, ondelete='CASCADE')
|
|
variant: int = Field(default=0, index=True)
|
|
position: str = Field(index=True, include=['P', 'C', '1B', '2B', '3B', 'SS', 'LF', 'CF', 'RF'])
|
|
innings: int = Field(default=0)
|
|
range: int = Field(default=5)
|
|
error: int = Field(default=0)
|
|
arm: int | None = Field(default=None)
|
|
pb: int | None = Field(default=None)
|
|
overthrow: int | None = Field(default=None)
|
|
created: datetime.datetime = Field(default_factory=datetime.datetime.now, nullable=True)
|
|
|
|
|
|
class PositionRating(PositionRatingBase, table=True):
|
|
player: Player = Relationship(back_populates='positions')
|
|
|
|
|
|
class Lineup(SQLModel, table=True):
|
|
id: int | None = Field(default=None, primary_key=True)
|
|
position: str = Field(index=True)
|
|
batting_order: int = Field(index=True)
|
|
after_play: int | None = Field(default=0)
|
|
replacing_id: int | None = Field(default=None)
|
|
active: bool = Field(default=True, index=True)
|
|
is_fatigued: bool | None = Field(default=None)
|
|
|
|
game_id: int = Field(foreign_key='game.id', index=True, ondelete='CASCADE')
|
|
game: Game = Relationship(back_populates='lineups')
|
|
|
|
team_id: int = Field(foreign_key='team.id', index=True, ondelete='CASCADE')
|
|
team: Team = Relationship(back_populates='lineups')
|
|
|
|
player_id: int = Field(foreign_key='player.id', index=True, ondelete='CASCADE')
|
|
player: Player = Relationship(back_populates='lineups')
|
|
|
|
card_id: int = Field(foreign_key='card.id', index=True, ondelete='CASCADE')
|
|
card: Card = Relationship(back_populates='lineups')
|
|
|
|
@field_validator('position')
|
|
def uppercase_strings(cls, value: str) -> str:
|
|
return value.upper()
|
|
|
|
# TODO: add function to return string value of game stats
|
|
|
|
|
|
class PlayBase(SQLModel):
|
|
id: int | None = Field(default=None, primary_key=True)
|
|
game_id: int = Field(foreign_key='game.id')
|
|
play_num: int
|
|
batter_id: int = Field(foreign_key='lineup.id')
|
|
pitcher_id: int = Field(foreign_key='lineup.id')
|
|
on_base_code: int = Field(default=0)
|
|
inning_half: str = Field(default='top')
|
|
inning_num: int = Field(default=1, ge=1)
|
|
batting_order: int = Field(default=1, ge=1, le=9)
|
|
starting_outs: int = Field(default=0, ge=0, le=2)
|
|
away_score: int = Field(default=0, ge=0)
|
|
home_score: int = Field(default=0, ge=0)
|
|
batter_pos: str | None = Field(default=None)
|
|
in_pow: bool = Field(default=False)
|
|
|
|
on_first_id: int | None = Field(default=None, foreign_key='lineup.id')
|
|
on_first_final: int | None = Field(default=None) # None = out, 1-4 = base
|
|
on_second_id: int | None = Field(default=None, foreign_key='lineup.id')
|
|
on_second_final: int | None = Field(default=None) # None = out, 1-4 = base
|
|
on_third_id: int | None = Field(default=None, foreign_key='lineup.id')
|
|
on_third_final: int | None = Field(default=None) # None = out, 1-4 = base
|
|
batter_final: int | None = Field(default=None) # None = out, 1-4 = base
|
|
|
|
pa: int = Field(default=1, ge=0, le=1)
|
|
ab: int = Field(default=1, ge=0, le=1)
|
|
run: int = Field(default=0, ge=0, le=1)
|
|
e_run: int = Field(default=0, ge=0, le=1)
|
|
hit: int = Field(default=0, ge=0, le=1)
|
|
rbi: int = Field(default=0)
|
|
double: int = Field(default=0, ge=0, le=1)
|
|
triple: int = Field(default=0, ge=0, le=1)
|
|
homerun: int = Field(default=0, ge=0, le=1)
|
|
bb: int = Field(default=0, ge=0, le=1)
|
|
so: int = Field(default=0, ge=0, le=1)
|
|
hbp: int = Field(default=0, ge=0, le=1)
|
|
sac: int = Field(default=0, ge=0, le=1)
|
|
ibb: int = Field(default=0, ge=0, le=1)
|
|
gidp: int = Field(default=0, ge=0, le=1)
|
|
bphr: int = Field(default=0, ge=0, le=1)
|
|
bpfo: int = Field(default=0, ge=0, le=1)
|
|
bp1b: int = Field(default=0, ge=0, le=1)
|
|
bplo: int = Field(default=0, ge=0, le=1)
|
|
sb: int = Field(default=0, ge=0, le=1)
|
|
cs: int = Field(default=0, ge=0, le=1)
|
|
outs: int = Field(default=0, ge=0, le=3)
|
|
|
|
wpa: float = Field(default=0)
|
|
re24: float = Field(default=0)
|
|
|
|
catcher_id: int = Field(foreign_key='lineup.id')
|
|
defender_id: int | None = Field(default=None, foreign_key='lineup.id')
|
|
runner_id: int | None = Field(default=None, foreign_key='lineup.id')
|
|
|
|
check_pos: str | None = Field(default=None)
|
|
error: int = Field(default=0)
|
|
wild_pitch: int = Field(default=0, ge=0, le=1)
|
|
passed_ball: int = Field(default=0, ge=0, le=1)
|
|
pick_off: int = Field(default=0, ge=0, le=1)
|
|
balk: int = Field(default=0, ge=0, le=1)
|
|
complete: bool = Field(default=False)
|
|
locked: bool = Field(default=False)
|
|
is_go_ahead: bool = Field(default=False)
|
|
is_tied: bool = Field(default=False)
|
|
is_new_inning: bool = Field(default=False)
|
|
managerai_id: int | None = Field(default=None, foreign_key='managerai.id')
|
|
|
|
@field_validator('inning_half')
|
|
def lowercase_strings(cls, value: str) -> str:
|
|
return value.lower()
|
|
|
|
@field_validator('check_pos', 'batter_pos')
|
|
def uppercase_strings(cls, value: str) -> str:
|
|
return value.upper()
|
|
|
|
def ai_run_diff(self):
|
|
if self.game.ai_team == 'away':
|
|
return self.away_score - self.home_score
|
|
else:
|
|
return self.home_score - self.away_score
|
|
|
|
|
|
class Play(PlayBase, table=True):
|
|
game: Game = Relationship(back_populates='plays')
|
|
batter: Lineup = Relationship(
|
|
sa_relationship_kwargs=dict(foreign_keys="[Play.batter_id]")
|
|
)
|
|
pitcher: Lineup = Relationship(
|
|
sa_relationship_kwargs=dict(foreign_keys="[Play.pitcher_id]")
|
|
)
|
|
on_first: Lineup = Relationship(
|
|
sa_relationship_kwargs=dict(foreign_keys="[Play.on_first_id]")
|
|
)
|
|
on_second: Lineup = Relationship(
|
|
sa_relationship_kwargs=dict(foreign_keys="[Play.on_second_id]")
|
|
)
|
|
on_third: Lineup = Relationship(
|
|
sa_relationship_kwargs=dict(foreign_keys="[Play.on_third_id]")
|
|
)
|
|
catcher: Lineup = Relationship(
|
|
sa_relationship_kwargs=dict(foreign_keys="[Play.catcher_id]")
|
|
)
|
|
defender: Lineup = Relationship(
|
|
sa_relationship_kwargs=dict(foreign_keys="[Play.defender_id]")
|
|
)
|
|
runner: Lineup = Relationship(
|
|
sa_relationship_kwargs=dict(foreign_keys="[Play.runner_id]")
|
|
)
|
|
managerai: ManagerAi = Relationship(back_populates='plays')
|
|
|
|
def init_ai(self, session: Session):
|
|
id = ((datetime.datetime.now().day * self.batter.team.id) % 3) + 1
|
|
if id > 3 or id < 1:
|
|
self.managerai_id = 1
|
|
else:
|
|
self.managerai_id = id
|
|
|
|
session.add(self)
|
|
session.commit()
|
|
|
|
@property
|
|
def scorebug_ascii(self):
|
|
occupied = '●'
|
|
unoccupied = '○'
|
|
|
|
first_base = unoccupied if not self.on_first else occupied
|
|
second_base = unoccupied if not self.on_second else occupied
|
|
third_base = unoccupied if not self.on_third else occupied
|
|
half = '▲' if self.inning_half == 'top' else '▼'
|
|
|
|
if self.game.active:
|
|
inning = f'{half} {self.inning_num}'
|
|
outs = f'{self.starting_outs} Out{"s" if self.starting_outs != 1 else ""}'
|
|
else:
|
|
inning = f'F/{self.inning_num if self.inning_half == "bot" else self.inning_num - 1}'
|
|
outs = ''
|
|
|
|
game_string = f'```\n' \
|
|
f'{self.game.away_team.abbrev.replace("Gauntlet-", ""): ^5}{self.away_score: ^3} {second_base}{inning: >10}\n' \
|
|
f'{self.game.home_team.abbrev.replace("Gauntlet-", ""): ^5}{self.home_score: ^3} {third_base} {first_base}{outs: >8}\n```'
|
|
|
|
return game_string
|
|
|
|
@property
|
|
def pitching_ai_note(self) -> str:
|
|
ai_note = ''
|
|
# Holding Baserunners
|
|
if self.starting_outs == 2 and self.on_base_code > 0:
|
|
if self.on_base_code == 1:
|
|
ai_note += f'- hold {self.on_first.player.name}\n'
|
|
elif self.on_base_code == 2:
|
|
ai_note += f'- hold {self.on_second.player.name}\n'
|
|
elif self.on_base_code in [4, 5, 7]:
|
|
ai_note += f'- hold {self.on_first.player.name} on first\n'
|
|
# elif self.on_base_code == 5:
|
|
# ai_note += f'- hold the runner on first\n'
|
|
elif self.on_base_code == 6:
|
|
ai_note += f'- hold {self.on_second.player.name} on 2nd\n'
|
|
elif self.on_base_code in [1, 5]:
|
|
runner = self.on_first.player
|
|
if self.on_first.card.batterscouting.battingcard.steal_auto:
|
|
ai_note += f'- hold {runner.name} on 1st\n'
|
|
elif self.on_base_code in [2, 4]:
|
|
if self.on_second.card.batterscouting.battingcard.steal_low + max(self.pitcher.card.pitcherscouting.pitchingcard.hold, 5) >= 14:
|
|
ai_note += f'- hold {self.on_second.player.name} on 2nd\n'
|
|
|
|
# Defensive Alignment
|
|
if self.on_third and self.starting_outs < 2:
|
|
if self.could_walkoff:
|
|
ai_note += f'- play the outfield and infield in'
|
|
elif abs(self.away_score - self.home_score) <= 3:
|
|
ai_note += f'- play the whole infield in\n'
|
|
else:
|
|
ai_note += f'- play the corners in\n'
|
|
|
|
if len(ai_note) == 0 and self.on_base_code > 0:
|
|
ai_note += f'- play straight up\n'
|
|
|
|
return ai_note
|
|
|
|
@property
|
|
def batting_ai_note(self) -> str:
|
|
ai_note = '' # TODO: migrate Manager AI to their own local model
|
|
|
|
return ai_note
|
|
|
|
@property
|
|
def ai_note(self) -> str: # TODO: test these three functions with specific OBCs
|
|
if self.inning_half == 'top':
|
|
if self.game.ai_team == 'away':
|
|
return self.batting_ai_note
|
|
else:
|
|
return self.pitching_ai_note
|
|
else:
|
|
if self.game.ai_team == 'away':
|
|
return self.pitching_ai_note
|
|
else:
|
|
return self.batting_ai_note
|
|
|
|
@property
|
|
def ai_is_batting(self) -> bool:
|
|
if self.game.ai_team is None:
|
|
return False
|
|
|
|
if (self.game.ai_team == 'away' and self.inning_half == 'top') or (self.game.ai_team == 'home' and self.inning_half == 'bot'):
|
|
return True
|
|
else:
|
|
return False
|
|
|
|
@property
|
|
def could_walkoff(self) -> bool:
|
|
if self.inning_half == 'bot' and self.on_third is not None:
|
|
runs_needed = self.away_score - self.home_score + 1
|
|
|
|
if runs_needed == 2 or (self.home_score - self.away_score == 9):
|
|
return True
|
|
|
|
else:
|
|
return False
|
|
|
|
|
|
"""
|
|
BEGIN DEVELOPMENT HELPERS
|
|
"""
|
|
|
|
|
|
def create_db_and_tables():
|
|
SQLModel.metadata.create_all(engine)
|
|
ManagerAi.create_ai()
|
|
|
|
|
|
def create_test_games():
|
|
with Session(engine) as session:
|
|
game_1 = Game(
|
|
away_team_id=1,
|
|
home_team_id=2,
|
|
channel_id=1234,
|
|
season=9,
|
|
)
|
|
game_2 = Game(
|
|
away_team_id=3,
|
|
home_team_id=4,
|
|
channel_id=5678,
|
|
season=9,
|
|
)
|
|
|
|
cardset_2024 = Cardset(name='2024 Season', ranked_legal=True)
|
|
cardset_2022 = Cardset(name='2022 Season', ranked_legal=False)
|
|
|
|
game_1_cardset_2024_link = GameCardsetLink(game=game_1, cardset=cardset_2024, priority=1)
|
|
game_1_cardset_2022_link = GameCardsetLink(game=game_1, cardset=cardset_2022, priority=2)
|
|
game_2_cardset_2024_link = GameCardsetLink(game=game_2, cardset=cardset_2024, priority=1)
|
|
|
|
for team_id in [1, 2]:
|
|
for (order, pos) in [(1, 'C'), (2, '1B'), (3, '2B'), (4, '3B'), (5, 'SS'), (6, 'LF'), (7, 'CF'), (8, 'RF'), (9, 'DH')]:
|
|
this_lineup = Lineup(team_id=team_id, card_id=order, player_id=68+order, position=pos, batting_order=order, game=game_1)
|
|
|
|
for team_id in [3, 4]:
|
|
for (order, pos) in [(1, 'C'), (2, '1B'), (3, '2B'), (4, '3B'), (5, 'SS'), (6, 'LF'), (7, 'CF'), (8, 'RF'), (9, 'DH')]:
|
|
this_lineup = Lineup(team_id=team_id, card_id=order, player_id=100+order, position=pos, batting_order=order, game=game_2)
|
|
|
|
session.add(game_1)
|
|
session.add(game_2)
|
|
session.commit()
|
|
|
|
|
|
def select_speed_testing():
|
|
with Session(engine) as session:
|
|
game_1 = session.exec(select(Game).where(Game.id == 1)).one()
|
|
ss_search_start = datetime.datetime.now()
|
|
man_ss = [x for x in game_1.lineups if x.position == 'SS' and x.active]
|
|
ss_search_end = datetime.datetime.now()
|
|
|
|
ss_query_start = datetime.datetime.now()
|
|
query_ss = session.exec(select(Lineup).where(Lineup.game == game_1, Lineup.position == 'SS', Lineup.active == True)).all()
|
|
ss_query_end = datetime.datetime.now()
|
|
|
|
manual_time = ss_search_end - ss_search_start
|
|
query_time = ss_query_end - ss_query_start
|
|
|
|
print(f'Manual Shortstops: time: {manual_time.microseconds} ms / {man_ss}')
|
|
print(f'Query Shortstops: time: {query_time.microseconds} ms / {query_ss}')
|
|
print(f'Game: {game_1}')
|
|
|
|
games = session.exec(select(Game).where(Game.active == True)).all()
|
|
print(f'len(games): {len(games)}')
|
|
|
|
|
|
def select_all_testing():
|
|
with Session(engine) as session:
|
|
game_search = session.exec(select(Team)).all()
|
|
for game in game_search:
|
|
print(f'Game: {game}')
|
|
|
|
|
|
# def select_specic_fields():
|
|
# with Session(engine) as session:
|
|
# games = session.exec(select(Game.id, Game.away_team, Game.home_team))
|
|
# print(f'Games: {games}')
|
|
# print(f'.all(): {games.all()}')
|
|
|
|
|
|
def main():
|
|
create_db_and_tables()
|
|
create_test_games()
|
|
# select_speed_testing()
|
|
# select_all_testing()
|
|
|
|
|
|
if __name__ == "__main__":
|
|
main()
|