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

New feature

Browse Source
This commit is contained in:
azlux
2018-05-17 23:55:42 +02:00
parent 54068a151e
commit ad378a586c
5 changed files with 193 additions and 29 deletions
Download Patch File Download Diff File Expand all files Collapse all files

119
mumbleBot.py
View File

@ -1,6 +1,8 @@
#!/usr/bin/python3
import threading
from __future__ import unicode_literals
import re
import threading
import time
import sys
import signal
@ -13,6 +15,9 @@ from os import listdir
import pymumble.pymumble_py3 as pymumble
import interface
import variables as var
import hashlib
import youtube_dl
import media
class MumbleBot:
@ -34,7 +39,24 @@ class MumbleBot:
self.channel = args.channel
var.current_music = None
######
## Format of the Playlist :
## [("<type>","<path>")]
## [("<radio>","<luna>"), ("<youtube>","<url>")]
## types : file, radio, url
######
######
## Format of the current_music variable
# len(var.current_music) = 4
# var.current_music[0] = <Type>
# var.current_music[1] = <url> if url of radio
# var.current_music[2] = <title>
# var.current_music[3] = <path> if url or file
var.playlist = []
var.user = args.user
var.music_folder = self.config.get('bot', 'music_folder')
var.is_proxified = self.config.getboolean("bot", "is_proxified")
@ -44,9 +66,9 @@ class MumbleBot:
interface.init_proxy()
t = threading.Thread(target=start_web_interface)
t.daemon = True
t.start()
# t = threading.Thread(target=start_web_interface)
# t.daemon = True
# t.start()
self.mumble = pymumble.Mumble(args.host, user=args.user, port=args.port, password=args.password,
debug=self.config.getboolean('debug', 'mumbleConnection'))
@ -91,10 +113,19 @@ class MumbleBot:
if "/" in parameter:
self.mumble.users[text.actor].send_message(self.config.get('strings', 'bad_file'))
elif os.path.isfile(path):
self.launch_play_file(path)
var.playlist.append(["file", path])
else:
self.mumble.users[text.actor].send_message(self.config.get('strings', 'bad_file'))
elif command == self.config.get('command', 'play_url') and parameter:
var.playlist.append(["url", parameter])
elif command == self.config.get('command', 'play_radio') and parameter:
var.playlist.append(["radio", parameter])
elif command == self.config.get('command', 'help'):
self.send_msg_channel(self.config.get('strings', 'help'))
elif command == self.config.get('command', 'stop'):
self.stop()
@ -123,17 +154,17 @@ class MumbleBot:
elif command == self.config.get('command', 'current_music'):
if var.current_music is not None:
self.send_msg_channel(var.current_music)
if var.current_music[0] == "radio":
self.send_msg_channel(media.get_radio_title(var.current_music[1]) + " sur " + var.current_music[2])
else:
self.send_msg_channel(var.current_music[2] + "<br />" + var.current_music[1])
else:
self.mumble.users[text.actor].send_message(self.config.get('strings', 'not_playing'))
elif command == self.config.get('command', 'list'):
folder_path = self.config.get('bot', 'music_folder')
files = [f for f in listdir(folder_path) if os.path.isfile(os.path.join(folder_path, f))]
if files:
self.mumble.users[text.actor].send_message('<br>'.join(files))
else:
self.mumble.users[text.actor].send_message(self.config.get('strings', 'no_file'))
elif command == self.config.get('command', 'next'):
var.current_music = var.playlist[0]
var.playlist.pop(0)
self.launch_next()
else:
self.mumble.users[text.actor].send_message(self.config.get('strings', 'bad_command'))
@ -145,21 +176,69 @@ class MumbleBot:
else:
return False
def launch_play_file(self, path=None):
if not path:
path = self.config.get('bot', 'music_folder') + var.current_music
def launch_next(self):
path = ""
title = ""
if var.current_music[0] == "url":
regex = re.compile("<a href=\"(.*?)\"")
m = regex.match(var.current_music[1])
url = m.group(1)
path, title = self.download_music(url)
var.current_music[1] = url
elif var.current_music[0] == "file":
path = self.config.get('bot', 'music_folder') + var.current_music[1]
title = var.current_music[1]
elif var.current_music[0] == "radio":
regex = re.compile("<a href=\"(.*?)\"")
m = regex.match(var.current_music[1])
url = m.group(1)
var.current_music[1] = url
path = url
title = media.get_radio_server_description(url)
if self.config.getboolean('debug', 'ffmpeg'):
ffmpeg_debug = "debug"
else:
ffmpeg_debug = "warning"
command = ["ffmpeg", '-v', ffmpeg_debug, '-nostdin', '-i', path, '-ac', '1', '-f', 's16le', '-ar', '48000', '-']
self.thread = sp.Popen(command, stdout=sp.PIPE, bufsize=480)
var.current_music = path
var.current_music.append(title)
var.current_music.append(path)
def download_music(self, url):
url_hash = hashlib.md5(url.encode()).hexdigest()
path = self.config.get('bot', 'tmp_folder') + url_hash + ".mp3"
ydl_opts = {
'format': 'bestaudio/best',
'outtmpl': path,
'noplaylist': True,
'postprocessors': [{
'key': 'FFmpegExtractAudio',
'preferredcodec': 'mp3',
'preferredquality': '192',
}]
}
video_title = ""
with youtube_dl.YoutubeDL(ydl_opts) as ydl:
for i in range(2):
try:
info_dict = ydl.extract_info(url, download=False)
video_title = info_dict['title']
ydl.download([url])
except youtube_dl.utils.DownloadError:
pass
else:
break
return path, video_title
def loop(self):
while not self.exit:
raw_music = ""
while not self.exit and self.mumble.isAlive():
while self.mumble.sound_output.get_buffer_size() > 0.5:
while self.mumble.sound_output.get_buffer_size() > 0.5 and not self.exit:
time.sleep(0.01)
if self.thread:
raw_music = self.thread.stdout.read(480)
@ -173,7 +252,7 @@ class MumbleBot:
if (self.thread is None or not raw_music) and len(var.playlist) != 0:
var.current_music = var.playlist[0]
var.playlist.pop(0)
self.launch_play_file()
self.launch_next()
while self.mumble.sound_output.get_buffer_size() > 0:
time.sleep(0.01)