Fix serialization

2022-08-25 23:05:07 +02:00 · 2022-08-25 23:05:07 +02:00 · ed9f9f4c5a
commit ed9f9f4c5a
parent f15c4cdcd9
1 changed files with 2 additions and 0 deletions
--- a/latecomers/serializer.py
+++ b/latecomers/serializer.py
@ -29,6 +29,8 @@ def to_excel(data: T.List[Details]) -> bytes:
    df = pd.DataFrame(data, columns=mapping)
    df["status"] = df["status"].map(lambda x: x.value)
    df.set_index("th_arrival", inplace=True)
+    df.rename(columns=mapping, inplace=True)
+    df = df.rename_axis(index=mapping["th_arrival"])

    with NamedTemporaryFile() as tmp:
        df.to_excel(tmp)