fblanke/tvstud-download-digital
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Encapsulate df creation

Browse Source
This commit is contained in:
Felix Blanke 2023-08-28 12:11:35 +02:00
parent ae6beafa3d
commit 7aca691596
1 changed files with 23 additions and 9 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

32
wsgi.py
View File

@ -36,15 +36,12 @@ def get_tables(url: str) -> tuple[pd.DataFrame, pd.DataFrame]:
return df, df_state return df, df_state
def plot( def create_plot_df(
current_df: pd.DataFrame | None = None, current_df: pd.DataFrame | None,
data_folder: str = "data", data_folder: str,
sheet_name: str = "digital", sheet_name: str,
total_target: int = 1500, curr_datetime,
alpha: float | None = None, ) -> pd.DataFrame:
) -> str:
curr_datetime = datetime.datetime.now()
data_dict = {} data_dict = {}
## Important: If multiple results are stored for the same date ## Important: If multiple results are stored for the same date
@ -79,6 +76,23 @@ def plot(
if pd.isna(df.loc[df.index.max()][0]): if pd.isna(df.loc[df.index.max()][0]):
df = df.drop([df.index.max()]) df = df.drop([df.index.max()])
return df
def plot(
current_df: pd.DataFrame | None = None,
data_folder: str = "data",
sheet_name: str = "digital",
total_target: int = 1500,
alpha: float | None = None,
) -> str:
curr_datetime = datetime.datetime.now()
df = create_plot_df(
current_df=current_df,
data_folder=data_folder,
sheet_name=sheet_name,
curr_datetime=curr_datetime,
)
fig = plt.figure(dpi=300) fig = plt.figure(dpi=300)
# fill weekends # fill weekends