Add 3G Adjl checking

queries/process_adjl.py

@@ -15,7 +15,9 @@ ADJL_GSM_COLUMNS = ["BSC", "BCF", "BTS", "ADJL", "earfcn", "lteAdjCellTac"]
 
 ADJL_WCDMA_COLUMNS = ["RNC", "WBTS", "WCEL", "ADJL", "AdjLEARFCN"]
 
-BTS_COLUMNS = ["ID_BTS", "name", "locationAreaIdLAC", "Code_Sector"]
+BTS_COLUMNS = ["ID_BTS", "name", "Code_Sector"]
+
+WCEL_COLUMNS = ["ID_WCEL", "name", "Code_Sector"]
 
 LTE_COLUMNS_CONFIG = ["Code_Sector", "site_config_band"]
 
@@ -87,10 +89,10 @@ def process_adjl_data(file_path: str) -> list[pd.DataFrame]:
     df_bts = process_gsm_data(file_path)[BTS_COLUMNS]
 
     # ------------------- WCEL -------------------
-    df_wcel = process_wcdma_data(file_path)
-    df_wcel["ID_WCEL"] = (
-        df_wcel[["RNC", "WBTS", "WCEL"]].astype(str).agg("_".join, axis=1)
-    )
+    df_wcel = process_wcdma_data(file_path)[WCEL_COLUMNS]
+    # df_wcel["ID_WCEL"] = (
+    #     df_wcel[["RNC", "WBTS", "WCEL"]].astype(str).agg("_".join, axis=1)
+    # )
 
     # ------------------- LTE -------------------
     lte_fdd_df, lte_tdd_df = process_lte_data(file_path)
@@ -164,7 +166,48 @@ def process_adjl_data(file_path: str) -> list[pd.DataFrame]:
     )
 
     # Mark existing BTS
-    df_bts["exists"] = df_bts["ID_BTS"].isin(gsm_adjl_df["ID_BTS"])
+    df_bts["adjl_exists"] = df_bts["ID_BTS"].isin(gsm_adjl_df["ID_BTS"])
+
+    # --- WCDMA ADJL ---
+    # Filter invalid rows
+    wcdma_adjl_df = wcdma_adjl_df[
+        wcdma_adjl_df["RNC"].notna()
+        & wcdma_adjl_df["WBTS"].notna()
+        & wcdma_adjl_df["WCEL"].notna()
+    ].reset_index(drop=True)
+
+    # Build IDs and bands
+    wcdma_adjl_df["ID_WCEL"] = (
+        wcdma_adjl_df[["RNC", "WBTS", "WCEL"]].astype(str).agg("_".join, axis=1)
+    )
+    wcdma_adjl_df["ID_WCEL"] = wcdma_adjl_df["ID_WCEL"].str.replace(
+        ".0", "", regex=False
+    )
+    wcdma_adjl_df["adjl_band"] = wcdma_adjl_df["AdjLEARFCN"].map(UtilsVars.lte_band)
+
+    # Merge WCEL info
+    wcdma_adjl_df = pd.merge(wcdma_adjl_df, df_wcel, on="ID_WCEL", how="left")
+
+    # Aggregate ADJL band info
+    wcdma_adjl_df_band = adjl_band(wcdma_adjl_df, "ID_WCEL", "adjl_band")
+    wcdma_adjl_df = pd.merge(
+        wcdma_adjl_df, wcdma_adjl_df_band, on="ID_WCEL", how="left"
+    )
+
+    # Build Code_Sector_band
+    wcdma_adjl_df["Code_Sector_band"] = (
+        wcdma_adjl_df[["Code_Sector", "adjl_band"]].astype(str).agg("_".join, axis=1)
+    )
+
+    # Merge LTE references
+    wcdma_adjl_df = wcdma_adjl_df.merge(lte_df_config, on="Code_Sector", how="left")
+
+    # Validations
+    wcdma_adjl_df["check_bands"] = wcdma_adjl_df.apply(check_bands, axis=1)
+    wcdma_adjl_df["missing_bands"] = wcdma_adjl_df.apply(missing_bands, axis=1)
+
+    # Mark existing WCEL
+    df_wcel["adjl_exists"] = df_wcel["ID_WCEL"].isin(wcdma_adjl_df["ID_WCEL"])
 
     return [gsm_adjl_df, wcdma_adjl_df, df_bts, df_wcel, lte_df]