Fixed is_banned check breaking on DM's

QuantumKat.py

@@ -24,7 +24,6 @@ async def init_models():
         await conn.run_sync(models.Base.metadata.create_all)
 
 
-
 log_helper = LogHelper()
 misc_helper = MiscHelper()
 discord_helper = DiscordHelper()
@@ -91,7 +90,7 @@ async def setup(bot: commands.Bot):
 
 
 async def is_authenticated(ctx: commands.Context) -> bool:
-    if ctx.author.id == int(OWNER_ID):
+    if ctx.author.id in bot.owner_ids:
         return True
     if not ctx.command.name.casefold() == "request_auth":
         authenticated_server_ids = await crud.get_authenticated_servers(
@@ -119,7 +118,7 @@ async def is_authenticated(ctx: commands.Context) -> bool:
 
 
 async def is_reboot_scheduled(ctx: commands.Context) -> bool:
-    if ctx.author.id == int(OWNER_ID):
+    if ctx.author.id in bot.owner_ids:
         return True
     if bot.reboot_scheduled:
         await ctx.reply(
@@ -131,26 +130,28 @@ async def is_reboot_scheduled(ctx: commands.Context) -> bool:
 
 
 async def is_banned(ctx: commands.Context) -> bool:
-    if ctx.author.id == int(OWNER_ID):
+    if ctx.author.id in bot.owner_ids:
         return True
     user = await crud.get_user(
         AsyncSessionLocal, schemas.User.Get(user_id=ctx.author.id)
     )
-    server = await crud.get_server(
-        AsyncSessionLocal, schemas.Server.Get(server_id=ctx.guild.id)
-    )
     if user and user.is_banned:
         await ctx.reply(
             "You have been banned from using QuantumKat. Please contact the bot owner for more information.",
             silent=True,
         )
         return False
-    if server and server.is_banned:
-        await ctx.reply(
-            "This server has been banned from using QuantumKat. Please contact the bot owner for more information.",
-            silent=True,
+
+    if not discord_helper.is_dm(ctx):
+        server = await crud.get_server(
+            AsyncSessionLocal, schemas.Server.Get(server_id=ctx.guild.id)
         )
-        return False
+        if server and server.is_banned:
+            await ctx.reply(
+                "This server has been banned from using QuantumKat. Please contact the bot owner for more information.",
+                silent=True,
+            )
+            return False
     return True