Update supabase_utils.py

supabase_utils.py

@@ -2,25 +2,17 @@ from supabase import create_client, Client
 import os
 import re
 import json
-from dotenv import load_dotenv
-
 
 # ---------------------------------------------
-# تهيئة Supabase
+# تحميل مفاتيح Supabase من Secrets
 # ---------------------------------------------
-#load_dotenv()
 SUPABASE_URL = os.environ.get("SUPABASE_URL")
 SUPABASE_KEY = os.environ.get("SUPABASE_SERVICE_ROLE_KEY")
 
-if not SUPABASE_URL or not SUPABASE_SERVICE_ROLE_KEY:
+if not SUPABASE_URL or not SUPABASE_KEY:
     raise ValueError("❌ لم يتم العثور على مفاتيح Supabase في Secrets")
 
-#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)
 
 
@@ -32,20 +24,28 @@ def save_law_to_supabase(law_json: dict):
     law_json: dict يحتوي على المفاتيح
         title, preamble, number, year, sections
     """
+
     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", [])
 
-    # 1) التحقق من وجود القانون
-    existing = supabase.table("laws").select("id").eq("title", title).execute()
+    # ---------------------------------------------------------
+    # 1) التحقق من وجود القانون سابقًا
+    # ---------------------------------------------------------
+    existing = (
+        supabase.table("laws")
+        .select("id")
+        .eq("title", title)
+        .execute()
+    )
 
     if existing.data:
         law_id = existing.data[0]["id"]
         print(f"⚠ القانون موجود مسبقاً، يتم حذفه: {title}")
 
-        # حذف المواد المرتبطة بالأقسام
+        # حذف المواد التابعة للأقسام
         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()
@@ -56,7 +56,9 @@ def save_law_to_supabase(law_json: dict):
         # حذف القانون نفسه
         supabase.table("laws").delete().eq("id", law_id).execute()
 
-    # 2) إدراج القانون الجديد
+    # ---------------------------------------------------------
+    # 2) إضافة القانون الجديد
+    # ---------------------------------------------------------
     res_law = (
         supabase.table("laws")
         .insert({
@@ -68,44 +70,49 @@ def save_law_to_supabase(law_json: dict):
         })
         .execute()
     )
+
     law_id = res_law.data[0]["id"]
     print(f"✔ تم إدراج القانون الجديد: {title}")
 
-    # 3) إدراج الأقسام والمواد
+    # ---------------------------------------------------------
+    # 3) إضافة الأقسام والمواد
+    # ---------------------------------------------------------
     for sec in sections:
         sec_title   = sec.get("title", "")
         sec_content = sec.get("content", "")
         sec_tables  = sec.get("tables", [])
 
+        # استخراج رقم القسم من العنوان
+        chapter_number = 0
+        match = re.search(r"\d+", sec_title or "")
+        if match:
+            chapter_number = int(match.group(0))
+
         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,
+                "chapter_number": chapter_number,
                 "tables":         sec_tables or [],
             })
             .execute()
         )
+
         section_id = res_sec.data[0]["id"]
 
+        # ----------------------- إضافة المواد -----------------------
         for art in sec.get("articles", []):
-            article_number   = int(art["number"]) if art.get("number") else None
-            article_content  = art.get("text", "")
-            article_tables   = art.get("tables", [])
+            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,
+                "chapter_id":     section_id,
                 "article_number": article_number,
-                "content":       article_content or "",
-                "tables":        article_tables or [],
+                "content":        article_content or "",
+                "tables":         article_tables or [],
             }).execute()
 
-    print(f"✔ تم حفظ جميع الأقسام والمواد للقانون: {title}")
-
-
-# ---------------------------------------------
-# مثال استخدام (يمكن حذفه أو تعليقه)
-# ---------------------------------------------
+    print(f"✔ تم حفظ جميع الأقسام والمواد للقانون: {title}")