From 7aca691596aff0dcd7e3e9135de5d4870a4aa8a8 Mon Sep 17 00:00:00 2001
From: Felix Blanke <felixblanke@uni-bonn.de>
Date: Mon, 28 Aug 2023 12:11:35 +0200
Subject: [PATCH] Encapsulate df creation

---
 wsgi.py | 32 +++++++++++++++++++++++---------
 1 file changed, 23 insertions(+), 9 deletions(-)

diff --git a/wsgi.py b/wsgi.py
index 25b0596..921af3e 100644
--- a/wsgi.py
+++ b/wsgi.py
@@ -36,15 +36,12 @@ def get_tables(url: str) -> tuple[pd.DataFrame, pd.DataFrame]:
     return df, df_state
 
 
-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()
-
+def create_plot_df(
+    current_df: pd.DataFrame | None,
+    data_folder: str,
+    sheet_name: str,
+    curr_datetime,
+) -> pd.DataFrame:
     data_dict = {}
 
     ## 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]):
             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)
 
     # fill weekends