Ki"jdZddlZddlZddlZddlZddlZddlZddlmZmZm Z ejjdejjejjeddlmZdZdZddd d d d d d d ddd d dd dZdZdZdZdZddZdZedk(reyy)z Fetch Referee Assignments from ESPN Gets referee/official data for NBA, NFL, NHL games. ESPN API: /sports/{sport}/{league}/summary?event={id} Officials are in header.competitions[0].officials Run: Daily after games complete N)datetimetimezone timedelta)normalize_to_fullz-https://site.api.espn.com/apis/site/v2/sportsg? basketballnba)sport espn_leaguefootballnflhockeynhlmens-college-basketballcollege-football)rr rncaabncaafc~d} t|d5}|D]S}|jds|jdddjjddccdddS dddt j jdd jddS#1swY;xYw#t$rYIwxYw) Nz/var/www/html/eventheodds/.envrzSPORTS_DATABASE_URL==?rSPORTS_DATABASE_URL)open startswithsplitstripFileNotFoundErrorosenvironget)env_pathflines scripts/fetch_espn_referees.py load_db_urlr&!s/H (C  GA G??#9:::c1-a0668>>sCAFF G G G G ::>>/ 4 : :3 ? BB  G G    s9 B0B$6B$ B0(B$*B0$B-)B00 B<;B<cj|jd|jd|jdy)z"Create GameReferee table if neededa CREATE TABLE IF NOT EXISTS "GameReferee" ( id BIGSERIAL PRIMARY KEY, league VARCHAR(50) NOT NULL, "gameId" BIGINT NOT NULL, "gameDate" TIMESTAMP, "refereeName" VARCHAR(200) NOT NULL, "refereeRole" VARCHAR(50), "refereeId" VARCHAR(100), "createdAt" TIMESTAMP DEFAULT NOW(), UNIQUE(league, "gameId", "refereeName") ) z`CREATE INDEX IF NOT EXISTS "GameReferee_league_ref_idx" ON "GameReferee" (league, "refereeName")zOCREATE INDEX IF NOT EXISTS "GameReferee_gameId_idx" ON "GameReferee" ("gameId")N)execute)curs r%ensure_referee_tabler*-s/KK   KKrsKKabcptd|d|d|}|dvr|dz } tj|d}|jdk7rgS|j }g}|jdgD]}|jd }|jd g} | s)| d } | jd g} d } d } | D]M}|jdi}|jddj }|jddk(r|} L|} O|s| s| s|j || | d|S#t$r }gcYd }~Sd }~wwxYw)z!Get all ESPN event IDs for a date/z/scoreboard?dates=)rrz&groups=50&limit=300timeouteventsid competitionsr competitorsNteam abbreviationrhomeAwayhome)r3r9away) ESPN_BASErequestsr! status_codejsonupperappend Exception)r r date_strurlrespdatar2eventevent_idr4compr5 home_team away_team competitorr6abbreves r%get_espn_events_for_daterN@s] Kqq -?z JCEE %%$||C,   s "Iyy{XXh+ EyyH 99^R8L?D((="5KII) ' !~~fb1."5;;=>>*-7 &I &I  'I) "%%) 4  s6'D B isinstancedictr@rstrrA)r r rGrCrDrErQrPr4officialrUrTrVref_id game_infoorMs r%fetch_officialsr_ms Kqq _XJ GC'||C,   s "Iyy{ (B'zz."5 (O// R@ ||M2Nhll:6N#<< B76@46Px||M2V^!d+$$ $ $-3c&k& HHZ, ! k26 H<< .J(,,z2JD $B1QvY$BB   JJL$LLR8<<]K5=\\$5G#hll401T" %C  sI'HB| jd} t!|d |d | } | D]} t#j$t&t)|d |d | d }|s5t+| d |}t+| d |}|j-d|||| f|j/}|s||\}}| dz } |D]8} |j-d||||d|d|d f|j0dkDr|dz }:| tdz } | |z j4dzdk(r(td|d| d| d|d |j| |kr>|j|j7|j7td|d|d| d|| dS#t2$r }Yd}~d}~wwxYw)zFetch referee data for a leaguezUnknown league: r)insertedgames)days z: Fetching referees from z to z%Y%m%dr r r3r9r:z SELECT id, "gameDate" FROM "SportsGame" WHERE league = %s AND "homeTeam" = %s AND "awayTeam" = %s AND "gameDate"::date = %s LIMIT 1 ra INSERT INTO "GameReferee" (league, "gameId", "gameDate", "refereeName", "refereeRole", "refereeId") VALUES (%s, %s, %s, %s, %s, %s) ON CONFLICT (league, "gameId", "refereeName") DO NOTHING rUrVNz z: Processed through z, z games, z refsz: Total  referees from  games) LEAGUE_CONFIGprintr&psycopg2connectcursorr*commitrnowrutcdaterr?strftimerNtimesleep REQUEST_DELAYr_rr(fetchonerowcountrArcclose)league days_backconfigdb_urlconnr)end_date start_datetotal_inserted total_games current_daterBr2rFrQ home_full away_fullgamegame_id game_dater[rMs r%fetch_referees_for_leaguers ]"  )*** 6 "F ]F   F #D ++-CKKM||HLL)..0HI955J Bv||~7 |4z RSNKL ( "((2)&/6-;PRZ[& E JJ} %'w 9NPUVZP[\I)%-@I)%-@I KK )Y =  ?<<>D!% GY 1 K%  KK!!'9hv6FQWHXZbcgZhi k ||a'&!+ 7& P  q)) : % + +a /1 4 Bvh2<.;-xXfWgglm n KKMa ( "d KKMIIKJJL Bvhh~.ok]& QR& ==!s5I%% I83I8ctjd}|jdtdd|jdtdd |j }t d t d t d tjtjjt d |jjd Dcgc] }|jj"}}d}d}|D])}t!||j"}||dz }||dz }+t dd t d|d|dt d ycc}w)NzFetch referee data from ESPN) descriptionz --leaguesz nba,nfl,nhlzComma-separated leagues)typedefaulthelpz--days<zDays to look backz<============================================================zFETCH ESPN REFEREE DATAzTime: ,r)ryrarbrdzTOTAL: rfrg)argparseArgumentParser add_argumentrZint parse_argsrirrnrro isoformatleaguesrrlowerrrc)parserargslrrrrxresults r%mainrs@  $ $1O PF  #}68 sB02    D (O #$ F8<< -779: ;< (O*.,,*<*rs 22277??277??8#<=>( ;  "% 8 6e 4#4M N!2D E   Cc&*Z+\O>d8 zFr+