a 8jn@sdZddlZddlmZmZmZddlmZddlmZm Z m Z m Z ddl Z ddlmZddlmZmZmZdd lmZmZmZmZed Zd Zeee jd d dZGdd d eZdS)zScraper for api.clubelo.com.N)datetime timedeltatimezone)Path)IOCallableOptionalUnion) unidecode)BaseRequestsReaderadd_alt_team_namesstandardize_colnames)DATA_DIRNOCACHENOSTORETEAMNAME_REPLACEMENTSClubElozhttp://api.clubelo.com)datareturncCstj|ddgddS)NFromTo%Y-%m-%d)Z parse_datesZ date_format)pdZread_csv)rr8E:\soccer\ml_env\Lib\site-packages\soccerdata/clubelo.py _parse_csvsrcseZdZdZdeeefeee e e e ge ffe e e dfdd Zd eee efejdddZde eeeefejd d d ZZS)raProvides pd.DataFrames from CSV API at http://api.clubelo.com. Data will be downloaded as necessary and cached locally in ``~/soccerdata/data/ClubElo``. Since the source does not provide league names, this class will not filter by league. League names will be inserted from the other sources where available. Leagues that are only covered by clubelo.com will have NaN values. Parameters ---------- proxy : 'tor' or or dict or list(dict) or callable, optional Use a proxy to hide your IP address. Valid options are: - "tor": Uses the Tor network. Tor should be running in the background on port 9050. - str: The address of the proxy server to use. - list(str): A list of proxies to choose from. A different proxy will be selected from this list after failed requests, allowing rotating proxies. - callable: A function that returns a valid proxy. This function will be called after failed requests, allowing rotating proxies. no_cache : bool If True, will not use cached data. no_store : bool If True, will not store downloaded data. data_dir : Path Path to directory where data will be cached. Nproxyno_cacheno_storedata_dircstj||||ddS)z Initialize a new ClubElo reader.rN)super__init__)selfrrr r! __class__rrr#5szClubElo.__init__)datercCs|stjtjd}nt|tr2t|dtj}t|tsDtd| d}|j |d}t d|}| ||}t |tjddiddtid td jd d d jdd d|jjdddS)abRetrieve ELO scores for all teams at specified date. Elo scores are available as early as 1939. Values before 1960 should be considered provisional. Parameters ---------- date : datetime object or string like 'YYYY-MM-DD' Date for which to retrieve ELO scores. If no date is specified, get today's scores. Raises ------ TypeError If date is not a date string or datetime object. ValueError If data is an invalid date string. Returns ------- pd.DataFrame )tzrz<'date' must be a datetime object or string like 'YYYY-MM-DD'.csv/clubteamcolumnsNonenancSs|ddSNrankfloatastypexrrrjz&ClubElo.read_by_date..r2cSs|dd|dtS)NZcountry_level)r5strr6rrrr8kr9)ZleagueT)Zdrop)rnowrutc isinstancer=strptime astimezone TypeErrorstrftimer! CLUB_ELO_APIgetrpiperrenamereplacerr3assignZ_translate_leagueZ reset_index set_index)r$r'Z datestringfilepathurlrrrr read_by_date?s<     zClubElo.read_by_dater )r,max_agerc Cst|}dd|D}|D]}|j|d}tdtdd|}||||}t|tj ddid  d t d j d d d d}t|dkr|j dtidd|Sqtd|dS)avRetrieve full ELO history for one club. For the exact spelling of a club's name, check the result of :func:`~soccerdata.ClubElo.read_by_date` or `clubelo.com `__. You can also use alternative team names specified in `teamname_replacements.json`. Values before 1960 should be considered provisional. Parameters ---------- team : str The club's name. max_age : int for age in days, or timedelta object The max. age of locally cached file before re-download. To disable re-downloading, set to None. Raises ------ TypeError If max_age is not an integer or timedelta object. ValueError If no ratings for the given team are available. Returns ------- pd.DataFrame cSsh|]}tddt|qS)z[\s'])resubr ).0r,rrr r9z,ClubElo.read_team_history..r)r*z [^a-zA-Z]rPr+r,r-r/r0cSs|ddSr1r4r6rrrr8r9z+ClubElo.read_team_history..r:fromrT)ZinplacezNo data found for team N)r r!rErQrRrFrrGrrHrIr3rJrKZ sort_indexlenr ValueError) r$r,rOZteams_to_checkZ_teamrLrMrZdfrrrread_team_historyqs,  zClubElo.read_team_history)N)r )__name__ __module__ __qualname____doc__rrCLUB_ELO_DATADIRrr r=listrboolrr#rr DataFramerNintrrX __classcell__rrr%rrs"  3)r\rQrrrpathlibrtypingrrrr Zpandasrr _commonr r r_configrrrrr]rEbytesr`rrrrrrs