Update supabase_utils.py

supabase_utils.py

@@ -23,68 +23,63 @@ supabase: Client = create_client(SUPABASE_URL, SUPABASE_KEY)
 # ---------------------------------------------
 def save_law_to_supabase(law_json):
     """
-    law_json: dict يحتوي على المفتاح 'title', 'preamble', 'sections'
+    law_json: dict يحتوي على المفتاح 'title', 'preamble', 'sections', 'number', 'year'
     """
     title = law_json.get("title", "عنوان غير معروف")
     preamble = law_json.get("preamble", "")
+    law_number = law_json.get("number")  # استخدم None بدل null
+    law_year = law_json.get("year")      
+
     sections = law_json.get("sections", [])
 
-    # 1) إدراج القانون إذا لم يكن موجودًا
-    existing_law = supabase.table("laws").select("id").eq("title", title).execute()
-    if existing_law.data and len(existing_law.data) > 0:
-        law_id = existing_law.data[0]["id"]
-        print(f"⚠ القانون موجود مسبقاً: {title}")
-    else:
-        res = (
-            supabase.table("laws")
-            .insert(
-                {
-                    "title": title or "",
-                    "preamble": preamble or "",
-                    "law_type": "def",
-                }
-            )
-            .execute()
+    # 1) UPSERT القانون مباشرة باستخدام المفتاح الفريد (number_law, year_law, title)
+    res = (
+        supabase.table("laws")
+        .upsert(
+            {
+                "title": title or "",
+                "number_law": law_number,
+                "year_law": law_year,
+                "preamble": preamble or "",
+                "law_type": "def",
+            },
+            on_conflict=["number_law", "year_law", "title"],  # المفتاح الفريد الثلاثي
         )
+        .execute()
+    )
+
+    if res.data and len(res.data) > 0:
         law_id = res.data[0]["id"]
-        print(f"✔ تم إدراج القانون: {title}")
+        print(f"✔ تم إدراج/تحديث القانون: {title}")
+    else:
+        raise Exception(f"❌ فشل إدراج/تحديث القانون: {title}")
 
-    # 2) إدراج الأقسام والمواد مع الجداول داخل العمود jsonb
+    # 2) إدراج الأقسام والمواد كما في الكود السابق
     for sec in sections:
         sec_title = sec.get("title", "")
         sec_content = sec.get("content", "")
         sec_tables = sec.get("tables", [])
 
-        # التحقق من وجود القسم مسبقاً
-        existing_section = (
+        # UPSERT الأقسام بناءً على law_id و title
+        res_sec = (
             supabase.table("sections")
-            .select("id")
-            .eq("law_id", law_id)
-            .eq("title", sec_title)
+            .upsert(
+                {
+                    "law_id": law_id,
+                    "title": sec_title or "",
+                    "content": sec_content or "",
+                    "chapter_number": (
+                        int(re.search(r"\d+", sec_title or "0").group(0))
+                        if re.search(r"\d+", sec_title or "")
+                        else 0
+                    ),
+                    "tables": sec_tables or [],
+                },
+                on_conflict=["law_id", "title"],  # المفتاح الفريد للأقسام
+            )
             .execute()
         )
-
-        if existing_section.data and len(existing_section.data) > 0:
-            section_id = existing_section.data[0]["id"]
-        else:
-            res_sec = (
-                supabase.table("sections")
-                .insert(
-                    {
-                        "law_id": law_id,
-                        "title": sec_title or "",
-                        "content": sec_content or "",
-                        "chapter_number": (
-                            int(re.search(r"\d+", sec_title or "0").group(0))
-                            if re.search(r"\d+", sec_title or "")
-                            else 0
-                        ),
-                        "tables": sec_tables or [],  # تخزين الجداول مباشرة في jsonb
-                    }
-                )
-                .execute()
-            )
-            section_id = res_sec.data[0]["id"]
+        section_id = res_sec.data[0]["id"]
 
         # إدراج المواد
         for art in sec.get("articles", []):
@@ -92,18 +87,20 @@ def save_law_to_supabase(law_json):
             article_content = art.get("text", "")
             article_tables = art.get("tables", [])
 
+            # UPSERT المواد بناءً على chapter_id و article_number
             res_art = (
                 supabase.table("articles")
-                .insert(
+                .upsert(
                     {
                         "chapter_id": section_id,
                         "article_number": article_number,
                         "content": article_content or "",
-                        "tables": article_tables or [],  # تخزين الجداول ه��ا أيضاً
-                    }
+                        "tables": article_tables or [],
+                    },
+                    on_conflict=["chapter_id", "article_number"],  # المفتاح الفريد للمواد
                 )
                 .execute()
             )
             article_id = res_art.data[0]["id"]
 
-    print(f"✔ تم حفظ جميع الأقسام والمواد والجداول للقانون: {title}")
+    print(f"✔ تم حفظ/تحديث جميع الأقسام والمواد والجداول للقانون: {title}")