2023-10-28 14:40:46 +01:00
|
|
|
import pandas as pd
|
|
|
|
|
import argparse
|
|
|
|
|
|
2023-10-28 19:38:42 +01:00
|
|
|
# Define Player attributes
|
|
|
|
|
|
|
|
|
|
goalkeeper = {
|
|
|
|
|
"role_name": "goalkeeper",
|
|
|
|
|
"primary_multiplier": 5,
|
|
|
|
|
"primary_attributes": ["Agi", "Ref"],
|
|
|
|
|
"secondary_multiplier": 3,
|
|
|
|
|
"secondary_attributes": ["1v1", "Ant", "Cmd", "Cnt", "Kic", "Pos"],
|
|
|
|
|
"tertiary_multiplier": 1,
|
|
|
|
|
"tertiary_attributes": ["Acc", "Aer", "Cmp", "Dec", "Fir", "Han", "Pas", "Thr", "Vis"]
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
def load_html_data_to_dataframe(filepath: str) -> pd.DataFrame:
|
2023-10-28 14:40:46 +01:00
|
|
|
"""Read HTML file exported by FM into a Dataframe
|
|
|
|
|
|
|
|
|
|
Keyword arguments:
|
|
|
|
|
filepath -- path to fm player html file
|
|
|
|
|
"""
|
|
|
|
|
player_df = pd.read_html(filepath, header=0, encoding="utf-8", keep_default_na=False)[0]
|
|
|
|
|
# Clean Dataframe to get rid of unknown values and ability ranges (takes the lowest value)
|
|
|
|
|
# This casts to a string to be able to split, so we have to cast back to an int later.
|
|
|
|
|
player_df = player_df.replace("-", 0)
|
|
|
|
|
player_df = player_df.map(lambda x: str(x).split("-")[0])
|
|
|
|
|
return player_df
|
|
|
|
|
|
2023-10-28 19:38:42 +01:00
|
|
|
def export_html_from_dataframe(player_df: pd.DataFrame, filepath: str) -> str:
|
|
|
|
|
"""Export Dataframe as html with jQuery Data Tables
|
|
|
|
|
Taken from: https://www.thepythoncode.com/article/convert-pandas-dataframe-to-html-table-python.
|
|
|
|
|
|
|
|
|
|
Keyword arguments:
|
|
|
|
|
filepath -- path to fm player html file
|
|
|
|
|
"""
|
|
|
|
|
table_html = player_df.to_html(table_id="table", index=False)
|
|
|
|
|
html = f"""
|
|
|
|
|
<html>
|
|
|
|
|
<header>
|
|
|
|
|
<link href="https://cdn.datatables.net/1.11.5/css/jquery.dataTables.min.css" rel="stylesheet">
|
|
|
|
|
</header>
|
|
|
|
|
<body>
|
|
|
|
|
{table_html}
|
|
|
|
|
<script src="https://code.jquery.com/jquery-3.6.0.slim.min.js" integrity="sha256-u7e5khyithlIdTpu22PHhENmPcRdFiHRjhAuHcs05RI=" crossorigin="anonymous"></script>
|
|
|
|
|
<script type="text/javascript" src="https://cdn.datatables.net/1.11.5/js/jquery.dataTables.min.js"></script>
|
|
|
|
|
<script>
|
|
|
|
|
$(document).ready( function () {{
|
|
|
|
|
$('#table').DataTable({{
|
|
|
|
|
paging: false,
|
|
|
|
|
order: [[12, 'desc']],
|
|
|
|
|
// scrollY: 400,
|
|
|
|
|
}});
|
|
|
|
|
}});
|
|
|
|
|
</script>
|
|
|
|
|
</body>
|
|
|
|
|
</html>
|
|
|
|
|
"""
|
|
|
|
|
open(filepath, "w", encoding="utf-8").write(html)
|
|
|
|
|
|
2023-10-28 14:40:46 +01:00
|
|
|
# TODO: Do I even want this?
|
|
|
|
|
def calc_composite_scores(player_df: pd.DataFrame) -> pd.DataFrame:
|
|
|
|
|
"""Calculate Speed, Workrate and Set Piece scores
|
|
|
|
|
|
|
|
|
|
Keyword arguments:
|
|
|
|
|
player_df: Dataframe of Players and Attributes
|
|
|
|
|
"""
|
|
|
|
|
player_df['Spd'] = ( player_df['Pac'] + player_df['Acc'] ) / 2
|
|
|
|
|
player_df['Work'] = ( player_df['Wor'] + player_df['Sta'] ) / 2
|
|
|
|
|
player_df['SetP'] = ( player_df['Jum'] + player_df['Bra'] ) / 2
|
|
|
|
|
return player_df
|
|
|
|
|
|
|
|
|
|
def sum_attributes(player_df: pd.DataFrame, role: str, attribute_type: str, attributes: [str]) -> pd.DataFrame:
|
|
|
|
|
"""Create a new Column containing the sum of provided attribute columns
|
|
|
|
|
|
|
|
|
|
Keyword arguments:
|
|
|
|
|
player_df: Dataframe of Players and Attributes
|
|
|
|
|
role: Name of role to be used as additional column in dataframe
|
|
|
|
|
attribute_type: Type of Attribute [Primary, Secondary, Tertiary]
|
|
|
|
|
attributes: List of Attributes to Sum
|
|
|
|
|
"""
|
|
|
|
|
player_df[f'{role}_{attribute_type}'] = 0
|
|
|
|
|
for attribute in attributes:
|
|
|
|
|
player_df[f'{role}_{attribute_type}'] += pd.to_numeric(player_df[attribute])
|
2023-10-28 19:38:42 +01:00
|
|
|
player_df[f'{role}_{attribute_type}'] = round(player_df[f'{role}_{attribute_type}'] / len(attributes), 2)
|
2023-10-28 14:40:46 +01:00
|
|
|
return player_df
|
|
|
|
|
|
2023-10-28 19:38:42 +01:00
|
|
|
def calc_role_scores(player_df: pd.DataFrame, role: dict) -> pd.DataFrame:
|
2023-10-28 14:40:46 +01:00
|
|
|
"""Calculate Player Role scores based on selected attributes.
|
|
|
|
|
|
|
|
|
|
Keyword arguments:
|
|
|
|
|
player_df: Dataframe of Players and Attributes
|
2023-10-28 19:38:42 +01:00
|
|
|
role: Dictionary containing role name, role attributes and role attribute weightings
|
2023-10-28 14:40:46 +01:00
|
|
|
"""
|
2023-10-28 19:38:42 +01:00
|
|
|
player_df = sum_attributes(player_df, role["role_name"], "primary", role["primary_attributes"])
|
|
|
|
|
player_df = sum_attributes(player_df, role["role_name"], "secondary", role["secondary_attributes"])
|
|
|
|
|
player_df = sum_attributes(player_df, role["role_name"], "tertiary", role["tertiary_attributes"])
|
|
|
|
|
divisor = role["primary_multiplier"] + role["secondary_multiplier"] + role["tertiary_multiplier"]
|
|
|
|
|
player_df[f'{role["role_name"]}'] = round((((player_df[f'{role["role_name"]}_primary'] * 5) + (player_df[f'{role["role_name"]}_secondary'] * 3) + (player_df[f'{role["role_name"]}_tertiary'] * 1)) / divisor ), 2)
|
2023-10-28 14:40:46 +01:00
|
|
|
return player_df
|
|
|
|
|
|
|
|
|
|
def calc_player_scores(player_df: pd.DataFrame):
|
|
|
|
|
# TODO: Create objects for each role that can be used here.
|
2023-10-28 19:38:42 +01:00
|
|
|
player_df = calc_role_scores(player_df, goalkeeper)
|
2023-10-28 14:40:46 +01:00
|
|
|
# TODO: Add roles.
|
2023-10-28 19:38:42 +01:00
|
|
|
return player_df
|
2023-10-28 14:40:46 +01:00
|
|
|
|
|
|
|
|
if __name__ == "__main__":
|
|
|
|
|
parser = argparse.ArgumentParser()
|
2023-10-28 19:38:42 +01:00
|
|
|
parser.add_argument("-i", "--input-filepath", type=str)
|
|
|
|
|
parser.add_argument("-o", "--output-filepath", type=str)
|
2023-10-28 14:40:46 +01:00
|
|
|
args = parser.parse_args()
|
2023-10-28 19:38:42 +01:00
|
|
|
input_filepath = args.input_filepath
|
|
|
|
|
output_filepath = args.output_filepath
|
2023-10-28 14:40:46 +01:00
|
|
|
|
2023-10-28 19:38:42 +01:00
|
|
|
player_df = load_html_data_to_dataframe(input_filepath)
|
|
|
|
|
player_df = calc_player_scores(player_df)
|
|
|
|
|
export_html_from_dataframe(player_df, output_filepath)
|
