Send query result to channel
This commit is contained in:
elpatron68
221
mumbleBot.py
221
mumbleBot.py
@ -27,6 +27,7 @@ import media.file
|
||||
import media.playlist
|
||||
import media.radio
|
||||
import media.system
|
||||
from radiobrowser import getstations_byname
|
||||
|
||||
"""
|
||||
FORMAT OF A MUSIC INTO THE PLAYLIST
|
||||
@ -70,13 +71,16 @@ class MumbleBot:
|
||||
# Set specific format for the log
|
||||
FORMAT = '%(asctime)s: %(message)s'
|
||||
if args.verbose:
|
||||
logging.basicConfig(format=FORMAT, level=logging.DEBUG, datefmt='%Y-%m-%d %H:%M:%S')
|
||||
logging.basicConfig(
|
||||
format=FORMAT, level=logging.DEBUG, datefmt='%Y-%m-%d %H:%M:%S')
|
||||
logging.debug("Starting in DEBUG loglevel")
|
||||
elif args.quiet:
|
||||
logging.basicConfig(format=FORMAT, level=logging.ERROR, datefmt='%Y-%m-%d %H:%M:%S')
|
||||
logging.basicConfig(
|
||||
format=FORMAT, level=logging.ERROR, datefmt='%Y-%m-%d %H:%M:%S')
|
||||
logging.error("Starting in ERROR loglevel")
|
||||
else:
|
||||
logging.basicConfig(format=FORMAT, level=logging.INFO, datefmt='%Y-%m-%d %H:%M:%S')
|
||||
logging.basicConfig(
|
||||
format=FORMAT, level=logging.INFO, datefmt='%Y-%m-%d %H:%M:%S')
|
||||
logging.info("Starting in INFO loglevel")
|
||||
|
||||
# the playlist is... a list (Surprise !!)
|
||||
@ -84,7 +88,8 @@ class MumbleBot:
|
||||
|
||||
var.user = args.user
|
||||
var.music_folder = var.config.get('bot', 'music_folder')
|
||||
var.is_proxified = var.config.getboolean("webinterface", "is_web_proxified")
|
||||
var.is_proxified = var.config.getboolean(
|
||||
"webinterface", "is_web_proxified")
|
||||
self.exit = False
|
||||
self.nb_exit = 0
|
||||
self.thread = None
|
||||
@ -94,7 +99,8 @@ class MumbleBot:
|
||||
wi_addr = var.config.get("webinterface", "listening_addr")
|
||||
wi_port = var.config.getint("webinterface", "listening_port")
|
||||
interface.init_proxy()
|
||||
tt = threading.Thread(target=start_web_interface, args=(wi_addr, wi_port))
|
||||
tt = threading.Thread(
|
||||
target=start_web_interface, args=(wi_addr, wi_port))
|
||||
tt.daemon = True
|
||||
tt.start()
|
||||
|
||||
@ -131,7 +137,8 @@ class MumbleBot:
|
||||
|
||||
self.mumble = pymumble.Mumble(host, user=self.username, port=port, password=password, tokens=tokens,
|
||||
debug=var.config.getboolean('debug', 'mumbleConnection'), certfile=certificate)
|
||||
self.mumble.callbacks.set_callback("text_received", self.message_received)
|
||||
self.mumble.callbacks.set_callback(
|
||||
"text_received", self.message_received)
|
||||
|
||||
self.mumble.set_codec_profile("audio")
|
||||
self.mumble.start() # start the mumble thread
|
||||
@ -146,7 +153,8 @@ class MumbleBot:
|
||||
|
||||
# Set the CTRL+C shortcut
|
||||
def ctrl_caught(self, signal, frame):
|
||||
logging.info("\nSIGINT caught, quitting, {} more to kill".format(2 - self.nb_exit))
|
||||
logging.info(
|
||||
"\nSIGINT caught, quitting, {} more to kill".format(2 - self.nb_exit))
|
||||
self.exit = True
|
||||
self.stop()
|
||||
if self.nb_exit > 1:
|
||||
@ -181,16 +189,19 @@ class MumbleBot:
|
||||
logging.info(command + ' - ' + parameter + ' by ' + user)
|
||||
|
||||
if command == var.config.get('command', 'joinme'):
|
||||
self.mumble.users.myself.move_in(self.mumble.users[text.actor]['channel_id'], token=parameter)
|
||||
self.mumble.users.myself.move_in(
|
||||
self.mumble.users[text.actor]['channel_id'], token=parameter)
|
||||
return
|
||||
|
||||
# Anti stupid guy function
|
||||
if not self.is_admin(user) and not var.config.getboolean('bot', 'allow_other_channel_message') and self.mumble.users[text.actor]['channel_id'] != self.mumble.users.myself['channel_id']:
|
||||
self.mumble.users[text.actor].send_text_message(var.config.get('strings', 'not_in_my_channel'))
|
||||
self.mumble.users[text.actor].send_text_message(
|
||||
var.config.get('strings', 'not_in_my_channel'))
|
||||
return
|
||||
|
||||
if not self.is_admin(user) and not var.config.getboolean('bot', 'allow_private_message') and text.session:
|
||||
self.mumble.users[text.actor].send_text_message(var.config.get('strings', 'pm_not_allowed'))
|
||||
self.mumble.users[text.actor].send_text_message(
|
||||
var.config.get('strings', 'pm_not_allowed'))
|
||||
return
|
||||
|
||||
###
|
||||
@ -198,49 +209,61 @@ class MumbleBot:
|
||||
###
|
||||
for i in var.db.items("user_ban"):
|
||||
if user.lower() == i[0]:
|
||||
self.mumble.users[text.actor].send_text_message(var.config.get('strings', 'user_ban'))
|
||||
self.mumble.users[text.actor].send_text_message(
|
||||
var.config.get('strings', 'user_ban'))
|
||||
return
|
||||
|
||||
if command == var.config.get('command', 'user_ban'):
|
||||
if self.is_admin(user):
|
||||
if parameter:
|
||||
self.mumble.users[text.actor].send_text_message(util.user_ban(parameter))
|
||||
self.mumble.users[text.actor].send_text_message(
|
||||
util.user_ban(parameter))
|
||||
else:
|
||||
self.mumble.users[text.actor].send_text_message(util.get_user_ban())
|
||||
self.mumble.users[text.actor].send_text_message(
|
||||
util.get_user_ban())
|
||||
else:
|
||||
self.mumble.users[text.actor].send_text_message(var.config.get('strings', 'not_admin'))
|
||||
self.mumble.users[text.actor].send_text_message(
|
||||
var.config.get('strings', 'not_admin'))
|
||||
return
|
||||
|
||||
elif command == var.config.get('command', 'user_unban'):
|
||||
if self.is_admin(user):
|
||||
if parameter:
|
||||
self.mumble.users[text.actor].send_text_message(util.user_unban(parameter))
|
||||
self.mumble.users[text.actor].send_text_message(
|
||||
util.user_unban(parameter))
|
||||
else:
|
||||
self.mumble.users[text.actor].send_text_message(var.config.get('strings', 'not_admin'))
|
||||
self.mumble.users[text.actor].send_text_message(
|
||||
var.config.get('strings', 'not_admin'))
|
||||
return
|
||||
|
||||
elif command == var.config.get('command', 'url_ban'):
|
||||
if self.is_admin(user):
|
||||
if parameter:
|
||||
self.mumble.users[text.actor].send_text_message(util.url_ban(self.get_url_from_input(parameter)))
|
||||
self.mumble.users[text.actor].send_text_message(
|
||||
util.url_ban(self.get_url_from_input(parameter)))
|
||||
else:
|
||||
self.mumble.users[text.actor].send_text_message(util.get_url_ban())
|
||||
self.mumble.users[text.actor].send_text_message(
|
||||
util.get_url_ban())
|
||||
else:
|
||||
self.mumble.users[text.actor].send_text_message(var.config.get('strings', 'not_admin'))
|
||||
self.mumble.users[text.actor].send_text_message(
|
||||
var.config.get('strings', 'not_admin'))
|
||||
return
|
||||
|
||||
elif command == var.config.get('command', 'url_unban'):
|
||||
if self.is_admin(user):
|
||||
if parameter:
|
||||
self.mumble.users[text.actor].send_text_message(util.url_unban(self.get_url_from_input(parameter)))
|
||||
self.mumble.users[text.actor].send_text_message(
|
||||
util.url_unban(self.get_url_from_input(parameter)))
|
||||
else:
|
||||
self.mumble.users[text.actor].send_text_message(var.config.get('strings', 'not_admin'))
|
||||
self.mumble.users[text.actor].send_text_message(
|
||||
var.config.get('strings', 'not_admin'))
|
||||
return
|
||||
|
||||
if parameter:
|
||||
for i in var.db.items("url_ban"):
|
||||
if self.get_url_from_input(parameter.lower()) == i[0]:
|
||||
self.mumble.users[text.actor].send_text_message(var.config.get('strings', 'url_ban'))
|
||||
self.mumble.users[text.actor].send_text_message(
|
||||
var.config.get('strings', 'url_ban'))
|
||||
return
|
||||
|
||||
###
|
||||
@ -259,16 +282,19 @@ class MumbleBot:
|
||||
var.playlist.append(music)
|
||||
else:
|
||||
# try to do a partial match
|
||||
matches = [file for file in util.get_recursive_filelist_sorted(music_folder) if parameter.lower() in file.lower()]
|
||||
matches = [file for file in util.get_recursive_filelist_sorted(
|
||||
music_folder) if parameter.lower() in file.lower()]
|
||||
if len(matches) == 0:
|
||||
self.send_msg(var.config.get('strings', 'no_file'), text)
|
||||
self.send_msg(var.config.get(
|
||||
'strings', 'no_file'), text)
|
||||
elif len(matches) == 1:
|
||||
music = {'type': 'file',
|
||||
'path': matches[0],
|
||||
'user': user}
|
||||
var.playlist.append(music)
|
||||
else:
|
||||
msg = var.config.get('strings', 'multiple_matches') + '<br />'
|
||||
msg = var.config.get(
|
||||
'strings', 'multiple_matches') + '<br />'
|
||||
msg += '<br />'.join(matches)
|
||||
self.send_msg(msg, text)
|
||||
else:
|
||||
@ -277,7 +303,8 @@ class MumbleBot:
|
||||
|
||||
elif command == var.config.get('command', 'play_url') and parameter:
|
||||
music = {'type': 'url',
|
||||
'url': self.get_url_from_input(parameter), # grab the real URL
|
||||
# grab the real URL
|
||||
'url': self.get_url_from_input(parameter),
|
||||
'user': user,
|
||||
'ready': 'validation'}
|
||||
var.playlist.append(music)
|
||||
@ -285,18 +312,21 @@ class MumbleBot:
|
||||
if media.url.get_url_info():
|
||||
if var.playlist[-1]['duration'] > var.config.getint('bot', 'max_track_duration'):
|
||||
var.playlist.pop()
|
||||
self.send_msg(var.config.get('strings', 'too_long'), text)
|
||||
self.send_msg(var.config.get(
|
||||
'strings', 'too_long'), text)
|
||||
else:
|
||||
for i in var.db.options("url_ban"):
|
||||
if var.playlist[-1]['url'] == i:
|
||||
self.mumble.users[text.actor].send_text_message(var.config.get('strings', 'url_ban'))
|
||||
self.mumble.users[text.actor].send_text_message(
|
||||
var.config.get('strings', 'url_ban'))
|
||||
var.playlist.pop()
|
||||
return
|
||||
var.playlist[-1]['ready'] = "no"
|
||||
self.async_download_next()
|
||||
else:
|
||||
var.playlist.pop()
|
||||
self.send_msg(var.config.get('strings', 'unable_download'), text)
|
||||
self.send_msg(var.config.get(
|
||||
'strings', 'unable_download'), text)
|
||||
|
||||
elif command == var.config.get('command', 'play_playlist') and parameter:
|
||||
offset = 1 # if you want to start the playlist at a specific index
|
||||
@ -310,7 +340,8 @@ class MumbleBot:
|
||||
elif command == var.config.get('command', 'play_radio'):
|
||||
if not parameter:
|
||||
all_radio = var.config.items('radio')
|
||||
msg = var.config.get('strings', 'preconfigurated_radio') + " :"
|
||||
msg = var.config.get(
|
||||
'strings', 'preconfigurated_radio') + " :"
|
||||
for i in all_radio:
|
||||
comment = ""
|
||||
if len(i[1].split(maxsplit=1)) == 2:
|
||||
@ -336,11 +367,16 @@ class MumbleBot:
|
||||
logging.info('rbquery without parameter')
|
||||
else:
|
||||
logging.info('Found query parameter: ' + parameter)
|
||||
stations = getstations_byname(parameter)
|
||||
for s in stations:
|
||||
msg += "<br />" + s
|
||||
self.send_msg(msg, text)
|
||||
|
||||
elif command == var.config.get('command', 'help'):
|
||||
self.send_msg(var.config.get('strings', 'help'), text)
|
||||
if self.is_admin(user):
|
||||
self.send_msg(var.config.get('strings', 'admin_help'), text)
|
||||
self.send_msg(var.config.get(
|
||||
'strings', 'admin_help'), text)
|
||||
|
||||
elif command == var.config.get('command', 'stop'):
|
||||
self.stop()
|
||||
@ -350,16 +386,19 @@ class MumbleBot:
|
||||
self.stop()
|
||||
self.exit = True
|
||||
else:
|
||||
self.mumble.users[text.actor].send_text_message(var.config.get('strings', 'not_admin'))
|
||||
self.mumble.users[text.actor].send_text_message(
|
||||
var.config.get('strings', 'not_admin'))
|
||||
|
||||
elif command == var.config.get('command', 'update'):
|
||||
if self.is_admin(user):
|
||||
self.mumble.users[text.actor].send_text_message("Starting the update")
|
||||
self.mumble.users[text.actor].send_text_message(
|
||||
"Starting the update")
|
||||
# Need to be improved
|
||||
msg = util.update(version)
|
||||
self.mumble.users[text.actor].send_text_message(msg)
|
||||
else:
|
||||
self.mumble.users[text.actor].send_text_message(var.config.get('strings', 'not_admin'))
|
||||
self.mumble.users[text.actor].send_text_message(
|
||||
var.config.get('strings', 'not_admin'))
|
||||
|
||||
elif command == var.config.get('command', 'stop_and_getout'):
|
||||
self.stop()
|
||||
@ -374,14 +413,16 @@ class MumbleBot:
|
||||
int(self.volume * 100), self.mumble.users[text.actor]['name']), text)
|
||||
var.db.set('bot', 'volume', str(self.volume))
|
||||
else:
|
||||
self.send_msg(var.config.get('strings', 'current_volume') % int(self.volume * 100), text)
|
||||
self.send_msg(var.config.get(
|
||||
'strings', 'current_volume') % int(self.volume * 100), text)
|
||||
|
||||
elif command == var.config.get('command', 'current_music'):
|
||||
if len(var.playlist) > 0:
|
||||
source = var.playlist[0]["type"]
|
||||
if source == "radio":
|
||||
reply = "[radio] {title} on {url} by {user}".format(
|
||||
title=media.radio.get_radio_title(var.playlist[0]["url"]),
|
||||
title=media.radio.get_radio_title(
|
||||
var.playlist[0]["url"]),
|
||||
url=var.playlist[0]["title"],
|
||||
user=var.playlist[0]["user"]
|
||||
)
|
||||
@ -411,19 +452,23 @@ class MumbleBot:
|
||||
self.send_msg(reply, text)
|
||||
|
||||
elif command == var.config.get('command', 'skip'):
|
||||
if parameter is not None and parameter.isdigit() and int(parameter) > 0: # Allow to remove specific music into the queue with a number
|
||||
# Allow to remove specific music into the queue with a number
|
||||
if parameter is not None and parameter.isdigit() and int(parameter) > 0:
|
||||
if int(parameter) < len(var.playlist):
|
||||
removed = var.playlist.pop(int(parameter))
|
||||
|
||||
# the Title isn't here if the music wasn't downloaded
|
||||
self.send_msg(var.config.get('strings', 'removing_item') % (removed['title'] if 'title' in removed else removed['url']), text)
|
||||
self.send_msg(var.config.get('strings', 'removing_item') % (
|
||||
removed['title'] if 'title' in removed else removed['url']), text)
|
||||
else:
|
||||
self.send_msg(var.config.get('strings', 'no_possible'), text)
|
||||
self.send_msg(var.config.get(
|
||||
'strings', 'no_possible'), text)
|
||||
elif self.next(): # Is no number send, just skip the current music
|
||||
self.launch_music()
|
||||
self.async_download_next()
|
||||
else:
|
||||
self.send_msg(var.config.get('strings', 'queue_empty'), text)
|
||||
self.send_msg(var.config.get(
|
||||
'strings', 'queue_empty'), text)
|
||||
self.stop()
|
||||
|
||||
elif command == var.config.get('command', 'list'):
|
||||
@ -439,10 +484,12 @@ class MumbleBot:
|
||||
if len(var.playlist) <= 1:
|
||||
msg = var.config.get('strings', 'queue_empty')
|
||||
else:
|
||||
msg = var.config.get('strings', 'queue_contents') + '<br />'
|
||||
msg = var.config.get(
|
||||
'strings', 'queue_contents') + '<br />'
|
||||
i = 1
|
||||
for value in var.playlist[1:]:
|
||||
msg += '[{}] ({}) {}<br />'.format(i, value['type'], value['title'] if 'title' in value else value['url'])
|
||||
msg += '[{}] ({}) {}<br />'.format(i, value['type'],
|
||||
value['title'] if 'title' in value else value['url'])
|
||||
i += 1
|
||||
|
||||
self.send_msg(msg, text)
|
||||
@ -451,7 +498,8 @@ class MumbleBot:
|
||||
var.playlist.append(var.playlist[0])
|
||||
|
||||
else:
|
||||
self.mumble.users[text.actor].send_text_message(var.config.get('strings', 'bad_command'))
|
||||
self.mumble.users[text.actor].send_text_message(
|
||||
var.config.get('strings', 'bad_command'))
|
||||
|
||||
@staticmethod
|
||||
def is_admin(user):
|
||||
@ -478,7 +526,8 @@ class MumbleBot:
|
||||
logging.debug("launch_music asked" + str(var.playlist[0]))
|
||||
if var.playlist[0]["type"] == "url":
|
||||
# Delete older music is the tmp folder is too big
|
||||
media.system.clear_tmp_folder(var.config.get('bot', 'tmp_folder'), var.config.getint('bot', 'tmp_folder_max_size'))
|
||||
media.system.clear_tmp_folder(var.config.get(
|
||||
'bot', 'tmp_folder'), var.config.getint('bot', 'tmp_folder_max_size'))
|
||||
|
||||
# Check if the music is ready to be played
|
||||
if var.playlist[0]["ready"] == "downloading":
|
||||
@ -493,9 +542,11 @@ class MumbleBot:
|
||||
audio = EasyID3(uri)
|
||||
title = ""
|
||||
if audio["title"]:
|
||||
title = audio["title"][0] # take the title from the file tag
|
||||
# take the title from the file tag
|
||||
title = audio["title"][0]
|
||||
|
||||
path_thumbnail = var.playlist[0]['path'][:-4] + '.jpg' # Remove .mp3 and add .jpg
|
||||
# Remove .mp3 and add .jpg
|
||||
path_thumbnail = var.playlist[0]['path'][:-4] + '.jpg'
|
||||
thumbnail_html = ""
|
||||
if os.path.isfile(path_thumbnail):
|
||||
# Create the image message
|
||||
@ -504,33 +555,39 @@ class MumbleBot:
|
||||
buffer = BytesIO()
|
||||
im.save(buffer, format="JPEG")
|
||||
thumbnail_base64 = base64.b64encode(buffer.getvalue())
|
||||
thumbnail_html = '<img - src="data:image/PNG;base64,' + thumbnail_base64.decode() + '"/>'
|
||||
thumbnail_html = '<img - src="data:image/PNG;base64,' + \
|
||||
thumbnail_base64.decode() + '"/>'
|
||||
|
||||
logging.debug("Thunbail data " + thumbnail_html)
|
||||
if var.config.getboolean('bot', 'announce_current_music'):
|
||||
self.send_msg(var.config.get('strings', 'now_playing') % (title, thumbnail_html))
|
||||
self.send_msg(var.config.get(
|
||||
'strings', 'now_playing') % (title, thumbnail_html))
|
||||
else:
|
||||
logging.error("Error with the path during launch_music")
|
||||
pass
|
||||
|
||||
elif var.playlist[0]["type"] == "file":
|
||||
uri = var.config.get('bot', 'music_folder') + var.playlist[0]["path"]
|
||||
uri = var.config.get('bot', 'music_folder') + \
|
||||
var.playlist[0]["path"]
|
||||
|
||||
elif var.playlist[0]["type"] == "radio":
|
||||
uri = var.playlist[0]["url"]
|
||||
title = media.radio.get_radio_server_description(uri)
|
||||
var.playlist[0]["title"] = title
|
||||
if var.config.getboolean('bot', 'announce_current_music'):
|
||||
self.send_msg(var.config.get('strings', 'now_playing') % (title, "URL : " + uri))
|
||||
self.send_msg(var.config.get('strings', 'now_playing') %
|
||||
(title, "URL : " + uri))
|
||||
|
||||
if var.config.getboolean('debug', 'ffmpeg'):
|
||||
ffmpeg_debug = "debug"
|
||||
else:
|
||||
ffmpeg_debug = "warning"
|
||||
|
||||
command = ("ffmpeg", '-v', ffmpeg_debug, '-nostdin', '-i', uri, '-ac', '1', '-f', 's16le', '-ar', '48000', '-')
|
||||
command = ("ffmpeg", '-v', ffmpeg_debug, '-nostdin', '-i',
|
||||
uri, '-ac', '1', '-f', 's16le', '-ar', '48000', '-')
|
||||
logging.info("FFmpeg command : " + " ".join(command))
|
||||
self.thread = sp.Popen(command, stdout=sp.PIPE, bufsize=480) # The ffmpeg process is a thread
|
||||
# The ffmpeg process is a thread
|
||||
self.thread = sp.Popen(command, stdout=sp.PIPE, bufsize=480)
|
||||
self.is_playing = True
|
||||
|
||||
def download_music(self, index):
|
||||
@ -539,7 +596,8 @@ class MumbleBot:
|
||||
if var.playlist[index]['duration'] > var.config.getint('bot', 'max_track_duration'):
|
||||
# Check the length, useful in case of playlist, it wasn't checked before)
|
||||
var.playlist.pop()
|
||||
logging.info("the music " + var.playlist[index]["url"] + " has a duration of " + var.playlist[index]['duration'] + "s -- too long")
|
||||
logging.info(
|
||||
"the music " + var.playlist[index]["url"] + " has a duration of " + var.playlist[index]['duration'] + "s -- too long")
|
||||
self.send_msg(var.config.get('strings', 'too_long'))
|
||||
return
|
||||
else:
|
||||
@ -580,9 +638,11 @@ class MumbleBot:
|
||||
'preferredquality': '192'},
|
||||
{'key': 'FFmpegMetadata'}]
|
||||
}
|
||||
self.send_msg(var.config.get('strings', "download_in_progress") % var.playlist[index]['title'])
|
||||
self.send_msg(var.config.get(
|
||||
'strings', "download_in_progress") % var.playlist[index]['title'])
|
||||
|
||||
logging.info("Information before start downloading :" + str(var.playlist[index]))
|
||||
logging.info("Information before start downloading :" +
|
||||
str(var.playlist[index]))
|
||||
with youtube_dl.YoutubeDL(ydl_opts) as ydl:
|
||||
for i in range(2): # Always try 2 times
|
||||
try:
|
||||
@ -600,7 +660,8 @@ class MumbleBot:
|
||||
# Do nothing in case the next music is already downloaded
|
||||
logging.info("Async download next asked")
|
||||
if len(var.playlist) > 1 and var.playlist[1]['type'] == 'url' and var.playlist[1]['ready'] in ["no", "validation"]:
|
||||
th = threading.Thread(target=self.download_music, kwargs={'index': 1})
|
||||
th = threading.Thread(
|
||||
target=self.download_music, kwargs={'index': 1})
|
||||
else:
|
||||
return
|
||||
logging.info("Start downloading next in thread")
|
||||
@ -632,7 +693,8 @@ class MumbleBot:
|
||||
raw_music = self.thread.stdout.read(480)
|
||||
if raw_music:
|
||||
# Adjust the volume and send it to mumble
|
||||
self.mumble.sound_output.add_sound(audioop.mul(raw_music, 2, self.volume))
|
||||
self.mumble.sound_output.add_sound(
|
||||
audioop.mul(raw_music, 2, self.volume))
|
||||
else:
|
||||
time.sleep(0.1)
|
||||
else:
|
||||
@ -686,30 +748,44 @@ def start_web_interface(addr, port):
|
||||
|
||||
|
||||
if __name__ == '__main__':
|
||||
parser = argparse.ArgumentParser(description='Bot for playing music on Mumble')
|
||||
parser = argparse.ArgumentParser(
|
||||
description='Bot for playing music on Mumble')
|
||||
|
||||
# General arguments
|
||||
parser.add_argument("--config", dest='config', type=str, default='configuration.ini', help='Load configuration from this file. Default: configuration.ini')
|
||||
parser.add_argument("--db", dest='db', type=str, default='db.ini', help='database file. Default db.ini')
|
||||
parser.add_argument("--config", dest='config', type=str, default='configuration.ini',
|
||||
help='Load configuration from this file. Default: configuration.ini')
|
||||
parser.add_argument("--db", dest='db', type=str,
|
||||
default='db.ini', help='database file. Default db.ini')
|
||||
|
||||
parser.add_argument("-q", "--quiet", dest="quiet", action="store_true", help="Only Error logs")
|
||||
parser.add_argument("-v", "--verbose", dest="verbose", action="store_true", help="Show debug log")
|
||||
parser.add_argument("-q", "--quiet", dest="quiet",
|
||||
action="store_true", help="Only Error logs")
|
||||
parser.add_argument("-v", "--verbose", dest="verbose",
|
||||
action="store_true", help="Show debug log")
|
||||
|
||||
# Mumble arguments
|
||||
parser.add_argument("-s", "--server", dest="host", type=str, help="Hostname of the Mumble server")
|
||||
parser.add_argument("-u", "--user", dest="user", type=str, help="Username for the bot")
|
||||
parser.add_argument("-P", "--password", dest="password", type=str, help="Server password, if required")
|
||||
parser.add_argument("-T", "--tokens", dest="tokens", type=str, help="Server tokens, if required")
|
||||
parser.add_argument("-p", "--port", dest="port", type=int, help="Port for the Mumble server")
|
||||
parser.add_argument("-c", "--channel", dest="channel", type=str, help="Default channel for the bot")
|
||||
parser.add_argument("-C", "--cert", dest="certificate", type=str, default=None, help="Certificate file")
|
||||
parser.add_argument("-s", "--server", dest="host",
|
||||
type=str, help="Hostname of the Mumble server")
|
||||
parser.add_argument("-u", "--user", dest="user",
|
||||
type=str, help="Username for the bot")
|
||||
parser.add_argument("-P", "--password", dest="password",
|
||||
type=str, help="Server password, if required")
|
||||
parser.add_argument("-T", "--tokens", dest="tokens",
|
||||
type=str, help="Server tokens, if required")
|
||||
parser.add_argument("-p", "--port", dest="port",
|
||||
type=int, help="Port for the Mumble server")
|
||||
parser.add_argument("-c", "--channel", dest="channel",
|
||||
type=str, help="Default channel for the bot")
|
||||
parser.add_argument("-C", "--cert", dest="certificate",
|
||||
type=str, default=None, help="Certificate file")
|
||||
|
||||
args = parser.parse_args()
|
||||
var.dbfile = args.db
|
||||
config = configparser.ConfigParser(interpolation=None, allow_no_value=True)
|
||||
parsed_configs = config.read(['configuration.default.ini', args.config], encoding='latin-1')
|
||||
parsed_configs = config.read(
|
||||
['configuration.default.ini', args.config], encoding='latin-1')
|
||||
|
||||
db = configparser.ConfigParser(interpolation=None, allow_no_value=True, delimiters='²')
|
||||
db = configparser.ConfigParser(
|
||||
interpolation=None, allow_no_value=True, delimiters='²')
|
||||
db.read(var.dbfile, encoding='latin-1')
|
||||
|
||||
if 'url_ban' not in db.sections():
|
||||
@ -720,7 +796,8 @@ if __name__ == '__main__':
|
||||
db.add_section('user_ban')
|
||||
|
||||
if len(parsed_configs) == 0:
|
||||
logging.error('Could not read configuration from file \"{}\"'.format(args.config), file=sys.stderr)
|
||||
logging.error('Could not read configuration from file \"{}\"'.format(
|
||||
args.config), file=sys.stderr)
|
||||
sys.exit()
|
||||
|
||||
var.config = config
|
||||
|
||||
Reference in New Issue
Block a user