prepare the help module and general cleanup of files
This commit is contained in:
65
cogs/flair.py
Normal file
65
cogs/flair.py
Normal file
@@ -0,0 +1,65 @@
|
||||
import discord
|
||||
from discord.ext import commands
|
||||
from logging import getLogger
|
||||
from jeevesbot import env
|
||||
|
||||
# setup logging
|
||||
log = getLogger(__name__)
|
||||
e = discord.Embed()
|
||||
|
||||
|
||||
class Flair(commands.Cog):
|
||||
""" This part of the bot is responsible for giving users roles! """
|
||||
def __init__(self, bot):
|
||||
self.bot = bot
|
||||
self.message_id = 1084937456819908679
|
||||
self.reactions = {
|
||||
'1️⃣' : env.FLAIRROLES[0],
|
||||
'2️⃣' : env.FLAIRROLES[1],
|
||||
'3️⃣' : env.FLAIRROLES[2],
|
||||
'4️⃣' : env.FLAIRROLES[3],
|
||||
'5️⃣' : env.FLAIRROLES[4],
|
||||
'6️⃣' : env.FLAIRROLES[5],
|
||||
'👍' : env.FLAIRROLES[6],
|
||||
'👎' : env.FLAIRROLES[7],
|
||||
'🔞' : env.FLAIRROLES[8],
|
||||
}
|
||||
|
||||
|
||||
@commands.Cog.listener()
|
||||
async def on_raw_reaction_add(self, payload):
|
||||
try:
|
||||
if payload.message_id == self.message_id and str(payload.emoji) in self.reactions:
|
||||
guild = self.bot.get_guild(payload.guild_id)
|
||||
member = guild.get_member(payload.user_id)
|
||||
role_id = self.reactions[str(payload.emoji)]
|
||||
role = guild.get_role(role_id)
|
||||
flairrole = guild.get_role(env.FLAIRROLE[0])
|
||||
await member.add_roles(flairrole)
|
||||
await member.add_roles(role)
|
||||
log.info(f'Added role "{role}" to {member}')
|
||||
except:
|
||||
log.debug("Reaction not found.")
|
||||
|
||||
|
||||
@commands.Cog.listener()
|
||||
async def on_raw_reaction_remove(self, payload):
|
||||
try:
|
||||
if payload.message_id == self.message_id and str(payload.emoji) in self.reactions:
|
||||
guild = self.bot.get_guild(payload.guild_id)
|
||||
member = guild.get_member(payload.user_id)
|
||||
role_id = self.reactions[str(payload.emoji)]
|
||||
role = guild.get_role(role_id)
|
||||
await member.remove_roles(role)
|
||||
log.info(f'Removed role "{role}" from {member}')
|
||||
except:
|
||||
log.debug("Could not remove role")
|
||||
|
||||
|
||||
@commands.Cog.listener()
|
||||
async def on_ready(self):
|
||||
log.info(f'module active')
|
||||
|
||||
|
||||
async def setup(bot):
|
||||
await bot.add_cog(Flair(bot))
|
||||
Reference in New Issue
Block a user