improve invunit

queries/process_invunit.py

@@ -75,13 +75,20 @@ def process_invunit_data(file_path: str) -> pd.DataFrame:
     )
     df_invunit = create_invunit_summary(df_invunit)
     df_invunit["code"] = df_invunit["MRBTS"].apply(extract_code_from_mrbts)
-    df_invunit = df_invunit[["MRBTS", "code", "invunit_summary"]]
+    # df_invunit = df_invunit[["MRBTS", "code", "invunit_summary"]]
     # df_invunit = df_invunit.sort_values(
     #     by=["MRBTS", "code"],
     # )
     # df_invunit["MRBTS_NUM"] = df_invunit.groupby("code").cumcount() + 1
     # df_invunit["MRBTS_NUM"] = "MRBTS_" + df_invunit["MRBTS_NUM"].astype(str)
 
+    # Move the columns to the desired position and keep all other columns
+    df_invunit = df_invunit[["MRBTS", "code", "invunit_summary"]].merge(
+        df_invunit.drop(["MRBTS", "code", "invunit_summary"], axis=1),
+        left_index=True,
+        right_index=True,
+    )
+
     UtilsVars.all_db_dfs.append(df_invunit)
     UtilsVars.all_db_dfs_names.append("INVUNIT")
     return df_invunit