60 lines
1.8 KiB
Python
60 lines
1.8 KiB
Python
import simplematrixbotlib as botlib
|
|
from re import search
|
|
from urllib.parse import quote as sanitize_url
|
|
import yt_dlp
|
|
|
|
from config import config
|
|
|
|
class Logger(object):
|
|
def debug(self, msg):
|
|
pass
|
|
|
|
def warning(self, msg):
|
|
pass
|
|
|
|
def error(self, msg):
|
|
print(msg)
|
|
|
|
def progress_hook(d):
|
|
if d['status'] == 'started':
|
|
print('Started downloading')
|
|
elif d['status'] == 'finished':
|
|
print('Done downloading')
|
|
|
|
def get_yt_vid_id(word):
|
|
if search('www.youtube.com/watch', word):
|
|
return word
|
|
elif search('youtu.be', word):
|
|
return word.replace('https://youtu.be/', 'https://www.youtube.com/watch?v=')
|
|
else:
|
|
return False
|
|
|
|
creds = botlib.Creds(config['url'], config['username'], config['password'])
|
|
bot = botlib.Bot(creds)
|
|
ydl_opts = {
|
|
'format': config['yt_dlp_format'],
|
|
'outtmpl': config['download_dir'] + '/%(title)s.%(ext)s',
|
|
'restrictfilenames': True,
|
|
'logger': Logger(),
|
|
'progress_hooks': [progress_hook],
|
|
}
|
|
|
|
@bot.listener.on_message_event
|
|
async def youtube(room, message):
|
|
match = botlib.MessageMatch(room, message, bot)
|
|
pieces = str(message).split()
|
|
try:
|
|
for i in pieces:
|
|
yt_url = get_yt_vid_id(i)
|
|
if yt_url:
|
|
with yt_dlp.YoutubeDL(ydl_opts) as ydl:
|
|
info_dict = ydl.extract_info(yt_url, download=False)
|
|
fname = sanitize_url(yt_dlp.utils.sanitize_filename(info_dict.get('title', None), restricted=True))
|
|
url = 'https://trygve.me/video_bot/' + fname + '.webm'
|
|
ydl.download([yt_url])
|
|
await bot.api.send_text_message(room.room_id, url)
|
|
except Exception as e:
|
|
await bot.api.send_text_message(room.room_id, str(e))
|
|
|
|
bot.run()
|