mirror of
https://github.com/AsgardEternal/DiscordWhitelist.git
synced 2024-09-29 04:04:26 -05:00
61 lines
2.4 KiB
Python
61 lines
2.4 KiB
Python
import jsonpickle
|
|
import discord
|
|
import logging
|
|
from autowl import config
|
|
from autowl.bot import Bot
|
|
from discord.ext import commands
|
|
from discord import app_commands
|
|
|
|
|
|
log = logging.getLogger(__name__)
|
|
|
|
|
|
class Whitelist(commands.Cog):
|
|
def __init__(self, client: Bot):
|
|
self.client = client
|
|
|
|
@app_commands.command()
|
|
async def register(self, interaction: discord.Interaction, steam64: int):
|
|
if not interaction.guild:
|
|
await interaction.response.send_message(
|
|
"This command must be ran within a discord server!"
|
|
)
|
|
return
|
|
|
|
if not len(self.client.whitelist.keys()):
|
|
await interaction.response.send_message(
|
|
"There are no Whitelist roles defined, unable to continue!"
|
|
)
|
|
return
|
|
|
|
steam64_updated = False
|
|
for role in interaction.user.roles:
|
|
for group in self.client.whitelist:
|
|
if role.id == self.client.whitelist[group].discord_role_id:
|
|
steam64_updated = True
|
|
self.client.whitelist[group].members[
|
|
str(interaction.user.id)
|
|
] = config.WhitelistMember(interaction.user.name, steam64)
|
|
|
|
if steam64_updated:
|
|
outFile = open("test.json", "w")
|
|
outFile.write(jsonpickle.encode(self.client.whitelist))
|
|
outFile.close()
|
|
wlFile = open("testwl", "w")
|
|
for wlgrp in self.client.whitelist:
|
|
wlFile.write(f"Group={wlgrp}:reserve\n")
|
|
for wlgrp in self.client.whitelist:
|
|
wlFile.write("\n")
|
|
wlFile.write(f"//{wlgrp}\n")
|
|
for wlmem in self.client.whitelist[wlgrp].members:
|
|
wlFile.write(f"Admin={self.client.whitelist[wlgrp].members[wlmem].steam64}:{wlgrp} //{self.client.whitelist[wlgrp].members[wlmem].discord_username} ({wlmem})\n")
|
|
wlFile.close()
|
|
log.info(
|
|
f"Updated {interaction.user.name}'s ({interaction.user.id}) whitelist steam64 to {steam64}"
|
|
)
|
|
await interaction.response.send_message(
|
|
f"Updated `{interaction.user.name}`'s whitelist steam64 to `{steam64}`!"
|
|
)
|
|
else:
|
|
await interaction.response.send_message(f"Unable to update `{interaction.user.name}`'s whitelist steam64 as they are not in a valid Whitelisted group")
|