Update supabase_utils.py

supabase_utils.py

@@ -1,10 +1,8 @@
 from supabase import create_client, Client
 import os
 import re
-import json
 from dotenv import load_dotenv
 
-
 # ---------------------------------------------
 # تهيئة Supabase
 # ---------------------------------------------
@@ -17,22 +15,22 @@ if not SUPABASE_URL or not SUPABASE_KEY:
 
 supabase: Client = create_client(SUPABASE_URL, SUPABASE_KEY)
 
-
 # ---------------------------------------------
-# حفظ القانون في قاعدة البيانات
+# حفظ القانون في قاعدة البيانات مع الأقسام والمواد
 # ---------------------------------------------
-def save_law_to_supabase(law_json):
+def save_law_to_supabase(law_json: dict):
     """
-    law_json: dict يحتوي على المفتاح 'title', 'preamble', 'sections', 'number', 'year'
+    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_number = law_json.get("number")
     law_year = law_json.get("year")      
-
     sections = law_json.get("sections", [])
 
-    # 1) UPSERT القانون مباشرة باستخدام المفتاح الفريد (number_law, year_law, title)
+    # -------------------------------
+    # 1) UPSERT القانون
+    # -------------------------------
     res = (
         supabase.table("laws")
         .upsert(
@@ -43,7 +41,7 @@ def save_law_to_supabase(law_json):
                 "preamble": preamble or "",
                 "law_type": "def",
             },
-            on_conflict=["number_law", "year_law", "title"],  # المفتاح الفريد الثلاثي
+            on_conflict=["number_law", "year_law", "title"],
         )
         .execute()
     )
@@ -54,13 +52,14 @@ def save_law_to_supabase(law_json):
     else:
         raise Exception(f"❌ فشل إدراج/تحديث القانون: {title}")
 
-    # 2) إدراج الأقسام والمواد كما في الكود السابق
+    # -------------------------------
+    # 2) UPSERT الأقسام
+    # -------------------------------
     for sec in sections:
         sec_title = sec.get("title", "")
         sec_content = sec.get("content", "")
         sec_tables = sec.get("tables", [])
 
-        # UPSERT الأقسام بناءً على law_id و title
         res_sec = (
             supabase.table("sections")
             .upsert(
@@ -75,34 +74,35 @@ def save_law_to_supabase(law_json):
                     ),
                     "tables": sec_tables or [],
                 },
-                on_conflict=["law_id", "title"],  # المفتاح الفريد للأقسام
+                on_conflict=["law_id", "title"],
             )
             .execute()
         )
+
         section_id = res_sec.data[0]["id"]
 
-        # إدراج المواد
-        # إدراج المواد
-for art in sec.get("articles", []):
-    article_number = int(art.get("number")) if art.get("number") else None
-    article_content = art.get("text", "")
-    article_tables = art.get("tables", [])
+        # -------------------------------
+        # 3) UPSERT المواد لكل قسم
+        # -------------------------------
+        for art in sec.get("articles", []):
+            article_number = int(art.get("number")) if art.get("number") else None
+            article_content = art.get("text", "")
+            article_tables = art.get("tables", [])
 
-    # UPSERT المواد بناءً على chapter_id و article_number مع id_law
-    res_art = (
-        supabase.table("articles")
-        .upsert(
-            {
-                "chapter_id": section_id,
-                "id_law": law_id,                 # ربط المادة بالقانون مباشرة
-                "article_number": article_number,
-                "content": article_content or "",
-                "tables": article_tables or [],
-            },
-            on_conflict=["chapter_id", "id_law", "article_number"],  # المفتاح الفريد المركب للمواد
-        )
-        .execute()
-    )
-    article_id = res_art.data[0]["id"]
+            res_art = (
+                supabase.table("articles")
+                .upsert(
+                    {
+                        "chapter_id": section_id,
+                        "id_law": law_id,                 # ربط المادة بالقانون مباشرة
+                        "article_number": article_number,
+                        "content": article_content or "",
+                        "tables": article_tables or [],
+                    },
+                    on_conflict=["chapter_id", "article_number"],  # المفتاح الفريد للمادة
+                )
+                .execute()
+            )
+            article_id = res_art.data[0]["id"]
 
     print(f"✔ تم حفظ/تحديث جميع الأقسام والمواد والجداول للقانون: {title}")