Добавлен TikTok загрузчик: обновлены конфигурации, добавлены функции для скачивания видео с TikTok и обновлены текстовые сообщения для поддержки нового источника.
This commit is contained in:
vrubelroman
parent
95d1ce4f9a
commit
da98462bbc
8 changed files with 315 additions and 0 deletions
67
bot.py
67
bot.py
|
|
@ -28,6 +28,7 @@ YOUTUBE_DOWNLOADER_URL = os.getenv('YOUTUBE_DOWNLOADER_URL', 'http://localhost:5
|
|||
INSTAGRAM_DOWNLOADER_URL = os.getenv('INSTAGRAM_DOWNLOADER_URL', 'http://localhost:5556')
|
||||
VK_DOWNLOADER_URL = os.getenv('VK_DOWNLOADER_URL', 'http://localhost:5555')
|
||||
YAPFILES_DOWNLOADER_URL = os.getenv('YAPFILES_DOWNLOADER_URL', 'http://localhost:5558')
|
||||
TIKTOK_DOWNLOADER_URL = os.getenv('TIKTOK_DOWNLOADER_URL', 'http://localhost:5559')
|
||||
|
||||
# Базовая директория проекта (абсолютный путь), чтобы не зависеть от рабочей директории процесса
|
||||
BASE_DIR = Path(__file__).resolve().parent
|
||||
|
|
@ -53,6 +54,7 @@ TEXTS = {
|
|||
"Поддерживаемые источники:\n"
|
||||
"• YouTube (youtube.com, youtu.be)\n"
|
||||
"• Instagram (instagram.com)\n"
|
||||
"• TikTok (tiktok.com)\n"
|
||||
"• VK (vk.com)\n"
|
||||
"• Yapfiles (yapfiles.ru)\n\n"
|
||||
"👥 Работа в группах:\n"
|
||||
|
|
@ -70,6 +72,7 @@ TEXTS = {
|
|||
"Этот бот позволяет скачивать видео из популярных источников:\n"
|
||||
"• YouTube — видео и shorts\n"
|
||||
"• Instagram — reels и посты с видео\n"
|
||||
"• TikTok — видео\n"
|
||||
"• VK — видеозаписи\n"
|
||||
"• Yapfiles — видеофайлы\n\n"
|
||||
"🔧 <b>Как использовать:</b>\n"
|
||||
|
|
@ -88,6 +91,7 @@ TEXTS = {
|
|||
"Поддерживаемые источники:\n"
|
||||
"• YouTube (youtube.com, youtu.be)\n"
|
||||
"• Instagram (instagram.com)\n"
|
||||
"• TikTok (tiktok.com)\n"
|
||||
"• VK (vk.com)\n"
|
||||
"• Yapfiles (yapfiles.ru)\n\n"
|
||||
"Для других источников: Пардон, не умеем 😅"
|
||||
|
|
@ -107,6 +111,7 @@ TEXTS = {
|
|||
"Supported sources:\n"
|
||||
"• YouTube (youtube.com, youtu.be)\n"
|
||||
"• Instagram (instagram.com)\n"
|
||||
"• TikTok (tiktok.com)\n"
|
||||
"• VK (vk.com)\n"
|
||||
"• Yapfiles (yapfiles.ru)\n\n"
|
||||
"👥 Group usage:\n"
|
||||
|
|
@ -124,6 +129,7 @@ TEXTS = {
|
|||
"This bot allows you to download videos from popular sources:\n"
|
||||
"• YouTube — videos and shorts\n"
|
||||
"• Instagram — reels and video posts\n"
|
||||
"• TikTok — videos\n"
|
||||
"• VK — video recordings\n"
|
||||
"• Yapfiles — video files\n\n"
|
||||
"🔧 <b>How to use:</b>\n"
|
||||
|
|
@ -142,6 +148,7 @@ TEXTS = {
|
|||
"Supported sources:\n"
|
||||
"• YouTube (youtube.com, youtu.be)\n"
|
||||
"• Instagram (instagram.com)\n"
|
||||
"• TikTok (tiktok.com)\n"
|
||||
"• VK (vk.com)\n"
|
||||
"• Yapfiles (yapfiles.ru)\n\n"
|
||||
"Other sources: Sorry, not supported 😅"
|
||||
|
|
@ -326,6 +333,8 @@ def detect_video_source(url: str) -> str:
|
|||
return 'vk'
|
||||
elif 'yapfiles.ru' in domain:
|
||||
return 'yapfiles'
|
||||
elif 'tiktok.com' in domain:
|
||||
return 'tiktok'
|
||||
else:
|
||||
return 'unknown'
|
||||
|
||||
|
|
@ -582,6 +591,62 @@ async def download_yapfiles_video(url: str, chat_id: int, max_retries: int = 3)
|
|||
raise last_error or Exception("Неизвестная ошибка при скачивании с Yapfiles через внешний сервис")
|
||||
|
||||
|
||||
async def download_tiktok_video(url: str, chat_id: int, max_retries: int = 3) -> str:
|
||||
"""Скачивает видео с TikTok через внешний сервис"""
|
||||
logger.info(f"TikTok: отправка запроса на внешний сервис {TIKTOK_DOWNLOADER_URL}")
|
||||
|
||||
last_error = None
|
||||
for attempt in range(max_retries):
|
||||
try:
|
||||
async with httpx.AsyncClient(timeout=600.0) as client:
|
||||
response = await client.post(
|
||||
f"{TIKTOK_DOWNLOADER_URL}/download/stream",
|
||||
json={"url": url},
|
||||
headers={"Content-Type": "application/json"}
|
||||
)
|
||||
|
||||
if response.status_code != 200:
|
||||
error_text = response.text
|
||||
try:
|
||||
error_json = response.json()
|
||||
error_text = error_json.get('error', error_text)
|
||||
except:
|
||||
pass
|
||||
raise Exception(f"TikTok сервис вернул ошибку {response.status_code}: {error_text}")
|
||||
|
||||
video_data = response.content
|
||||
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}_tiktok_video.{video_ext}'
|
||||
|
||||
video_path = DOWNLOADS_DIR / video_filename
|
||||
with open(video_path, 'wb') as f:
|
||||
f.write(video_data)
|
||||
|
||||
logger.info(f"TikTok: видео скачано через внешний сервис: {video_path}")
|
||||
return str(video_path)
|
||||
|
||||
except httpx.TimeoutException:
|
||||
last_error = Exception(f"Таймаут при запросе к TikTok сервису (попытка {attempt + 1}/{max_retries})")
|
||||
logger.warning(f"TikTok: таймаут при запросе к сервису: {last_error}")
|
||||
except Exception as e:
|
||||
last_error = e
|
||||
logger.warning(f"TikTok: попытка {attempt + 1}/{max_retries} не удалась: {e}")
|
||||
|
||||
if attempt < max_retries - 1:
|
||||
await asyncio.sleep((attempt + 1) * 2)
|
||||
|
||||
raise last_error or Exception("Неизвестная ошибка при скачивании с TikTok через внешний сервис")
|
||||
|
||||
|
||||
async def download_video(url: str, chat_id: int, locale: str, max_retries: int = 3) -> str:
|
||||
"""Главная функция скачивания - вызывает нужную функцию в зависимости от источника"""
|
||||
source = detect_video_source(url)
|
||||
|
|
@ -595,6 +660,8 @@ async def download_video(url: str, chat_id: int, locale: str, max_retries: int =
|
|||
return await download_vk_video(url, chat_id, max_retries)
|
||||
elif source == 'yapfiles':
|
||||
return await download_yapfiles_video(url, chat_id, max_retries)
|
||||
elif source == 'tiktok':
|
||||
return await download_tiktok_video(url, chat_id, max_retries)
|
||||
else:
|
||||
raise Exception(get_text(locale, 'error_unknown_source'))
|
||||
|
||||
|
|
|
|||
Loading…
Add table
Add a link
Reference in a new issue