adding NEI symetrique checking

queries/process_neighbors.py

@@ -33,6 +33,31 @@ WCEL_TARGET = [
 ]
 
 
+def check_symmetry(df: pd.DataFrame):
+    """
+    Check for symmetric relationships in a dataframe of network neighbors.
+    For each source-target pair, checks if the reverse target-source pair exists.
+
+    Args:
+        df: pandas DataFrame with columns 'SOURCE_NAME' and 'TARGET_NAME'
+
+    Returns:
+        DataFrame with added 'SYMETRIQUE' column ('YES' if symmetric, 'NO' otherwise)
+    """
+    # Create a set of all (source, target) pairs for fast lookup
+    pairs = set(zip(df["SOURCE_NAME"], df["TARGET_NAME"]))
+
+    # Check for each row if the reverse relationship exists
+    df["SYMETRIQUE"] = df.apply(
+        lambda row: (
+            "YES" if (row["TARGET_NAME"], row["SOURCE_NAME"]) in pairs else "NO"
+        ),
+        axis=1,
+    )
+
+    return df
+
+
 def process_neighbors_data(file_path: str):
     """
     Process data from the specified file path.
@@ -84,6 +109,7 @@ def process_neighbors_data(file_path: str):
     df_adce_final.rename(
         columns={"ID_BTS": "SOURCE_ID", "lac_id": "TARGET_LAC_ID"}, inplace=True
     )
+    df_adce_final = check_symmetry(df_adce_final)
 
     # process ADJS data
     df_adjs = dfs["ADJS"]
@@ -120,6 +146,7 @@ def process_neighbors_data(file_path: str):
     df_adjs_final.rename(
         columns={"ID_WCEL": "SOURCE_ID", "lac_id": "TARGET_LAC_ID"}, inplace=True
     )
+    df_adjs_final = check_symmetry(df_adjs_final)
 
     # process ADJI DATA
     df_adji = dfs["ADJI"]
@@ -140,6 +167,7 @@ def process_neighbors_data(file_path: str):
     df_adji_final.rename(
         columns={"ID_WCEL": "SOURCE_ID", "lac_id": "TARGET_LAC_ID"}, inplace=True
     )
+    df_adji_final = check_symmetry(df_adji_final)
 
     # process ADJG DATA
     df_adjg = dfs["ADJG"]