Reorganizing

This commit is contained in:
Trygve 2022-07-31 22:59:30 +02:00
parent c86eecfad1
commit a864af61b3

22
main.py
View File

@ -15,24 +15,12 @@ class MyLogger(object):
def error(self, msg): def error(self, msg):
print(msg) print(msg)
creds = botlib.Creds(config['url'], config['username'], config['password'])
bot = botlib.Bot(creds)
def progress_hook(d): def progress_hook(d):
if d['status'] == 'started': if d['status'] == 'started':
print('Started downloading') print('Started downloading')
elif d['status'] == 'finished': elif d['status'] == 'finished':
print('Done downloading') print('Done downloading')
ydl_opts = {
'format': config['yt_dlp_format'],
'outtmpl': config['download_dir'] + '/%(title)s.%(ext)s',
'restrictfilenames': True,
'logger': MyLogger(),
'progress_hooks': [progress_hook],
}
def get_yt_vid_id(word): def get_yt_vid_id(word):
if search('www.youtube.com/watch', word): if search('www.youtube.com/watch', word):
return word return word
@ -58,4 +46,14 @@ async def youtube(room, message):
except Exception as e: except Exception as e:
await bot.api.send_text_message(room.room_id, str(e)) await bot.api.send_text_message(room.room_id, str(e))
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': MyLogger(),
'progress_hooks': [progress_hook],
}
bot.run() bot.run()