From bd4432bb47178677463e3c0ae1886c1b79f364b7 Mon Sep 17 00:00:00 2001
From: JulienCorny <40057801+JulienCorny@users.noreply.github.com>
Date: Wed, 11 Dec 2024 18:45:47 +0100
Subject: [PATCH] fix(backend): rb and dep filter errors (#87)

@JulienCorny @edelclaux
---
 backend/gn_module_zh/search.py | 11 +++++++----
 1 file changed, 7 insertions(+), 4 deletions(-)

diff --git a/backend/gn_module_zh/search.py b/backend/gn_module_zh/search.py
index c111c761..eba676c1 100644
--- a/backend/gn_module_zh/search.py
+++ b/backend/gn_module_zh/search.py
@@ -154,7 +154,11 @@ def filter_area_size(query, json: dict):
 
 
 def filter_area(query, json: dict, type_code: str):
-    codes = [area.get("code", None) for area in json]
+    if type_code == "COM":
+        codes = [area.get("code", None) for area in json]
+    else:
+        codes = json["code"].split()
+
     if any(code is None for code in codes):
         return query
 
@@ -193,9 +197,8 @@ def filter_hydro(query, json):
     return query
 
 
-def filter_basin(query, basin):
-    code = basin.get("code", None)
-
+def filter_basin(query, json):
+    code = json["code"]
     if code is not None:
         subquery = (
             select(