Update app.py

app.py

@@ -5,24 +5,53 @@ from datetime import datetime
 from pathlib import Path
 from telethon import TelegramClient, events
 from telethon.sessions import StringSession
-from typing import Optional
+import nest_asyncio
+
+# فعال کردن nest_asyncio برای محیط‌هایی مثل Colab/Jupyter
+nest_asyncio.apply()
 
 # تنظیمات لاگ
-logging.basicConfig(level=logging.INFO)
+logging.basicConfig(
+    level=logging.INFO,
+    format='%(asctime)s - %(name)s - %(levelname)s - %(message)s'
+)
 logger = logging.getLogger(__name__)
 
-# تنظیمات از متغیرهای محیطی (در Spaces از Secrets استفاده می‌شود)
-API_ID = os.environ.get("API_ID", "your_api_id_here")
-API_HASH = os.environ.get("API_HASH", "your_api_hash_here")
-BOT_TOKEN = os.environ.get("BOT_TOKEN", "your_bot_token_here")
+# تنظیمات از متغیرهای محیطی
+API_ID = os.environ.get("API_ID")
+API_HASH = os.environ.get("API_HASH")
+BOT_TOKEN = os.environ.get("BOT_TOKEN")
 
 # مسیر ذخیره‌سازی در فضای HuggingFace
 BASE_DIR = Path("/tmp/telegram_bot_files")
-BASE_DIR.makedirs(parents=True, exist_ok=True)
+BASE_DIR.mkdir(parents=True, exist_ok=True)
 
 # دیکشنری برای ذخیره اطلاعات کاربران
 user_data = {}
 
+def check_config():
+    """بررسی تنظیمات لازم"""
+    missing_configs = []
+    
+    if not API_ID:
+        missing_configs.append("API_ID")
+    if not API_HASH:
+        missing_configs.append("API_HASH")
+    if not BOT_TOKEN:
+        missing_configs.append("BOT_TOKEN")
+    
+    if missing_configs:
+        logger.error(f"❌ تنظیمات زیر تنظیم نشده‌اند: {', '.join(missing_configs)}")
+        logger.info("📝 لطفاً در HuggingFace Spaces از Secrets استفاده کنید:")
+        logger.info("1. به Settings > Secrets بروید")
+        logger.info("2. متغیرهای زیر را اضافه کنید:")
+        logger.info("   - API_ID: API ID از my.telegram.org")
+        logger.info("   - API_HASH: API Hash از my.telegram.org")
+        logger.info("   - BOT_TOKEN: توکن ربات از @BotFather")
+        return False
+    
+    return True
+
 class TelegramFileBot:
     def __init__(self):
         self.client = None
@@ -31,6 +60,8 @@ class TelegramFileBot:
     async def start(self):
         """شروع ربات"""
         try:
+            logger.info("🚀 در حال راه‌اندازی ربات...")
+            
             self.client = TelegramClient(
                 StringSession(),
                 int(API_ID),
@@ -39,17 +70,59 @@ class TelegramFileBot:
             
             await self.client.start(bot_token=BOT_TOKEN)
             self.running = True
-            logger.info("🤖 ربات تلگرام با موفقیت شروع به کار کرد!")
+            
+            me = await self.client.get_me()
+            logger.info(f"🤖 ربات با موفقیت شروع به کار کرد: @{me.username}")
+            logger.info(f"📁 مسیر ذخیره‌سازی: {BASE_DIR}")
+            
+            # ایجاد پوشه ذخیره‌سازی
+            BASE_DIR.mkdir(parents=True, exist_ok=True)
+            
+            # نمایش اطلاعات فضای ذخیره‌سازی
+            await self.show_storage_info()
             
             # ثبت هندلرها
             await self._setup_handlers()
             
+            # اطلاع‌رسانی شروع
+            await self.notify_admin("✅ ربات با موفقیت راه‌اندازی شد!")
+            
             # اجرای ربات
             await self.client.run_until_disconnected()
             
         except Exception as e:
-            logger.error(f"خطا در شروع ربات: {e}")
+            logger.error(f"خطا در شروع ربات: {e}", exc_info=True)
             self.running = False
+            
+    async def show_storage_info(self):
+        """نمایش اطلاعات فضای ذخیره‌سازی"""
+        try:
+            total_size = 0
+            file_count = 0
+            
+            for file_path in BASE_DIR.glob("*"):
+                if file_path.is_file():
+                    total_size += file_path.stat().st_size
+                    file_count += 1
+            
+            logger.info(f"💾 فضای ذخیره‌سازی:")
+            logger.info(f"   📁 تعداد فایل‌ها: {file_count}")
+            logger.info(f"   📊 حجم استفاده شده: {total_size / 1024 / 1024:.2f} MB")
+            logger.info(f"   📍 مسیر: {BASE_DIR}")
+            
+        except Exception as e:
+            logger.error(f"خطا در دریافت اطلاعات فضای ذخیره‌سازی: {e}")
+    
+    async def notify_admin(self, message):
+        """ارسال اطلاع به ادمین"""
+        try:
+            # در اینجا می‌توانید چت آیدی ادمین را اضافه کنید
+            # به عنوان مثال: ADMIN_CHAT_ID = "123456789"
+            # if ADMIN_CHAT_ID:
+            #     await self.client.send_message(int(ADMIN_CHAT_ID), message)
+            pass
+        except Exception as e:
+            logger.error(f"خطا در ارسال اطلاع به ادمین: {e}")
     
     async def _setup_handlers(self):
         """تنظیم هندلرهای ربات"""
@@ -57,6 +130,9 @@ class TelegramFileBot:
         @self.client.on(events.NewMessage(pattern='/start'))
         async def start_handler(event):
             """هندلر دستور /start"""
+            user = await event.get_sender()
+            logger.info(f"کاربر {user.id} ({user.username}) دستور /start را فرستاد")
+            
             welcome_msg = """
             🌟 **به ربات آپلود فایل خوش آمدید!**
             
@@ -76,6 +152,7 @@ class TelegramFileBot:
             /list - نمایش فایل‌های شما
             /clean - پاکسازی فایل‌های قدیمی
             /stats - آمار ربات
+            /download [کد] - دانلود فایل
             
             **⚠️ توجه:**
             • فایل‌ها به صورت موقت ذخیره می‌شوند
@@ -92,22 +169,24 @@ class TelegramFileBot:
             **🔹 نحوه کار:**
             1. هر فایلی (عکس، ویدیو، سند، ...) را برای ربات بفرستید
             2. ربات فایل را دریافت و ذخیره می‌کند
-            3. لینک دانلود فایل برای شما ارسال می‌شود
+            3. یک کد یکتا برای فایل ایجاد می‌شود
+            4. با دستور /download می‌توانید فایل را دانلود کنید
             
             **🔹 دستورات موجود:**
-            • `/list` - نمایش تمام فایل‌های آپلود شده شما
-            • `/clean` - پاک کردن فایل‌های قدیمی شما
+            • `/list` - نمایش 10 فایل آخر شما
+            • `/download کد` - دانلود فایل با کد
+            • `/clean` - پاک کردن تمام فایل‌های شما
             • `/stats` - مشاهده آمار ربات
             
-            **🔹 محدودیت‌ها:**
-            • حداکثر حجم فایل: 2GB
-            • فضای ذخیره‌سازی: موقتی
-            • فرمت‌ها: تمامی فرمت‌ها پشتیبانی می‌شوند
+            **🔹 مثال:**
+            ```
+            /download 123456_abc
+            ```
             
             **🔹 نکات مهم:**
-            • فایل‌ها فقط تا زمان فعالیت Space ذخیره می‌شوند
-            • برای نگهداری طولانی‌مدت، فایل‌ها را دانلود کنید
-            • از ارسال فایل‌های حساس خودداری کنید
+            • حداکثر حجم فایل: 2GB
+            • فایل‌ها موقتی ذخیره می‌شوند
+            • کد فایل را در مکانی امن ذخیره کنید
             """
             await event.reply(help_text)
         
@@ -125,12 +204,17 @@ class TelegramFileBot:
                 
                 response = "📁 **فایل‌های شما:**\n\n"
                 for idx, file_info in enumerate(files[-10:], 1):  # آخرین 10 فایل
-                    response += f"{idx}. **{file_info['name']}**\n"
+                    response += f"**{idx}. {file_info['name']}**\n"
                     response += f"   📏 حجم: {file_info['size_mb']:.2f} MB\n"
                     response += f"   🕐 زمان: {file_info['time']}\n"
                     response += f"   🔗 کد: `{file_info['code']}`\n\n"
                 
-                response += "📥 برای دانلود از دستور /download کد_فایل استفاده کنید."
+                response += "📥 برای دانلود:\n"
+                response += "```\n"
+                response += "/download کد_فایل\n"
+                response += "```\n\n"
+                response += "مثال: `/download abc123`"
+                
                 await event.reply(response)
             else:
                 await event.reply("📭 **هیچ فایلی از شما آپلود نشده است.**")
@@ -140,18 +224,23 @@ class TelegramFileBot:
             """پاکسازی فایل‌های کاربر"""
             user_id = event.sender_id
             
-            if user_id in user_data:
+            if user_id in user_data and "files" in user_data[user_id]:
+                file_count = len(user_data[user_id]["files"])
+                
                 # پاک کردن فایل‌های فیزیکی
-                if "files" in user_data[user_id]:
-                    for file_info in user_data[user_id]["files"]:
-                        file_path = BASE_DIR / file_info["code"]
-                        if file_path.exists():
+                for file_info in user_data[user_id]["files"]:
+                    file_path = BASE_DIR / file_info["code"]
+                    if file_path.exists():
+                        try:
                             file_path.unlink()
+                            logger.info(f"فایل پاک شد: {file_info['code']}")
+                        except Exception as e:
+                            logger.error(f"خطا در پاک کردن فایل {file_info['code']}: {e}")
                 
                 # پاک کردن داده‌های کاربر
                 user_data[user_id]["files"] = []
                 
-                await event.reply("🗑️ **تمام فایل‌های شما با موفقیت پاک شدند.**")
+                await event.reply(f"🗑️ **{file_count} فایل با موفقیت پاک شدند.**")
             else:
                 await event.reply("📭 **هیچ فایلی برای پاک کردن وجود ندارد.**")
         
@@ -168,19 +257,32 @@ class TelegramFileBot:
                     for file_info in data["files"]:
                         total_size += file_info.get("size_bytes", 0)
             
+            # محاسبه فضای باقی‌مانده
+            try:
+                statvfs = os.statvfs('/tmp')
+                free_space = (statvfs.f_bavail * statvfs.f_frsize) / 1024 / 1024 / 1024  # GB
+                total_space = (statvfs.f_blocks * statvfs.f_frsize) / 1024 / 1024 / 1024  # GB
+                used_space = total_space - free_space
+            except:
+                free_space = used_space = total_space = 0
+            
             stats_msg = f"""
             📊 **آمار ربات:**
             
             👥 کاربران فعال: {total_users}
             📁 فایل‌های آپلود شده: {total_files}
             💾 فضای استفاده شده: {total_size / 1024 / 1024:.2f} MB
-            🗂️ مسیر ذخیره‌سازی: `{BASE_DIR}`
             
-            ⚠️ **توجه:** فایل‌ها موقتی هستند و بعد از ری‌استارت پاک می‌شوند.
+            🗂️ **فضای ذخیره‌سازی:**
+            📊 استفاده شده: {used_space:.2f} GB
+            📈 کل فضای /tmp: {total_space:.2f} GB
+            📉 فضای آزاد: {free_space:.2f} GB
+            
+            ⚠️ **توجه:** فایل‌ها در `/tmp` ذخیره می‌شوند و موقتی هستند.
             """
             await event.reply(stats_msg)
         
-        @self.client.on(events.NewMessage(pattern='/download'))
+        @self.client.on(events.NewMessage(pattern=r'/download\s+\S+'))
         async def download_handler(event):
             """دانلود فایل با کد"""
             try:
@@ -207,29 +309,32 @@ class TelegramFileBot:
                 
                 if file_info:
                     # ارسال فایل به کاربر
-                    caption = f"📄 **{file_info['name']}**\n📏 حجم: {file_info['size_mb']:.2f} MB"
+                    caption = f"📄 **{file_info['name']}**\n📏 حجم: {file_info['size_mb']:.2f} MB\n🕐 زمان آپلود: {file_info['time']}"
                     await event.reply("📤 **در حال ارسال فایل...**")
-                    await event.respond(file=file_path, caption=caption)
+                    
+                    # ارسال فایل
+                    await event.respond(
+                        file=file_path,
+                        caption=caption,
+                        force_document=True
+                    )
+                    
+                    logger.info(f"فایل ارسال شد: {file_code} به کاربر {event.sender_id}")
                 else:
                     await event.reply("⚠️ **اطلاعات فایل یافت نشد.**")
                     
             except Exception as e:
-                logger.error(f"خطا در دانلود: {e}")
+                logger.error(f"خطا در دانلود: {e}", exc_info=True)
                 await event.reply(f"❌ **خطا در پردازش:** {str(e)}")
         
         @self.client.on(events.NewMessage)
         async def message_handler(event):
             """هندلر پیام‌های معمولی و فایل‌ها"""
-            # اگر پیام دستور باشد، نادیده بگیر (دستورات بالا مدیریت می‌شوند)
+            # اگر پیام دستور باشد، نادیده بگیر
             if event.message.text and event.message.text.startswith('/'):
                 return
             
             user_id = event.sender_id
-            chat_id = event.chat_id
-            
-            # ایجاد پوشه کاربر اگر وجود نداشته باشد
-            if user_id not in user_data:
-                user_data[user_id] = {"files": [], "chat_id": chat_id}
             
             # اگر پیام حاوی فایل باشد
             if event.file:
@@ -244,10 +349,16 @@ class TelegramFileBot:
                 # اطلاعات فایل
                 file_name = event.file.name or f"file_{datetime.now().strftime('%Y%m%d_%H%M%S')}"
                 file_size = event.file.size
-                file_ext = os.path.splitext(file_name)[1] or ".bin"
+                
+                # بررسی حجم فایل (حداکثر 2GB)
+                max_size = 2 * 1024 * 1024 * 1024  # 2GB
+                if file_size > max_size:
+                    await event.reply(f"❌ **حجم فایل بسیار بزرگ است!**\nحداکثر حجم مجاز: 2GB\nحجم فایل شما: {file_size / 1024 / 1024 / 1024:.2f} GB")
+                    return
                 
                 # ایجاد کد یکتا برای فایل
-                file_code = f"{user_id}_{int(datetime.now().timestamp())}_{os.urandom(4).hex()}"
+                timestamp = int(datetime.now().timestamp())
+                file_code = f"{user_id}_{timestamp}_{os.urandom(4).hex()}"
                 save_path = BASE_DIR / file_code
                 
                 # اطلاع‌رسانی شروع دریافت
@@ -259,7 +370,7 @@ class TelegramFileBot:
                 )
                 
                 # دانلود فایل
-                download_path = await event.download_media(file=save_path)
+                download_path = await event.download_media(file=str(save_path))
                 
                 # ذخیره اطلاعات فایل
                 file_info = {
@@ -269,10 +380,13 @@ class TelegramFileBot:
                     "size_bytes": file_size,
                     "size_mb": file_size / 1024 / 1024,
                     "time": datetime.now().strftime("%Y-%m-%d %H:%M:%S"),
-                    "extension": file_ext,
                     "message_id": event.id
                 }
                 
+                # ایجاد یا به‌روزرسانی داده کاربر
+                if user_id not in user_data:
+                    user_data[user_id] = {"files": []}
+                
                 user_data[user_id]["files"].append(file_info)
                 
                 # فقط آخرین 20 فایل را نگه دار
@@ -282,6 +396,7 @@ class TelegramFileBot:
                     old_path = BASE_DIR / old_file["code"]
                     if old_path.exists():
                         old_path.unlink()
+                        logger.info(f"فایل قدیمی پاک شد: {old_file['code']}")
                 
                 # پاسخ موفقیت‌آمیز
                 success_msg = (
@@ -290,8 +405,9 @@ class TelegramFileBot:
                     f"📏 حجم فایل: {file_size / 1024 / 1024:.2f} MB\n"
                     f"🔢 کد فایل: `{file_code}`\n\n"
                     f"📥 **برای دانلود:**\n"
-                    f"1. از دستور `/download {file_code}` استفاده کنید\n"
-                    f"2. یا از منوی /list فایل خود را انتخاب کنید\n\n"
+                    f"```\n"
+                    f"/download {file_code}\n"
+                    f"```\n\n"
                     f"📋 **تعداد فایل‌های شما:** {len(user_data[user_id]['files'])}"
                 )
                 
@@ -300,38 +416,43 @@ class TelegramFileBot:
                 logger.info(f"فایل دریافت شد: {file_name} ({file_code}) توسط کاربر {user_id}")
                 
             except Exception as e:
-                logger.error(f"خطا در دریافت فایل: {e}")
+                logger.error(f"خطا در دریافت فایل: {e}", exc_info=True)
                 await event.reply(f"❌ **خطا در دریافت فایل:**\n{str(e)}")
 
-# تابع اصلی
 async def main():
     """تابع اصلی اجرای ربات"""
-    bot = TelegramFileBot()
-    
-    # چک کردن تنظیمات
-    if API_ID == "your_api_id_here" or API_HASH == "your_api_hash_here" or BOT_TOKEN == "your_bot_token_here":
-        logger.error("❌ لطفا API_ID, API_HASH و BOT_TOKEN را تنظیم کنید!")
-        logger.info("📝 در HuggingFace Spaces از Secrets استفاده کنید.")
+    # بررسی تنظیمات
+    if not check_config():
         return
     
+    # ایجاد مسیر ذخیره‌سازی
+    BASE_DIR.mkdir(parents=True, exist_ok=True)
+    
+    # نمایش اطلاعات راه‌اندازی
+    logger.info("=" * 50)
+    logger.info("🤖 ربات آپلود فایل تلگرام")
+    logger.info(f"📁 مسیر ذخیره‌سازی: {BASE_DIR}")
+    
+    # محاسبه فضای موجود
+    try:
+        statvfs = os.statvfs('/tmp')
+        free_gb = (statvfs.f_bavail * statvfs.f_frsize) / 1024 / 1024 / 1024
+        logger.info(f"💾 فضای آزاد /tmp: {free_gb:.2f} GB")
+    except:
+        logger.info("💾 فضای /tmp: نامشخص")
+    
+    logger.info("=" * 50)
+    
+    # ایجاد و اجرای ربات
+    bot = TelegramFileBot()
+    
     try:
         await bot.start()
     except KeyboardInterrupt:
         logger.info("🛑 ربات متوقف شد.")
     except Exception as e:
-        logger.error(f"خطای غیرمنتظره: {e}")
+        logger.error(f"خطای غیرمنتظره: {e}", exc_info=True)
 
 # اجرای ربات
 if __name__ == "__main__":
-    # ایجاد مسیرهای لازم
-    BASE_DIR.makedirs(parents=True, exist_ok=True)
-    
-    # چاپ اطلاعات
-    logger.info("=" * 50)
-    logger.info("🤖 ربات آپلود فایل تلگرام")
-    logger.info(f"📁 مسیر ذخیره‌سازی: {BASE_DIR}")
-    logger.info(f"💾 فضای موجود: {sum(f.stat().st_size for f in BASE_DIR.glob('**/*') if f.is_file()) / 1024 / 1024:.2f} MB")
-    logger.info("=" * 50)
-    
-    # اجرای ربات
     asyncio.run(main())