добавлена локализация и команда /support
This commit is contained in:
vrubelroman
parent
cb1458bd93
commit
77fd4b15a3
2 changed files with 233 additions and 86 deletions
313
bot.py
313
bot.py
|
|
@ -41,6 +41,142 @@ DATA_DIR = BASE_DIR / 'data'
|
|||
DATA_DIR.mkdir(parents=True, exist_ok=True)
|
||||
DB_FILE = DATA_DIR / 'bot.db'
|
||||
|
||||
# ============================================================================
|
||||
# ЛОКАЛИЗАЦИЯ
|
||||
# ============================================================================
|
||||
|
||||
TEXTS = {
|
||||
'ru': {
|
||||
'start': (
|
||||
"👋 Привет! Я бот для скачивания видео.\n\n"
|
||||
"Просто отправь мне ссылку на видео, и я скачаю его для тебя.\n\n"
|
||||
"Поддерживаемые источники:\n"
|
||||
"• YouTube (youtube.com, youtu.be)\n"
|
||||
"• Instagram (instagram.com)\n"
|
||||
"• VK (vk.com)\n"
|
||||
"• Yapfiles (yapfiles.ru)\n\n"
|
||||
"👥 Работа в группах:\n"
|
||||
"Добавь меня в группу и дай права администратора (нужно право на удаление сообщений). "
|
||||
"После этого я буду автоматически находить ссылки на видео в сообщениях участников, "
|
||||
"скачивать их и отправлять прямо в группу, заменяя исходное сообщение со ссылкой.\n\n"
|
||||
"Команды:\n"
|
||||
"/start - Начать работу\n"
|
||||
"/stat - Статистика бота\n"
|
||||
"/support - Поддержка и информация\n\n"
|
||||
"Отправь ссылку на видео:"
|
||||
),
|
||||
'support': (
|
||||
"ℹ️ <b>О боте</b>\n\n"
|
||||
"Этот бот позволяет скачивать видео из популярных источников:\n"
|
||||
"• YouTube — видео и shorts\n"
|
||||
"• Instagram — reels и посты с видео\n"
|
||||
"• VK — видеозаписи\n"
|
||||
"• Yapfiles — видеофайлы\n\n"
|
||||
"🔧 <b>Как использовать:</b>\n"
|
||||
"1. Отправьте ссылку на видео в личный чат с ботом\n"
|
||||
"2. Дождитесь скачивания\n"
|
||||
"3. Получите видео прямо в Telegram!\n\n"
|
||||
"👥 <b>В группах:</b>\n"
|
||||
"Добавьте бота в группу с правами администратора — "
|
||||
"он будет автоматически скачивать видео из сообщений участников.\n\n"
|
||||
"❓ Есть вопросы или предложения?\n"
|
||||
"Пишите автору: @rvrubel"
|
||||
),
|
||||
'stat': "📊 Статистика бота:\n\n👥 Всего пользователей: {users}\n📹 Всего скачано видео: {downloads}",
|
||||
'send_link': (
|
||||
"Пожалуйста, отправьте ссылку на видео.\n"
|
||||
"Поддерживаемые источники:\n"
|
||||
"• YouTube (youtube.com, youtu.be)\n"
|
||||
"• Instagram (instagram.com)\n"
|
||||
"• VK (vk.com)\n"
|
||||
"• Yapfiles (yapfiles.ru)\n\n"
|
||||
"Для других источников: Пардон, не умеем 😅"
|
||||
),
|
||||
'unsupported_source': "Пардон, не умеем работать с этим источником 😅",
|
||||
'processing': "🔍 Обрабатываю ссылку...",
|
||||
'downloading': "⬇️ Скачиваю видео...",
|
||||
'sending': "📤 Отправляю видео...",
|
||||
'caption': "Видео скачано с @{bot_username}",
|
||||
'error': "❌ Произошла ошибка при обработке видео:\n{error}",
|
||||
'error_unknown_source': "Пардон, не умеем работать с этим источником",
|
||||
},
|
||||
'en': {
|
||||
'start': (
|
||||
"👋 Hi! I'm a video download bot.\n\n"
|
||||
"Just send me a video link, and I'll download it for you.\n\n"
|
||||
"Supported sources:\n"
|
||||
"• YouTube (youtube.com, youtu.be)\n"
|
||||
"• Instagram (instagram.com)\n"
|
||||
"• VK (vk.com)\n"
|
||||
"• Yapfiles (yapfiles.ru)\n\n"
|
||||
"👥 Group usage:\n"
|
||||
"Add me to a group with admin rights (message deletion required). "
|
||||
"I'll automatically find video links in messages, "
|
||||
"download them and send directly to the group.\n\n"
|
||||
"Commands:\n"
|
||||
"/start - Get started\n"
|
||||
"/stat - Bot statistics\n"
|
||||
"/support - Support and info\n\n"
|
||||
"Send a video link:"
|
||||
),
|
||||
'support': (
|
||||
"ℹ️ <b>About the bot</b>\n\n"
|
||||
"This bot allows you to download videos from popular sources:\n"
|
||||
"• YouTube — videos and shorts\n"
|
||||
"• Instagram — reels and video posts\n"
|
||||
"• VK — video recordings\n"
|
||||
"• Yapfiles — video files\n\n"
|
||||
"🔧 <b>How to use:</b>\n"
|
||||
"1. Send a video link in a private chat with the bot\n"
|
||||
"2. Wait for the download\n"
|
||||
"3. Get the video right in Telegram!\n\n"
|
||||
"👥 <b>In groups:</b>\n"
|
||||
"Add the bot to a group with admin rights — "
|
||||
"it will automatically download videos from participants' messages.\n\n"
|
||||
"❓ Questions or suggestions?\n"
|
||||
"Contact the author: @rvrubel"
|
||||
),
|
||||
'stat': "📊 Bot statistics:\n\n👥 Total users: {users}\n📹 Total downloads: {downloads}",
|
||||
'send_link': (
|
||||
"Please send a video link.\n"
|
||||
"Supported sources:\n"
|
||||
"• YouTube (youtube.com, youtu.be)\n"
|
||||
"• Instagram (instagram.com)\n"
|
||||
"• VK (vk.com)\n"
|
||||
"• Yapfiles (yapfiles.ru)\n\n"
|
||||
"Other sources: Sorry, not supported 😅"
|
||||
),
|
||||
'unsupported_source': "Sorry, this source is not supported 😅",
|
||||
'processing': "🔍 Processing link...",
|
||||
'downloading': "⬇️ Downloading video...",
|
||||
'sending': "📤 Sending video...",
|
||||
'caption': "Video downloaded via @{bot_username}",
|
||||
'error': "❌ Error processing video:\n{error}",
|
||||
'error_unknown_source': "Sorry, this source is not supported",
|
||||
}
|
||||
}
|
||||
|
||||
|
||||
def get_locale_from_language_code(language_code: str | None) -> str:
|
||||
"""Определяет локаль на основе language_code из Telegram"""
|
||||
if language_code and language_code.lower().startswith('ru'):
|
||||
return 'ru'
|
||||
return 'en'
|
||||
|
||||
|
||||
def get_text(locale: str, key: str, **kwargs) -> str:
|
||||
"""Возвращает локализованный текст"""
|
||||
if locale not in TEXTS:
|
||||
locale = 'en'
|
||||
text = TEXTS[locale].get(key, TEXTS['en'].get(key, key))
|
||||
if kwargs:
|
||||
text = text.format(**kwargs)
|
||||
return text
|
||||
|
||||
|
||||
# ============================================================================
|
||||
# БАЗА ДАННЫХ
|
||||
# ============================================================================
|
||||
|
||||
def init_database():
|
||||
"""Инициализирует базу данных и создает таблицы если их нет"""
|
||||
|
|
@ -59,6 +195,13 @@ def init_database():
|
|||
)
|
||||
''')
|
||||
|
||||
# Проверяем, есть ли колонка locale (миграция для существующей базы)
|
||||
cursor.execute("PRAGMA table_info(users)")
|
||||
columns = [col[1] for col in cursor.fetchall()]
|
||||
if 'locale' not in columns:
|
||||
cursor.execute("ALTER TABLE users ADD COLUMN locale TEXT DEFAULT 'en'")
|
||||
logger.info("Добавлена колонка locale в таблицу users")
|
||||
|
||||
# Таблица статистики
|
||||
cursor.execute('''
|
||||
CREATE TABLE IF NOT EXISTS stats (
|
||||
|
|
@ -78,6 +221,7 @@ def init_database():
|
|||
except Exception as e:
|
||||
logger.error(f"Ошибка при инициализации базы данных: {e}")
|
||||
|
||||
|
||||
def get_total_downloads() -> int:
|
||||
"""Возвращает общее количество скачанных видео"""
|
||||
try:
|
||||
|
|
@ -91,6 +235,7 @@ def get_total_downloads() -> int:
|
|||
logger.error(f"Ошибка при получении количества скачанных видео: {e}")
|
||||
return 0
|
||||
|
||||
|
||||
def increment_downloads():
|
||||
"""Увеличивает счетчик скачанных видео"""
|
||||
try:
|
||||
|
|
@ -104,6 +249,7 @@ def increment_downloads():
|
|||
except Exception as e:
|
||||
logger.error(f"Ошибка при увеличении счетчика скачанных видео: {e}")
|
||||
|
||||
|
||||
def get_total_users() -> int:
|
||||
"""Возвращает общее количество уникальных пользователей"""
|
||||
try:
|
||||
|
|
@ -117,7 +263,22 @@ def get_total_users() -> int:
|
|||
logger.error(f"Ошибка при получении количества пользователей: {e}")
|
||||
return 0
|
||||
|
||||
def add_user(chat_id: int, username: str = None, first_name: str = None):
|
||||
|
||||
def get_user_locale(chat_id: int) -> str:
|
||||
"""Возвращает локаль пользователя из базы данных"""
|
||||
try:
|
||||
conn = sqlite3.connect(str(DB_FILE))
|
||||
cursor = conn.cursor()
|
||||
cursor.execute('SELECT locale FROM users WHERE chat_id = ?', (chat_id,))
|
||||
result = cursor.fetchone()
|
||||
conn.close()
|
||||
return result[0] if result and result[0] else 'en'
|
||||
except Exception as e:
|
||||
logger.error(f"Ошибка при получении локали пользователя: {e}")
|
||||
return 'en'
|
||||
|
||||
|
||||
def add_user(chat_id: int, username: str = None, first_name: str = None, locale: str = 'en'):
|
||||
"""Добавляет пользователя в базу данных или обновляет информацию о нем"""
|
||||
try:
|
||||
now = datetime.now().isoformat()
|
||||
|
|
@ -129,19 +290,19 @@ def add_user(chat_id: int, username: str = None, first_name: str = None):
|
|||
exists = cursor.fetchone()
|
||||
|
||||
if exists:
|
||||
# Обновляем last_seen
|
||||
# Обновляем last_seen и locale
|
||||
cursor.execute(
|
||||
'UPDATE users SET last_seen = ?, username = ?, first_name = ? WHERE chat_id = ?',
|
||||
(now, username, first_name, chat_id)
|
||||
'UPDATE users SET last_seen = ?, username = ?, first_name = ?, locale = ? WHERE chat_id = ?',
|
||||
(now, username, first_name, locale, chat_id)
|
||||
)
|
||||
else:
|
||||
# Добавляем нового пользователя
|
||||
cursor.execute(
|
||||
'INSERT INTO users (chat_id, username, first_name, first_seen, last_seen) VALUES (?, ?, ?, ?, ?)',
|
||||
(chat_id, username, first_name, now, now)
|
||||
'INSERT INTO users (chat_id, username, first_name, first_seen, last_seen, locale) VALUES (?, ?, ?, ?, ?, ?)',
|
||||
(chat_id, username, first_name, now, now, locale)
|
||||
)
|
||||
total_users = get_total_users()
|
||||
logger.info(f"Добавлен новый пользователь (chat_id: {chat_id}). Всего пользователей: {total_users}")
|
||||
logger.info(f"Добавлен новый пользователь (chat_id: {chat_id}, locale: {locale}). Всего пользователей: {total_users}")
|
||||
|
||||
conn.commit()
|
||||
conn.close()
|
||||
|
|
@ -149,6 +310,10 @@ def add_user(chat_id: int, username: str = None, first_name: str = None):
|
|||
logger.error(f"Ошибка при добавлении пользователя: {e}")
|
||||
|
||||
|
||||
# ============================================================================
|
||||
# УТИЛИТЫ
|
||||
# ============================================================================
|
||||
|
||||
def detect_video_source(url: str) -> str:
|
||||
"""Определяет источник видео по URL"""
|
||||
domain = urlparse(url).netloc.lower()
|
||||
|
|
@ -167,7 +332,6 @@ def detect_video_source(url: str) -> str:
|
|||
|
||||
def extract_urls_from_text(text: str) -> list[str]:
|
||||
"""Извлекает все URL из текста сообщения"""
|
||||
# Регулярное выражение для поиска URL (http/https)
|
||||
url_pattern = r'https?://[^\s<>"{}|\\^`\[\]]+'
|
||||
urls = re.findall(url_pattern, text)
|
||||
return urls
|
||||
|
|
@ -180,7 +344,6 @@ def cleanup_old_files():
|
|||
if not file_path.is_file():
|
||||
continue
|
||||
|
||||
# Удаляем только .part файлы (недокачанные)
|
||||
if file_path.suffix == '.part':
|
||||
try:
|
||||
file_path.unlink()
|
||||
|
|
@ -191,6 +354,10 @@ def cleanup_old_files():
|
|||
logger.error(f"Ошибка при очистке .part файлов: {e}")
|
||||
|
||||
|
||||
# ============================================================================
|
||||
# ФУНКЦИИ СКАЧИВАНИЯ
|
||||
# ============================================================================
|
||||
|
||||
async def download_youtube_video(url: str, chat_id: int, max_retries: int = 3) -> str:
|
||||
"""Скачивает видео с YouTube через внешний сервис"""
|
||||
logger.info(f"YouTube: отправка запроса на внешний сервис {YOUTUBE_DOWNLOADER_URL}")
|
||||
|
|
@ -198,8 +365,7 @@ async def download_youtube_video(url: str, chat_id: int, max_retries: int = 3) -
|
|||
last_error = None
|
||||
for attempt in range(max_retries):
|
||||
try:
|
||||
async with httpx.AsyncClient(timeout=600.0) as client: # Увеличенный таймаут для YouTube
|
||||
# Отправляем запрос на YouTube сервис
|
||||
async with httpx.AsyncClient(timeout=600.0) as client:
|
||||
response = await client.post(
|
||||
f"{YOUTUBE_DOWNLOADER_URL}/download/stream",
|
||||
json={"url": url},
|
||||
|
|
@ -215,23 +381,19 @@ async def download_youtube_video(url: str, chat_id: int, max_retries: int = 3) -
|
|||
pass
|
||||
raise Exception(f"YouTube сервис вернул ошибку {response.status_code}: {error_text}")
|
||||
|
||||
# Сохраняем видео во временный файл
|
||||
video_data = response.content
|
||||
video_ext = 'mp4' # По умолчанию mp4
|
||||
video_ext = 'mp4'
|
||||
|
||||
# Пробуем определить расширение из заголовков
|
||||
content_type = response.headers.get('Content-Type', '')
|
||||
if 'video/' in content_type:
|
||||
video_ext = content_type.split('/')[-1].split(';')[0]
|
||||
|
||||
# Получаем имя файла из заголовка или создаем случайное
|
||||
filename = response.headers.get('Content-Disposition', '')
|
||||
if filename and 'filename=' in filename:
|
||||
video_filename = filename.split('filename=')[1].strip('"\'')
|
||||
else:
|
||||
video_filename = f'{chat_id}_youtube_video.{video_ext}'
|
||||
|
||||
# Сохраняем файл
|
||||
video_path = DOWNLOADS_DIR / video_filename
|
||||
with open(video_path, 'wb') as f:
|
||||
f.write(video_data)
|
||||
|
|
@ -259,8 +421,7 @@ async def download_instagram_video(url: str, chat_id: int, max_retries: int = 3)
|
|||
last_error = None
|
||||
for attempt in range(max_retries):
|
||||
try:
|
||||
async with httpx.AsyncClient(timeout=600.0) as client: # Увеличенный таймаут для Instagram
|
||||
# Отправляем запрос на Instagram сервис
|
||||
async with httpx.AsyncClient(timeout=600.0) as client:
|
||||
response = await client.post(
|
||||
f"{INSTAGRAM_DOWNLOADER_URL}/download/stream",
|
||||
json={"url": url},
|
||||
|
|
@ -276,23 +437,19 @@ async def download_instagram_video(url: str, chat_id: int, max_retries: int = 3)
|
|||
pass
|
||||
raise Exception(f"Instagram сервис вернул ошибку {response.status_code}: {error_text}")
|
||||
|
||||
# Сохраняем видео во временный файл
|
||||
video_data = response.content
|
||||
video_ext = 'mp4' # По умолчанию mp4
|
||||
video_ext = 'mp4'
|
||||
|
||||
# Пробуем определить расширение из заголовков
|
||||
content_type = response.headers.get('Content-Type', '')
|
||||
if 'video/' in content_type:
|
||||
video_ext = content_type.split('/')[-1].split(';')[0]
|
||||
|
||||
# Получаем имя файла из заголовка или создаем случайное
|
||||
filename = response.headers.get('Content-Disposition', '')
|
||||
if filename and 'filename=' in filename:
|
||||
video_filename = filename.split('filename=')[1].strip('"\'')
|
||||
else:
|
||||
video_filename = f'{chat_id}_instagram_video.{video_ext}'
|
||||
|
||||
# Сохраняем файл
|
||||
video_path = DOWNLOADS_DIR / video_filename
|
||||
with open(video_path, 'wb') as f:
|
||||
f.write(video_data)
|
||||
|
|
@ -320,8 +477,7 @@ async def download_vk_video(url: str, chat_id: int, max_retries: int = 3) -> str
|
|||
last_error = None
|
||||
for attempt in range(max_retries):
|
||||
try:
|
||||
async with httpx.AsyncClient(timeout=600.0) as client: # Увеличенный таймаут для VK
|
||||
# Отправляем запрос на VK сервис
|
||||
async with httpx.AsyncClient(timeout=600.0) as client:
|
||||
response = await client.post(
|
||||
f"{VK_DOWNLOADER_URL}/download/stream",
|
||||
json={"url": url},
|
||||
|
|
@ -337,23 +493,19 @@ async def download_vk_video(url: str, chat_id: int, max_retries: int = 3) -> str
|
|||
pass
|
||||
raise Exception(f"VK сервис вернул ошибку {response.status_code}: {error_text}")
|
||||
|
||||
# Сохраняем видео во временный файл
|
||||
video_data = response.content
|
||||
video_ext = 'mp4' # По умолчанию mp4
|
||||
video_ext = 'mp4'
|
||||
|
||||
# Пробуем определить расширение из заголовков
|
||||
content_type = response.headers.get('Content-Type', '')
|
||||
if 'video/' in content_type:
|
||||
video_ext = content_type.split('/')[-1].split(';')[0]
|
||||
|
||||
# Получаем имя файла из заголовка или создаем случайное
|
||||
filename = response.headers.get('Content-Disposition', '')
|
||||
if filename and 'filename=' in filename:
|
||||
video_filename = filename.split('filename=')[1].strip('"\'')
|
||||
else:
|
||||
video_filename = f'{chat_id}_vk_video.{video_ext}'
|
||||
|
||||
# Сохраняем файл
|
||||
video_path = DOWNLOADS_DIR / video_filename
|
||||
with open(video_path, 'wb') as f:
|
||||
f.write(video_data)
|
||||
|
|
@ -430,7 +582,7 @@ async def download_yapfiles_video(url: str, chat_id: int, max_retries: int = 3)
|
|||
raise last_error or Exception("Неизвестная ошибка при скачивании с Yapfiles через внешний сервис")
|
||||
|
||||
|
||||
async def download_video(url: str, chat_id: int, max_retries: int = 3) -> str:
|
||||
async def download_video(url: str, chat_id: int, locale: str, max_retries: int = 3) -> str:
|
||||
"""Главная функция скачивания - вызывает нужную функцию в зависимости от источника"""
|
||||
source = detect_video_source(url)
|
||||
logger.info(f"Определен источник: {source} для URL: {url}")
|
||||
|
|
@ -444,9 +596,13 @@ async def download_video(url: str, chat_id: int, max_retries: int = 3) -> str:
|
|||
elif source == 'yapfiles':
|
||||
return await download_yapfiles_video(url, chat_id, max_retries)
|
||||
else:
|
||||
raise Exception("Пардон, не умеем работать с этим источником")
|
||||
raise Exception(get_text(locale, 'error_unknown_source'))
|
||||
|
||||
|
||||
# ============================================================================
|
||||
# ОБРАБОТЧИКИ КОМАНД И СООБЩЕНИЙ
|
||||
# ============================================================================
|
||||
|
||||
async def handle_message(update: Update, context: ContextTypes.DEFAULT_TYPE):
|
||||
"""Обрабатывает сообщения от пользователей"""
|
||||
if not update.message or not update.message.text:
|
||||
|
|
@ -454,27 +610,23 @@ async def handle_message(update: Update, context: ContextTypes.DEFAULT_TYPE):
|
|||
|
||||
text = update.message.text.strip()
|
||||
chat_id = update.message.chat_id
|
||||
chat_type = update.message.chat.type # 'private', 'group', 'supergroup'
|
||||
chat_type = update.message.chat.type
|
||||
username = update.message.from_user.username if update.message.from_user else None
|
||||
first_name = update.message.from_user.first_name if update.message.from_user else None
|
||||
language_code = update.message.from_user.language_code if update.message.from_user else None
|
||||
|
||||
# Добавляем пользователя в статистику при первом взаимодействии
|
||||
add_user(chat_id, username, first_name)
|
||||
# Определяем локаль
|
||||
locale = get_locale_from_language_code(language_code)
|
||||
|
||||
# Добавляем пользователя в статистику
|
||||
add_user(chat_id, username, first_name, locale)
|
||||
|
||||
# Извлекаем все URL из текста
|
||||
urls = extract_urls_from_text(text)
|
||||
|
||||
# Если это личный чат и нет ссылок, отправляем инструкцию
|
||||
if not urls and chat_type == 'private':
|
||||
await update.message.reply_text(
|
||||
"Пожалуйста, отправьте ссылку на видео.\n"
|
||||
"Поддерживаемые источники:\n"
|
||||
"• YouTube (youtube.com, youtu.be)\n"
|
||||
"• Instagram (instagram.com)\n"
|
||||
"• VK (vk.com)\n"
|
||||
"• Yapfiles (yapfiles.ru)\n\n"
|
||||
"Для других источников: Пардон, не умеем 😅"
|
||||
)
|
||||
await update.message.reply_text(get_text(locale, 'send_link'))
|
||||
return
|
||||
|
||||
# Если нет ссылок в группе, просто игнорируем сообщение
|
||||
|
|
@ -487,24 +639,23 @@ async def handle_message(update: Update, context: ContextTypes.DEFAULT_TYPE):
|
|||
# Проверяем источник до начала обработки
|
||||
source = detect_video_source(url)
|
||||
if source == 'unknown':
|
||||
# В группах не отвечаем на неподдерживаемые источники, чтобы не спамить
|
||||
if chat_type == 'private':
|
||||
await update.message.reply_text("Пардон, не умеем работать с этим источником 😅")
|
||||
await update.message.reply_text(get_text(locale, 'unsupported_source'))
|
||||
return
|
||||
|
||||
# Отправляем сообщение о начале обработки
|
||||
status_message = await update.message.reply_text("🔍 Обрабатываю ссылку...")
|
||||
status_message = await update.message.reply_text(get_text(locale, 'processing'))
|
||||
|
||||
try:
|
||||
# Скачиваем видео
|
||||
await status_message.edit_text("⬇️ Скачиваю видео...")
|
||||
video_path = await download_video(url, chat_id)
|
||||
await status_message.edit_text(get_text(locale, 'downloading'))
|
||||
video_path = await download_video(url, chat_id, locale)
|
||||
|
||||
# Отправляем файл пользователю
|
||||
await status_message.edit_text("📤 Отправляю видео...")
|
||||
await status_message.edit_text(get_text(locale, 'sending'))
|
||||
|
||||
video_file = open(video_path, 'rb')
|
||||
caption = f"Видео скачано с @{TELEGRAM_BOT_USERNAME}"
|
||||
caption = get_text(locale, 'caption', bot_username=TELEGRAM_BOT_USERNAME)
|
||||
await update.message.reply_video(
|
||||
video=video_file,
|
||||
caption=caption,
|
||||
|
|
@ -515,7 +666,6 @@ async def handle_message(update: Update, context: ContextTypes.DEFAULT_TYPE):
|
|||
# Увеличиваем счетчик скачанных видео
|
||||
increment_downloads()
|
||||
|
||||
# Сохраняем видео в папку video (не удаляем)
|
||||
logger.info(f"Видео сохранено: {video_path}")
|
||||
|
||||
# Удаляем статусное сообщение и исходное сообщение со ссылкой
|
||||
|
|
@ -525,20 +675,16 @@ async def handle_message(update: Update, context: ContextTypes.DEFAULT_TYPE):
|
|||
logger.info(f"Удалено сообщение пользователя с ссылкой (chat_id: {chat_id}, тип чата: {chat_type})")
|
||||
except Exception as e:
|
||||
logger.warning(f"Не удалось удалить сообщение: {e}")
|
||||
# Если не удалось удалить (нет прав), просто логируем
|
||||
|
||||
except Exception as e:
|
||||
logger.error(f"Ошибка: {e}")
|
||||
error_msg = f"❌ Произошла ошибка при обработке видео:\n{str(e)}"
|
||||
error_msg = get_text(locale, 'error', error=str(e))
|
||||
try:
|
||||
await status_message.edit_text(error_msg)
|
||||
except:
|
||||
# Если status_message не существует, создаем новое сообщение
|
||||
await update.message.reply_text(error_msg)
|
||||
|
||||
# При ошибке тоже пытаемся удалить временные файлы
|
||||
try:
|
||||
# Удаляем все .part файлы для этого chat_id
|
||||
for part_file in DOWNLOADS_DIR.glob(f'{chat_id}_*.part'):
|
||||
part_file.unlink()
|
||||
logger.info(f"Удален .part файл после ошибки: {part_file.name}")
|
||||
|
|
@ -548,43 +694,45 @@ async def handle_message(update: Update, context: ContextTypes.DEFAULT_TYPE):
|
|||
|
||||
async def start_command(update: Update, context: ContextTypes.DEFAULT_TYPE):
|
||||
"""Обрабатывает команду /start"""
|
||||
# Добавляем пользователя в статистику
|
||||
chat_id = update.message.chat_id
|
||||
username = update.message.from_user.username if update.message.from_user else None
|
||||
first_name = update.message.from_user.first_name if update.message.from_user else None
|
||||
add_user(chat_id, username, first_name)
|
||||
language_code = update.message.from_user.language_code if update.message.from_user else None
|
||||
|
||||
await update.message.reply_text(
|
||||
"👋 Привет! Я бот для скачивания видео.\n\n"
|
||||
"Просто отправь мне ссылку на видео, и я скачаю его для тебя.\n\n"
|
||||
"Поддерживаемые источники:\n"
|
||||
"• YouTube (youtube.com, youtu.be)\n"
|
||||
"• Instagram (instagram.com)\n"
|
||||
"• VK (vk.com)\n"
|
||||
"• Yapfiles (yapfiles.ru)\n\n"
|
||||
"👥 Работа в группах:\n"
|
||||
"Добавь меня в группу и дай права администратора (нужно право на удаление сообщений). "
|
||||
"После этого я буду автоматически находить ссылки на видео в сообщениях участников, "
|
||||
"скачивать их и отправлять прямо в группу, заменяя исходное сообщение со ссылкой.\n\n"
|
||||
"Команды:\n"
|
||||
"/start - Начать работу\n"
|
||||
"/stat - Статистика скачанных видео\n\n"
|
||||
"Отправь ссылку на видео:"
|
||||
)
|
||||
locale = get_locale_from_language_code(language_code)
|
||||
add_user(chat_id, username, first_name, locale)
|
||||
|
||||
await update.message.reply_text(get_text(locale, 'start'))
|
||||
|
||||
|
||||
async def stat_command(update: Update, context: ContextTypes.DEFAULT_TYPE):
|
||||
"""Обрабатывает команду /stat"""
|
||||
language_code = update.message.from_user.language_code if update.message.from_user else None
|
||||
locale = get_locale_from_language_code(language_code)
|
||||
|
||||
total_downloads = get_total_downloads()
|
||||
total_users = get_total_users()
|
||||
|
||||
await update.message.reply_text(
|
||||
f"📊 Статистика бота:\n\n"
|
||||
f"👥 Всего пользователей: {total_users}\n"
|
||||
f"📹 Всего скачано видео: {total_downloads}"
|
||||
get_text(locale, 'stat', users=total_users, downloads=total_downloads)
|
||||
)
|
||||
|
||||
|
||||
async def support_command(update: Update, context: ContextTypes.DEFAULT_TYPE):
|
||||
"""Обрабатывает команду /support"""
|
||||
language_code = update.message.from_user.language_code if update.message.from_user else None
|
||||
locale = get_locale_from_language_code(language_code)
|
||||
|
||||
await update.message.reply_text(
|
||||
get_text(locale, 'support'),
|
||||
parse_mode='HTML'
|
||||
)
|
||||
|
||||
|
||||
# ============================================================================
|
||||
# MAIN
|
||||
# ============================================================================
|
||||
|
||||
def main():
|
||||
"""Главная функция для запуска бота"""
|
||||
if not TELEGRAM_BOT_TOKEN:
|
||||
|
|
@ -596,7 +744,7 @@ def main():
|
|||
|
||||
# Очищаем .part файлы при старте
|
||||
logger.info("Очистка .part файлов при старте...")
|
||||
cleanup_old_files() # Удаляем только недокачанные .part файлы
|
||||
cleanup_old_files()
|
||||
|
||||
# Создаем приложение
|
||||
application = Application.builder().token(TELEGRAM_BOT_TOKEN).build()
|
||||
|
|
@ -605,14 +753,14 @@ def main():
|
|||
application.add_handler(MessageHandler(filters.TEXT & ~filters.COMMAND, handle_message))
|
||||
application.add_handler(CommandHandler("start", start_command))
|
||||
application.add_handler(CommandHandler("stat", stat_command))
|
||||
application.add_handler(CommandHandler("support", support_command))
|
||||
|
||||
# Запускаем периодическую очистку файлов
|
||||
async def post_init(application: Application):
|
||||
"""Выполняется после инициализации приложения"""
|
||||
# Запускаем периодическую очистку .part файлов (каждые 6 часов)
|
||||
async def periodic_cleanup():
|
||||
while True:
|
||||
await asyncio.sleep(6 * 3600) # 6 часов
|
||||
await asyncio.sleep(6 * 3600)
|
||||
cleanup_old_files()
|
||||
logger.info("Периодическая очистка .part файлов выполнена")
|
||||
|
||||
|
|
@ -628,4 +776,3 @@ def main():
|
|||
|
||||
if __name__ == '__main__':
|
||||
main()
|
||||
|
||||
|
|
|
|||
Loading…
Add table
Add a link
Reference in a new issue