78 lines
2.6 KiB
Python
78 lines
2.6 KiB
Python
from argparse import ArgumentParser
|
|
from config import Config as cfg
|
|
from datetime import datetime
|
|
from espn_api.football import League
|
|
from gspread_dataframe import set_with_dataframe
|
|
from service import gc
|
|
from tabulate import tabulate
|
|
from time import strftime
|
|
|
|
import pandas as pd
|
|
|
|
|
|
def check_int(s):
|
|
if s is None:
|
|
return s
|
|
if s[0] in ('-', '+'):
|
|
return s[1:].isdigit()
|
|
return s.isdigit()
|
|
|
|
|
|
def extract_positional_data(lineup: list, position: str) -> int:
|
|
response = 0
|
|
for player in lineup:
|
|
if player.lineupSlot == position:
|
|
response = response + player.points
|
|
return response
|
|
|
|
|
|
def extract_matchup_box_scores(league: League) -> dict:
|
|
result = []
|
|
for week in range(1, league.current_week + 1):
|
|
matchups = league.box_scores(week=week)
|
|
for matchup in matchups:
|
|
result.append({
|
|
'WEEK #': week,
|
|
'AWAY TEAM': matchup.away_team.team_name,
|
|
'AWAY TEAM SCORE': matchup.away_score,
|
|
'AWAY TEAM KICKER': extract_positional_data(matchup.away_lineup, 'K'),
|
|
'AWAY TEAM BENCH': extract_positional_data(matchup.away_lineup, 'BE'),
|
|
'HOME TEAM': matchup.home_team.team_name,
|
|
'HOME TEAM SCORE': matchup.home_score,
|
|
'HOME TEAM KICKER': extract_positional_data(matchup.home_lineup, 'K'),
|
|
'HOME TEAM BENCH': extract_positional_data(matchup.home_lineup, 'BE')
|
|
})
|
|
return result
|
|
|
|
|
|
def write_to_google_spreadsheet(df: pd.DataFrame):
|
|
spreadsheet = gc.open_by_key(cfg.SPREADSHEET_ID)
|
|
worksheet = spreadsheet.worksheet("FantasyData")
|
|
set_with_dataframe(worksheet, df)
|
|
|
|
|
|
if __name__ == "__main__":
|
|
parser = ArgumentParser(description='Weekly Report Generator for Fantasy Football')
|
|
parser.add_argument('-csv', action='store_true')
|
|
parser.add_argument('-gs', action='store_true')
|
|
args = vars(parser.parse_args())
|
|
timestamp = strftime("%Y%m%d_%H%M%S")
|
|
|
|
league = League(league_id=cfg.LEAGUE_ID,
|
|
year=datetime.now().year,
|
|
espn_s2=cfg.ESPN_S2,
|
|
swid=cfg.SWID,
|
|
fetch_league=cfg.FETCH_LEAGUE)
|
|
|
|
matchup = extract_matchup_box_scores(league=league)
|
|
df = pd.DataFrame.from_dict(matchup)
|
|
print(tabulate(df, headers='keys', tablefmt='psql', showindex=False))
|
|
|
|
if args['csv']:
|
|
print("Saving to CSV...")
|
|
df.to_csv(f'espn_fantasy_football_weekly_report_{timestamp}.csv', index=False, float_format='%.2f')
|
|
|
|
if args['gs']:
|
|
print("Saving to Google Sheets...")
|
|
write_to_google_spreadsheet(df=df)
|