From 583873bb78beb76976070d9f6d975b9f942abb62 Mon Sep 17 00:00:00 2001 From: Cal Corum Date: Sat, 25 Feb 2023 18:37:04 -0600 Subject: [PATCH] Update ai_manager.py remove deprecated bullpen check --- ai_manager.py | 22 +++++++++++++++------- 1 file changed, 15 insertions(+), 7 deletions(-) diff --git a/ai_manager.py b/ai_manager.py index 9d9a9dd..563fd90 100644 --- a/ai_manager.py +++ b/ai_manager.py @@ -5,7 +5,6 @@ import random import pydantic from db_calls_gameplay import StratPlay, get_team_lineups from db_calls import db_get, db_post -import db_calls_card_creation as scouting from peewee import * 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 = [] + 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 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( endpoint='players', - 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') - ], + params=params, timeout=10 )['players'] except ConnectionError as e: