Update main.py

main.py

@@ -92,118 +92,163 @@ def gregorian_to_jalali(gy, gm, gd):
     return jy, jm, jd
 
 # ==============================================================================
-# 🟢 پارت 4: دیتابیس SQLite (نسخه ضد خرابی و خود-تعمیرگر)
+# 🟢 پارت 4: راه‌اندازی دیتابیس هوشمند و پرسرعت SQLite + انتقال خودکار کاربران قدیمی
 # ==============================================================================
+import os
+import sqlite3
+import json
+import copy
+
+# آدرس فایل‌های دیتابیس در پوشه Mount شده فضای ذخیره‌سازی ابری
 DB_FILE = "/data/users_db.db"
 OLD_JSON_FILE = "/data/users_db.json"
+
+# این متغیر برای نگهداری آخرین وضعیت کاربران استفاده می‌شود تا فقط تغییرات جدید در دیتابیس نوشته شوند
 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, timeout=30.0) # افزایش تایم‌اوت برای جلوگیری از قفل شدن
+        
+        conn = sqlite3.connect(DB_FILE)
         cursor = conn.cursor()
         
-        # ۱. بررسی سلامت دیتابیس و تعمیر خودکار در صورت خرابی (Malformed)
-        try:
-            cursor.execute('PRAGMA integrity_check;')
-            result = cursor.fetchone()
-            if result[0] != "ok":
-                print("⚠️ دیتابیس دچار خرابی شده، در حال تعمیر...")
-                cursor.execute('PRAGMA journal_mode=DELETE;') # موقتاً تغییر حالت برای تعمیر
-                cursor.execute('VACUUM;') # بازسازی کامل فایل دیتابیس
-                print("✅ دیتابیس با موفقیت بازسازی و تعمیر شد.")
-        except Exception as e:
-            print(f"🛠 تلاش برای تعمیر اضطراری: {e}")
-
-        # ۲. فعال‌سازی حالت بهینه
+        # فعال‌سازی WAL (Write-Ahead Logging) برای جلوگیری از قفل شدن (Lock) دیتابیس
         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('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()
     except Exception as e:
-        print("خطا در راه‌اندازی دیتابیس:", e)
+        print("خطا در ساخت دیتابیس SQLite:", e)
 
 def migrate_old_json_to_sqlite():
+    """انتقال خودکار و امن اطلاعات فایل قدیمی JSON (اشتراک‌ها و سکه‌ها) به دیتابیس جدید SQLite"""
     if os.path.exists(OLD_JSON_FILE):
-        print("⚠️ انتقال اطلاعات از JSON قدیمی...")
+        print("⚠️ در حال انتقال اطلاعات کاربران از فایل JSON قدیمی به دیتابیس SQLite...")
         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)
-                    conn.execute("INSERT OR IGNORE INTO users (chat_id, user_data) VALUES (?, ?)", (str(chat_id), user_data_str))
+                    cursor.execute("INSERT OR IGNORE INTO users (chat_id, user_data) VALUES (?, ?)", (str(chat_id), user_data_str))
+                
                 conn.commit()
                 conn.close()
-            os.rename(OLD_JSON_FILE, OLD_JSON_FILE + ".bak")
-            print("✅ انتقال تمام شد.")
+                print(f"✅ انتقال اطلاعات {len(old_data)} کاربر قدیمی با موفقیت انجام شد.")
+            
+            # تغییر نام فایل قدیمی به نسخه پشتیبان (.bak) تا دیگر خوانده نشود و امنیت حفظ شود
+            backup_file = OLD_JSON_FILE + ".bak"
+            os.rename(OLD_JSON_FILE, backup_file)
+            print(f"✅ فایل قدیمی به {backup_file} تغییر نام یافت.")
         except Exception as e:
-            print(f"❌ خطا در مهاجرت: {e}")
+            print(f"❌ خطا در انتقال اطلاعات فایل JSON قدیمی: {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)
         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])
+        rows = cursor.fetchall()
+        for row in rows:
+            chat_id, user_data_str = row
+            db_dict[chat_id] = json.loads(user_data_str)
         conn.close()
+        
+        # یک کپی دقیق از وضعیت اولیه می‌گیریم تا بعداً فقط تغییرات را تشخیص دهیم
         last_saved_state = copy.deepcopy(db_dict)
-        print(f"✅ دیتابیس لود شد. کاربران: {len(db_dict)}")
+        print(f"✅ دیتابیس SQLite با موفقیت لود شد. (تعداد کاربران فعلی: {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 = []
-        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)))
+        changed_records =[]
         
-        if not changed: return
+        # تشخیص کاربرانی که اطلاعاتشان تغییر کرده
+        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))
+        
+        if not changed_records:
+            return
 
-        conn = sqlite3.connect(DB_FILE, timeout=30.0)
-        conn.execute('PRAGMA journal_mode=WAL;')
-        conn.executemany("INSERT OR REPLACE INTO users (chat_id, user_data) VALUES (?, ?)", changed)
+        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)
         conn.commit()
         conn.close()
         
-        for cid, _ in changed:
-            last_saved_state[cid] = copy.deepcopy(db_data[cid])
+        for chat_id, _ in changed_records:
+            last_saved_state[chat_id] = copy.deepcopy(db_data[chat_id])
+            
     except Exception as e:
-        print("❌ خطا در ذخیره SQLite:", e)
+        print("خطا در آپدیت دیتابیس SQLite:", e)
 
 def is_message_processed(message_id):
+    """بررسی سریع اینکه آیا پیام قبلاً در دیتابیس ثبت شده است یا خیر"""
     try:
-        conn = sqlite3.connect(DB_FILE, timeout=20.0)
+        conn = sqlite3.connect(DB_FILE, timeout=10.0)
         cursor = conn.cursor()
         cursor.execute("SELECT 1 FROM processed_messages WHERE message_id = ?", (str(message_id),))
-        res = cursor.fetchone()
+        result = cursor.fetchone()
         conn.close()
-        return res is not None
-    except: return False
+        return result is not None
+    except Exception:
+        return False
 
 def mark_message_processed(message_id):
+    """ثبت همیشگی شناسه پیام در دیتابیس"""
     try:
-        conn = sqlite3.connect(DB_FILE, timeout=20.0)
-        conn.execute("INSERT OR IGNORE INTO processed_messages (message_id) VALUES (?)", (str(message_id),))
+        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.commit()
         conn.close()
-    except: pass
+    except Exception:
+        pass
 
+# جایگذاری متغیر اصلی دیتابیس با مکانیزم جدید
 user_credits_db = load_db()