'ri dZddlZddlZddlZddlmZmZejdZejdejZ ejdZ ejdZ iddd d d d d d d d d d ddddddddddddddddddddddiddddddddddddddd d d!d!d"d"d#d#d$d$d%d%d&d&d'd'd(d(d)d)id*d*d+d+d,d,d-d-d.d.d/d0d0d0d1d1d2d2d3d3d4d4d5d5d6d6d7d7d8d8d9d9d:d:id;d;dd>d?d?d@dAdAdAdBdBdCdCdDdEdEdEdFdFdGdHdHdHdIdJdJdJdKdKdd8dLdd7d:dd"dddd$d%dM ZidNddOd dPd dQd dRd dSd dTddUddVddWddXddYddZdd[dd\dd]dd^did_dd`ddaddbddcdddddeddfddgddhd did!djd"dkd#dld$dmd%dnd&dod'idpd dqd(drd)dsd*dtd dud+dvd,dwddxddyd-dzd.d{d0d|d1d}d1d~d2dd3dd4idd5dd6dddd7dd9dd8dd:dd;dd#dd$dd$dd<dd=dddd>dd'ddidd?dd ddddFddLdd dd+ddddddEddddddddddddCddBdddd7dJdHd8d:dAdKd#dd Zd;d3d0d9dddAdEdHdJdddZiZej%D]%\ZZej+dZeseevs!eee<'dedefdZdedefdZddedeedefdZdedefdZddedeedefdZdededefdZdededefdZdedeeeffdZ y)z9Player/team name normalization for cross-source matching.N)OptionalTuplez pub-integrityz\s+(Jr\.?|Sr\.?|III|II|IV|V)$z[^a-z\s]z\s+ATLBOSBKNCHACHICLEDALDENDETGSWHOUINDLACLALMEMMIAMILMINNOPNYKOKCORLPHIPHXPORSACSASTORUTAWASANAARIBUFCGYCARCOLCBJEDMFLALALAKMTLNSHNJDNYINYROTTPITSEASJSSTLTBLVANVGKWPGBALTBTBRNYYNYMKCKCRCWSSFSFGSDSDPTEXCIN) rr5rIrr4r7GSSANONYPHOUTAHWASHz atlanta hawkszboston celticsz brooklyn netszcharlotte hornetsz chicago bullszcleveland cavalierszdallas maverickszdenver nuggetszdetroit pistonszgolden state warriorszhouston rocketszindiana pacersz la clipperszlos angeles clippersz la lakerszlos angeles lakerszmemphis grizzliesz miami heatzmilwaukee buckszminnesota timberwolvesznew orleans pelicansznew york knickszoklahoma city thunderz orlando magiczphiladelphia 76ersz phoenix sunszportland trail blazerszsacramento kingszsan antonio spursztoronto raptorsz utah jazzzwashington wizardsz anaheim duckszarizona coyotesz boston bruinszbuffalo sabreszcalgary flameszcarolina hurricaneszchicago blackhawkszcolorado avalanchezcolumbus blue jacketsz dallas starszdetroit red wingszedmonton oilerszflorida pantherszlos angeles kingszmontreal canadiensumontréal canadiensznashville predatorsznew jersey devilsznew york islandersznew york rangerszottawa senatorszphiladelphia flyerszpittsburgh penguinszsan jose sharkszseattle krakenzst. louis bluesztampa bay lightningztoronto maple leafszutah hockey clubz utah mammothzvancouver canuckszvegas golden knightszwashington capitalsWSHz winnipeg jetszarizona diamondbackszatlanta braveszbaltimore orioleszboston red soxz chicago cubsCHCzchicago white soxzcincinnati redszcleveland guardianszcolorado rockieszdetroit tigerszhouston astroszkansas city royalszlos angeles angelsLAAzlos angeles dodgersLADz miami marlinszmilwaukee brewerszminnesota twinsz new york metsznew york yankeesOAK) athleticszoakland athleticszphiladelphia phillieszpittsburgh pirateszsan diego padreszsan francisco giantszseattle marinerszst. louis cardinalsztampa bay raysz texas rangersztoronto blue jayszwashington nationals)r=NJr,SJrQ)r=rArDrFrQ)nhlmlbvaluereturnc|sytjd|}djd|D}|jj }t j dd|}tj d|j }|S)NNFKDc3LK|]}tj|r|ywN unicodedata combining.0cs -scripts/publishing_integrity/normalization.py z'_normalize_team_text..nIq 0E0Ea0HI$$z [^a-z0-9\s.] )rd normalizejoinlowerstripresub _MULTI_SPACE)r\nfkd ascii_namelowereds ri_normalize_team_textrxjsw    /DIDIIJ &&(Gff_c73GsG,224G NnamecD|sytjd|}djd|D}|jj }t j d|}tj d|}tj d|j }|S)zICanonical player name: lowercase, no accents, no suffixes, single spaces.r_r`c3LK|]}tj|r|ywrbrcrfs rirjz(normalize_player_name..{rkrlrm) rdrnrorprq _SUFFIXESrs _NON_ALPHArt)rzrurvresults rinormalize_player_namerus    .DIDIIJ     % % 'F ]]2v &F ^^B 'F   c6 * 0 0 2F Mryabbrevleaguec |sy|jj}|xsdjj}|r)tj |ij |}|r|St j ||S)zBNormalize a team abbreviation or full team name to canonical form.r_)upperrqrpLEAGUE_ABBREV_ALIASESgetTEAM_ABBREV_MAP)rrr league_keyaliass rinormalize_team_abbrevrsr  LLN "E,B%%'--/J%))*b9==eD L   ue ,,ryc|syt|S)z9Lowercase, stripped team display name for fuzzy matching.r_)rx)rzs rinormalize_team_namers   %%ryct|}|sytj|}|r|S|jd}|tvr t|St ||S)zOResolve abbreviations, full team names, and nicknames to a canonical team code.r_r[)rx TEAM_NAME_MAPrsplitNICKNAME_ABBREV_MAPr)r\r normalized direct_namenicknames ricanonicalize_team_identifierrs_%e,J ##J/K!"%H&&"8,,  //ryname_aname_bc0t|t|k(S)z-Compare two player names after normalization.)r)rrs riplayer_names_matchrs  (,A&,I IIry player_namect|}|j}t|dk\r|dd|dd|jS|d|jS)z2Generate a canonical ID from player name + league.r_r[)rrlenrp)rrnormpartss riextract_player_canonical_idrsd  -D JJLE 5zQ(1U2YKq17799V1VH  # # %%rysgo_idc|sy|jd}t|dk\rN|dj}|dd}|r|djr|dd}dj d|D}||fS|dfS) zEParse SGO playerID (FIRST_LAST_N_LEAGUE) into (display_name, league).)r_r_rr[Nrmc3<K|]}|jywrb) capitalize)rgps rirjz&parse_sgo_player_id..s>a1<<>>sr_)rrrpisdigitro)rrr name_partsdisplays riparse_sgo_player_idrs  LL E 5zQr"3BZ *R.002#CRJ((>:>>  B<ryrb)!__doc__rrrdloggingtypingrr getLoggerlogcompile IGNORECASEr}r~rtrrrritems team_name team_abbrevrrstrrxrrrrboolrrrryrirs] ? "g( BJJ7 G RZZ $ rzz&!  5 %u.3UJ5&u&7&@VW\&u&6u&?Nu&%&"0&9QRW&& 3E&LU/&01&0!71&0@UV[1&2U3&6E7&6$4U7&6=PQV7&8e9&8,U9&85H9&:u;&:4U;&:=OPU;&<e=&<.u=&<7KE=&>%?&>"7?&>@OPU?&@A&@!25A&@;J5A&BC&B-2"%UZ!uUZe%!K& T    "+1134I{ $HH$77(3H%4 $ -# -x} - -&c&c&00Xc]0c0"JsJCJDJ &S&#&#&  c3h ry