Update main.py

main.py

@@ -93,7 +93,7 @@ def gregorian_to_jalali(gy, gm, gd):
     return jy, jm, jd
 
 # ==============================================================================
-# 🟢 پارت 4: دیتابیس SQLite (نسخه فوق‌هوشمند با استخراج خط‌به‌خط و بدون باگ ابری)
+# 🟢 پارت 4: دیتابیس SQLite (نسخه نجات جراحی - پریدن از روی سکتورهای خراب)
 # ==============================================================================
 import os
 import sqlite3
@@ -101,8 +101,8 @@ import json
 import copy
 import threading
 
-# 🚀 انتقال نهایی به v5 برای داشتن یک دیتابیس ۱۰۰٪ سالم و بدون سابقه خرابی
-DB_FILE = "/data/users_v5.db" 
+# 🚀 انتقال به v6 برای انجام مجدد عملیات نجات (این بار به صورت خط‌به‌خط جراحی)
+DB_FILE = "/data/users_v6.db" 
 OLD_DB_V3 = "/data/users_v3.db"
 
 last_saved_state = {}
@@ -121,49 +121,62 @@ def init_sqlite_db():
         conn.execute('PRAGMA journal_mode=TRUNCATE;') 
         conn.execute('PRAGMA synchronous=FULL;')
         
-        # ایجاد جدول کاربران (هیچ اثری از جدول پیام‌های قدیمی نیست)
+        # ایجاد جدول کاربران
         conn.execute('CREATE TABLE IF NOT EXISTS users (chat_id TEXT PRIMARY KEY, user_data TEXT)')
         conn.commit()
 
         if is_first_run:
-            print("🚨 عملیات استخراج از فایل v3 آغاز شد...")
-            robust_salvage(OLD_DB_V3, conn)
+            print("🚨 عملیات نجات جراحی (دور زدن خرابی‌ها) از فایل v3 آغاز شد...")
+            surgical_salvage(OLD_DB_V3, conn)
         
         conn.close()
     except Exception as e:
         print(f"❌ خطا در راه اندازی دیتابیس: {e}")
 
-# 🧲 تابع فوق‌هوشمند استخراج خط به خط (عبور از بخش‌های خراب دیتابیس)
-def robust_salvage(old_file, new_conn):
+# 🧲 تابع استخراج جراحی (رکورد به رکورد) برای پریدن از روی سکتورهای خراب دیتابیس
+def surgical_salvage(old_file, new_conn):
     if not os.path.exists(old_file): 
         print(f"⚠️ فایل قدیمی {old_file} یافت نشد.")
         return
         
-    print(f"🔍 در حال استخراج اطلاعات کاربران از {old_file} ...")
+    print(f"🔍 در حال اسکن عمیق و جراحی رکوردهای {old_file} ...")
     extracted_count = 0
+    corrupt_count = 0
     try:
         old_conn = sqlite3.connect(old_file, timeout=20.0)
         cursor = old_conn.cursor()
         
-        cursor.execute("SELECT chat_id, user_data FROM users")
+        # ابتدا بالاترین شماره سطر (ROWID) را پیدا می‌کنیم تا بدانیم چقدر باید بگردیم
+        try:
+            cursor.execute("SELECT MAX(rowid) FROM users")
+            max_id = cursor.fetchone()[0]
+            if not max_id: max_id = 150000
+        except:
+            max_id = 150000 # اگر فایل خیلی خراب بود، تا 150 هزار رکورد را دستی می‌گردیم
+            
+        print(f"تعداد رکوردهای احتمالی جهت اسکن: {max_id}. لطفا چند ثانیه صبر کنید...")
         
-        # استخراج دانه به دانه برای جلوگیری از ارور malformed
-        while True:
+        # جستجوی دانه به دانه: اگر یکی خراب بود، فقط همون رو رد میکنه و میره بعدی!
+        for i in range(1, max_id + 1):
             try:
+                cursor.execute("SELECT chat_id, user_data FROM users WHERE rowid = ?", (i,))
                 row = cursor.fetchone()
-                if row is None:
-                    break
-                new_conn.execute("INSERT OR IGNORE INTO users VALUES (?, ?)", row)
-                extracted_count += 1
-            except sqlite3.DatabaseError as db_err:
-                print(f"⚠️ خرابی در بخشی از فایل قدیمی شناسایی شد. عملیات در همین نقطه متوقف شد.")
-                break # خروج امن از حلقه در صورت برخورد به بخش خراب فایل
+                if row:
+                    new_conn.execute("INSERT OR IGNORE INTO users VALUES (?, ?)", row)
+                    extracted_count += 1
+            except sqlite3.DatabaseError:
+                # این رکورد روی سکتور خراب هارد افتاده است -> رد میشویم
+                corrupt_count += 1
+                continue
+            except Exception:
+                corrupt_count += 1
+                continue
                 
         new_conn.commit()
         old_conn.close()
-        print(f"✅ موفقیت! اطلاعات {extracted_count} کاربر سالم استخراج و به دیتابیس v5 منتقل شد.")
+        print(f"✅ شاهکار نجات! {extracted_count} کاربر کاملا سالم استخراج شد و از روی {corrupt_count} رکورد خراب با موفقیت پریدیم.")
     except Exception as e:
-        print(f"⚠️ توقف کامل استخراج: {e} | تعداد نجات‌یافته تا این لحظه: {extracted_count}")
+        print(f"⚠️ توقف استخراج: {e} | تعداد نجات‌یافته تا این لحظه: {extracted_count}")
 
 def load_db():
     global last_saved_state, recent_messages_dict