o g/h @s4ddlZddlZddlZddZddZddZdS)Nc Csgd}t|ddd}|d}tjtj|ddt|ddd d }t|}tj||d d }||D]B}|d dkrBd|d <n0|d dkrMd|d <n%|d dkrXd|d <n|d dkrcd|d <n|d dkrnd|d <nt d| |q5dS)z Creates a file with only important data columns for each player Args: filename (str): Name of the file that contains the full data for each player ) first_name second_name goals_scoredassists total_pointsminutesgoals_conceded creativity influencethreatbonusbps ict_index clean_sheets red_cards yellow_cardsselected_by_percentnow_cost element_typer+utf-8encodingzcleaned_players.csvTexist_okw+rnewlineignore extrasactionr1GK2DEF3MID4FWD5AMzUnknown element typeN) openosmakedirspathdirnamecsv DictReader DictWriter writeheaderprintwriterow) filename base_filenameheadersfinoutnamefoutreaderwriterliner@6/var/www/html/fpl_model_app/scripts/utills/cleaners.py clean_playerss,            rBc Cs~gd}t|ddd}|d}tjtj|ddt|ddd d }t|}tj||d d }||D]}| |q5d S)z Creates a file that contains the name to id mappings for each player Args: players_filename (str): Name of the file that contains the full data for each player )rridrrrplayer_idlist.csvTrrrrrr N) r,r-r.r/r0r1r2r3r4r6) players_filenamer8r9r:r;r<r=r>r?r@r@rA id_players"s  rFcCsZ|d}t|ddd}t|}i}|D]}t|d}|dd|d}|||<q|S) z6 Gets the list of all player ids and player names rDrrrrCr_r)r,r1r2int)r8r7r:r= player_idsr?kvr@r@rAget_player_ids3s   rL)r1mathr-rBrFrLr@r@r@rAs