adding some columns to gsm and wcdma

queries/process_gsm.py

@@ -31,8 +31,6 @@ BTS_COLUMNS = [
     "fddRscpMin",
     "gprsEnabled",
     "locationAreaIdLAC",
-    "locationAreaIdMCC",
-    "locationAreaIdMNC",
     "rac",
     "rachDropRxLevelThreshold",
     "sectorId",
@@ -42,6 +40,9 @@ BTS_COLUMNS = [
     "gsmPriority",
     "segmentName",
     "Code_Sector",
+    "band_frequence",
+    "type_cellule",
+    "configuration_schema",
 ]
 
 BCF_COLUMNS = [
@@ -78,6 +79,18 @@ def process_gsm_data(file_path: str):
     df_bts["SectorId2"] = (
         df_bts["sectorId"].map(UtilsVars.sector_mapping).fillna(df_bts["sectorId"])
     )
+    df_bts["band_frequence"] = (
+        df_bts["frequencyBandInUse"].map(UtilsVars.band_frequence).fillna("not found")
+    )
+    df_bts["type_cellule"] = (
+        df_bts["frequencyBandInUse"].map(UtilsVars.type_cellule).fillna("not found")
+    )
+    df_bts["configuration_schema"] = (
+        df_bts["frequencyBandInUse"]
+        .map(UtilsVars.configuration_schema)
+        .fillna("not found")
+    )
+
     df_bts["ID_BCF"] = df_bts[["BSC", "BCF"]].astype(str).apply("_".join, axis=1)
     df_bts["Code_Sector"] = (
         df_bts[["code", "SectorId2"]].astype(str).apply("_".join, axis=1)

queries/process_wcdma.py

@@ -38,6 +38,7 @@ WCEL_COLUMNS = [
     "SectorID",
     "Code_Sector",
     "code_wcel",
+    "porteuse",
 ]
 
 WBTS_COLUMNS = [
@@ -92,6 +93,9 @@ def process_wcdma_data(file_path: str):
 
     df_wcel["Code_Sector"] = df_wcel["Code_Sector"].str.replace(".0", "")
 
+    df_wcel["porteuse"] = (
+        df_wcel["UARFCN"].map(UtilsVars.porteuse_mapping).fillna("not found")
+    )
     # Process WBTS data
     df_wbts = dfs["WBTS"]
     df_wbts.columns = df_wbts.columns.str.replace(r"[ ]", "", regex=True)

utils/check_sheet_exist.py

@@ -43,11 +43,11 @@ def check_sheets(technology_attr, sheet_list, file_path):
     missing_sheets = [sheet for sheet in sheet_list if sheet not in available_sheets]
     if not missing_sheets:
         setattr(Technology, technology_attr, True)
-        print(getattr(Technology, technology_attr))
-        print("All sheets exist")
-    else:
-        print(f"Missing sheets: {missing_sheets}")
-        print(getattr(Technology, technology_attr))
+    #     print(getattr(Technology, technology_attr))
+    #     print("All sheets exist")
+    # else:
+    #     print(f"Missing sheets: {missing_sheets}")
+    #     print(getattr(Technology, technology_attr))
 
 
 # Check each technology's sheets

utils/utils_vars.py

@@ -18,7 +18,17 @@ def get_physical_db():
 
 class UtilsVars:
     sector_mapping = {4: 1, 5: 2, 6: 3, 11: 1, 12: 2, 13: 3}
+    type_cellule = {1: "Macro Cell 1800", 0: "Macro Cell 900"}
+    band_frequence = {1: "OML BAND GSM 1800", 0: "OML BAND GSM 900"}
+    configuration_schema = {1: "EGPRS 1800", 0: "EGPRS 900"}
     channeltype_mapping = {4: "BCCH", 3: "TCH"}
+    porteuse_mapping = {
+        3004: "OML UTRA Band VIII",
+        3006: "OML UTRA Band VIII",
+        10812: "OML UTRA Band I",
+        10787: "OML UTRA Band I",
+        10837: "OML UTRA Band I",
+    }
     final_lte_database = ""
     final_gsm_database = ""
     final_wcdma_database = ""