Update main.py

main.py

@@ -92,163 +92,144 @@ def gregorian_to_jalali(gy, gm, gd):
     return jy, jm, jd
 
 # ==============================================================================
-# 🟢 پارت 4: راه‌اندازی دیتابیس هوشمند و پرسرعت SQLite + انتقال خودکار کاربران قدیمی
+# 🟢 پارت 4: دیتابیس SQLite (نسخه نهایی - ضد خرابی در شبکه Hugging Face)
 # ==============================================================================
 import os
 import sqlite3
 import json
 import copy
 
-# آدرس فایل‌های دیتابیس در پوشه Mount شده فضای ذخیره‌سازی ابری
 DB_FILE = "/data/users_db.db"
-OLD_JSON_FILE = "/data/users_db.json"
+BACKUP_JSON_FILE = "/data/users_db.json.bak"
 
-# این متغیر برای نگهداری آخرین وضعیت کاربران استفاده می‌شود تا فقط تغییرات جدید در دیتابیس نوشته شوند
 last_saved_state = {}
 
 def init_sqlite_db():
-    """ساخت جداول دیتابیس در صورت عدم وجود و فعال‌سازی حالت پرسرعت WAL"""
+    """راه‌اندازی دیتابیس با تنظیمات مخصوص درایوهای شبکه"""
     try:
-        # اطمینان از وجود پوشه data
         os.makedirs(os.path.dirname(DB_FILE), exist_ok=True)
         
-        conn = sqlite3.connect(DB_FILE)
+        # ۱. حذف فایل‌های موقت WAL که باعث خرابی می‌شوند
+        for ext in ["-wal", "-shm"]:
+            if os.path.exists(DB_FILE + ext):
+                try: os.remove(DB_FILE + ext)
+                except: pass
+
+        # ۲. بررسی سلامت و بازسازی در صورت خرابی
+        if os.path.exists(DB_FILE):
+            conn_test = sqlite3.connect(DB_FILE, timeout=10)
+            try:
+                conn_test.execute("PRAGMA integrity_check").fetchone()
+                conn_test.close()
+            except:
+                conn_test.close()
+                print("🚨 دیتابیس غیرقابل تعمیر است. در حال جایگزینی با فایل سالم...")
+                os.rename(DB_FILE, DB_FILE + ".old") # فایل خراب را کنار بگذار
+
+        # ۳. اتصال با تنظیمات امن (حالت TRUNCATE به جای WAL)
+        conn = sqlite3.connect(DB_FILE, timeout=60.0)
         cursor = conn.cursor()
         
-        # فعال‌سازی WAL (Write-Ahead Logging) برای جلوگیری از قفل شدن (Lock) دیتابیس
-        cursor.execute('PRAGMA journal_mode=WAL;')
-        
-        # جدول کاربران
-        cursor.execute('''
-            CREATE TABLE IF NOT EXISTS users (
-                chat_id TEXT PRIMARY KEY,
-                user_data TEXT
-            )
-        ''')
+        # این تنظیمات برای Hugging Face حیاتی است:
+        cursor.execute('PRAGMA journal_mode=TRUNCATE;') 
+        cursor.execute('PRAGMA synchronous=FULL;')
         
-        # جدول شناسه‌های پیام پردازش شده (ضد تکرار قطعی پیام)
-        cursor.execute('''
-            CREATE TABLE IF NOT EXISTS processed_messages (
-                message_id TEXT PRIMARY KEY
-            )
-        ''')
+        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()
+        print("✅ دیتابیس با حالت امن شبکه (TRUNCATE) آماده شد.")
     except Exception as e:
-        print("خطا در ساخت دیتابیس SQLite:", e)
+        print(f"❌ خطا در راه اندازی دیتابیس: {e}")
 
-def migrate_old_json_to_sqlite():
-    """انتقال خودکار و امن اطلاعات فایل قدیمی JSON (اشتراک‌ها و سکه‌ها) به دیتابیس جدید SQLite"""
-    if os.path.exists(OLD_JSON_FILE):
-        print("⚠️ در حال انتقال اطلاعات کاربران از فایل JSON قدیمی به دیتابیس SQLite...")
+def restore_credits_only():
+    """بازیابی اعتبار کاربران از بک‌آپ JSON"""
+    if os.path.exists(BACKUP_JSON_FILE):
+        print("🛠 در حال بازیابی اعتبارها از فایل بک‌آپ...")
         try:
-            with open(OLD_JSON_FILE, "r", encoding="utf-8") as f:
-                old_data = json.load(f)
-            
-            if old_data:
-                conn = sqlite3.connect(DB_FILE)
-                cursor = conn.cursor()
-                
-                # انتقال تمام کاربران قدیمی به دیتابیس جدید (بدون پاک شدن اشتراک‌ها)
-                for chat_id, data in old_data.items():
-                    user_data_str = json.dumps(data, ensure_ascii=False)
-                    cursor.execute("INSERT OR IGNORE INTO users (chat_id, user_data) VALUES (?, ?)", (str(chat_id), user_data_str))
-                
+            with open(BACKUP_JSON_FILE, "r", encoding="utf-8") as f:
+                backup_data = json.load(f)
+            if backup_data:
+                conn = sqlite3.connect(DB_FILE, timeout=60.0)
+                for chat_id, data in backup_data.items():
+                    conn.execute("INSERT OR REPLACE INTO users (chat_id, user_data) VALUES (?, ?)", 
+                                 (str(chat_id), json.dumps(data, ensure_ascii=False)))
                 conn.commit()
                 conn.close()
-                print(f"✅ انتقال اطلاعات {len(old_data)} کاربر قدیمی با موفقیت انجام شد.")
-            
-            # تغییر نام فایل قدیمی به نسخه پشتیبان (.bak) تا دیگر خوانده نشود و امنیت حفظ شود
-            backup_file = OLD_JSON_FILE + ".bak"
-            os.rename(OLD_JSON_FILE, backup_file)
-            print(f"✅ فایل قدیمی به {backup_file} تغییر نام یافت.")
+                print(f"✅ اعتبار {len(backup_data)} کاربر با موفقیت بازگشت.")
         except Exception as e:
-            print(f"❌ خطا در انتقال اطلاعات فایل JSON قدیمی: {e}")
+            print(f"❌ خطا در بازیابی: {e}")
 
 def load_db():
-    """لود کردن اطلاعات کاربران از SQLite به داخل حافظه رم ربات در لحظه استارت"""
     global last_saved_state
-    print(f"در حال خواندن دیتابیس SQLite از مسیر {DB_FILE} ...")
-    
     init_sqlite_db()
     
-    # 🟢 فراخوانی سیستم انتقال خودکار (فقط یک‌بار اجرا می‌شود و اگر فایل قدیمی نباشد نادیده گرفته می‌شود)
-    migrate_old_json_to_sqlite()
-    
     db_dict = {}
     try:
-        conn = sqlite3.connect(DB_FILE)
+        conn = sqlite3.connect(DB_FILE, timeout=60.0)
         cursor = conn.cursor()
+        
+        # چک کردن تعداد کاربران
+        cursor.execute("SELECT count(*) FROM users")
+        if cursor.fetchone()[0] < 50: # اگر دیتابیس خالی است، بازیابی کن
+            conn.close()
+            restore_credits_only()
+            conn = sqlite3.connect(DB_FILE, timeout=60.0)
+            cursor = conn.cursor()
+
         cursor.execute("SELECT chat_id, user_data FROM users")
-        rows = cursor.fetchall()
-        for row in rows:
-            chat_id, user_data_str = row
-            db_dict[chat_id] = json.loads(user_data_str)
-        conn.close()
+        for row in cursor.fetchall():
+            db_dict[row[0]] = json.loads(row[1])
         
-        # یک کپی دقیق از وضعیت اولیه می‌گیریم تا بعداً فقط تغییرات را تشخیص دهیم
+        conn.close()
         last_saved_state = copy.deepcopy(db_dict)
-        print(f"✅ دیتابیس SQLite با موفقیت لود شد. (تعداد کاربران فعلی: {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):
-    """ذخیره‌سازی هوشمند و فوق‌سریع کاربران"""
     global last_saved_state
     try:
-        changed_records =[]
-        
-        # تشخیص کاربرانی که اطلاعاتشان تغییر کرده
-        for chat_id, data in db_data.items():
-            if chat_id not in last_saved_state or last_saved_state[chat_id] != data:
-                user_data_str = json.dumps(data, ensure_ascii=False)
-                changed_records.append((str(chat_id), user_data_str))
+        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)))
         
-        if not changed_records:
-            return
+        if not changed: return
 
-        conn = sqlite3.connect(DB_FILE, timeout=15.0) 
-        cursor = conn.cursor()
-        cursor.execute('PRAGMA journal_mode=WAL;')
-        
-        cursor.executemany("INSERT OR REPLACE INTO users (chat_id, user_data) VALUES (?, ?)", changed_records)
+        # استفاده از TRUNCATE برای امنیت در شبکه
+        conn = sqlite3.connect(DB_FILE, timeout=60.0)
+        conn.execute('PRAGMA journal_mode=TRUNCATE;')
+        conn.executemany("INSERT OR REPLACE INTO users (chat_id, user_data) VALUES (?, ?)", changed)
         conn.commit()
         conn.close()
         
-        for chat_id, _ in changed_records:
-            last_saved_state[chat_id] = copy.deepcopy(db_data[chat_id])
-            
+        for cid, _ in changed:
+            last_saved_state[cid] = copy.deepcopy(db_data[cid])
     except Exception as e:
-        print("خطا در آپدیت دیتابیس SQLite:", e)
+        print(f"❌ خطا در ذخیره اعتبار: {e}")
 
 def is_message_processed(message_id):
-    """بررسی سریع اینکه آیا پیام قبلاً در دیتابیس ثبت شده است یا خیر"""
     try:
-        conn = sqlite3.connect(DB_FILE, timeout=10.0)
+        conn = sqlite3.connect(DB_FILE, timeout=30.0)
         cursor = conn.cursor()
         cursor.execute("SELECT 1 FROM processed_messages WHERE message_id = ?", (str(message_id),))
-        result = cursor.fetchone()
+        res = cursor.fetchone()
         conn.close()
-        return result is not None
-    except Exception:
-        return False
+        return res is not None
+    except: return False
 
 def mark_message_processed(message_id):
-    """ثبت همیشگی شناسه پیام در دیتابیس"""
     try:
-        conn = sqlite3.connect(DB_FILE, timeout=10.0)
-        cursor = conn.cursor()
-        cursor.execute('PRAGMA journal_mode=WAL;')
-        cursor.execute("INSERT OR IGNORE INTO processed_messages (message_id) VALUES (?)", (str(message_id),))
+        conn = sqlite3.connect(DB_FILE, timeout=30.0)
+        conn.execute("INSERT OR IGNORE INTO processed_messages (message_id) VALUES (?)", (str(message_id),))
         conn.commit()
         conn.close()
-    except Exception:
-        pass
+    except: pass
 
-# جایگذاری متغیر اصلی دیتابیس با مکانیزم جدید
 user_credits_db = load_db()