Update main.py

main.py

@@ -92,17 +92,17 @@ def gregorian_to_jalali(gy, gm, gd):
     return jy, jm, jd
 
 # ==============================================================================
-# 🟢 پارت 4: دیتابیس SQLite (نسخه فوق‌امن - بازسازی کامل برای رفع Malformed)
+# 🟢 پارت 4: دیتابیس SQLite (نسخه نجات همه‌جانبه: اعتبارها + پیام‌های قدیمی)
 # ==============================================================================
 import os
 import sqlite3
 import json
 import copy
 
-# تغییر نام فایل اصلی برای اطمینان از حذف کش‌های خراب سیستم‌عامل
-DB_FILE = "/data/users_v2.db" 
-OLD_DB_FILE = "/data/users_db.db"
-BACKUP_JSON_FILE = "/data/users_db.json.bak"
+# استفاده از نام جدید v3 برای شروع کاملاً پاک و بدون قفل
+DB_FILE = "/data/users_v3.db" 
+CORRUPT_FILE = "/data/users_db.db" # فایل ۱۱.۷ مگابایتی شما
+BACKUP_JSON = "/data/users_db.json.bak" # فایل ۴.۷ مگابایتی اعتبارها
 
 last_saved_state = {}
 
@@ -110,49 +110,32 @@ def init_sqlite_db():
     try:
         os.makedirs(os.path.dirname(DB_FILE), exist_ok=True)
         
-        # حذف فایل‌های موقت WAL قدیمی اگر وجود دارند
-        for f in [OLD_DB_FILE+"-wal", OLD_DB_FILE+"-shm", DB_FILE+"-wal", DB_FILE+"-shm"]:
-            if os.path.exists(f): 
-                try: os.remove(f)
-                except: pass
-
-        # بررسی سلامت فایل فعلی
-        should_rebuild = False
-        if not os.path.exists(DB_FILE):
-            should_rebuild = True
-        else:
-            try:
-                conn = sqlite3.connect(DB_FILE, timeout=10)
-                conn.execute("PRAGMA integrity_check").fetchone()
-                conn.close()
-            except:
-                should_rebuild = True
-                print("🚨 فایل v2 هم خراب شده، نیاز به بازسازی مجدد...")
-
-        if should_rebuild:
-            print("🛠 در حال ساخت دیتابیس جدید و سالم...")
-            if os.path.exists(DB_FILE):
-                os.rename(DB_FILE, DB_FILE + ".broken")
-            
-            conn = sqlite3.connect(DB_FILE, timeout=60.0)
-            cursor = conn.cursor()
-            cursor.execute('PRAGMA journal_mode=DELETE;') # امن‌ترین حالت برای درایو شبکه
-            cursor.execute('CREATE TABLE users (chat_id TEXT PRIMARY KEY, user_data TEXT)')
-            cursor.execute('CREATE TABLE processed_messages (message_id TEXT PRIMARY KEY)')
-            conn.commit()
-            conn.close()
-            
-            # بلافاصله اطلاعات کاربران را از بک‌آپ سالم برگردان
-            restore_from_json()
+        # اگر فایل v3 وجود ندارد، یعنی بار اول است و باید نجات را شروع کنیم
+        is_first_run = not os.path.exists(DB_FILE)
+
+        conn = sqlite3.connect(DB_FILE, timeout=60.0)
+        cursor = conn.cursor()
+        cursor.execute('PRAGMA journal_mode=DELETE;') # امن‌ترین حالت برای شبکه
+        cursor.execute('CREATE TABLE IF NOT EXISTS users (chat_id TEXT PRIMARY KEY, user_data TEXT)')
+        cursor.execute('CREATE TABLE IF NOT EXISTS processed_messages (message_id TEXT PRIMARY KEY)')
+        conn.commit()
+        conn.close()
+
+        if is_first_run:
+            print("🚨 عملیات نجات آغاز شد...")
+            # ۱. بازیابی اعتبارها از JSON (۱۰۰ درصد مطمئن)
+            restore_users_from_json()
+            # ۲. نجات شناسه‌های پیام از فایل ۱۱.۷ مگابایتی (تا جایی که راه بدهد)
+            salvage_messages_from_corrupt_db()
 
     except Exception as e:
-        print(f"❌ خطا در راه اندازی: {e}")
+        print(f"❌ خطا در راه اندازی دیتابیس: {e}")
 
-def restore_from_json():
-    if os.path.exists(BACKUP_JSON_FILE):
-        print("📥 در حال تزریق اشتراک‌ها به فایل جدید...")
+def restore_users_from_json():
+    if os.path.exists(BACKUP_JSON):
+        print("📥 در حال بازیابی اعتبار کاربران از فایل پشتیبان...")
         try:
-            with open(BACKUP_JSON_FILE, "r", encoding="utf-8") as f:
+            with open(BACKUP_JSON, "r", encoding="utf-8") as f:
                 data = json.load(f)
             if data:
                 conn = sqlite3.connect(DB_FILE, timeout=60.0)
@@ -160,9 +143,27 @@ def restore_from_json():
                     conn.execute("INSERT OR REPLACE INTO users VALUES (?, ?)", (str(cid), json.dumps(udata)))
                 conn.commit()
                 conn.close()
-                print(f"✅ اطلاعات {len(data)} کاربر بازیابی شد.")
+                print(f"✅ اعتبار {len(data)} کاربر با موفقیت بازیابی شد.")
+        except Exception as e:
+            print(f"❌ خطا در بازیابی اعتبارها: {e}")
+
+def salvage_messages_from_corrupt_db():
+    if os.path.exists(CORRUPT_FILE):
+        print(f"🔍 در حال تلاش برای نجات شناسه‌های پیام از فایل {CORRUPT_FILE} ...")
+        try:
+            # وصل کردن فایل خراب به دیتابیس جدید
+            conn = sqlite3.connect(DB_FILE, timeout=60.0)
+            conn.execute(f"ATTACH DATABASE '{CORRUPT_FILE}' AS old_db")
+            
+            # کپی کردن شناسه‌ها (با استفاده از INSERT OR IGNORE که اگر خراب بود رد شود)
+            # این دستور تلاش می‌کند حتی اگر فایل Malformed باشد، ردیف‌های سالم را بردارد
+            conn.execute("INSERT OR IGNORE INTO processed_messages SELECT message_id FROM old_db.processed_messages")
+            
+            conn.commit()
+            conn.close()
+            print("✅ شناسه‌های پیام قدیمی با موفقیت نجات یافتند و به دیتابیس جدید منتقل شدند.")
         except Exception as e:
-            print(f"❌ خطا در بازیابی بک‌آپ: {e}")
+            print(f"⚠️ هشدار: برخی پیام‌ها به دلیل خرابی شدید فایل اصلی قابل نجات نبودند، اما نگران نباشید، ربات دوباره آن‌ها را شناسایی می‌کند. ارور: {e}")
 
 def load_db():
     global last_saved_state
@@ -176,10 +177,10 @@ def load_db():
             db_dict[row[0]] = json.loads(row[1])
         conn.close()
         last_saved_state = copy.deepcopy(db_dict)
-        print(f"✅ دیتابیس با موفقیت لود شد. کاربران: {len(db_dict)}")
+        print(f"🚀 دیتابیس نهایی آماده شد. تعداد کاربران: {len(db_dict)}")
         return db_dict
     except Exception as e:
-        print(f"⚠️ ارور بحرانی در لود نهایی: {e}")
+        print(f"⚠️ ارور در لود نهایی: {e}")
         return {}
 
 def save_db(db_data):
@@ -192,7 +193,7 @@ def save_db(db_data):
         if not changed: return
         conn = sqlite3.connect(DB_FILE, timeout=60.0)
         conn.execute('PRAGMA journal_mode=DELETE;')
-        conn.executemany("INSERT OR REPLACE INTO users VALUES (?, ?)", changed)
+        conn.executemany("INSERT OR REPLACE INTO users (chat_id, user_data) VALUES (?, ?)", changed)
         conn.commit()
         conn.close()
         for cid, _ in changed: