Update ai_manager.py

remove deprecated bullpen check
This commit is contained in:
Cal Corum 2023-02-25 18:37:04 -06:00
parent e31dda1d22
commit 583873bb78

View File

@ -5,7 +5,6 @@ import random
import pydantic import pydantic
from db_calls_gameplay import StratPlay, get_team_lineups from db_calls_gameplay import StratPlay, get_team_lineups
from db_calls import db_get, db_post from db_calls import db_get, db_post
import db_calls_card_creation as scouting
from peewee import * from peewee import *
from typing import Optional, Literal from typing import Optional, Literal
@ -215,16 +214,25 @@ def build_lineup(team_object: dict, game_id: int, league_name: str, vs_hand: str
} }
p_names = [] p_names = []
set_params = [('cardset_id_exclude', 2)]
if league_name == 'minor-league':
set_params = [('cardset_id', 1), ('cardset_id', 3)]
elif league_name == 'major-league':
set_params = [
('cardset_id', 1), ('cardset_id', 3), ('cardset_id', 4), ('cardset_id', 5)
]
# Pull players sorted by current cost # Pull players sorted by current cost
try: try:
params = [
('mlbclub', team_object['lname']), ('pos_include', 'C'), ('pos_include', '1B'), ('pos_include', '2B'),
('pos_include', '3B'), ('pos_include', 'SS'), ('pos_include', 'LF'), ('pos_include', 'CF'),
('pos_include', 'RF'), ('pos_include', 'DH'), ('inc_dex', False), ('sort_by', 'cost-desc')
]
params.extend(set_params)
all_players = db_get( all_players = db_get(
endpoint='players', endpoint='players',
params=[ params=params,
('mlbclub', team_object['lname']), ('pos_include', 'C'), ('pos_include', '1B'), ('pos_include', '2B'),
('pos_include', '3B'), ('pos_include', 'SS'), ('pos_include', 'LF'), ('pos_include', 'CF'),
('pos_include', 'RF'), ('pos_include', 'DH'), ('cardset_id_exclude', 2), ('cardset_id_exclude', 4),
('inc_dex', False), ('sort_by', 'cost-desc')
],
timeout=10 timeout=10
)['players'] )['players']
except ConnectionError as e: except ConnectionError as e: