improve invunit

queries/process_invunit.py

@@ -1,9 +1,51 @@
 import pandas as pd
 
-from utils.convert_to_excel import convert_dfs, save_dataframe
+from utils.convert_to_excel import convert_invunit_dfs, save_dataframe
 from utils.extract_code import extract_code_from_mrbts
 from utils.utils_vars import UtilsVars
 
+RF_UNIT = [
+    "AHDA",
+    "AHEGB",
+    "AHEGC",
+    "AHEGHA",
+    "AHGA",
+    "AHMA",
+    "AHPMDA",
+    "AHPMDG",
+    "AHPMDI",
+    "ARDA",
+    "AREA",
+    "ARGA",
+    "ARMA",
+    "AZNA",
+    "FHDB",
+    "FHEB",
+    "FHEL",
+    "FRGU",
+    "FRGY",
+    "FRMB",
+    "FRMF",
+    "FXDB",
+    "FXED",
+    "FZNI",
+]
+
+BB_UNIT_LIST = [
+    "FBBA",
+    "FBBC",
+    "FSMF",
+    "ABIA",
+    "total_number_of_subunit",
+]
+
+BB_UNIT_VALUES = {
+    "FBBA": 6,
+    "FBBC": 6,
+    "FSMF": 5.5,
+    "ABIA": 8,
+}
+
 
 def create_invunit_summary(df: pd.DataFrame) -> pd.DataFrame:
     """
@@ -75,20 +117,22 @@ 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.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,
+
+    # Compute total_number_of_subunit
+    df_invunit["total_number_of_subunit"] = sum(
+        df_invunit[col].fillna(0) * val for col, val in BB_UNIT_VALUES.items()
     )
 
+    # Start order with "MRBTS", "code", "invunit_summary" follow by bb_unit  , rf_unit and then all other columns
+    df_invunit = df_invunit[
+        ["MRBTS", "code", "invunit_summary"]
+        + BB_UNIT_LIST
+        + RF_UNIT
+        + df_invunit.columns.difference(
+            ["MRBTS", "code", "invunit_summary"] + BB_UNIT_LIST + RF_UNIT
+        ).tolist()
+    ]
+
     UtilsVars.all_db_dfs.append(df_invunit)
     UtilsVars.all_db_dfs_names.append("INVUNIT")
     return df_invunit
@@ -102,4 +146,4 @@ def process_invunit_data_to_excel(file_path: str) -> None:
         file_path (str): The path to the file.
     """
     invunit_df = process_invunit_data(file_path)
-    UtilsVars.final_invunit_database = convert_dfs([invunit_df], ["INVUNIT"])
+    UtilsVars.final_invunit_database = convert_invunit_dfs([invunit_df], ["INVUNIT"])

utils/convert_to_excel.py

@@ -146,6 +146,36 @@ def get_format_map_by_format_type(formats: dict, format_type: str) -> dict:
             "number_trx_per_cell": formats["blue_light"],
             "number_trx_per_bcf": formats["blue_light"],
             "number_trx_per_site": formats["blue_light"],
+            # invunit part in database
+            "FBBA": formats["blue_light"],
+            "FBBC": formats["blue_light"],
+            "FSMF": formats["blue_light"],
+            "ABIA": formats["blue_light"],
+            "total_number_of_subunit": formats["blue_light"],
+            "AHDA": formats["beurre"],
+            "AHEGB": formats["beurre"],
+            "AHEGC": formats["beurre"],
+            "AHEGHA": formats["beurre"],
+            "AHGA": formats["beurre"],
+            "AHMA": formats["beurre"],
+            "AHPMDA": formats["beurre"],
+            "AHPMDG": formats["beurre"],
+            "AHPMDI": formats["beurre"],
+            "ARDA": formats["beurre"],
+            "AREA": formats["beurre"],
+            "ARGA": formats["beurre"],
+            "ARMA": formats["beurre"],
+            "AZNA": formats["beurre"],
+            "FHDB": formats["beurre"],
+            "FHEB": formats["beurre"],
+            "FHEL": formats["beurre"],
+            "FRGU": formats["beurre"],
+            "FRGY": formats["beurre"],
+            "FRMB": formats["beurre"],
+            "FRMF": formats["beurre"],
+            "FXDB": formats["beurre"],
+            "FXED": formats["beurre"],
+            "FZNI": formats["beurre"],
         }
     elif format_type == "LTE_Analysis":
         return {
@@ -228,6 +258,40 @@ def get_format_map_by_format_type(formats: dict, format_type: str) -> dict:
             "final_comments": formats["green"],
         }
 
+    elif format_type == "invunit":
+        return {
+            "code": formats["blue"],
+            "FBBA": formats["blue_light"],
+            "FBBC": formats["blue_light"],
+            "FSMF": formats["blue_light"],
+            "ABIA": formats["blue_light"],
+            "total_number_of_subunit": formats["blue_light"],
+            "AHDA": formats["beurre"],
+            "AHEGB": formats["beurre"],
+            "AHEGC": formats["beurre"],
+            "AHEGHA": formats["beurre"],
+            "AHGA": formats["beurre"],
+            "AHMA": formats["beurre"],
+            "AHPMDA": formats["beurre"],
+            "AHPMDG": formats["beurre"],
+            "AHPMDI": formats["beurre"],
+            "ARDA": formats["beurre"],
+            "AREA": formats["beurre"],
+            "ARGA": formats["beurre"],
+            "ARMA": formats["beurre"],
+            "AZNA": formats["beurre"],
+            "FHDB": formats["beurre"],
+            "FHEB": formats["beurre"],
+            "FHEL": formats["beurre"],
+            "FRGU": formats["beurre"],
+            "FRGY": formats["beurre"],
+            "FRMB": formats["beurre"],
+            "FRMF": formats["beurre"],
+            "FXDB": formats["beurre"],
+            "FXED": formats["beurre"],
+            "FZNI": formats["beurre"],
+        }
+
     else:
         return {}  # No formatting if format_type not matched
 
@@ -285,6 +349,11 @@ def convert_database_dfs(dfs, sheet_names) -> bytes:
     return _write_to_excel(dfs, sheet_names, index=True, format_type="database")
 
 
+@st.cache_data
+def convert_invunit_dfs(dfs, sheet_names) -> bytes:
+    return _write_to_excel(dfs, sheet_names, index=True, format_type="invunit")
+
+
 def save_dataframe(df: pd.DataFrame, sheet_name: str):
     """
     Save the dataframe to a csv file.