fix: Resolve flooddb crash and correct admin permissions

DragMusic/utils/database.py

@@ -1,8 +1,9 @@
 import random
+from functools import wraps
 from typing import Dict, List, Union
 
 from DragMusic import userbot
-from DragMusic.core.mongo import mongodb
+from DragMusic.core.mongo import mongodb, flooddb, sudoersdb
 
 authdb = mongodb.adminauth
 authuserdb = mongodb.authuser
@@ -19,7 +20,6 @@ onoffdb = mongodb.onoffper
 playmodedb = mongodb.playmode
 playtypedb = mongodb.playtypedb
 skipdb = mongodb.skipmode
-sudoersdb = mongodb.sudoers
 usersdb = mongodb.tgusersdb
 
 # Shifting to memory [mongo sucks often]

DragMusic/utils/decorators/admins.py

@@ -1,4 +1,4 @@
-from pyrogram.enums import ChatType
+from pyrogram.enums import ChatType, ChatMemberStatus
 from pyrogram.types import InlineKeyboardButton, InlineKeyboardMarkup
 
 from DragMusic import app
@@ -126,11 +126,17 @@ def AdminActual(mystic):
         if message.from_user.id not in SUDOERS:
             try:
                 member = await app.get_chat_member(message.chat.id, message.from_user.id)
-                if not member.privileges.can_promote_members:
-                    return await message.reply_text(_["admin_18"])
-            except:
-                return await message.reply_text("You are not an admin.")
+                
+                if member.status == ChatMemberStatus.OWNER:
+                    pass  # Owner is always allowed
+                elif member.status == ChatMemberStatus.ADMINISTRATOR and member.privileges.can_promote_members:
+                    pass  # Admin with promote rights is allowed
+                else:
+                    return await message.reply_text(_["admin_18"]) # "You don't have permission to promote users."
 
+            except Exception:
+                return await message.reply_text("You are not an admin in this chat.")
+        
         return await mystic(client, message, _)
 
     return wrapper