CopyBot/Red-DiscordBot
1
0
Fork 0
mirror of https://github.com/Cog-Creators/Red-DiscordBot.git synced 2025-11-08 04:08:56 -05:00
Code Issues Packages Projects Releases Wiki Activity

pep8 for trivia module

Browse Source
This commit is contained in:
Twentysix 2016-03-05 10:31:58 +01:00
parent b959065667
commit 5710540eea
1 changed files with 78 additions and 78 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

156
cogs/trivia.py
View File

@ -24,22 +24,22 @@ class Trivia:
""" """
message = ctx.message message = ctx.message
if list_name == None: if list_name == None:
await self.triviaList(ctx.message.author) await self.trivia_list(ctx.message.author)
elif list_name.lower() == "stop": elif list_name.lower() == "stop":
if await getTriviabyChannel(message.channel): if await get_trivia_by_channel(message.channel):
s = await getTriviabyChannel(message.channel) s = await get_trivia_by_channel(message.channel)
await s.endGame() await s.end_game()
await self.bot.say("`Trivia stopped.`") await self.bot.say("Trivia stopped.")
else: else:
await self.bot.say("`There's no trivia session ongoing in this channel.`") await self.bot.say("There's no trivia session ongoing in this channel.")
elif not await getTriviabyChannel(message.channel): elif not await get_trivia_by_channel(message.channel):
t = TriviaSession(message) t = TriviaSession(message)
self.trivia_sessions.append(t) self.trivia_sessions.append(t)
await t.loadQuestions(message.content) await t.load_questions(message.content)
else: else:
await self.bot.say("`A trivia session is already ongoing in this channel.`") await self.bot.say("A trivia session is already ongoing in this channel.")
async def triviaList(self, author): async def trivia_list(self, author):
msg = "**Available trivia lists:** \n\n```" msg = "**Available trivia lists:** \n\n```"
lists = os.listdir("data/trivia/") lists = os.listdir("data/trivia/")
if lists: if lists:
@ -63,51 +63,51 @@ class Trivia:
class TriviaSession(): class TriviaSession():
def __init__(self, message): def __init__(self, message):
self.gaveAnswer = ["I know this one! {}!", "Easy: {}.", "Oh really? It's {} of course."] self.gave_answer = ["I know this one! {}!", "Easy: {}.", "Oh really? It's {} of course."]
self.currentQ = None # {"QUESTION" : "String", "ANSWERS" : []} self.current_q = None # {"QUESTION" : "String", "ANSWERS" : []}
self.questionList = "" self.question_list = ""
self.channel = message.channel self.channel = message.channel
self.scoreList = {} self.score_list = {}
self.status = None self.status = None
self.timer = None self.timer = None
self.count = 0 self.count = 0
async def loadQuestions(self, msg): async def load_questions(self, msg):
msg = msg.split(" ") msg = msg.split(" ")
if len(msg) == 2: if len(msg) == 2:
_, qlist = msg _, qlist = msg
if qlist == "random": if qlist == "random":
chosenList = randchoice(glob.glob("data/trivia/*.txt")) chosen_list = randchoice(glob.glob("data/trivia/*.txt"))
self.questionList = self.loadList(chosenList) self.question_list = self.load_list(chosen_list)
self.status = "new question" self.status = "new question"
self.timeout = time.perf_counter() self.timeout = time.perf_counter()
if self.questionList: await self.newQuestion() if self.question_list: await self.new_question()
else: else:
if os.path.isfile("data/trivia/" + qlist + ".txt"): if os.path.isfile("data/trivia/" + qlist + ".txt"):
self.questionList = self.loadList("data/trivia/" + qlist + ".txt") self.question_list = self.load_list("data/trivia/" + qlist + ".txt")
self.status = "new question" self.status = "new question"
self.timeout = time.perf_counter() self.timeout = time.perf_counter()
if self.questionList: await self.newQuestion() if self.question_list: await self.new_question()
else: else:
await triviaManager.bot.say("`There is no list with that name.`") await trivia_manager.bot.say("There is no list with that name.")
await self.stopTrivia() await self.stop_trivia()
else: else:
await triviaManager.bot.say("`trivia [list name]`") await trivia_manager.bot.say("trivia [list name]")
async def stopTrivia(self): async def stop_trivia(self):
self.status = "stop" self.status = "stop"
triviaManager.trivia_sessions.remove(self) trivia_manager.trivia_sessions.remove(self)
async def endGame(self): async def end_game(self):
self.status = "stop" self.status = "stop"
if self.scoreList: if self.score_list:
await self.sendTable() await self.send_table()
triviaManager.trivia_sessions.remove(self) trivia_manager.trivia_sessions.remove(self)
def loadList(self, qlist): def load_list(self, qlist):
with open(qlist, "r", encoding="ISO-8859-1") as f: with open(qlist, "r", encoding="ISO-8859-1") as f:
qlist = f.readlines() qlist = f.readlines()
parsedList = [] parsed_list = []
for line in qlist: for line in qlist:
if "`" in line and len(line) > 4: if "`" in line and len(line) > 4:
line = line.replace("\n", "") line = line.replace("\n", "")
@ -118,98 +118,98 @@ class TriviaSession():
answers.append(l.lower()) answers.append(l.lower())
if len(line) >= 2: if len(line) >= 2:
line = {"QUESTION" : question, "ANSWERS": answers} #string, list line = {"QUESTION" : question, "ANSWERS": answers} #string, list
parsedList.append(line) parsed_list.append(line)
if parsedList != []: if parsed_list != []:
return parsedList return parsed_list
else: else:
self.stopTrivia() self.stop_trivia()
return None return None
async def newQuestion(self): async def new_question(self):
for score in self.scoreList.values(): for score in self.score_list.values():
if score == settings["TRIVIA_MAX_SCORE"]: if score == settings["TRIVIA_MAX_SCORE"]:
await self.endGame() await self.end_game()
return True return True
if self.questionList == []: if self.question_list == []:
await self.endGame() await self.end_game()
return True return True
self.currentQ = randchoice(self.questionList) self.current_q = randchoice(self.question_list)
self.questionList.remove(self.currentQ) self.question_list.remove(self.current_q)
self.status = "waiting for answer" self.status = "waiting for answer"
self.count += 1 self.count += 1
self.timer = int(time.perf_counter()) self.timer = int(time.perf_counter())
await triviaManager.bot.say("**Question number {}!**\n\n{}".format(str(self.count), self.currentQ["QUESTION"])) await trivia_manager.bot.say("**Question number {}!**\n\n{}".format(str(self.count), self.current_q["QUESTION"]))
while self.status != "correct answer" and abs(self.timer - int(time.perf_counter())) <= settings["TRIVIA_DELAY"]: while self.status != "correct answer" and abs(self.timer - int(time.perf_counter())) <= settings["TRIVIA_DELAY"]:
if abs(self.timeout - int(time.perf_counter())) >= settings["TRIVIA_TIMEOUT"]: if abs(self.timeout - int(time.perf_counter())) >= settings["TRIVIA_TIMEOUT"]:
await triviaManager.bot.say("Guys...? Well, I guess I'll stop then.") await trivia_manager.bot.say("Guys...? Well, I guess I'll stop then.")
await self.stopTrivia() await self.stop_trivia()
return True return True
await asyncio.sleep(1) #Waiting for an answer or for the time limit await asyncio.sleep(1) #Waiting for an answer or for the time limit
if self.status == "correct answer": if self.status == "correct answer":
self.status = "new question" self.status = "new question"
await asyncio.sleep(3) await asyncio.sleep(3)
if not self.status == "stop": if not self.status == "stop":
await self.newQuestion() await self.new_question()
elif self.status == "stop": elif self.status == "stop":
return True return True
else: else:
msg = randchoice(self.gaveAnswer).format(self.currentQ["ANSWERS"][0]) msg = randchoice(self.gave_answer).format(self.current_q["ANSWERS"][0])
if settings["TRIVIA_BOT_PLAYS"]: if settings["TRIVIA_BOT_PLAYS"]:
msg += " **+1** for me!" msg += " **+1** for me!"
self.addPoint(self.bot.user.name) self.add_point(self.bot.user.name)
self.currentQ["ANSWERS"] = [] self.current_q["ANSWERS"] = []
await triviaManager.bot.say(msg) await trivia_manager.bot.say(msg)
await triviaManager.bot.send_typing(self.channel) await trivia_manager.bot.send_typing(self.channel)
await asyncio.sleep(3) await asyncio.sleep(3)
if not self.status == "stop": if not self.status == "stop":
await self.newQuestion() await self.new_question()
async def sendTable(self): async def send_table(self):
self.scoreList = sorted(self.scoreList.items(), reverse=True, key=lambda x: x[1]) # orders score from lower to higher self.score_list = sorted(self.score_list.items(), reverse=True, key=lambda x: x[1]) # orders score from lower to higher
t = "```Scores: \n\n" t = "```Scores: \n\n"
for score in self.scoreList: for score in self.score_list:
t += score[0] # name t += score[0] # name
t += "\t" t += "\t"
t += str(score[1]) # score t += str(score[1]) # score
t += "\n" t += "\n"
t += "```" t += "```"
await triviaManager.bot.say(t) await trivia_manager.bot.say(t)
async def checkAnswer(self, message): async def check_answer(self, message):
self.timeout = time.perf_counter() self.timeout = time.perf_counter()
for answer in self.currentQ["ANSWERS"]: for answer in self.current_q["ANSWERS"]:
if answer in message.content.lower(): if answer in message.content.lower():
self.currentQ["ANSWERS"] = [] self.current_q["ANSWERS"] = []
self.status = "correct answer" self.status = "correct answer"
self.addPoint(message.author.name) self.add_point(message.author.name)
await triviaManager.bot.send_message(message.channel, "You got it {}! **+1** to you!".format(message.author.name)) await trivia_manager.bot.send_message(message.channel, "You got it {}! **+1** to you!".format(message.author.name))
await triviaManager.bot.send_typing(self.channel) await trivia_manager.bot.send_typing(self.channel)
return True return True
def addPoint(self, user): def add_point(self, user):
if user in self.scoreList: if user in self.score_list:
self.scoreList[user] += 1 self.score_list[user] += 1
else: else:
self.scoreList[user] = 1 self.score_list[user] = 1
def getTriviaQuestion(self): def get_trivia_question(self):
q = randchoice(list(trivia_questions.keys())) q = randchoice(list(trivia_questions.keys()))
return q, trivia_questions[q] # question, answer return q, trivia_questions[q] # question, answer
async def getTriviabyChannel(channel): async def get_trivia_by_channel(channel):
for t in triviaManager.trivia_sessions: for t in trivia_manager.trivia_sessions:
if t.channel == channel: if t.channel == channel:
return t return t
return False return False
async def checkMessages(message): async def check_messages(message):
if message.author.id != triviaManager.bot.user.id: if message.author.id != trivia_manager.bot.user.id:
if await getTriviabyChannel(message.channel): if await get_trivia_by_channel(message.channel):
trvsession = await getTriviabyChannel(message.channel) trvsession = await get_trivia_by_channel(message.channel)
await trvsession.checkAnswer(message) await trvsession.check_answer(message)
def setup(bot): def setup(bot):
global triviaManager global trivia_manager
bot.add_listener(checkMessages, "on_message") bot.add_listener(check_messages, "on_message")
triviaManager = Trivia(bot) trivia_manager = Trivia(bot)
bot.add_cog(triviaManager) bot.add_cog(trivia_manager)