
    Y/h
                     .    S SK r S SKrS SKrS rS rS rg)    Nc                    / SQn[        U SSS9nUS-   n[        R                  " [        R                  R	                  U5      SS9  [        USSS	S
9n[
        R                  " U5      n[
        R                  " XRSS9nUR                  5         U Hj  nUS   S:X  a  SUS'   OGUS   S:X  a  SUS'   O8US   S:X  a  SUS'   O)US   S:X  a  SUS'   OUS   S:X  a  SUS'   O[        S5        UR                  U5        Ml     g)zCreates 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_scoredassiststotal_pointsminutesgoals_conceded
creativity	influencethreatbonusbps	ict_indexclean_sheets	red_cardsyellow_cardsselected_by_percentnow_costelement_typer+utf-8encodingzcleaned_players.csvTexist_okw+ r   newlineignoreextrasactionr   1GK2DEF3MID4FWD5AMzUnknown element typeN)openosmakedirspathdirnamecsv
DictReader
DictWriterwriteheaderprintwriterow)	filenamebase_filenameheadersfinoutnamefoutreaderwriterlines	            @g:\Projects\Python Projects\fpl_model\scripts\utills\cleaners.pyclean_playersrC      s
    VG
x
0C33GKK(48<D^^C F^^DAF
3&#'D .!S(#(D .!S(#(D .!S(#(D .!S(#'D ()     c                 R   / SQn[        U SSS9nUS-   n[        R                  " [        R                  R	                  U5      SS9  [        USSS	S
9n[
        R                  " U5      n[
        R                  " XRSS9nUR                  5         U H  nUR                  U5        M     g)zCreates 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
)r   r   idr   r   r   player_idlist.csvTr   r   r   r   r!   r"   N)
r.   r/   r0   r1   r2   r3   r4   r5   r6   r8   )	players_filenamer:   r;   r<   r=   r>   r?   r@   rA   s	            rB   
id_playersrI   "   s     2G

8C11GKK(48<D^^C F^^DAF
 rD   c                     U S-   n[        USSS9n[        R                  " U5      n0 nU H#  n[        US   5      nUS   S-   US   -   nXtU'   M%     U$ )	z5Gets the list of all player ids and player names
    rG   r   r   r   rF   r   _r   )r.   r3   r4   int)r:   r9   r<   r?   
player_idsrA   kvs           rB   get_player_idsrP   3   so     22H
x
0C^^C FJT
O$tM'::1  rD   )r3   mathr/   rC   rI   rP    rD   rB   <module>rS      s    
  	:"rD   