0a17745389
Standardize formatting with black and apply ruff auto-fixes. Co-Authored-By: Claude Opus 4.6 <noreply@anthropic.com>
127 lines
4.5 KiB
Python
127 lines
4.5 KiB
Python
import asyncio
|
|
import sys
|
|
|
|
import pandas as pd
|
|
|
|
from creation_helpers import get_args, pd_players_df, pd_positions_df
|
|
from db_calls import db_get, db_patch, DB_URL
|
|
from exceptions import logger
|
|
|
|
CARD_BASE_URL = f"{DB_URL}/v2/players"
|
|
|
|
|
|
async def main(args):
|
|
arg_data = get_args(args)
|
|
post_players = (
|
|
False
|
|
if "post_players" not in arg_data or arg_data["post_players"].lower() == "false"
|
|
else True
|
|
)
|
|
|
|
# cardset_name = input(f'What is the name of this Cardset? ')
|
|
cardset_name = arg_data["cardset_name"]
|
|
print(f"Searching for cardset: {cardset_name}")
|
|
c_query = await db_get("cardsets", params=[("name", cardset_name)])
|
|
|
|
if c_query["count"] == 0:
|
|
print(f"I do not see a cardset named {cardset_name}")
|
|
return
|
|
cardset = c_query["cardsets"][0]
|
|
del c_query
|
|
|
|
print(f'Pulling {cardset["name"]} players now...')
|
|
base_df = await pd_players_df(cardset["id"])
|
|
all_players = base_df.set_index("bbref_id", drop=False)
|
|
|
|
print(f'Pulling {cardset["name"]} card positions now...')
|
|
all_pos = await pd_positions_df(cardset["id"])
|
|
|
|
print("Running player position updates..")
|
|
p_query = await db_get("pitchingcards", params=[("cardset_id", cardset["id"])])
|
|
if p_query["count"] == 0:
|
|
raise ValueError(f'No pitching cards returned for cardset {cardset["id"]}')
|
|
|
|
for x in p_query["cards"]:
|
|
x["player_id"] = x["player"]["player_id"]
|
|
x["bbref_id"] = x["player"]["bbref_id"]
|
|
del x["player"]
|
|
|
|
raw_df = pd.DataFrame(p_query["cards"])
|
|
pit_pos_df = raw_df.set_index("bbref_id", drop=False)
|
|
|
|
player_updates = {}
|
|
|
|
def set_all_positions(df_data):
|
|
def pos_match(old_pos, new_pos):
|
|
if old_pos is None and new_pos == "False":
|
|
return True
|
|
elif old_pos is None:
|
|
return False
|
|
else:
|
|
return old_pos.lower() == new_pos.lower()
|
|
|
|
pos_series = all_pos.query(f'player_id == {df_data["player_id"]}')["position"]
|
|
this_player = all_players.loc[df_data["bbref_id"]]
|
|
pos_updates = []
|
|
count = 1
|
|
for this_pos in pos_series:
|
|
if this_pos == "P":
|
|
this_pitcher = pit_pos_df.loc[df_data["bbref_id"]]
|
|
if this_pitcher["starter_rating"] > 3:
|
|
if not pos_match(this_player[f"pos_{count}"], "SP"):
|
|
pos_updates.append((f"pos_{count}", "SP"))
|
|
count += 1
|
|
if this_pitcher["relief_rating"] > 1 or not pd.isna(
|
|
this_pitcher["closer_rating"]
|
|
):
|
|
if not pos_match(this_player[f"pos_{count}"], "RP"):
|
|
pos_updates.append((f"pos_{count}", "RP"))
|
|
count += 1
|
|
else:
|
|
if not pos_match(this_player[f"pos_{count}"], "RP"):
|
|
pos_updates.append((f"pos_{count}", "RP"))
|
|
count += 1
|
|
|
|
if not pd.isna(this_pitcher["closer_rating"]):
|
|
if not pos_match(this_player[f"pos_{count}"], "CP"):
|
|
pos_updates.append((f"pos_{count}", "CP"))
|
|
count += 1
|
|
else:
|
|
if not pos_match(this_player[f"pos_{count}"], this_pos):
|
|
pos_updates.append((f"pos_{count}", this_pos))
|
|
count += 1
|
|
|
|
if count == 1:
|
|
if not pos_match(this_player[f"pos_{count}"], "DH"):
|
|
pos_updates.append(("pos_1", "DH"))
|
|
count += 1
|
|
|
|
while count < 9:
|
|
if not pos_match(this_player[f"pos_{count}"], "False"):
|
|
pos_updates.append((f"pos_{count}", "False"))
|
|
count += 1
|
|
|
|
if len(pos_updates) > 0:
|
|
if df_data.player_id not in player_updates.keys():
|
|
player_updates[df_data.player_id] = pos_updates
|
|
else:
|
|
player_updates[df_data.player_id].extend(pos_updates)
|
|
|
|
all_players.apply(set_all_positions, axis=1)
|
|
|
|
print(f"Sending {len(player_updates)} player updates to PD database...")
|
|
if post_players:
|
|
for x in player_updates:
|
|
await db_patch("players", object_id=x, params=player_updates[x])
|
|
else:
|
|
update_string = "\n".join(player_updates)
|
|
logger.info(f"All player updates:{update_string}")
|
|
print(
|
|
"Printed player updates to log since 'post_players' argument was not true"
|
|
)
|
|
print("Player updates are complete\n")
|
|
|
|
|
|
if __name__ == "__main__":
|
|
asyncio.run(main(sys.argv[1:]))
|