print errors to stderr and format
crispy-caesus 114518720+crispy-caesus@users.noreply.github.com
Mon, 11 Nov 2024 00:14:30 +0100
1 files changed,
117 insertions(+),
96 deletions(-)
jump to
M
bot.py
→
bot.py
@@ -3,6 +3,7 @@ import discord.ext.commands
import asyncio from varname import nameof import emoji +import sys import bot_token import db as database@@ -19,9 +20,11 @@ member_converter = discord.ext.commands.MemberConverter()
# ======================= INITIALIZATION ========================== # + @bot.event async def on_ready(): print(f'LOG: bot has logged in as {bot.user}') + @bot.listen() async def on_guild_join(guild):@@ -31,7 +34,7 @@ print(f"LOG: guild {guild} joined")
# ====================== DB INIT ========================= # -async def init_db(ctx, guild=None)-> database.Database: +async def init_db(ctx, guild=None) -> database.Database: if guild: return database.Database(f"{guild.id}.db") return database.Database(f"{ctx.guild.id}.db")@@ -39,28 +42,32 @@
# ======================== ERROR HANDLING ============================= # +def eprint(*args, **kwargs): + print(*args, file=sys.stderr, **kwargs) + + async def dm(ctx, message: str, err): user = await bot.fetch_user(621759962280099840) if user: messageDraft = message - if err != None: + if err is not None: messageDraft += "\n```{err}```" try: await user.send(messageDraft) - await ctx.send("Der Entwickler wurde kontaktiert und wird sich sobald wie möglich darum kümmern") + await ctx.send("Der Entwickler wurde kontaktiert und wird sich sobald wie möglich darum kümmern") except discord.Forbidden: - print(f"{ctx.guild.id} | ERROR: Can't DM user") + eprint(f"{ctx.guild.id} | ERROR: Can't DM user") await ctx.send("Bitte kontaktiere die Serverleitung") else: - print(f"{ctx.guild.id} | ERROR: Can't find user to DM") + eprint(f"{ctx.guild.id} | ERROR: Can't find user to DM") await ctx.send("Bitte kontaktiere die Serverleitung") -async def error(ctx, message: str, err = None): +async def error(ctx, message: str, err=None): decoratedMessage = f"{ctx.guild.id} | ERROR: {message}\n{err}" - print(decoratedMessage) + eprint(decoratedMessage) await dm(ctx, decoratedMessage, err=err) - + await ctx.send(":x: Interner Fehler")@@ -71,46 +78,51 @@
async def is_admin(ctx): if not ctx.author.guild_permissions.administrator: - ctx.respond(":x: Du musst Administrator sein, um diesen Command auszuführen") + ctx.respond( + ":x: Du musst Administrator sein, um diesen Command auszuführen") return False else: return True - # ======================= SETUP ========================= # -@bot.slash_command(description="Gibt alle Commands zurück, die für die Initialisation des Bots nötig sind") +@bot.slash_command(description="Gibt alle Commands zurück, die für die Initialisation des Bots nötig sind") async def setup(ctx): - if await is_admin(ctx) == False: + if await is_admin(ctx) is False: return db = await init_db(ctx) ids = await db.get_discord_ids() - + checks = [] for id in ids: - if id[2] == None: + if id[2] is None: checks.append("❌") else: checks.append("✅") - await ctx.respond( \ - f"{checks[0]} Setze die Booster Rolle mit `/{nameof(setze_booster_rolle)}`\n" \ - f"{checks[1]} Setze den Verteiler Channel mit `/{nameof(setze_verteiler_channel)}`\n" \ - f"{checks[2]} Setze die Kategorie, in der die Clubs erstellt werden sollen mit `/{nameof(setze_club_kategorie)}`\n" \ - f"{checks[3]} Setze die Rolle, welche als Clubrollenheader dient mit `/{nameof(setze_clubrollenheader_rolle)}`" \ - ) + await ctx.respond( + f"{checks[0] + } Setze die Booster Rolle mit `/{nameof(setze_booster_rolle)}`\n" + f"{checks[1]} Setze den Verteiler Channel mit `/{ + nameof(setze_verteiler_channel)}`\n" + f"{checks[2]} Setze die Kategorie, in der die Clubs erstellt werden sollen mit `/{ + nameof(setze_club_kategorie)}`\n" + f"{checks[3]} Setze die Rolle, welche als Clubrollenheader dient mit `/{ + nameof(setze_clubrollenheader_rolle)}`" + ) return @bot.slash_command(description="Setzt die Booster Rolle intern im Bot") async def setze_booster_rolle(ctx, booster_rolle): - if await is_admin(ctx) == False: + if await is_admin(ctx) is False: return db = await init_db(ctx) - err = await db.add_id("booster_role_id", booster_rolle[3:-1]) # [3:-1] to remove @<> around role ping + # [3:-1] to remove @<> around role ping + err = await db.add_id("booster_role_id", booster_rolle[3:-1]) if err: await error(ctx, "Set-Booster-Role: DB error", err) return@@ -122,11 +134,11 @@
@bot.slash_command(description="Setzt den Verteiler Channel intern im Bot") async def setze_verteiler_channel(ctx, verteiler_channel_id): - if await is_admin(ctx) == False: + if await is_admin(ctx) is False: return db = await init_db(ctx) - err = await db.add_id("distributor_channel_id", verteiler_channel_id) + err = await db.add_id("distributor_channel_id", verteiler_channel_id) if err: await error(ctx, "Set-Distributor-Channel: DB error", err) return@@ -138,10 +150,10 @@
@bot.slash_command(description="Setzt die Kategorie, in der die Clubs erstellt werden sollen") async def setze_club_kategorie(ctx, kategorie_id): - if await is_admin(ctx) == False: + if await is_admin(ctx) is False: return db = await init_db(ctx) - err = await db.add_id("new_channel_category_id", kategorie_id) + err = await db.add_id("new_channel_category_id", kategorie_id) if err: await error(ctx, "Set-New-Channel-Category: DB error", err) return@@ -152,19 +164,18 @@
@bot.slash_command(description="Setzt die Rolle für den Clubrollen-Header") async def setze_clubrollenheader_rolle(ctx, clubrollenheader_rolle: str): - if await is_admin(ctx) == False: + if await is_admin(ctx) is False: return clubrollenheader_rolle = clubrollenheader_rolle.strip() db = await init_db(ctx) err = await db.add_id("club_role_header_role_id", int(clubrollenheader_rolle[3:-1])) - if err != None: + if err is not None: await error(ctx, "Database error when adding club role header role id", err) return - + await ctx.respond(f"✅ Clubrollenheader {clubrollenheader_rolle} registriert") - """@@ -177,36 +188,41 @@
@bot.slash_command() async def test(ctx): - await ctx.guild.create_role(name = "hi", color=int("0x"+("#5460D9"[1:]), 16)+0x200, mentionable = False) + await ctx.guild.create_role(name="hi", color=int("0x"+("#5460D9"[1:]), 16)+0x200, mentionable=False) await ctx.respond("hi") # ======================== ADD CLUB ================================ # + async def check_club_parameters(ctx, db: database.Database, channelName="", channelEmoji="", roleName="", roleColor=""): if channelEmoji != "": if len(channelEmoji) != 1: - ctx.respond("❌ Error! Das Emoji-Feld darf nicht länger oder kürzer als 1 sein") - return - if emoji.is_emoji(channelEmoji) != True: - ctx.respond("❌ Error! Das Emoji-Feld muss mit einem Emoji gefüllt werden") - return + ctx.respond( + "❌ Error! Das Emoji-Feld darf nicht länger oder kürzer als 1 sein") + return + if emoji.is_emoji(channelEmoji) is not True: + ctx.respond( + "❌ Error! Das Emoji-Feld muss mit einem Emoji gefüllt werden") + return if channelName != "": if emoji.emoji_count(channelName) > 0: ctx.respond("❌ Error! Der Kanalname darf keine Emojis enthalten") - return + return combined_channel_name = f"「{channelEmoji}」{channelName}" existing_club = await db.select_club_by_channel_name(combined_channel_name) - if existing_club != None: - ctx.respond("❌ Error! Es gibt bereits einen Club mit diesem Kanalnamen") + if existing_club is not None: + ctx.respond( + "❌ Error! Es gibt bereits einen Club mit diesem Kanalnamen") return if roleName != "": existing_club_role = await db.select_club_by_role_name(roleName) - if existing_club_role != None: - ctx.respond("❌ Error! Es existiert bereits ein Club mit diesem Rollennamen") - return + if existing_club_role is not None: + ctx.respond( + "❌ Error! Es existiert bereits ein Club mit diesem Rollennamen") + return if roleColor != "": if roleColor[0] == '#':@@ -214,9 +230,9 @@ roleColor = roleColor[1:]
try: roleColor = int("0x"+roleColor, 16) except: - return("❌ Error! Farbformat falsch angegeben") + return ("❌ Error! Farbformat falsch angegeben") - return(roleName, roleColor) + return (roleName, roleColor) @bot.slash_command(description="Erstellt einen Booster Club")@@ -225,22 +241,22 @@ db = await init_db(ctx)
kanalemoji = kanalemoji.strip() boosterRoleId = await db.get_booster_role_id() - if boosterRoleId == None: + if boosterRoleId is None: await error(ctx, "Role-Creation: No booster role id foundin DB") return - if ctx.author.get_role(boosterRoleId) == None: + if ctx.author.get_role(boosterRoleId) is None: await ctx.respond(":x: Du bist kein Booster") return - if await db.select_role_id_by_owner(ctx.author.id) != None: + if await db.select_role_id_by_owner(ctx.author.id) is None: await ctx.respond("❌ Error! Du hast bereits einen Club") return check_return = await check_club_parameters(ctx, db, kanalname, kanalemoji, rollenname, rollenfarbe) - if check_return != None: + if check_return is not None: roleName, roleColor = check_return - createdRole = await ctx.guild.create_role(name = roleName, color = roleColor, mentionable = False) - if createdRole == None: + createdRole = await ctx.guild.create_role(name=roleName, color=roleColor, mentionable=False) + if createdRole is None: await error(ctx, "Role-Creation: Couldn't create role on discord") return err = await db.create_club(f"「{kanalemoji}」{kanalname}", ctx.author.id, createdRole.id, roleName)@@ -272,50 +288,56 @@ if rollenfarbe != "":
pass # ========================= dsfsf ======================= # + + @bot.slash_command(description="Fügt Member zu eigenem Club hinzu") async def mitglied_hinzufuegen(ctx, member): db = await init_db(ctx) response = await db.add_member(member[2:-1], ctx.author.id) - if response != None: + if response is not None: await ctx.respond(response) return member = await member_converter.convert(ctx, member) response = await db.select_role_id_by_owner(ctx.author.id) - if response == None: + if response is None: await ctx.respond("Du hast keinen Club") return role = discord.utils.get(ctx.guild.roles, id=response) - await member.add_roles(role) await ctx.respond(":white_check_mark:") await log(ctx, f"Added {member} to {role}") -@bot.command(description="Sends the bot's latency.") # this decorator makes a slash command -async def ping(ctx): # a slash command will be created with the name "ping" +# this decorator makes a slash command +@bot.command(description="Sends the bot's latency.") +async def ping(ctx): # a slash command will be created with the name "ping" await ctx.respond(f"Pong! Latency is {bot.latency}") # ==================== distributor vc ======================= # + async def send_club_list(ctx, user, db_response): clubs = "" for i in range(len(db_response)): - clubs += f"\n**{i+1}.** {db_response[i][0]}" # create a new numbered line for every club + # create a new numbered line for every club + clubs += f"\n**{i+1}.** {db_response[i][0]}" if clubs.strip() == "": - await ctx.send(f":x: {user.name}, du bist in keinen Clubs") # if no clubs + # if no clubs + await ctx.send(f":x: {user.name}, du bist in keinen Clubs") return else: await ctx.send(f" {user.name}, welchen Club-Kanal willst du öffnen?" + clubs) + async def get_and_check_user_response(ctx, user, db_response): def check(m): return m.author == user and m.channel == ctx - + cycle = 1 while cycle < 3:@@ -325,7 +347,7 @@ except asyncio.TimeoutError:
await ctx.send(":x: Zu spät (joine dem Kanal noch einmal, um die Dialogauswahl wieder zu erhalten)") return else: - try: + try: int(response.content) except ValueError: await ctx.send(":x: Konnte nicht in ganze Zahl umwandeln")@@ -334,18 +356,20 @@ if int(response.content) > len(db_response) and int(response.content) <= 0:
await ctx.send(":x: Nicht zulässige Zahl") else: return response - + cycle += 1 - ctx.send(":x: Zu viele Versuche, trete dem Channel erneut bei, um noch einmal auszuwählen") + ctx.send( + ":x: Zu viele Versuche, trete dem Channel erneut bei, um noch einmal auszuwählen") return None - + async def gather_arguments_for_channel_creation(ctx, db, db_response, userResponse): channel_name = db_response[int(userResponse.content)-1][0] new_channel_category_id = await db.get_discord_id("new_channel_category_id") - category = discord.utils.get(ctx.guild.categories, id=new_channel_category_id) + category = discord.utils.get( + ctx.guild.categories, id=new_channel_category_id) if category is None: await error(ctx, f"Distributor-Channel: Can't find new channel category ({new_channel_category_id}) on Discord")@@ -365,25 +389,25 @@ return (channel_name, category, role, bot_member, club_owner)
async def create_permission_overwrites(ctx, role, bot_member, club_owner): - bot_overwrites = discord.PermissionOverwrite ( - move_members = True, - view_channel = True, - manage_channels = True + bot_overwrites = discord.PermissionOverwrite( + move_members=True, + view_channel=True, + manage_channels=True ) - club_member_overwrites = discord.PermissionOverwrite ( - view_channel = True + club_member_overwrites = discord.PermissionOverwrite( + view_channel=True ) - club_owner_overwrites = discord.PermissionOverwrite ( - manage_channels = True, - mute_members = True, - deafen_members = True, - move_members = True + club_owner_overwrites = discord.PermissionOverwrite( + manage_channels=True, + mute_members=True, + deafen_members=True, + move_members=True ) - default_overwrites = discord.PermissionOverwrite ( - view_channel = False + default_overwrites = discord.PermissionOverwrite( + view_channel=False ) overwrites = {@@ -400,29 +424,30 @@ async def distributor_channel(user, after, db):
ctx = after.channel # ctx is distributor channel await log(ctx, f"{user.name} joined distributor channel {ctx.name}") db_response = await db.get_channel_name_role_name_by_member(user.id) - + await send_club_list(ctx, user, db_response) userResponse = await get_and_check_user_response(ctx, user, db_response) - if userResponse == None: + if userResponse is None: return channel_name, category, role, bot_member, club_owner = await gather_arguments_for_channel_creation(ctx, db, db_response, userResponse) - overwrites = await create_permission_overwrites(ctx, role, bot_member, club_owner) + overwrites = await create_permission_overwrites(ctx, role, bot_member, club_owner) voice_channel = await category.create_voice_channel( - name = channel_name, + name=channel_name, overwrites=overwrites ) await log(ctx, f"New channel {voice_channel} created") distributor_vcs.append(voice_channel.id) await log(ctx, f"The List of existing club channels is now: {distributor_vcs}") - + if user.voice: await user.move_to(voice_channel) await log(ctx, f"Moved {user} into {voice_channel}") distributor_vcs = [] + @bot.event async def on_voice_state_update(user, before, after):@@ -433,43 +458,42 @@ db = await init_db(after.channel)
if after.channel.id == await db.get_discord_id("distributor_channel_id"): await distributor_channel(user, after, db) - - - if before.channel and before.channel.id in distributor_vcs: # if old channel exists and is a distributor vc + # if old channel exists and is a distributor vc + if before.channel and before.channel.id in distributor_vcs: if len(before.channel.members) == 0: await before.channel.delete(reason="Niemand ist mehr verbunden") distributor_vcs.remove(before.channel.id) await log(before.channel, f"Deleted {before.channel} because it was empty") - + @bot.slash_command(description="Löscht eigenen Club") -async def club_löschen(ctx, owner = None): +async def club_löschen(ctx, owner=None): db = database.Database(f"{ctx.guild.id}.db") - if owner == None: + if owner is None: owner = ctx.author.id else: owner = owner[2:-1] club_role_id = await db.select_role_id_by_owner(owner) - if club_role_id == None: + if club_role_id is None: await ctx.respond(":x: Der angegebene Benutzer besitzt keinen Club") return - + err = await db.delete_club(owner) - if err != None: + if err is not None: await error(ctx, "Club-Deletion: Club from {owner} could not be deleted from the db", err) return else: print(f"LOG: club from {owner} has been deleted from the db") role = discord.utils.get(ctx.guild.roles, id=club_role_id) - + # Check if the role exists - if role == None: + if role is None: await error(ctx, f"Club-Deletion: Role {club_role_id} could not be found") return - + # Attempt to delete the role try: await role.delete()@@ -481,9 +505,6 @@ return
except discord.HTTPException as e: await error(ctx, "Club-Deletion: Error when deleting role", e) return - - - bot.run(bot_token.token)