浏览代码

working on reaction roles

BaumSplitter41 1 周之前
父节点
当前提交
015f0a5bfc
共有 1 个文件被更改,包括 4 次插入7 次删除
  1. 4 7
      VPD_BOT/cogs/reaction_roles.py

+ 4 - 7
VPD_BOT/cogs/reaction_roles.py

@@ -107,11 +107,9 @@ class reactionroles(commands.Cog):
 
     #Add role to user
     @commands.Cog.listener()
-    async def on_raw_reaction_add(self, payload):
+    async def on_raw_reaction_add(self, payload: discord.RawReactionActionEvent):
         #Get variables
-        user = self.bot.get_user(payload.user_id)
-        user = self.bot.get(payload.user_id)
-        guild = self.bot.get_guild(payload.guild_id)
+        user = self.parse_reaction_payload(payload)
         if user.bot:
             return
         message_id = self._get_message_id()
@@ -137,10 +135,9 @@ class reactionroles(commands.Cog):
 
     #Remove role from user
     @commands.Cog.listener()
-    async def on_raw_reaction_remove(self, payload):
+    async def on_raw_reaction_remove(self, payload: discord.RawReactionActionEvent):
         #Get variables
-        user = self.bot._get_user(payload.user_id)
-        guild = self.bot._get_guild(payload.guild_id)
+        user = self.parse_reaction_payload(payload)
         if user.bot:
             return
         message_id = self._get_message_id()