mirror of
https://github.com/AsgardEternal/DiscordWhitelist.git
synced 2024-09-29 12:14:27 -05:00
82 lines
2.7 KiB
Python
82 lines
2.7 KiB
Python
import os.path
|
|
|
|
import jsonpickle
|
|
import logging
|
|
import discord
|
|
from discord.ext import commands
|
|
from autowl import config
|
|
|
|
log = logging.getLogger(__name__)
|
|
|
|
|
|
class Bot(commands.Bot):
|
|
whitelistGrps = {}
|
|
|
|
if not os.path.exists('./wlgrps'):
|
|
os.makedirs('./wlgrps')
|
|
else:
|
|
for wlfile in os.listdir('./wlgrps'):
|
|
filename = os.fsdecode(wlfile)
|
|
if filename.endswith('.json'):
|
|
file = open(f'./wlgrps/{filename}', 'r')
|
|
wlgrp: config.WhitelistGroup = jsonpickle.decode(file.read())
|
|
whitelistGrps[wlgrp.discord_role_id] = wlgrp
|
|
|
|
def __init__(self, config: config.DiscordClientConfig):
|
|
self.config = config
|
|
intents = discord.Intents.default()
|
|
intents.message_content = True
|
|
intents.members = True
|
|
super().__init__(
|
|
command_prefix=commands.when_mentioned_or("&"),
|
|
intents=intents,
|
|
help_command=commands.DefaultHelpCommand(dm_help=True),
|
|
)
|
|
|
|
async def on_command(self, ctx: commands.Context):
|
|
log.info(f"{ctx.author} ({ctx.author.id}) invoked command: {ctx.command.name}, {ctx.message}")
|
|
|
|
async def on_ready(self):
|
|
log.info(f"Logged in as '{self.user}' ({self.user.id})")
|
|
log.info(
|
|
f"Have access to the following guilds: "
|
|
f"{', '.join([str(guild.name) + ' (' + str(guild.id) + ')' for guild in self.guilds])}"
|
|
)
|
|
|
|
for guild in self.guilds:
|
|
self.tree.copy_global_to(guild=guild)
|
|
await self.tree.sync(guild=guild)
|
|
log.info(f"Synced guild: {guild.name}")
|
|
|
|
await self.tree.sync()
|
|
|
|
async def on_member_update(self, before: discord.Member, after: discord.Member):
|
|
log.info(f"Updating {after.name} ({after.id})")
|
|
rmroles = []
|
|
for befrole in before.roles:
|
|
rmroles.append(befrole.id)
|
|
for aftrole in after.roles:
|
|
for befrole in before.roles:
|
|
if befrole.id == aftrole.id:
|
|
rmroles.remove(aftrole.id)
|
|
log.info(f"roles found to remove from {after.name}: {rmroles}")
|
|
for rmroleid in rmroles:
|
|
self.whitelistGrps[rmroleid].delMember(before.id)
|
|
|
|
async def setup_hook(self):
|
|
log.info("Setting up bot")
|
|
from autowl import Cogs
|
|
|
|
await self.add_cog(Cogs.Whitelist(self))
|
|
await self.add_cog(Cogs.Group(self))
|
|
|
|
def start_bot(self):
|
|
log.info("Starting discord bot")
|
|
|
|
try:
|
|
self.run(self.config.login_token, log_handler=None)
|
|
except discord.errors.LoginFailure as e:
|
|
log.debug(f"Received login failure: {e}")
|
|
log.error("Failed to login to discord, check your discord token!")
|
|
raise e
|