Refactor

wsgi.py

@@ -8,23 +8,33 @@ from download_digital import construct_dataframe, get_bez_data
 app = Flask(__name__)
 
 
-@app.route("/")
-def tables(
-    url: str = "https://beschaeftigtenbefragung.verdi.de/",
-):
+def get_tables(url: str) -> tuple[pd.DataFrame, pd.DataFrame]:
     bez_data = get_bez_data(["bez_data_0", "bez_data_2"], url)
 
     df = construct_dataframe(
         bez_data=bez_data[0],
         grouped=False,
         special_tag="stud",
-    ).sort_values(
-        ["Digitale Befragung", "Bundesland", "Bezirk"], ascending=[False, True, True]
     )
 
     df_state = construct_dataframe(
         bez_data=bez_data[1], grouped=False, no_processing=True
-    ).sort_values("Landesbezirk")
+    )
+
+    return df, df_state
+
+
+@app.route("/")
+def tables(
+    url: str = "https://beschaeftigtenbefragung.verdi.de/",
+):
+    df, df_state = get_tables(url)
+
+    df = df.sort_values(
+        ["Digitale Befragung", "Bundesland", "Bezirk"], ascending=[False, True, True]
+    )
+
+    df_state = df_state.sort_values("Landesbezirk")
 
     output_str = []