diff --git a/wsgi.py b/wsgi.py index fff3e74..bd1caba 100644 --- a/wsgi.py +++ b/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 = []