Update supabase_utils.py

supabase_utils.py

@@ -1,24 +1,48 @@
-def save_law_to_supabase(law_json):
+from supabase import create_client, Client
+import os
+import re
+import json
+from dotenv import load_dotenv
+
+
+# ---------------------------------------------
+# تهيئة Supabase
+# ---------------------------------------------
+load_dotenv()
+SUPABASE_URL   = os.getenv("SUPABASE_URL")
+SUPABASE_KEY   = os.getenv("SUPABASE_SERVICE_ROLE_KEY")
+
+if not SUPABASE_URL or not SUPABASE_KEY:
+    raise Exception("يجب تعريف SUPABASE_URL و SUPABASE_SERVICE_ROLE_KEY في .env")
+
+supabase: Client = create_client(SUPABASE_URL, SUPABASE_KEY)
+
+
+# ---------------------------------------------
+# حفظ / استبدال القانون في قاعدة البيانات
+# ---------------------------------------------
+def save_law_to_supabase(law_json: dict):
     """
-    law_json: dict يحتوي على المفتاح 'title', 'preamble', 'sections'
+    law_json: dict يحتوي على المفاتيح
+        title, preamble, number, year, sections
     """
-    title = law_json.get("title", "عنوان غير معروف")
-    preamble = law_json.get("preamble", "")
+    title      = law_json.get("title", "عنوان غير معروف")
+    preamble   = law_json.get("preamble", "")
     law_number = law_json.get("number")
-    law_year = law_json.get("year")
-    sections = law_json.get("sections", [])
+    law_year   = law_json.get("year")
+    sections   = law_json.get("sections", [])
 
     # 1) التحقق من وجود القانون
-    existing_law = supabase.table("laws").select("id").eq("title", title).execute()
+    existing = 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"]
+    if existing.data:
+        law_id = existing.data[0]["id"]
         print(f"⚠ القانون موجود مسبقاً، يتم حذفه: {title}")
 
         # حذف المواد المرتبطة بالأقسام
-        sections_res = supabase.table("sections").select("id").eq("law_id", law_id).execute()
-        for sec in sections_res.data:
-            supabase.table("articles").delete().eq("chapter_id", sec["id"]).execute()
+        secs = supabase.table("sections").select("id").eq("law_id", law_id).execute()
+        for s in secs.data:
+            supabase.table("articles").delete().eq("chapter_id", s["id"]).execute()
 
         # حذف الأقسام
         supabase.table("sections").delete().eq("law_id", law_id).execute()
@@ -27,59 +51,55 @@ def save_law_to_supabase(law_json):
         supabase.table("laws").delete().eq("id", law_id).execute()
 
     # 2) إدراج القانون الجديد
-    res = (
+    res_law = (
         supabase.table("laws")
-        .insert(
-            {
-                "title": title or "",
-                "number_law": law_number,
-                "year_law": law_year,
-                "preamble": preamble or "",
-                "law_type": "def",
-            }
-        )
+        .insert({
+            "title":      title or "",
+            "number_law": law_number,
+            "year_law":   law_year,
+            "preamble":   preamble or "",
+            "law_type":   "def",
+        })
         .execute()
     )
-    law_id = res.data[0]["id"]
+    law_id = res_law.data[0]["id"]
     print(f"✔ تم إدراج القانون الجديد: {title}")
 
     # 3) إدراج الأقسام والمواد
     for sec in sections:
-        sec_title = sec.get("title", "")
+        sec_title   = sec.get("title", "")
         sec_content = sec.get("content", "")
-        sec_tables = sec.get("tables", [])
+        sec_tables  = sec.get("tables", [])
 
         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 [],
-                }
-            )
+            .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 [],
+            })
             .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", [])
-
-            supabase.table("articles").insert(
-                {
-                    "chapter_id": section_id,
-                    "article_number": article_number,
-                    "content": article_content or "",
-                    "tables": article_tables or [],
-                }
-            ).execute()
-
-    print(f"✔ تم حفظ القانون الجديد مع جميع الأقسام والمواد: {title}")
+            article_number   = int(art["number"]) if art.get("number") else None
+            article_content  = art.get("text", "")
+            article_tables   = art.get("tables", [])
+
+            supabase.table("articles").insert({
+                "chapter_id":    section_id,
+                "article_number": article_number,
+                "content":       article_content or "",
+                "tables":        article_tables or [],
+            }).execute()
+
+    print(f"✔ تم حفظ جميع الأقسام والمواد للقانون: {title}")
+
+
+# ---------------------------------------------
+# مثال استخدام (يمكن حذفه أو تعليقه)
+# ---------------------------------------------