Add admin check for locks

userge/plugins/admin/locks.py

@@ -15,6 +15,18 @@
 CHANNEL = userge.getCLogger(__name__)
 
 
+async def is_admin(message: Message):
+    check_user = await userge.get_chat_member(message.chat.id, message.from_user.id)
+    user_type = check_user.status
+    if user_type == "member":
+        return False
+    if user_type == "administrator":
+        return True
+    if user_type == "creator":
+        return True
+    return False
+
+
 @userge.on_cmd("lock", about={
     'header': "use this to lock group permissions",
     'description': "Allows you to lock some common permission types in the chat.\n"
@@ -28,6 +40,10 @@ async def lock_perm(message: Message):
     """
     lock chat permissions from tg group
     """
+    if not await is_admin(message):
+        await message.edit("Are you even Admin of this chat?")
+        return
+
     msg = ""
     media = ""
     stickers = ""
@@ -168,6 +184,10 @@ async def unlock_perm(message: Message):
     """
     unlock chat permissions from tg group
     """
+    if not await is_admin(message):
+        await message.edit("Are you even Admin of this chat?")
+        return
+
     umsg = ""
     umedia = ""
     ustickers = ""
@@ -313,6 +333,9 @@ async def view_perm(message: Message):
     """
     check chat permissions from tg group
     """
+    if not await is_admin(message):
+        await message.edit("Are you even Admin of this chat?")
+        return
 
     v_perm = ""
     vmsg = ""