crayonix/pyvis
0
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

feat: add ability to exclusive command on channel

Browse Source
This commit is contained in:
Giovani
2021-07-30 17:24:21 -04:00
parent db1bb161d6
commit bfc675df79
3 changed files with 4 additions and 7 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
cogs/fun.py
View File

@@ -115,8 +115,6 @@ class Fun(commands.Cog):
# get poll from cache # get poll from cache
poll = Poll.create_from(json.loads(polls[event.message_id])) poll = Poll.create_from(json.loads(polls[event.message_id]))
print(event)
# check reaction emoji type # check reaction emoji type
if event.event_type == "REACTION_ADD": if event.event_type == "REACTION_ADD":
poll.add_vote(event.user_id, event.emoji.name) poll.add_vote(event.user_id, event.emoji.name)

4
model/poll.py
View File

@@ -32,8 +32,6 @@ class Poll:
# only add vote if user has not already voted # only add vote if user has not already voted
if user_id not in self.votes[option]: if user_id not in self.votes[option]:
self.votes[option].append(user_id) self.votes[option].append(user_id)
print(self.votes) # TODO: remove
def remove_vote(self, user_id: str, emoji: str) -> None: def remove_vote(self, user_id: str, emoji: str) -> None:
option = self._get_option(emoji) option = self._get_option(emoji)
@@ -41,8 +39,6 @@ class Poll:
# only remove if user has voted # only remove if user has voted
if user_id in self.votes[option]: if user_id in self.votes[option]:
self.votes[option].remove(user_id) self.votes[option].remove(user_id)
print(self.votes) # TODO: remove
def get_emoji(self, option: str) -> str: def get_emoji(self, option: str) -> str:
return Poll.OPTION_EMOJIS[self.options.index(option)] return Poll.OPTION_EMOJIS[self.options.index(option)]

5
pyvis.py
View File

@@ -55,7 +55,10 @@ async def on_message(message):
if message.author == bot.user: if message.author == bot.user:
return return
# TODO: if channel is a proposal channel check for poll command # if channel is proposals and message is not a poll command then delete
if message.channel.name == "📝proposals" and not message.content.startswith("!poll create") and not message.content.startswith("!poll results"):
await message.delete()
return
# repsond with command prefix # repsond with command prefix
if bot.user.mentioned_in(message) and not message.mention_everyone: if bot.user.mentioned_in(message) and not message.mention_everyone: