storm/bragi
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

fix for #5

Browse Source 
check if thumbnail exist
This commit is contained in:
azlux 2018-06-06 00:06:34 +02:00
parent 4f1372697b
commit 558d1e5bd3
1 changed files with 12 additions and 9 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

21
mumbleBot.py
View File

@ -207,10 +207,10 @@ class MumbleBot:
folder_path = var.config.get('bot', 'music_folder') folder_path = var.config.get('bot', 'music_folder')
files = util.get_recursive_filelist_sorted(folder_path) files = util.get_recursive_filelist_sorted(folder_path)
if files : if files:
self.mumble.users[text.actor].send_message('<br>'.join(files)) self.mumble.users[text.actor].send_message('<br>'.join(files))
else : else:
self.mumble.users[text.actor].send_message(var.config.get('strings', 'no_file')) self.mumble.users[text.actor].send_message(var.config.get('strings', 'no_file'))
elif command == var.config.get('command', 'queue'): elif command == var.config.get('command', 'queue'):
if len(var.playlist) == 0: if len(var.playlist) == 0:
@ -252,12 +252,15 @@ class MumbleBot:
path, title = self.download_music(url) path, title = self.download_music(url)
var.current_music[1] = url var.current_music[1] = url
im = Image.open(var.config.get('bot', 'tmp_folder') + hashlib.md5(url.encode()).hexdigest() + '.jpg') path_thumbnail = var.config.get('bot', 'tmp_folder') + hashlib.md5(url.encode()).hexdigest() + '.jpg'
im.thumbnail((100, 100), Image.ANTIALIAS) thumbnail_html = ""
buffer = BytesIO() if os.path.isfile(path_thumbnail):
im.save(buffer, format="JPEG") im = Image.open(path_thumbnail)
thumbnail_base64 = base64.b64encode(buffer.getvalue()) im.thumbnail((100, 100), Image.ANTIALIAS)
thumbnail_html = '<img - src="data:image/PNG;base64,' + thumbnail_base64.decode() + '"/>' buffer = BytesIO()
im.save(buffer, format="JPEG")
thumbnail_base64 = base64.b64encode(buffer.getvalue())
thumbnail_html = '<img - src="data:image/PNG;base64,' + thumbnail_base64.decode() + '"/>'
logging.debug(thumbnail_html) logging.debug(thumbnail_html)
self.send_msg_channel(var.config.get('strings', 'now_playing') % (title, thumbnail_html)) self.send_msg_channel(var.config.get('strings', 'now_playing') % (title, thumbnail_html))