94 lines
4.0 KiB
Python
94 lines
4.0 KiB
Python
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):
|
||
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.command(name='flairsetup', hidden=True)
|
||
@commands.is_owner()
|
||
async def flairsetup(self, ctx):
|
||
""" Setup reaction post as embed. """
|
||
await ctx.message.delete()
|
||
embed = discord.Embed(title="Kies je rollen!",
|
||
description="Klik op de emoji onder het bericht om de rol te krijgen.")
|
||
embed.add_field(name="", value="", inline=False)
|
||
embed.add_field(name="Voornaamwoorden:", value="", inline=False)
|
||
embed.add_field(name=":one:", value="hij/hem")
|
||
embed.add_field(name=":two:", value="zij/haar")
|
||
embed.add_field(name=":three:", value="hen/hun")
|
||
embed.add_field(name=":four:", value="die/hun")
|
||
embed.add_field(name=":five:", value="die/diens")
|
||
embed.add_field(name=":six:", value="iedere/all")
|
||
embed.add_field(name="Sta je open voor Direct Messages van andere serverleden?", value="", inline=False)
|
||
embed.add_field(name=":thumbsup:", value="DM: ja")
|
||
embed.add_field(name=":thumbsdown:", value="DM: nee")
|
||
embed.add_field(name="Rollen voor toegang tot opt-in kanalen.", value="", inline=False)
|
||
embed.add_field(name=":underage:", value="serieuze-onderwerpen")
|
||
embed.set_footer(text="Mis je een voornaamwoord of heb je ideeën voor een andere rol? Laat het de mods weten in de #ideeënbus!\nJe kan zelf je rollen verwijderen door opnieuw op de emoji te drukken.")
|
||
message = await ctx.send(embed=embed)
|
||
await message.add_reaction("1️⃣")
|
||
await message.add_reaction("2️⃣")
|
||
await message.add_reaction("3️⃣")
|
||
await message.add_reaction("4️⃣")
|
||
await message.add_reaction("5️⃣")
|
||
await message.add_reaction("6️⃣")
|
||
await message.add_reaction("👍")
|
||
await message.add_reaction("👎")
|
||
await message.add_reaction("🔞")
|
||
log.warn(f'{ctx.message.author} reset the flair embed, new message_id = {message.id}')
|
||
|
||
@commands.Cog.listener()
|
||
async def on_ready(self):
|
||
log.info(f'module active')
|
||
|
||
async def setup(bot):
|
||
await bot.add_cog(Flair(bot))
|