Загрузить файлы в «/»
Модуль для загрузки из популярный соц сетей видео (tiktok,YouTube,vk,Instagram)
This commit is contained in:
yanistyle
+271
@@ -0,0 +1,271 @@
|
|||||||
|
# meta developer: @origsama
|
||||||
|
from .. import loader, utils
|
||||||
|
import requests
|
||||||
|
import re
|
||||||
|
import os
|
||||||
|
import subprocess
|
||||||
|
import tempfile
|
||||||
|
import urllib.parse
|
||||||
|
import json
|
||||||
|
|
||||||
|
@loader.tds
|
||||||
|
class SocialDownloaderMod(loader.Module):
|
||||||
|
"""Скачивание видео из TikTok, Instagram, YouTube, VK"""
|
||||||
|
strings = {"name": "SocialDownloader"}
|
||||||
|
|
||||||
|
def __init__(self):
|
||||||
|
self.config = loader.ModuleConfig(
|
||||||
|
loader.ConfigValue(
|
||||||
|
"max_size",
|
||||||
|
50,
|
||||||
|
"Максимальный размер видео в МБ",
|
||||||
|
validator=loader.validators.Integer()
|
||||||
|
)
|
||||||
|
)
|
||||||
|
|
||||||
|
async def dlcmd(self, message):
|
||||||
|
"""<ссылка> - Скачать видео из TikTok/Instagram/YouTube/VK"""
|
||||||
|
args = utils.get_args_raw(message)
|
||||||
|
if not args:
|
||||||
|
await utils.answer(message,
|
||||||
|
"❌ Укажи ссылку на видео\n\n"
|
||||||
|
"Поддерживаемые сайты:\n"
|
||||||
|
"• TikTok: https://vm.tiktok.com/...\n"
|
||||||
|
"• Instagram: https://www.instagram.com/p/...\n"
|
||||||
|
"• YouTube: https://youtu.be/...\n"
|
||||||
|
"• VK: https://vk.com/video...\n"
|
||||||
|
"• Twitter/X: https://twitter.com/...\n"
|
||||||
|
"• Reddit: https://reddit.com/..."
|
||||||
|
)
|
||||||
|
return
|
||||||
|
|
||||||
|
# Определяем источник
|
||||||
|
source = self.detect_source(args)
|
||||||
|
if not source:
|
||||||
|
await utils.answer(message, "❌ Неподдерживаемая ссылка")
|
||||||
|
return
|
||||||
|
|
||||||
|
status = await utils.answer(message, f"📥 Скачиваю с {source}...")
|
||||||
|
|
||||||
|
try:
|
||||||
|
# Скачиваем видео
|
||||||
|
video_path, info = await self.download_video(args, source)
|
||||||
|
|
||||||
|
if not video_path:
|
||||||
|
await status.edit("❌ Не удалось скачать видео")
|
||||||
|
return
|
||||||
|
|
||||||
|
# Формируем описание
|
||||||
|
caption = f"<b>📹 Видео с {source}</b>\n\n"
|
||||||
|
if info.get('title'):
|
||||||
|
caption += f"📝 {info['title']}\n"
|
||||||
|
if info.get('author'):
|
||||||
|
caption += f"👤 {info['author']}\n"
|
||||||
|
if info.get('duration'):
|
||||||
|
caption += f"⏱ {info['duration']}\n"
|
||||||
|
|
||||||
|
caption += f"\n🔗 <a href='{args}'>Оригинал</a>"
|
||||||
|
|
||||||
|
# Отправляем
|
||||||
|
await status.delete()
|
||||||
|
|
||||||
|
await message.client.send_file(
|
||||||
|
message.chat_id,
|
||||||
|
video_path,
|
||||||
|
caption=caption,
|
||||||
|
reply_to=message.reply_to_msg_id or message.id
|
||||||
|
)
|
||||||
|
|
||||||
|
# Чистим
|
||||||
|
os.remove(video_path)
|
||||||
|
|
||||||
|
except Exception as e:
|
||||||
|
await status.edit(f"❌ Ошибка: {str(e)}")
|
||||||
|
|
||||||
|
def detect_source(self, url):
|
||||||
|
"""Определяет источник по ссылке"""
|
||||||
|
url = url.lower()
|
||||||
|
|
||||||
|
sources = {
|
||||||
|
'tiktok': ['tiktok.com', 'vm.tiktok.com'],
|
||||||
|
'instagram': ['instagram.com', 'instagr.am'],
|
||||||
|
'youtube': ['youtube.com', 'youtu.be', 'm.youtube.com'],
|
||||||
|
'vk': ['vk.com', 'vkontakte.ru'],
|
||||||
|
'twitter': ['twitter.com', 'x.com'],
|
||||||
|
'reddit': ['reddit.com', 'redd.it'],
|
||||||
|
'facebook': ['facebook.com', 'fb.com'],
|
||||||
|
'pinterest': ['pinterest.com', 'pin.it']
|
||||||
|
}
|
||||||
|
|
||||||
|
for source, domains in sources.items():
|
||||||
|
for domain in domains:
|
||||||
|
if domain in url:
|
||||||
|
return source.capitalize()
|
||||||
|
|
||||||
|
return None
|
||||||
|
|
||||||
|
async def download_video(self, url, source):
|
||||||
|
"""Скачивает видео через yt-dlp"""
|
||||||
|
try:
|
||||||
|
# Создаем временную папку
|
||||||
|
with tempfile.TemporaryDirectory() as temp_dir:
|
||||||
|
output_template = os.path.join(temp_dir, '%(title)s.%(ext)s')
|
||||||
|
|
||||||
|
# Формируем команду
|
||||||
|
cmd = [
|
||||||
|
'yt-dlp',
|
||||||
|
'-f', 'best[filesize<50M]', # Лучшее качество до 50 МБ
|
||||||
|
'--max-filesize', f"{self.config['max_size']}M",
|
||||||
|
'-o', output_template,
|
||||||
|
url
|
||||||
|
]
|
||||||
|
|
||||||
|
# Добавляем специфичные опции
|
||||||
|
if source.lower() == 'tiktok':
|
||||||
|
cmd.extend(['--cookies', 'tiktok_cookies.txt']) # Опционально
|
||||||
|
elif source.lower() == 'instagram':
|
||||||
|
cmd.extend(['--cookies', 'instagram_cookies.txt'])
|
||||||
|
|
||||||
|
# Запускаем
|
||||||
|
result = subprocess.run(cmd, capture_output=True, text=True)
|
||||||
|
|
||||||
|
if result.returncode != 0:
|
||||||
|
return None, {}
|
||||||
|
|
||||||
|
# Ищем скачанный файл
|
||||||
|
for file in os.listdir(temp_dir):
|
||||||
|
file_path = os.path.join(temp_dir, file)
|
||||||
|
if os.path.isfile(file_path):
|
||||||
|
# Копируем в постоянное место
|
||||||
|
import shutil
|
||||||
|
final_path = os.path.join(tempfile.gettempdir(), file)
|
||||||
|
shutil.copy2(file_path, final_path)
|
||||||
|
|
||||||
|
# Получаем информацию
|
||||||
|
info = await self.get_video_info(url)
|
||||||
|
|
||||||
|
return final_path, info
|
||||||
|
|
||||||
|
return None, {}
|
||||||
|
|
||||||
|
except Exception as e:
|
||||||
|
print(f"Download error: {e}")
|
||||||
|
return None, {}
|
||||||
|
|
||||||
|
async def get_video_info(self, url):
|
||||||
|
"""Получает информацию о видео"""
|
||||||
|
try:
|
||||||
|
cmd = [
|
||||||
|
'yt-dlp',
|
||||||
|
'--dump-json',
|
||||||
|
'--no-playlist',
|
||||||
|
url
|
||||||
|
]
|
||||||
|
|
||||||
|
result = subprocess.run(cmd, capture_output=True, text=True)
|
||||||
|
|
||||||
|
if result.returncode == 0:
|
||||||
|
data = json.loads(result.stdout)
|
||||||
|
|
||||||
|
duration = data.get('duration', 0)
|
||||||
|
minutes = duration // 60
|
||||||
|
seconds = duration % 60
|
||||||
|
|
||||||
|
return {
|
||||||
|
'title': data.get('title', ''),
|
||||||
|
'author': data.get('uploader', data.get('channel', '')),
|
||||||
|
'duration': f"{minutes}:{seconds:02d}" if duration else ''
|
||||||
|
}
|
||||||
|
|
||||||
|
return {}
|
||||||
|
except:
|
||||||
|
return {}
|
||||||
|
|
||||||
|
async def tiktokcmd(self, message):
|
||||||
|
"""<ссылка> - Скачать видео с TikTok"""
|
||||||
|
args = utils.get_args_raw(message)
|
||||||
|
if not args:
|
||||||
|
await utils.answer(message, "❌ Укажи ссылку на TikTok")
|
||||||
|
return
|
||||||
|
|
||||||
|
await self.dlcmd(message)
|
||||||
|
|
||||||
|
async def instacmd(self, message):
|
||||||
|
"""<ссылка> - Скачать видео/фото с Instagram"""
|
||||||
|
args = utils.get_args_raw(message)
|
||||||
|
if not args:
|
||||||
|
await utils.answer(message, "❌ Укажи ссылку на Instagram")
|
||||||
|
return
|
||||||
|
|
||||||
|
await self.dlcmd(message)
|
||||||
|
|
||||||
|
async def ytcmd(self, message):
|
||||||
|
"""<ссылка> [качество] - Скачать видео с YouTube"""
|
||||||
|
args = utils.get_args_raw(message)
|
||||||
|
if not args:
|
||||||
|
await utils.answer(message, "❌ Укажи ссылку на YouTube")
|
||||||
|
return
|
||||||
|
|
||||||
|
# Разбираем качество если указано
|
||||||
|
parts = args.split()
|
||||||
|
url = parts[0]
|
||||||
|
quality = parts[1] if len(parts) > 1 else 'best'
|
||||||
|
|
||||||
|
await utils.answer(message, f"📥 Скачиваю с YouTube ({quality})...")
|
||||||
|
|
||||||
|
try:
|
||||||
|
with tempfile.TemporaryDirectory() as temp_dir:
|
||||||
|
output_template = os.path.join(temp_dir, '%(title)s.%(ext)s')
|
||||||
|
|
||||||
|
cmd = [
|
||||||
|
'yt-dlp',
|
||||||
|
'-f', f'best[height<={quality}]' if quality != 'best' else 'best',
|
||||||
|
'--max-filesize', f"{self.config['max_size']}M",
|
||||||
|
'-o', output_template,
|
||||||
|
url
|
||||||
|
]
|
||||||
|
|
||||||
|
result = subprocess.run(cmd, capture_output=True, text=True)
|
||||||
|
|
||||||
|
if result.returncode != 0:
|
||||||
|
await utils.answer(message, "❌ Ошибка скачивания")
|
||||||
|
return
|
||||||
|
|
||||||
|
for file in os.listdir(temp_dir):
|
||||||
|
file_path = os.path.join(temp_dir, file)
|
||||||
|
if os.path.isfile(file_path):
|
||||||
|
import shutil
|
||||||
|
final_path = os.path.join(tempfile.gettempdir(), file)
|
||||||
|
shutil.copy2(file_path, final_path)
|
||||||
|
|
||||||
|
info = await self.get_video_info(url)
|
||||||
|
|
||||||
|
caption = f"<b>🎥 YouTube</b>\n"
|
||||||
|
if info.get('title'):
|
||||||
|
caption += f"📝 {info['title']}\n"
|
||||||
|
if info.get('author'):
|
||||||
|
caption += f"👤 {info['author']}\n"
|
||||||
|
if info.get('duration'):
|
||||||
|
caption += f"⏱ {info['duration']}\n"
|
||||||
|
caption += f"\n⚙️ {quality}"
|
||||||
|
|
||||||
|
await message.client.send_file(
|
||||||
|
message.chat_id,
|
||||||
|
final_path,
|
||||||
|
caption=caption
|
||||||
|
)
|
||||||
|
|
||||||
|
os.remove(final_path)
|
||||||
|
return
|
||||||
|
|
||||||
|
except Exception as e:
|
||||||
|
await utils.answer(message, f"❌ Ошибка: {str(e)}")
|
||||||
|
|
||||||
|
async def vkcmd(self, message):
|
||||||
|
"""<ссылка> - Скачать видео с VK"""
|
||||||
|
args = utils.get_args_raw(message)
|
||||||
|
if not args:
|
||||||
|
await utils.answer(message, "❌ Укажи ссылку на VK видео")
|
||||||
|
return
|
||||||
|
|
||||||
|
await self.dlcmd(message)
|
||||||
Reference in New Issue
Block a user