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

[General] Multi character prefix support for polls (#689)

Browse Source
This commit is contained in:
flapjax 2017-06-08 19:05:22 -04:00 committed by Twentysix
parent 820f62f94f
commit 3b93bb776a
1 changed files with 3 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

7
cogs/general.py
View File

@ -339,7 +339,7 @@ class General:
if "@everyone" in check or "@here" in check: if "@everyone" in check or "@here" in check:
await self.bot.say("Nice try.") await self.bot.say("Nice try.")
return return
p = NewPoll(message, self) p = NewPoll(message, " ".join(text), self)
if p.valid: if p.valid:
self.poll_sessions.append(p) self.poll_sessions.append(p)
await p.start() await p.start()
@ -377,13 +377,12 @@ class General:
return user.joined_at return user.joined_at
class NewPoll(): class NewPoll():
def __init__(self, message, main): def __init__(self, message, text, main):
self.channel = message.channel self.channel = message.channel
self.author = message.author.id self.author = message.author.id
self.client = main.bot self.client = main.bot
self.poll_sessions = main.poll_sessions self.poll_sessions = main.poll_sessions
msg = message.content[6:] msg = [ans.strip() for ans in text.split(";")]
msg = msg.split(";")
if len(msg) < 2: # Needs at least one question and 2 choices if len(msg) < 2: # Needs at least one question and 2 choices
self.valid = False self.valid = False
return None return None