67 lines
1.7 KiB
Python
67 lines
1.7 KiB
Python
from __future__ import unicode_literals
|
|
import simplematrixbotlib as botlib
|
|
from re import search
|
|
import urllib.parse
|
|
import youtube_dl
|
|
import os
|
|
|
|
try:
|
|
passfile = open("pass.txt", "r")
|
|
passw = passfile.read().translate({ord(c): None for c in '\n'})
|
|
except:
|
|
print("Put the password in pass.txt in this directory")
|
|
|
|
creds = botlib.Creds("https://chat.trygve.me", "video_bot", passw)
|
|
bot = botlib.Bot(creds)
|
|
|
|
prefix = 'https://'
|
|
|
|
class MyLogger(object):
|
|
def debug(self, msg):
|
|
pass
|
|
|
|
def warning(self, msg):
|
|
pass
|
|
|
|
def error(self, msg):
|
|
print(msg)
|
|
|
|
|
|
def my_hook(d):
|
|
if d['status'] == 'finished':
|
|
print('Done downloading')
|
|
|
|
|
|
ydl_opts = {
|
|
'format': 'bestvideo[ext=webm]+bestaudio[ext=webm]/best[ext=mp4]/best',
|
|
'outtmpl': '/var/www/html/video_bot/%(title)s.%(ext)s',
|
|
'restrictfilenames': False,
|
|
'logger': MyLogger(),
|
|
'progress_hooks': [my_hook],
|
|
}
|
|
|
|
def filterfunc(variable):
|
|
if search('www.youtube.com/watch', variable):
|
|
return True
|
|
else:
|
|
return False
|
|
|
|
async def youtube(room, message):
|
|
match = botlib.MessageMatch(room, message, bot)
|
|
pieces = str(message).split()
|
|
links = filter(filterfunc, pieces)
|
|
try:
|
|
for i in links:
|
|
with youtube_dl.YoutubeDL(ydl_opts) as ydl:
|
|
info_dict = ydl.extract_info(i, download=False)
|
|
fname = urllib.parse.quote(info_dict.get('title', None).replace('|', '_'))
|
|
url = 'https://trygve.me/video_bot/' + fname + '.webm'
|
|
ydl.download([i])
|
|
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.add_message_listener(youtube)
|
|
|
|
bot.run()
|