You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
85 lines
2.8 KiB
85 lines
2.8 KiB
6 years ago
|
import discord
|
||
|
|
from redbot.core import Config, checks, commands
|
||
|
|
|
||
|
|
|
||
|
|
class ExclusiveRole:
|
||
|
|
"""
|
||
|
|
Custom commands
|
||
|
|
Creates commands used to display text and adjust roles
|
||
|
|
"""
|
||
|
|
|
||
|
|
def __init__(self, bot):
|
||
|
|
self.bot = bot
|
||
|
|
self.config = Config.get_conf(self, identifier=9999114111108101)
|
||
|
|
default_guild = {
|
||
|
|
"role_set": set()
|
||
|
|
}
|
||
|
|
|
||
|
|
self.config.register_guild(**default_guild)
|
||
|
|
|
||
|
|
@commands.group(no_pm=True)
|
||
|
|
async def exclusive(self, ctx):
|
||
|
|
"""Base command for managing exclusive roles"""
|
||
|
|
|
||
|
|
if not ctx.invoked_subcommand:
|
||
|
|
pass
|
||
|
|
|
||
|
|
@exclusive.command(name="add")
|
||
|
|
@checks.mod_or_permissions(administrator=True)
|
||
|
|
async def exclusive_add(self, ctx, role: discord.Role):
|
||
|
|
"""Adds an exclusive role"""
|
||
|
|
if role.id in (await self.config.guild(ctx.guild).role_list()):
|
||
|
|
await ctx.send("That role is already exclusive")
|
||
|
|
return
|
||
|
|
|
||
|
|
async with self.config.guild(ctx.guild).role_set() as rs:
|
||
|
|
rs.add(role.id)
|
||
|
|
|
||
|
|
await self.check_guild(ctx.guild)
|
||
|
|
|
||
|
|
await ctx.send("Exclusive role added")
|
||
|
|
|
||
|
|
@exclusive.command(name="delete")
|
||
|
|
@checks.mod_or_permissions(administrator=True)
|
||
|
|
async def exclusive_delete(self, ctx, role: discord.Role):
|
||
|
|
"""Deletes an exclusive role"""
|
||
|
|
if role.id not in (await self.config.guild(ctx.guild).role_list()):
|
||
|
|
await ctx.send("That role is not exclusive")
|
||
|
|
return
|
||
|
|
|
||
|
|
async with self.config.guild(ctx.guild).role_set() as rs:
|
||
|
|
rs.remove(role.id)
|
||
|
|
|
||
|
|
await ctx.send("Exclusive role removed")
|
||
|
|
|
||
|
|
async def check_guild(self, guild: discord.Guild):
|
||
|
|
role_set = await self.config.guild(guild).role_set()
|
||
|
|
for member in guild.members:
|
||
|
|
try:
|
||
|
|
await self.remove_non_exclusive_roles(member, role_set=role_set)
|
||
|
|
except discord.Forbidden:
|
||
|
|
pass
|
||
|
|
|
||
|
|
async def remove_non_exclusive_roles(self, member: discord.Member, role_set=None):
|
||
|
|
if role_set is None:
|
||
|
|
role_set = await self.config.guild(member.guild).role_set()
|
||
|
|
|
||
|
|
member_set = set([role.id for role in member.roles])
|
||
|
|
to_remove = member_set - role_set
|
||
|
|
|
||
|
|
if to_remove and member_set & role_set:
|
||
|
|
await member.remove_roles(*to_remove, "Exclusive roles")
|
||
|
|
|
||
|
|
async def on_member_update(self, before: discord.Member, after: discord.Member):
|
||
|
|
if before.roles == after.roles:
|
||
|
|
return
|
||
|
|
|
||
|
|
role_set = await self.config.guild(after.guild).role_set()
|
||
|
|
member_set = set([role.id for role in after.roles])
|
||
|
|
|
||
|
|
if role_set & member_set and member_set - role_set:
|
||
|
|
try:
|
||
|
|
await self.remove_non_exclusive_roles(after, role_set=role_set)
|
||
|
|
except discord.Forbidden:
|
||
|
|
pass
|