Fix permission admins

akn/Akeno/admin.py

@@ -127,7 +127,7 @@ async def member_ban_user(client: Client, message: Message):
     user_id, reason = await extract_user_and_reason(message, sender_chat=True)
     rd = await message.edit_text("`Processing...`")
     bot = (await client.get_chat_member(message.chat.id, client.me.id)).privileges
-    if not bot.can_restrict_members:
+    if not bot or not bot.can_restrict_members:
         return await rd.edit_text("I don't have enough permissions")
     if not user_id:
         return await rd.edit_text("I can't find that user.")
@@ -165,8 +165,7 @@ async def member_ban_user(client: Client, message: Message):
 async def member_unban_user(client: Client, message: Message):
     reply = message.reply_to_message
     rd = await message.edit_text("`Processing...`")
-    bot = (await client.get_chat_member(message.chat.id, client.me.id)).privileges
-    if not bot.can_restrict_members:
+    if not bot or not bot.can_restrict_members:
         return await rd.edit_text("I don't have enough permissions")
     if reply and reply.sender_chat and reply.sender_chat != message.chat.id:
         return await rd.edit_text("You cannot unban a channel")
@@ -192,7 +191,7 @@ async def pin_message(client: Client, message: Message):
         return await message.edit_text("Reply to a message to pin/unpin it.")
     rd = await message.edit_text("`Processing...`")
     bot = (await client.get_chat_member(message.chat.id, client.me.id)).privileges
-    if not bot.can_pin_messages:
+    if not bot or not bot.can_pin_messages:
         return await rd.edit_text("I don't have enough permissions")
     r = message.reply_to_message
     if message.command[0][0] == "u":
@@ -214,7 +213,7 @@ async def mute_user(client: Client, message: Message):
     user_id, reason = await extract_user_and_reason(message)
     rd = await message.edit_text("`Processing...`")
     bot = (await client.get_chat_member(message.chat.id, client.me.id)).privileges
-    if not bot.can_restrict_members:
+    if not bot or not bot.can_restrict_members:
         return await rd.edit_text("I don't have enough permissions")
     if not user_id:
         return await rd.edit_text("I can't find that user.")
@@ -245,7 +244,7 @@ async def unmute_user(client: Client, message: Message):
     user_id = await extract_user(message)
     rd = await message.edit_text("`Processing...`")
     bot = (await client.get_chat_member(message.chat.id, client.me.id)).privileges
-    if not bot.can_restrict_members:
+    if not bot or not bot.can_restrict_members:
         return await rd.edit_text("I don't have enough permissions")
     if not user_id:
         return await rd.edit_text("I can't find that user.")
@@ -260,7 +259,7 @@ async def kick_user(client: Client, message: Message):
     user_id, reason = await extract_user_and_reason(message)
     rd = await message.edit_text("`Processing...`")
     bot = (await client.get_chat_member(message.chat.id, client.me.id)).privileges
-    if not bot.can_restrict_members:
+    if not bot or not bot.can_restrict_members:
         return await rd.edit_text("I don't have enough permissions")
     if not user_id:
         return await rd.edit_text("I can't find that user.")
@@ -298,7 +297,7 @@ async def promotte_user(client: Client, message: Message):
     if not user_id:
         return await rd.edit_text("I can't find that user.")
     bot = (await client.get_chat_member(message.chat.id, client.me.id)).privileges
-    if not bot.can_promote_members:
+    if not bot or not bot.can_promote_members:
         return await rd.edit_text("I don't have enough permissions")
     if message.command[0][0] == "f":
         await message.chat.promote_member(
@@ -341,6 +340,9 @@ async def demote_user(client: Client, message: Message):
         return await rd.edit_text("I can't find that user.")
     if user_id == client.me.id:
         return await rd.edit_text("I can't demote myself.")
+    bot = (await client.get_chat_member(message.chat.id, client.me.id)).privileges
+    if not bot or not bot.can_promote_members:
+        return await rd.edit_text("I don't have enough permissions")
     await message.chat.promote_member(
         user_id,
         privileges=ChatPrivileges(