подключил япфайл
This commit is contained in:
parent
377fa3246e
commit
cb1458bd93
9 changed files with 393 additions and 6 deletions
67
bot.py
67
bot.py
|
|
@ -27,6 +27,7 @@ TELEGRAM_BOT_USERNAME = os.getenv('TELEGRAM_BOT_USERNAME', 'vrubelVideoDownload_
|
|||
YOUTUBE_DOWNLOADER_URL = os.getenv('YOUTUBE_DOWNLOADER_URL', 'http://localhost:5557')
|
||||
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')
|
||||
|
||||
# Базовая директория проекта (абсолютный путь), чтобы не зависеть от рабочей директории процесса
|
||||
BASE_DIR = Path(__file__).resolve().parent
|
||||
|
|
@ -158,6 +159,8 @@ def detect_video_source(url: str) -> str:
|
|||
return 'instagram'
|
||||
elif 'vk.com' in domain or 'vkontakte.ru' in domain:
|
||||
return 'vk'
|
||||
elif 'yapfiles.ru' in domain:
|
||||
return 'yapfiles'
|
||||
else:
|
||||
return 'unknown'
|
||||
|
||||
|
|
@ -371,6 +374,62 @@ async def download_vk_video(url: str, chat_id: int, max_retries: int = 3) -> str
|
|||
raise last_error or Exception("Неизвестная ошибка при скачивании с VK через внешний сервис")
|
||||
|
||||
|
||||
async def download_yapfiles_video(url: str, chat_id: int, max_retries: int = 3) -> str:
|
||||
"""Скачивает видео с Yapfiles через внешний сервис"""
|
||||
logger.info(f"Yapfiles: отправка запроса на внешний сервис {YAPFILES_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"{YAPFILES_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"Yapfiles сервис вернул ошибку {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}_yapfiles_video.{video_ext}'
|
||||
|
||||
video_path = DOWNLOADS_DIR / video_filename
|
||||
with open(video_path, 'wb') as f:
|
||||
f.write(video_data)
|
||||
|
||||
logger.info(f"Yapfiles: видео скачано через внешний сервис: {video_path}")
|
||||
return str(video_path)
|
||||
|
||||
except httpx.TimeoutException:
|
||||
last_error = Exception(f"Таймаут при запросе к Yapfiles сервису (попытка {attempt + 1}/{max_retries})")
|
||||
logger.warning(f"Yapfiles: таймаут при запросе к сервису: {last_error}")
|
||||
except Exception as e:
|
||||
last_error = e
|
||||
logger.warning(f"Yapfiles: попытка {attempt + 1}/{max_retries} не удалась: {e}")
|
||||
|
||||
if attempt < max_retries - 1:
|
||||
await asyncio.sleep((attempt + 1) * 2)
|
||||
|
||||
raise last_error or Exception("Неизвестная ошибка при скачивании с Yapfiles через внешний сервис")
|
||||
|
||||
|
||||
async def download_video(url: str, chat_id: int, max_retries: int = 3) -> str:
|
||||
"""Главная функция скачивания - вызывает нужную функцию в зависимости от источника"""
|
||||
source = detect_video_source(url)
|
||||
|
|
@ -382,6 +441,8 @@ async def download_video(url: str, chat_id: int, max_retries: int = 3) -> str:
|
|||
return await download_instagram_video(url, chat_id, max_retries)
|
||||
elif source == 'vk':
|
||||
return await download_vk_video(url, chat_id, max_retries)
|
||||
elif source == 'yapfiles':
|
||||
return await download_yapfiles_video(url, chat_id, max_retries)
|
||||
else:
|
||||
raise Exception("Пардон, не умеем работать с этим источником")
|
||||
|
||||
|
|
@ -410,7 +471,8 @@ async def handle_message(update: Update, context: ContextTypes.DEFAULT_TYPE):
|
|||
"Поддерживаемые источники:\n"
|
||||
"• YouTube (youtube.com, youtu.be)\n"
|
||||
"• Instagram (instagram.com)\n"
|
||||
"• VK (vk.com)\n\n"
|
||||
"• VK (vk.com)\n"
|
||||
"• Yapfiles (yapfiles.ru)\n\n"
|
||||
"Для других источников: Пардон, не умеем 😅"
|
||||
)
|
||||
return
|
||||
|
|
@ -498,7 +560,8 @@ async def start_command(update: Update, context: ContextTypes.DEFAULT_TYPE):
|
|||
"Поддерживаемые источники:\n"
|
||||
"• YouTube (youtube.com, youtu.be)\n"
|
||||
"• Instagram (instagram.com)\n"
|
||||
"• VK (vk.com)\n\n"
|
||||
"• VK (vk.com)\n"
|
||||
"• Yapfiles (yapfiles.ru)\n\n"
|
||||
"👥 Работа в группах:\n"
|
||||
"Добавь меня в группу и дай права администратора (нужно право на удаление сообщений). "
|
||||
"После этого я буду автоматически находить ссылки на видео в сообщениях участников, "
|
||||
|
|
|
|||
Loading…
Add table
Add a link
Reference in a new issue