# -*- coding: utf-8 -*- ################################ ######## Red - Discord bot ##### ################################ # made by Twentysix # # import discord import logging import time import datetime import requests import traceback import re import youtube_dl import os import asyncio import glob from os import path from random import choice, randint, shuffle import dataIO #IO settings, proverbs, etc import economy #Credits import youtubeparser from sys import modules help = """**Commands list:** !flip - Flip a coin !rps [rock or paper o scissors] - Play rock paper scissors !proverb !choose option1 or option2 or option3 (...) - Random choice !8 [question] - Ask 8 ball !sw - Start/stop the stopwatch !avatar [name or mention] - Shows user's avatar !trivia start - Start a trivia session !trivia stop - Stop a trivia session !twitch [stream] - Check if stream is online !twitchalert [stream] - Whenever the stream is online the bot will send an alert in the channel (admin only) !stoptwitchalert [stream] - Stop sending alerts about the specified stream in the channel (admin only) !roll [number] - Random number between 0 and [number] !gif [text] - GIF search !customcommands - Custom commands' list !addcom [command] [text] - Add a custom command !editcom [command] [text] - Edit a custom command !delcom [command] - Delete a custom command !audio help - Audio related commands !economy - Economy explanation, if available """ youtube_dl_options = { 'format': 'bestaudio/best', 'extractaudio': True, 'audioformat': "mp3", 'outtmpl': '%(id)s', 'noplaylist': True, 'nocheckcertificate': True, 'ignoreerrors': True, 'quiet': True, 'no_warnings': True, 'outtmpl': "cache/%(id)s"} audio_help = """ **General audio help commands:** !next or !skip - Next song !prev - Previous song !pause - Pause song !resume - Resume song !repeat or !replay - Replay current song !title or !song - Current song's title + link !youtube [link] - Play a youtube video in a voice channel !sing - Make Red sing !stop - Stop any voice channel activity !volume [0-1] - Sets the volume !downloadmode - Disables/enables download mode (admin only) **Playlist commands:** !play [playlist_name] - Play chosen playlist !playlists - Playlists' list !shuffle - Mix music list !addplaylist [name] [link] - Add a youtube playlist. Link format example: https://www.youtube.com/playlist?list=PLe8jmEHFkvsaDOOWcREvkgFoj6MD0pXXX !delplaylist [name] - Delete a youtube playlist. Limited to author and admins. !getplaylist - Receive the current playlist through DM. This also works with favorites. **Local commands:** !local [playlist_name] - Play chosen local playlist !locallist or !local or !locals - Local playlists' list **Favorites:** !addfavorite - Add song to your favorites !delfavorite - Remove song from your favorites !playfavorites - Play your favorites **You can submit your own playlist by doing the following:** 1) Make a txt file. Name must be only letters, numbers and underscores. It will be your playlist's name, so choose wisely. 2) One youtube link each line. 3) Send me the txt. If any line is incorrect I will reject it. 4) Listen to it with !play [playlist_name]! """ client = discord.Client() if not discord.opus.is_loaded(): discord.opus.load_opus('libopus-0.dll') @client.async_event async def on_message(message): await gameSwitcher.changeGame() if message.channel.is_private and message.attachments != []: await transferPlaylist(message) if not message.channel.is_private and message.author.id != client.user.id: if settings["FILTER"] and not isMemberAdmin(message): if await checkFilter(message) or await checkRegex(message): return False #exits without checking for commands if message.channel.id in shush_list and message.content == "!talk": await talk(message) if message.channel.id not in shush_list: if message.content == client.user.name.upper() or message.content == client.user.name.upper() + "?": await client.send_message(message.channel, "`" + choice(greetings_caps) + "`") elif message.content.lower() == client.user.name.lower() + "?": await client.send_message(message.channel, "`" + choice(greetings) + "`") elif message.content == client.user.mention + " ?" or message.content == client.user.mention + "?": await client.send_message(message.channel, "`" + choice(greetings) + "`") elif message.content == "!flip": await client.send_message(message.channel, "*flips a coin and... " + choice(["HEADS!*", "TAILS!*"])) elif message.content.startswith("!rps"): await rpsgame(message) elif message.content == "!proverb": await client.send_message(message.channel, "`" + choice(proverbs) + "`") elif message.content == "!help": await client.send_message(message.author, help) await client.send_message(message.channel, "{} `Check your DMs for the command list.`".format(message.author.mention)) elif message.content.startswith('!choose'): await randomchoice(message) elif message.content.startswith('!8 ') and message.content.endswith("?") and len(message.content) > 5: await client.send_message(message.channel, "{}: ".format(message.author.mention) + "`" + choice(ball) + "`") elif message.content.startswith('!roll'): await roll(message) elif message.content.startswith('!addcom'): await addcom(message) elif message.content.startswith('!editcom'): await editcom(message) elif message.content.startswith('!delcom'): await delcom(message) elif message.content == "!customcommands": await listCustomCommands(message) elif message.content.startswith('!sw'): await stopwatch(message) elif message.content.startswith('!id'): await client.send_message(message.channel, "{} `Your id is {}`".format(message.author.mention, message.author.id)) elif message.content.startswith('!twitchalert'): await addTwitchAlert(message) elif message.content.startswith('!stoptwitchalert'): await removeTwitchAlert(message) elif message.content.startswith('!twitch'): await twitchCheck(message) elif message.content.startswith('!image'): #image(message) pass elif message.content.startswith('!gif'): await gif(message) elif message.content.startswith('!uptime'): await uptime(message) elif message.content.startswith('!avatar'): await avatar(message) ################## music ####################### elif message.content == "!sing": await playPlaylist(message, sing=True) elif message.content.startswith('!youtube'): await playVideo(message) elif message.content.startswith('!play '): await playPlaylist(message) elif message.content.startswith('!local '): await playLocal(message) elif message.content == "!local" or message.content == "!locallist" or message.content == "!locals": await listLocal(message) await client.send_message(message.channel, "{} `Check your DMs for the local playlists list.`".format(message.author.mention)) elif message.content == "!stop": await leaveVoice() elif message.content == "!playlist" or message.content == "!playlists": await listPlaylists(message) await client.send_message(message.channel, "{} `Check your DMs for the playlists list.`".format(message.author.mention)) elif message.content == "!skip" or message.content == "!next": if currentPlaylist: currentPlaylist.nextSong(currentPlaylist.getNextSong()) elif message.content == "!prev" or message.content == "!previous": if currentPlaylist: currentPlaylist.nextSong(currentPlaylist.getPreviousSong()) elif message.content == "!repeat" or message.content == "!replay": if currentPlaylist: currentPlaylist.nextSong(currentPlaylist.current) elif message.content == "!pause": if currentPlaylist: currentPlaylist.pause() elif message.content == "!resume": if currentPlaylist: currentPlaylist.resume() elif message.content == "!shuffle": if currentPlaylist: currentPlaylist.shuffle() elif message.content == "!song" or message.content == "!title" : if currentPlaylist: await getSongTitle(message) elif message.content == "!audio help": await client.send_message(message.author, audio_help) await client.send_message(message.channel, "{} `Check your DMs for the audio help.`".format(message.author.mention)) elif message.content.startswith("!addplaylist"): await addPlaylist(message) elif message.content.startswith("!delplaylist"): await delPlaylist(message) elif message.content == "!addfavorite": await addToFavorites(message) elif message.content == "!delfavorite": await removeFromFavorites(message) elif message.content == "!playfavorites": await playFavorites(message) elif message.content == "!getplaylist": await sendPlaylist(message) elif message.content.startswith("!volume"): await setVolume(message) elif message.content == "!downloadmode": await downloadMode(message) ################################################ elif message.content == "!trivia start": if checkAuth("Trivia", message, settings): if not getTriviabyChannel(message.channel): #th = threading.Thread(target=controlTrivia, args=[message, True]) #th.start() #await controlTrivia(message, True) await client.send_message(message.channel, "`Trivia is currently out of service. Sorry.`") else: await client.send_message(message.channel, "`A trivia session is already ongoing in this channel.`") else: await client.send_message(message.channel, "`Trivia is currently admin-only.`") elif message.content == "!trivia stop": if checkAuth("Trivia", message, settings): await controlTrivia(message, False) else: await client.send_message(message.channel, "`Trivia is currently admin-only.`") elif message.content == "!trivia": pass ######## Admin commands ####################### elif message.content.startswith('!addwords'): await addBadWords(message) elif message.content.startswith('!removewords'): await removeBadWords(message) elif message.content.startswith('!addregex ') and len(message.content) > 11: await addRegex(message) elif message.content.startswith('!removeregex ') and len(message.content) > 14: await removeRegex(message) elif message.content == "!shutdown": await shutdown(message) elif message.content.startswith('!join'): await join(message) elif message.content == "!leaveserver": await leave(message) elif message.content == "!shush": await shush(message) elif message.content == "!talk": #prevents !talk custom command pass elif message.content == "!reload": await reloadSettings(message) elif message.content.startswith("!name"): await changeName(message) elif message.content.startswith("!cleanup"): await cleanup(message) ################################### elif getTriviabyChannel(message.channel): #check if trivia is ongoing in the channel trvsession = getTriviabyChannel(message.channel) await trvsession.checkAnswer(message) elif "economy" in modules: await economy.checkCommands(message) if message.content.startswith('!') and len(message.content) > 2 and settings["CUSTOMCOMMANDS"]: await customCommand(message) @client.async_event async def on_ready(): logger.info("I'm online " + "(" + client.user.id + ")") await gameSwitcher.changeGame(now=True) # cns = threading.Thread(target=console, args=[]) # cns.start() # console, WIP @client.async_event def on_message_delete(message): # WIP. Need to check for permissions #await client.send_message(message.channel, "{} `I have deleted your message.`".format(message.author.mention)) pass @client.async_event async def on_message_edit(before, message): if message.author.id != client.user.id and settings["FILTER"] and not isMemberAdmin(message) and not message.channel.is_private: await checkFilter(message) await checkRegex(message) def loggerSetup(): #api wrapper logger = logging.getLogger('discord') logger.setLevel(logging.WARNING) handler = logging.FileHandler(filename='wrapper.log', encoding='utf-8', mode='a') handler.setFormatter(logging.Formatter('%(asctime)s:%(levelname)s:%(name)s: %(message)s', datefmt="[%d/%m/%Y %H:%M]")) logger.addHandler(handler) #Red logger = logging.getLogger(__name__) logger.setLevel(logging.DEBUG) handler = logging.StreamHandler() handler.setFormatter(logging.Formatter('%(asctime)s %(levelname)s: %(message)s', datefmt="[%d/%m/%Y %H:%M]")) logger.addHandler(handler) file_handler = logging.FileHandler(filename="red.log", mode='a') file_formatter = logging.Formatter('%(asctime)s %(levelname)s: %(message)s', datefmt="[%d/%m/%Y %H:%M]") file_handler.setFormatter(file_formatter) logger.addHandler(file_handler) return logger class Trivia(): def __init__(self, message): self.channel = message.channel self.stop = False self.currentQ = "" self.currentA = "" self.scorelist = {} self.Atimestamp = 99 #var initialization self.answered = False self.noAnswers = ["`No one got it. Disappointing.`", "`Oook... Next one...`", "`Maybe next time I'll tell you the solution.`", "`*sighs*`", "`I have no words. Next one...`", "`Come on, that one was easy.`", "`That was one of the easiest ones.`", "`I knew the answer.`", "`Is the game almost over? I have places to be.`", "`I'm on the verge of tears...`", "`:'(`", "`Really? No one?`", "`Crazy.`", "`I expected nothing and I'm still disappointed.`", "`*facepalm*`"] logger.info("Trivia started in channel " + self.channel.id) async def game(self): global trivia_sessions while not self.stop: self.answered = False self.currentQ, self.currentA = self.getTriviaQuestion() # [question, answer] logger.debug(self.currentA) await client.send_message(self.channel, "`" + self.currentQ + "`") time.sleep(settings["TRIVIADELAY"]) await self.checkScore() if time.perf_counter() - self.Atimestamp < 3: time.sleep(2) # waits for 2 seconds if the answer was given very recently if not self.answered and not self.stop: await client.send_message(self.channel, choice(self.noAnswers)) # client.send_typing(self.channel) #doesn't work for some reason time.sleep(3) trivia_sessions.remove(self) async def checkAnswer(self, message): if self.currentA.lower() == message.content.lower() and not self.answered: if message.author.name in self.scorelist: self.scorelist[message.author.name] += 1 else: self.scorelist[message.author.name] = 1 await client.send_message(self.channel, "{} `Correct answer! +1 point! ({} points)`".format(message.author.mention, str(self.scorelist[message.author.name]))) self.Atimestamp = time.perf_counter() self.answered = True async def checkScore(self): for k, v in self.scorelist.items(): if self.scorelist[k] == settings["TRIVIAMAXSCORE"]: await client.send_message(self.channel, "`Congratulations {}! You win!`".format(k)) self.sendTable() self.stop = True async def sendTable(self): self.scorelist = sorted(self.scorelist.items(), reverse=True, key=lambda x: x[1]) # orders score from lower to higher t = "```Scores: \n" for score in self.scorelist: t += score[0] # name t += "\t" t += str(score[1]) # score t += "\n" t += "```" await client.send_message(self.channel, t) async def stopp(self): self.stop = True await client.send_message(self.channel, "`Trivia stopped!`") logger.info("Trivia stopped in channel " + self.channel.id) def getTriviaQuestion(self): q = choice(list(trivia_questions.keys())) return q, trivia_questions[q] # question, answer class botPlays(): def __init__(self): self.games = dataIO.fileIO("games.json", "load") self.lastChanged = int(time.perf_counter()) self.delay = 300 async def changeGame(self, now=False): if abs(self.lastChanged - int(time.perf_counter())) >= self.delay or now: self.lastChanged = int(time.perf_counter()) await client.change_status(discord.Game(name=choice(self.games))) class Playlist(): def __init__(self, filename=None): #a playlist with a single song is just there to make !addfavorite work with !youtube command self.filename = filename self.current = 0 self.stop = False self.lastAction = 999 self.currentTitle = "" self.type = filename["type"] if filename["type"] == "playlist": self.playlist = dataIO.fileIO("playlists/" + filename["filename"] + ".txt", "load")["playlist"] elif filename["type"] == "favorites": self.playlist = dataIO.fileIO("favorites/" + filename["filename"] + ".txt", "load") elif filename["type"] == "local": self.playlist = filename["filename"] elif filename["type"] == "singleSong": self.playlist = [filename["filename"]] self.playSingleSong(self.playlist[0]) else: raise("Invalid playlist call.") if filename["type"] != "singleSong": self.nextSong(0) def nextSong(self, nextTrack, lastError=False): global musicPlayer if not self.passedTime() < 1 and not self.stop: #direct control if musicPlayer: musicPlayer.stop() self.lastAction = int(time.perf_counter()) try: if isPlaylistValid([self.playlist[nextTrack]]): #Checks if it's a valid youtube link if settings["DOWNLOADMODE"]: path = self.getVideo(self.playlist[nextTrack]) try: logger.info("Starting track...") musicPlayer = client.voice.create_ffmpeg_player("cache/" + path, options='''-filter:a "volume={}"'''.format(settings["VOLUME"])) musicPlayer.start() except: logger.warning("Something went wrong with track " + self.playlist[self.current]) if not lastError: #prevents error loop self.lastAction = 999 self.nextSong(self.getNextSong(), lastError=True) else: #Stream mode. Buggy. musicPlayer = client.voice.create_ytdl_player(self.playlist[nextTrack], options=youtube_dl_options) musicPlayer.start() else: # must be a local playlist then musicPlayer = client.voice.create_ffmpeg_player(self.playlist[nextTrack], options='''-filter:a "volume={}"'''.format(settings["VOLUME"])) musicPlayer.start() except Exception as e: logger.warning("Something went wrong with track " + self.playlist[self.current]) if not lastError: #prevents error loop self.lastAction = 999 self.nextSong(self.getNextSong(), lastError=True) def getVideo(self, url): try: yt = youtube_dl.YoutubeDL(youtube_dl_options) v = yt.extract_info(url, download=False) if not os.path.isfile("cache/" + v["id"]): logger.info("Track not in cache, downloading...") v = yt.extract_info(url, download=True) self.currentTitle = v["title"] return v["id"] except Exception as e: logger.error(e) return False def playSingleSong(self, url): global musicPlayer if settings["DOWNLOADMODE"]: v = self.getVideo(url) if musicPlayer: if musicPlayer.is_playing(): musicPlayer.stop() if v: musicPlayer = client.voice.create_ffmpeg_player("cache/" + v, options='''-filter:a "volume={}"'''.format(settings["VOLUME"])) musicPlayer.start() else: if musicPlayer: if musicPlayer.is_playing(): musicPlayer.stop() musicPlayer = client.voice.create_ytdl_player(self.playlist[0], options=youtube_dl_options) musicPlayer.start() async def songSwitcher(self): while not self.stop: if musicPlayer.is_done() and not self.stop: self.nextSong(self.getNextSong()) await asyncio.sleep(0.5) def passedTime(self): return abs(self.lastAction - int(time.perf_counter())) def getPreviousSong(self): try: song = self.playlist[self.current-1] self.current -= 1 return self.current except: #if the current song was the first song, returns the last in the playlist song = self.playlist[len(self.current)-1] self.current -= 1 return self.current def getNextSong(self): try: song = self.playlist[self.current+1] self.current += 1 return self.current except: #if the current song was the last song, returns the first in the playlist song = self.playlist[0] self.current = 0 return self.current def pause(self): if musicPlayer.is_playing() and not self.stop: musicPlayer.pause() def resume(self): if not self.stop: musicPlayer.resume() def shuffle(self): if not self.stop: shuffle(self.playlist) async def addcom(message): if checkAuth("ModifyCommands", message, settings): msg = message.content.split() if len(msg) > 2: msg = message.content[8:] # removes !addcom newcmd = msg[:msg.find(" ")] # extracts custom command customtext = msg[msg.find(" ") + 1:] # extracts [text] if len(newcmd) > 1 and newcmd.find(" ") == -1: if not message.channel.server.id in commands: commands[message.channel.server.id] = {} cmdlist = commands[message.channel.server.id] if newcmd not in cmdlist: cmdlist[newcmd] = customtext commands[message.channel.server.id] = cmdlist dataIO.fileIO("commands.json", "save", commands) logger.info("Saved commands database.") await client.send_message(message.channel, "`Custom command successfully added.`") else: await client.send_message(message.channel, "`This command already exists. Use !editcom [command] [text]`") else: await client.send_message(message.channel, "`!addcom [command] [text]`") else: await client.send_message(message.channel, "`You don't have permissions to edit custom commands.`") async def editcom(message): if checkAuth("ModifyCommands", message, settings): msg = message.content.split() if len(msg) > 2: msg = message.content[9:] # removes !editcom cmd = msg[:msg.find(" ")] # extracts custom command customtext = msg[msg.find(" ") + 1:] # extracts [text] if message.channel.server.id in commands: cmdlist = commands[message.channel.server.id] if cmd in cmdlist: cmdlist[cmd] = customtext commands[message.channel.server.id] = cmdlist dataIO.fileIO("commands.json", "save", commands) logger.info("Saved commands database.") await client.send_message(message.channel, "`Custom command successfully edited.`") else: await client.send_message(message.channel, "`That command doesn't exist. Use !addcom [command] [text]`") else: await client.send_message(message.channel, "`There are no custom commands in this server. Use !addcom [command] [text]`") else: await client.send_message(message.channel, "`!editcom [command] [text]`") else: await client.send_message(message.channel, "`You don't have permissions to edit custom commands.`") async def delcom(message): if checkAuth("ModifyCommands", message, settings): msg = message.content.split() if len(msg) == 2: if message.channel.server.id in commands: cmdlist = commands[message.channel.server.id] if msg[1] in cmdlist: cmdlist.pop(msg[1], None) commands[message.channel.server.id] = cmdlist dataIO.fileIO("commands.json", "save", commands) logger.info("Saved commands database.") await client.send_message(message.channel, "`Custom command successfully deleted.`") else: await client.send_message(message.channel, "`That command doesn't exist.`") else: await client.send_message(message.channel, "`There are no custom commands in this server. Use !addcom [command] [text]`") else: await client.send_message(message.channel, "`!delcom [command]`") else: await client.send_message(message.channel, "`You don't have permissions to edit custom commands.`") async def listCustomCommands(message): msg = "Custom commands: \n\n```" cmds = commands[message.channel.server.id].keys() if cmds: for i, d in enumerate(cmds): if i % 4 == 0 and i != 0: msg = msg + d + "\n" else: msg = msg + d + "\t" msg += "```" await client.send_message(message.author, msg) else: await client.send_message(message.author, "There are no custom commands.") def checkAuth(cmd, message, settings): #checks if those settings are on. If they are, it checks if the user is a owner if cmd == "ModifyCommands": if settings["EDIT_CC_ADMIN_ONLY"]: if isMemberAdmin(message): return True else: return False else: return True elif cmd == "Trivia": if settings["TRIVIA_ADMIN_ONLY"]: if isMemberAdmin(message): return True else: return False else: return True else: logger.error("Invalid call to checkAuth") return False async def rpsgame(message): rps = {"rock" : ":moyai:", "paper": ":page_facing_up:", "scissors":":scissors:" } msg = message.content.lower().split(" ") if len(msg) == 2: _, userchoice = msg if userchoice in rps.keys(): botchoice = choice(list(rps.keys())) msgs = { "win": " You win {}!".format(message.author.mention), "square": " We're square {}!".format(message.author.mention), "lose": " You lose {}!".format(message.author.mention) } if userchoice == botchoice: await client.send_message(message.channel, rps[botchoice] + msgs["square"]) elif userchoice == "rock" and botchoice == "paper": await client.send_message(message.channel, rps[botchoice] + msgs["lose"]) elif userchoice == "rock" and botchoice == "scissors": await client.send_message(message.channel, rps[botchoice] + msgs["win"]) elif userchoice == "paper" and botchoice == "rock": await client.send_message(message.channel, rps[botchoice] + msgs["win"]) elif userchoice == "paper" and botchoice == "scissors": await client.send_message(message.channel, rps[botchoice] + msgs["lose"]) elif userchoice == "scissors" and botchoice == "rock": await client.send_message(message.channel, rps[botchoice] + msgs["lose"]) elif userchoice == "scissors" and botchoice == "paper": await client.send_message(message.channel, rps[botchoice] + msgs["win"]) else: await client.send_message(message.channel, "`!rps [rock or paper or scissors]`") else: await client.send_message(message.channel, "`!rps [rock or paper or scissors]`") async def randomchoice(message): sentences = ["Mmm... I think I'll choose ", "I choose ", "I prefer ", "This one is best: ", "This: "] msg = message.content[8:] # removes !choose msg = msg.split(" or ") if len(msg) == 1: await client.send_message(message.channel, "`!choose option1 or option2 or option3 (...)`") elif len(msg) >= 2: await client.send_message(message.channel, "`" + choice(sentences) + choice(msg) + "`") else: await client.send_message(message.channel, "`The options must be at least two.`") async def stopwatch(message): global stopwatches if message.author.id in stopwatches: tmp = abs(stopwatches[message.author.id] - int(time.perf_counter())) tmp = str(datetime.timedelta(seconds=tmp)) await client.send_message(message.channel, "`Stopwatch stopped! Time: " + str(tmp) + " `") stopwatches.pop(message.author.id, None) else: stopwatches[message.author.id] = int(time.perf_counter()) await client.send_message(message.channel, "`Stopwatch started! Use !sw to stop it.`") async def image(message): # API's dead. msg = message.content.split() if len(msg) > 1: if len(msg[1]) > 1 and len([msg[1]]) < 20: try: msg.remove(msg[0]) msg = "+".join(msg) search = "http://ajax.googleapis.com/ajax/services/search/images?v=1.0&q=" + msg + "&start=0" result = requests.get(search).json() url = result["responseData"]["results"][0]["url"] await client.send_message(message.channel, url) except: await client.send_message(message.channel, "Error.") else: await client.send_message(message.channel, "Invalid search.") else: await client.send_message(message.channel, "!image [text]") async def gif(message): msg = message.content.split() if len(msg) > 1: if len(msg[1]) > 1 and len([msg[1]]) < 20: try: msg.remove(msg[0]) msg = "+".join(msg) search = "http://api.giphy.com/v1/gifs/search?q=" + msg + "&api_key=dc6zaTOxFJmzC" result = requests.get(search).json() if result["data"] != []: url = result["data"][0]["url"] await client.send_message(message.channel, url) else: await client.send_message(message.channel, "Your search terms gave no results.") except: await client.send_message(message.channel, "Error.") else: await client.send_message(message.channel, "Invalid search.") else: await client.send_message(message.channel, "!gif [text]") async def controlTrivia(message, b): global trivia_sessions if b: #new trivia tmp = Trivia(message) trivia_sessions.append(tmp) await client.send_message(message.channel, "`Trivia started! First question...`") time.sleep(3) await tmp.game() else: await stopTriviabyChannel(message.channel) async def stopTriviabyChannel(channel): global trivia_sessions for t in trivia_sessions: if t.channel == channel: await t.stopp() async def avatar(message): if message.mentions: m = message.mentions[0] await client.send_message(message.channel, "{}'s avatar: {}".format(m.name, m.avatar_url)) else: if len(message.content.split(" ")) >= 2: name = message.content[8:] member = discord.utils.get(message.server.members, name=name) if member != None: await client.send_message(message.channel, "{}'s avatar: {}".format(member.name, member.avatar_url)) else: await client.send_message(message.channel, "`User not found.`") else: await client.send_message(message.channel, "`!avatar [name or mention]`") def getTriviabyChannel(channel): for t in trivia_sessions: if t.channel == channel: return t return False async def roll(message): msg = message.content.split() if len(msg) == 2: if msg[1].isdigit(): msg[1] = int(msg[1]) if msg[1] < 99999 and msg[1] > 1: await client.send_message(message.channel, "{} :game_die: `{}` :game_die:".format(message.author.mention, str(randint(1, msg[1])))) else: await client.send_message(message.channel, "{} `A number between 1 and 99999, maybe? :)`".format(message.author.mention)) else: await client.send_message(message.channel, "`!roll [number]`") else: await client.send_message(message.channel, "`!roll [number]`") async def checkFilter(message): #WIP msg = message.content.lower() if message.server.id in badwords: for word in badwords[message.server.id]: if msg.find(word.lower()) != -1: if canDeleteMessages(message): await client.delete_message(message) logger.info("Message eliminated.") return True else: logger.info("Couldn't delete message. I need permissions.") return False return False async def checkRegex(message): #WIP msg = message.content #.lower()? if message.server.id in badwords_regex: for pattern in badwords_regex[message.server.id]: rr = re.search(pattern, msg, re.I | re.U) if rr != None: if canDeleteMessages(message): await client.delete_message(message) logger.info("Message eliminated. Regex: " + pattern) return True else: logger.info("Couldn't delete message. I need permissions.") return False return False async def twitchCheck(message): msg = message.content.split() if len(msg) == 2: try: url = "https://api.twitch.tv/kraken/streams/" + msg[1] data = requests.get(url).json() if "error" in data: await client.send_message(message.channel, "{} `There is no streamer named {}`".format(message.author.mention, msg[1])) elif "stream" in data: if data["stream"] != None: await client.send_message(message.channel, "{} `{} is online!` {}".format(message.author.mention, msg[1], "http://www.twitch.tv/" + msg[1])) else: await client.send_message(message.channel, "{} `{} is offline.`".format(message.author.mention, msg[1])) else: await client.send_message(message.channel, "{} `There is no streamer named {}`".format(message.author.mention, msg[1])) except: await client.send_message(message.channel, "{} `Error.`".format(message.author.mention)) else: await client.send_message(message.channel, "{} `!twitch [stream]`".format(message.author.mention)) async def uptime(message): up = abs(uptime_timer - int(time.perf_counter())) up = str(datetime.timedelta(seconds=up)) await client.send_message(message.channel, "`Uptime: {}`".format(up)) async def checkVoice(message): if not client.is_voice_connected(): if message.author.voice_channel: await client.join_voice_channel(message.author.voice_channel) else: await client.send_message(message.channel, "{} `You need to join a voice channel first.`".format(message.author.mention)) return False return True async def playVideo(message): global musicPlayer, currentPlaylist toDelete = None if await checkVoice(message): pattern = "(?:youtube\.com\/watch\?v=)(.*)|(?:youtu.be/)(.*)" rr = re.search(pattern, message.content, re.I | re.U) if rr.group(1) != None: id = rr.group(1) elif rr.group(2) != None: id = rr.group(2) else: await client.send_message(message.channel, "{} `Invalid link.`".format(message.author.mention)) return False stopMusic() if canDeleteMessages(message): await client.send_message(message.channel, "`Playing youtube video {} requested by {}`".format(id, message.author.name)) await client.delete_message(message) if settings["DOWNLOADMODE"]: toDelete = await client.send_message(message.channel, "`I'm in download mode. It might take a bit for me to start. I'll delete this message as soon as I'm ready.`".format(id, message.author.name)) data = {"filename" : 'https://www.youtube.com/watch?v=' + id, "type" : "singleSong"} currentPlaylist = Playlist(data) if toDelete: await client.delete_message(toDelete) # currentPlaylist.playlist = ['https://www.youtube.com/watch?v=' + id] # musicPlayer = client.voice.create_ytdl_player('https://www.youtube.com/watch?v=' + id, options=youtube_dl_options) # musicPlayer.start() #!addfavorite compatibility stuff async def playPlaylist(message, sing=False): global musicPlayer, currentPlaylist msg = message.content toDelete = None if not sing: if msg != "!play" or msg != "play ": if await checkVoice(message): msg = message.content[6:] if dataIO.fileIO("playlists/" + msg + ".txt", "check"): stopMusic() data = {"filename" : msg, "type" : "playlist"} if settings["DOWNLOADMODE"]: toDelete = await client.send_message(message.channel, "`I'm in download mode. It might take a bit for me to start and switch between tracks. I'll delete this message as soon as the current playlist stops.`".format(id, message.author.name)) currentPlaylist = Playlist(data) await asyncio.sleep(2) await currentPlaylist.songSwitcher() if toDelete: await client.delete_message(toDelete) else: await client.send_message(message.channel, "{} `That playlist doesn't exist.`".format(message.author.mention)) else: if await checkVoice(message): stopMusic() msg = ["Sure why not? :microphone:", "*starts singing* :microphone:", "*starts humming* :notes:"] playlist = ["https://www.youtube.com/watch?v=zGTkAVsrfg8", "https://www.youtube.com/watch?v=cGMWL8cOeAU", "https://www.youtube.com/watch?v=vFrjMq4aL-g", "https://www.youtube.com/watch?v=WROI5WYBU_A", "https://www.youtube.com/watch?v=41tIUr_ex3g", "https://www.youtube.com/watch?v=f9O2Rjn1azc"] song = choice(playlist) data = {"filename" : song, "type" : "singleSong"} if settings["DOWNLOADMODE"]: toDelete = await client.send_message(message.channel, "`I'm in download mode. It might take a bit for me to start. I'll delete this message as soon as I'm ready.`".format(id, message.author.name)) currentPlaylist = Playlist(data) # currentPlaylist.playlist = [song] # musicPlayer = client.voice.create_ytdl_player(song, options=youtube_dl_options) # musicPlayer.start() if toDelete: await client.delete_message(toDelete) await client.send_message(message.channel, choice(msg)) async def playLocal(message): global currentPlaylist msg = message.content.split(" ") if await checkVoice(message): if len(msg) == 2: localplaylists = getLocalPlaylists() if localplaylists and ("/" not in msg[1] and "\\" not in msg[1]): if msg[1] in localplaylists: files = [] if glob.glob("localtracks\\" + msg[1] + "\\*.mp3"): files.extend(glob.glob("localtracks\\" + msg[1] + "\\*.mp3")) if glob.glob("localtracks\\" + msg[1] + "\\*.flac"): files.extend(glob.glob("localtracks\\" + msg[1] + "\\*.flac")) stopMusic() data = {"filename" : files, "type" : "local"} currentPlaylist = Playlist(data) await asyncio.sleep(2) await currentPlaylist.songSwitcher() else: await client.send_message(message.channel, "`There is no local playlist called {}. !local or !locallist to receive the list.`".format(msg[1])) else: await client.send_message(message.channel, "`There are no valid playlists in the localtracks folder.`") else: await client.send_message(message.channel, "`!local [playlist]`") def getLocalPlaylists(): dirs = [] files = os.listdir("localtracks/") for f in files: if os.path.isdir("localtracks/" + f) and " " not in f: if glob.glob("localtracks/" + f + "/*.mp3") != []: dirs.append(f) elif glob.glob("localtracks/" + f + "/*.flac") != []: dirs.append(f) if dirs != []: return dirs else: return False async def leaveVoice(): if client.is_voice_connected(): stopMusic() await client.voice.disconnect() async def listPlaylists(message): msg = "Available playlists: \n\n```" files = os.listdir("playlists/") if files: for i, f in enumerate(files): if f.endswith(".txt"): if i % 4 == 0 and i != 0: msg = msg + f.replace(".txt", "") + "\n" else: msg = msg + f.replace(".txt", "") + "\t" msg += "```" await client.send_message(message.author, msg) else: await client.send_message(message.author, "There are no playlists.") async def listLocal(message): msg = "Available local playlists: \n\n```" dirs = getLocalPlaylists() if dirs: for i, d in enumerate(dirs): if i % 4 == 0 and i != 0: msg = msg + d + "\n" else: msg = msg + d + "\t" msg += "```" await client.send_message(message.author, msg) else: await client.send_message(message.author, "There are no local playlists.") def stopMusic(): global musicPlayer, currentPlaylist if currentPlaylist != None: currentPlaylist.stop = True if musicPlayer != None: musicPlayer.stop() async def transferPlaylist(message): msg = message.attachments[0] if msg["filename"].endswith(".txt"): if not dataIO.fileIO("playlists/" + msg["filename"], "check"): #returns false if file already exists r = requests.get(msg["url"]) data = r.text.replace("\r", "") data = data.split() if isPlaylistValid(data) and isPlaylistNameValid(msg["filename"].replace(".txt", "")): data = { "author" : message.author.id, "playlist": data} dataIO.fileIO("playlists/" + msg["filename"], "save", data) await client.send_message(message.channel, "`Playlist added. Name: {}`".format(msg["filename"].replace(".txt", ""))) else: await client.send_message(message.channel, "`Something is wrong with the playlist or its filename. Type !audio help to read how to format it properly.`") else: await client.send_message(message.channel, "`A playlist with that name already exists. Change the filename and resubmit it.`") def isPlaylistValid(data): data = [y for y in data if y != ""] # removes all empty elements data = [y for y in data if y != "\n"] for link in data: pattern = "^(https:\/\/www\.youtube\.com\/watch\?v=...........*)|^(https:\/\/youtu.be\/...........*)|^(https:\/\/youtube\.com\/watch\?v=...........*)" rr = re.search(pattern, link, re.I | re.U) if rr == None: return False return True def isPlaylistNameValid(name): for l in name: if l.isdigit() or l.isalpha() or l == "_": pass else: return False return True def isPlaylistLinkValid(link): pattern = "^https:\/\/www.youtube.com\/playlist\?list=(.[^:/]*)" rr = re.search(pattern, link, re.I | re.U) if not rr == None: return rr.group(1) else: return False async def addPlaylist(message): msg = message.content.split(" ") if len(msg) == 3: _, name, link = msg if isPlaylistNameValid(name) and len(name) < 25 and isPlaylistLinkValid(link): if dataIO.fileIO("playlists/" + name + ".txt", "check"): await client.send_message(message.channel, "`A playlist with that name already exists.`") return False links = youtubeparser.parsePlaylist(link) if links: data = { "author" : message.author.id, "playlist": links} dataIO.fileIO("playlists/" + name + ".txt", "save", data) await client.send_message(message.channel, "`Playlist added. Name: {}`".format(name)) else: await client.send_message(message.channel, "`Something went wrong. Either the link was incorrect or I was unable to retrieve the page.`") else: await client.send_message(message.channel, "`Something is wrong with the playlist's link or its filename. Remember, the name must be with only numbers, letters and underscores. Link must be this format: https://www.youtube.com/playlist?list=PLe8jmEHFkvsaDOOWcREvkgFoj6MD0pXXX`") else: await client.send_message(message.channel, "`!addplaylist [name] [link]`") async def delPlaylist(message): msg = message.content.split(" ") if len(msg) == 2: _, filename = msg if dataIO.fileIO("playlists/" + filename + ".txt", "check"): authorid = dataIO.fileIO("playlists/" + filename + ".txt", "load")["author"] if message.author.id == authorid or isMemberAdmin(message): os.remove("playlists/" + filename + ".txt") await client.send_message(message.channel, "`Playlist {} removed.`".format(filename)) else: await client.send_message(message.channel, "`Only the playlist's author and admins can do that.`") else: await client.send_message(message.channel, "`There is no playlist with that name.`") else: await client.send_message(message.channel, "`!delplaylist [name]`") async def getSongTitle(message): title = youtubeparser.getTitle(currentPlaylist.playlist[currentPlaylist.current]) if title: await client.send_message(message.channel, "`Current song: {}\n{}`".format(title, currentPlaylist.playlist[currentPlaylist.current])) else: await client.send_message(message.channel, "`I couldn't retrieve the current song's title.`") async def addToFavorites(message): if currentPlaylist: if dataIO.fileIO("favorites/" + message.author.id + ".txt", "check"): data = dataIO.fileIO("favorites/" + message.author.id + ".txt", "load") else: data = [] data.append(currentPlaylist.playlist[currentPlaylist.current]) dataIO.fileIO("favorites/" + message.author.id + ".txt", "save", data) await client.send_message(message.channel, "{} `This song has been added to your favorites.`".format(message.author.mention)) else: await client.send_message(message.channel, "{} `No song is being played`".format(message.author.mention)) async def removeFromFavorites(message): if currentPlaylist: if dataIO.fileIO("favorites/" + message.author.id + ".txt", "check"): data = dataIO.fileIO("favorites/" + message.author.id + ".txt", "load") if currentPlaylist.playlist[currentPlaylist.current] in data: data.remove(currentPlaylist.playlist[currentPlaylist.current]) dataIO.fileIO("favorites/" + message.author.id + ".txt", "save", data) await client.send_message(message.channel, "{} `This song has been removed from your favorites.`".format(message.author.mention)) else: await client.send_message(message.channel, "{} `This song isn't in your favorites.`".format(message.author.mention)) else: await client.send_message(message.channel, "{} `You don't have any favorites yet. Start adding them with !addfavorite`".format(message.author.mention)) else: await client.send_message(message.channel, "{} `No song is being played`".format(message.author.mention)) async def playFavorites(message): global musicPlayer, currentPlaylist if await checkVoice(message): if dataIO.fileIO("favorites/" + message.author.id + ".txt", "check"): data = {"filename" : message.author.id, "type" : "favorites"} stopMusic() currentPlaylist = Playlist(data) await asyncio.sleep(2) await currentPlaylist.songSwitcher() else: await client.send_message(message.channel, "{} `You don't have any favorites yet. Start adding them with !addfavorite`".format(message.author.mention)) async def sendPlaylist(message): if currentPlaylist: msg = "Here's the current playlist:\n```" for track in currentPlaylist.playlist: msg += track msg += "\n" if len(msg) >= 1900: msg += "```" await client.send_message(message.author, msg) msg = "```" if msg != "```": msg += "```" await client.send_message(message.author, msg) async def setVolume(message): global settings msg = message.content if len(msg.split(" ")) == 2: msg = msg.split(" ") try: vol = float(msg[1]) if vol >= 0 and vol <= 1: settings["VOLUME"] = vol await(client.send_message(message.channel, "`Volume set. Next track will have the desired volume.`")) dataIO.fileIO("settings.json", "save", settings) else: await(client.send_message(message.channel, "`Volume must be between 0 and 1. Example: !volume 0.50`")) except: await(client.send_message(message.channel, "`Volume must be between 0 and 1. Example: !volume 0.15`")) else: await(client.send_message(message.channel, "`Volume must be between 0 and 1. Example: !volume 0.15`")) async def downloadMode(message): if isMemberAdmin(message): if settings["DOWNLOADMODE"]: settings["DOWNLOADMODE"] = False await(client.send_message(message.channel, "`Download mode disabled. This mode is unstable and tracks might interrupt. Also, the volume settings will not have any effect.`")) else: settings["DOWNLOADMODE"] = True await(client.send_message(message.channel, "`Download mode enabled.`")) dataIO.fileIO("settings.json", "save", settings) else: await(client.send_message(message.channel, "`I don't take orders from you.`")) ############## ADMIN COMMANDS ################### async def shutdown(message): if isMemberAdmin(message): await client.send_message(message.channel, client.user.name + " shutting down... See you soon. :hand:") await client.logout() try: exit(1) except SystemExit: #clean exit logger.info("Shutting down as requested by " + message.author.id + "...") pass else: await client.send_message(message.channel, "`I don't take orders from you.`") async def join(message): if isMemberAdmin(message): msg = message.content.split() if len(msg) > 1: await client.accept_invite(msg[1]) else: logger.warning("Join: missing parameters") else: await client.send_message(message.channel, "`I don't take orders from you.`") async def leave(message): if isMemberAdmin(message): await client.send_message(message.channel, "`Bye.`") await client.leave_server(message.channel.server) else: await client.send_message(message.channel, "`I don't take orders from you.`") async def shush(message): global shush_list if isMemberAdmin(message): await client.send_message(message.channel, "`Ok, I'll ignore this channel.`") shush_list.append(message.channel.id) dataIO.fileIO("shushlist.json", "save", shush_list) logger.info("Saved silenced channels database.") else: await client.send_message(message.channel, "`I don't take orders from you.`") async def talk(message): if isMemberAdmin(message): if message.channel.id in shush_list: shush_list.remove(message.channel.id) dataIO.fileIO("shushlist.json", "save", shush_list) logger.info("Saved silenced channels database.") await client.send_message(message.channel, "`Aaand I'm back.`") else: await client.send_message(message.channel, "`I don't take orders from you.`") async def addBadWords(message): global badwords if isMemberAdmin(message): msg = message.content.split() if len(msg) >= 2: del msg[0] if not message.server.id in badwords: badwords[message.server.id] = [] for word in msg: if word.find("/") != -1: word = word.replace("/", " ") badwords[message.server.id].append(word) await client.send_message(message.channel, "`Updated banned words database.`") dataIO.fileIO("filter.json", "save", badwords) logger.info("Saved filter words.") else: await client.send_message(message.channel, "`!addwords [word1] [word2] [phrase/with/many/words] (...)`") else: await client.send_message(message.channel, "`I don't take orders from you.`") async def removeBadWords(message): global badwords if isMemberAdmin(message): msg = message.content.split() if len(msg) >= 2: del msg[0] if message.server.id in badwords: for w in msg: try: if w.find("/") != -1: w = w.replace("/", " ") badwords[message.server.id].remove(w) except: pass await client.send_message(message.channel, "`Updated banned words database.`") dataIO.saveFilter(badwords) else: await client.send_message(message.channel, "`!removewords [word1] [word2] [phrase/with/many/words](...)`") else: await client.send_message(message.channel, "`I don't take orders from you.`") async def changeName(message): global settings if isMemberAdmin(message): msg = message.content.split() if len(msg) == 2: try: await client.edit_profile(settings["PASSWORD"], username=msg[1]) except Exception as e: logger.error(e) else: await client.send_message(message.channel, "`!name [new name]`") else: await client.send_message(message.channel, "`I don't take orders from you.`") async def addRegex(message): global badwords_regex if isMemberAdmin(message): msg = message.content msg = msg[10:] if not message.server.id in badwords_regex: badwords_regex[message.server.id] = [] badwords_regex[message.server.id].append(msg) await client.send_message(message.channel, "`Updated regex filter database.`") dataIO.fileIO("regex_filter.json", "save", badwords_regex) logger.info("Saved regex filter database.") else: await client.send_message(message.channel, "`I don't take orders from you.`") async def removeRegex(message): global badwords_regex if isMemberAdmin(message): msg = message.content msg = msg[13:] if message.server.id in badwords_regex: if msg in badwords_regex[message.server.id]: badwords_regex[message.server.id].remove(msg) await client.send_message(message.channel, "`Updated regex filter database.`") dataIO.saveRegex(badwords_regex) else: await client.send_message(message.channel, "`No match.`") else: await client.send_message(message.channel, "`I don't take orders from you.`") async def reloadSettings(message): if isMemberAdmin(message): loadDataFromFiles(True) await client.send_message(message.channel, "`Settings and files reloaded.`") else: await client.send_message(message.channel, "`I don't take orders from you.`") async def cleanup(message): if isMemberAdmin(message): if canDeleteMessages(message): msg = message.content.split() if len(msg) == 2: if msg[1].isdigit(): n = int(msg[1]) for x in await client.logs_from(message.channel, limit=n+1): await client.delete_message(x) else: await client.send_message(message.channel, "`!cleanup [number]`") else: await client.send_message(message.channel, "`!cleanup [number]`") else: await client.send_message(message.channel, "`I need permissions to delete messages.`") else: await client.send_message(message.channel, "`I don't take orders from you.`") def isMemberAdmin(message): if not message.channel.is_private: if discord.utils.get(message.author.roles, name=settings["ADMINROLE"]) != None: return True else: return False else: return False def canDeleteMessages(message): return message.channel.permissions_for(message.server.me).manage_messages async def addTwitchAlert(message): global twitchStreams added = False if isMemberAdmin(message): msg = message.content.split(" ") if len(msg) == 2: for i, stream in enumerate(twitchStreams): if stream["NAME"] == msg[1] and message.channel.id in stream["CHANNELS"]: await client.send_message(message.channel, "`I'm already monitoring that stream in this channel.`") return False for stream in twitchStreams: if stream["NAME"] == msg[1] and message.channel.id not in stream["CHANNELS"]: # twitchAlert is already monitoring this streamer but not in this channel twitchStreams[i]["CHANNELS"].append(message.channel.id) added = True if not added: # twitchAlert wasn't monitoring this streamer twitchStreams.append({"CHANNELS" : [message.channel.id], "NAME" : msg[1], "ALREADY_ONLINE" : False}) dataIO.fileIO("twitch.json", "save", twitchStreams) await client.send_message(message.channel, "`I will always send an alert in this channel whenever {}'s stream is online. Use !stoptwitchalert [name] to stop it.`".format(msg[1])) else: await client.send_message(message.channel, "`!twitchalert [name]`") else: await client.send_message(message.channel, "`I don't take orders from you.`") async def removeTwitchAlert(message): global twitchStreams if isMemberAdmin(message): msg = message.content.split(" ") if len(msg) == 2: for i, stream in enumerate(twitchStreams): if stream["NAME"] == msg[1] and message.channel.id in stream["CHANNELS"]: if len(stream["CHANNELS"]) == 1: twitchStreams.remove(stream) else: twitchStreams[i]["CHANNELS"].remove(message.channel.id) dataIO.fileIO("twitch.json", "save", twitchStreams) await client.send_message(message.channel, "`I will stop sending alerts about {}'s stream in this channel.`".format(msg[1])) return True await client.send_message(message.channel, "`There's no alert for {}'s stream in this channel.`".format(msg[1])) else: await client.send_message(message.channel, "`!stoptwitchalert [name]`") else: await client.send_message(message.channel, "`I don't take orders from you.`") ################################################ @asyncio.coroutine async def twitchAlert(): global twitchStreams CHECK_DELAY = 10 while True: if twitchStreams and client.is_logged_in: to_delete = [] save = False consistency_check = twitchStreams for i, stream in enumerate(twitchStreams): if twitchStreams == consistency_check: #prevents buggy behavior if twitchStreams gets modified during the iteration try: url = "https://api.twitch.tv/kraken/streams/" + stream["NAME"] data = requests.get(url).json() if "error" in data: #Stream doesn't exist, remove from list to_delete.append(stream) elif "stream" in data: if data["stream"] != None: if not stream["ALREADY_ONLINE"]: for channel in stream["CHANNELS"]: try: await client.send_message(client.get_channel(channel), "`{} is online!` {}".format(stream["NAME"], "http://www.twitch.tv/" + stream["NAME"])) except: #In case of missing permissions pass twitchStreams[i]["ALREADY_ONLINE"] = True save = True else: if stream["ALREADY_ONLINE"]: twitchStreams[i]["ALREADY_ONLINE"] = False save = True except Exception as e: logger.warning(e) if save: #Saves online status, in case the bot needs to be restarted it can prevent message spam dataIO.fileIO("twitch.json", "save", twitchStreams) save = False await asyncio.sleep(CHECK_DELAY) else: break if to_delete: for invalid_stream in to_delete: twitchStreams.remove(invalid_stream) dataIO.fileIO("twitch.json", "save", twitchStreams) else: await asyncio.sleep(5) async def customCommand(message): msg = message.content[1:] if message.channel.server.id in commands: cmdlist = commands[message.channel.server.id] if msg in cmdlist: await client.send_message(message.channel, cmdlist[msg] ) def console(): while True: try: exec(input("")) except Exception: traceback.print_exc() print("\n") def loadDataFromFiles(loadsettings=False): global proverbs, commands, trivia_questions, badwords, badwords_regex, shush_list, twitchStreams proverbs = dataIO.loadProverbs() logger.info("Loaded " + str(len(proverbs)) + " proverbs.") commands = dataIO.fileIO("commands.json", "load") logger.info("Loaded " + str(len(commands)) + " lists of custom commands.") # trivia_questions = dataIO.loadTrivia() # logger.info("Loaded " + str(len(trivia_questions)) + " questions.") badwords = dataIO.fileIO("filter.json", "load") logger.info("Loaded " + str(len(badwords)) + " lists of filtered words.") badwords_regex = dataIO.fileIO("regex_filter.json", "load") logger.info("Loaded " + str(len(badwords_regex)) + " regex lists.") shush_list = dataIO.fileIO("shushlist.json", "load") logger.info("Loaded " + str(len(shush_list)) + " silenced channels.") twitchStreams = dataIO.fileIO("twitch.json", "load") logger.info("Loaded " + str(len(twitchStreams)) + " streams to monitor.") if loadsettings: global settings settings = dataIO.fileIO("settings.json", "load") def main(): global ball, greetings, greetings_caps, stopwatches, trivia_sessions, message, gameSwitcher, uptime_timer, musicPlayer, currentPlaylist global logger, settings logger = loggerSetup() dataIO.logger = logger if not os.path.isfile("twitch.json"): logger.info("Missing twitch.json. Creating it...") dataIO.fileIO("twitch.json", "save", []) settings = dataIO.loadAndCheckSettings() loadDataFromFiles() ball = ["As I see it, yes", "It is certain", "It is decidedly so", "Most likely", "Outlook good", "Signs point to yes", "Without a doubt", "Yes", "Yes – definitely", "You may rely on it", "Reply hazy, try again", "Ask again later", "Better not tell you now", "Cannot predict now", "Concentrate and ask again", "Don't count on it", "My reply is no", "My sources say no", "Outlook not so good", "Very doubtful"] greetings = ["Hey.", "Yes?", "Hi.", "I'm listening.", "Hello.", "I'm here."] greetings_caps = ["DON'T SCREAM", "WHAT", "WHAT IS IT?!", "ì_ì", "NO CAPS LOCK"] stopwatches = {} trivia_sessions = [] message = "" gameSwitcher = botPlays() if "economy" in modules: economy.initialize(client) uptime_timer = int(time.perf_counter()) musicPlayer = None currentPlaylist = None if not os.path.exists("cache/"): #Stores youtube audio for DOWNLOADMODE os.makedirs("cache") loop.create_task(twitchAlert()) #client.run(settings["EMAIL"], settings["PASSWORD"]) yield from client.login(settings["EMAIL"], settings["PASSWORD"]) yield from client.connect() if __name__ == '__main__': loop = asyncio.get_event_loop() try: loop.run_until_complete(main()) except: loop.run_until_complete(client.logout()) finally: loop.close()