From bfc675df7990b78c53f8250bb97d7fbcbfd42ea1 Mon Sep 17 00:00:00 2001 From: Giovani Date: Fri, 30 Jul 2021 17:24:21 -0400 Subject: [PATCH] feat: add ability to exclusive command on channel --- cogs/fun.py | 2 -- model/poll.py | 4 ---- pyvis.py | 5 ++++- 3 files changed, 4 insertions(+), 7 deletions(-) diff --git a/cogs/fun.py b/cogs/fun.py index da0238e..61d2403 100644 --- a/cogs/fun.py +++ b/cogs/fun.py @@ -115,8 +115,6 @@ class Fun(commands.Cog): # get poll from cache poll = Poll.create_from(json.loads(polls[event.message_id])) - print(event) - # check reaction emoji type if event.event_type == "REACTION_ADD": poll.add_vote(event.user_id, event.emoji.name) diff --git a/model/poll.py b/model/poll.py index 8dbf61d..853e308 100644 --- a/model/poll.py +++ b/model/poll.py @@ -32,8 +32,6 @@ class Poll: # only add vote if user has not already voted if user_id not in self.votes[option]: self.votes[option].append(user_id) - - print(self.votes) # TODO: remove def remove_vote(self, user_id: str, emoji: str) -> None: option = self._get_option(emoji) @@ -41,8 +39,6 @@ class Poll: # only remove if user has voted if user_id in self.votes[option]: self.votes[option].remove(user_id) - - print(self.votes) # TODO: remove def get_emoji(self, option: str) -> str: return Poll.OPTION_EMOJIS[self.options.index(option)] diff --git a/pyvis.py b/pyvis.py index c69360d..0e27590 100644 --- a/pyvis.py +++ b/pyvis.py @@ -55,7 +55,10 @@ async def on_message(message): if message.author == bot.user: 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 if bot.user.mentioned_in(message) and not message.mention_everyone: