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,12 +101,14 @@ import json
 import copy
 import threading
 
-DB_FILE = "/data/users_v3.db" 
+# تغییر نام به v4 برای فرار از فایل‌های قفل شده قبلی و شروع یکپارچه
+DB_FILE = "/data/users_v4.db" 
+OLD_DB_V3 = "/data/users_v3.db"
 CORRUPT_FILE = "/data/users_db.db" 
 BACKUP_JSON = "/data/users_db.json.bak" 
 
 last_saved_state = {}
-# حافظه کش برای پیام‌های پردازش شده صرفاً در RAM (با ری‌استارت اسپیس کاملاً پاک می‌شود)
+# حافظه کش فقط در RAM (با ری‌استارت اسپیس پاک می‌شود)
 recent_messages_dict = {}
 # قفل‌ها برای جلوگیری از تداخل رشته‌ها (Threads)
 db_lock = threading.Lock()
@@ -118,38 +120,53 @@ def init_sqlite_db():
         is_first_run = not os.path.exists(DB_FILE)
 
         conn = sqlite3.connect(DB_FILE, timeout=60.0)
-        cursor = conn.cursor()
-        # 🚀 استفاده از WAL برای پشتیبانی از هزاران کاربر همزمان
-        cursor.execute('PRAGMA journal_mode=WAL;') 
-        cursor.execute('PRAGMA synchronous=NORMAL;')
-        cursor.execute('CREATE TABLE IF NOT EXISTS users (chat_id TEXT PRIMARY KEY, user_data TEXT)')
         
-        # 🧹 پاکسازی دیتابیس: حذف جدول ذخیره پیام‌ها (طبق درخواست، پیام‌ها دیگر در هارد ذخیره نمی‌شوند)
-        cursor.execute('DROP TABLE IF EXISTS processed_messages')
+        # 🚀 حذف WAL چون در هاست‌های ابری (NFS) باعث خطای malformed می‌شود
+        # با خاموش کردن آن، ارورها برای همیشه از بین می‌روند.
+        conn.execute('PRAGMA journal_mode=DELETE;') 
         
+        # اطمینان از وجود جدول کاربران
+        conn.execute('CREATE TABLE IF NOT EXISTS users (chat_id TEXT PRIMARY KEY, user_data TEXT)')
         conn.commit()
-        conn.close()
 
         if is_first_run:
-            print("🚨 عملیات نجات آغاز شد...")
-            restore_users_from_json()
-
+            print("🚨 عملیات استخراج و انتقال به دیتابیس ابری جدید آغاز شد...")
+            robust_salvage(CORRUPT_FILE, conn)  # استخراج از فایل 30 مگابایتی
+            robust_salvage(OLD_DB_V3, conn)     # استخراج از فایل v3
+            restore_users_from_json(conn)       # استخراج از فایل بکاپ JSON
+        
+        conn.close()
     except Exception as e:
         print(f"❌ خطا در راه اندازی دیتابیس: {e}")
 
-def restore_users_from_json():
+# 🧲 تابع قدرتمند برای کشیدن اطلاعات از فایل‌هایی که ارور malformed می‌دهند
+def robust_salvage(old_file, new_conn):
+    if not os.path.exists(old_file): return
+    print(f"🔍 در حال استخراج اطلاعات از فایل {old_file} ...")
+    try:
+        old_conn = sqlite3.connect(old_file)
+        cursor = old_conn.cursor()
+        cursor.execute("SELECT chat_id, user_data FROM users")
+        rows = cursor.fetchall()
+        
+        # فقط درج اطلاعات کاربر (بدون دست زدن به پیام‌های قدیمی)
+        new_conn.executemany("INSERT OR IGNORE INTO users VALUES (?, ?)", rows)
+        new_conn.commit()
+        old_conn.close()
+        print(f"✅ اطلاعات {len(rows)} کاربر از فایل قدیمی با موفقیت استخراج شد.")
+    except Exception as e:
+        print(f"⚠️ استخراج از فایل قدیمی در یک نقطه متوقف شد (احتمالاً به بخش خراب رسیده است): {e}")
+
+def restore_users_from_json(new_conn):
     if os.path.exists(BACKUP_JSON):
-        print("📥 در حال بازیابی اعتبار کاربران از فایل پشتیبان...")
+        print("📥 در حال بازیابی اعتبار کاربران از فایل پشتیبان JSON...")
         try:
             with open(BACKUP_JSON, "r", encoding="utf-8") as f:
                 data = json.load(f)
             if data:
-                conn = sqlite3.connect(DB_FILE, timeout=60.0)
-                conn.execute('PRAGMA journal_mode=WAL;')
                 for cid, udata in data.items():
-                    conn.execute("INSERT OR REPLACE INTO users VALUES (?, ?)", (str(cid), json.dumps(udata)))
-                conn.commit()
-                conn.close()
+                    new_conn.execute("INSERT OR REPLACE INTO users VALUES (?, ?)", (str(cid), json.dumps(udata)))
+                new_conn.commit()
                 print(f"✅ اعتبار {len(data)} کاربر با موفقیت بازیابی شد.")
         except Exception as e:
             print(f"❌ خطا در بازیابی اعتبارها: {e}")
@@ -160,30 +177,28 @@ def load_db():
     db_dict = {}
     try:
         conn = sqlite3.connect(DB_FILE, timeout=60.0)
-        conn.execute('PRAGMA journal_mode=WAL;')
+        conn.execute('PRAGMA journal_mode=DELETE;')
         cursor = conn.cursor()
         
-        # لود کردن کاربران
+        # 🟢 فقط لود کردن اطلاعات کاربران
         cursor.execute("SELECT chat_id, user_data FROM users")
         for row in cursor.fetchall():
             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 background_save_worker(changed_data):
     with db_lock:
         try:
             conn = sqlite3.connect(DB_FILE, timeout=60.0)
-            conn.execute('PRAGMA journal_mode=WAL;')
-            conn.execute('PRAGMA synchronous=NORMAL;')
+            conn.execute('PRAGMA journal_mode=DELETE;')
             conn.executemany("INSERT OR REPLACE INTO users (chat_id, user_data) VALUES (?, ?)", changed_data)
             conn.commit()
             conn.close()
@@ -192,7 +207,7 @@ def background_save_worker(changed_data):
 
 def save_db(db_data):
     global last_saved_state
-    changed = []
+    changed =[]
     for cid, data in db_data.items():
         if cid not in last_saved_state or last_saved_state[cid] != data:
             changed.append((str(cid), json.dumps(data, ensure_ascii=False)))
@@ -203,12 +218,12 @@ def save_db(db_data):
     for cid, _ in changed:
         last_saved_state[cid] = copy.deepcopy(db_data[cid])
         
-    # 🚀 پرتاب عملیات سنگین به یک پردازشگر دیگر تا ربات متوقف نشود
+    # ارسال عملیات ذخیره به یک Thread دیگر تا ربات قفل نکند
     threading.Thread(target=background_save_worker, args=(changed,), daemon=True).start()
 
-# --- سیستم بررسی پیام تکراری (اکنون ۱۰۰٪ فقط در RAM کار می‌کند) ---
+# --- سیستم ضد تکرار پیام کاملاً منتقل شده به RAM ---
 def is_message_processed(message_id):
-    # چک کردن داخل RAM (زمان اجرا: 0.0001 ثانیه)
+    # چک کردن داخل RAM در کسری از ثانیه
     return str(message_id) in recent_messages_dict
 
 def mark_message_processed(message_id):
@@ -216,7 +231,8 @@ def mark_message_processed(message_id):
     with msg_cache_lock:
         # ثبت فقط در RAM
         recent_messages_dict[msg_id_str] = True
-        # جلوگیری از پر شدن بیش از حد RAM در استفاده‌های طولانی
+        
+        # جلوگیری از پر شدن بیش از حد RAM (نگه داشتن فقط 15000 پیام آخر)
         if len(recent_messages_dict) > 15000:
             oldest_key = next(iter(recent_messages_dict))
             del recent_messages_dict[oldest_key]