Changed according to `new pyrogram version`

Powers/plugins/admin.py

@@ -2,6 +2,7 @@ from os import remove
 from html import escape
 from asyncio import sleep
 from pyrogram import filters
+from pyrogram.enums import ChatMemberStatus as CMS
 from Powers.vars import Config
 from traceback import format_exc
 from Powers.bot_class import Gojo
@@ -40,7 +41,8 @@ async def adminlist_show(_, m: Message):
         adminstr = f"Admins in <b>{m.chat.title}</b>:" + "\n\n"
 
         bot_admins = [i for i in admin_list if (i[1].lower()).endswith("bot")]
-        user_admins = [i for i in admin_list if not (i[1].lower()).endswith("bot")]
+        user_admins = [i for i in admin_list if not (
+            i[1].lower()).endswith("bot")]
 
         # format is like: (user_id, username/name,anonyamous or not)
         mention_users = [
@@ -189,7 +191,7 @@ async def fullpromote_usr(c: Gojo, m: Message):
         )  # This should be here
 
     user = await c.get_chat_member(m.chat.id, m.from_user.id)
-    if m.from_user.id not in [DEV_USERS, OWNER_ID] and user.status != "creator":
+    if m.from_user.id != OWNER_ID and user.status != CMS.OWNER:
         return await m.reply_text("This command can only be used by chat owner.")
     # If user is alreay admin
     try:
@@ -396,7 +398,7 @@ async def get_invitelink(c: Gojo, m: Message):
     if m.from_user.id not in DEV_LEVEL:
         user = await m.chat.get_member(m.from_user.id)
 
-        if not user.can_invite_users and user.status != "creator":
+        if not user.can_invite_users and user.status != CMS.OWNER:
             await m.reply_text(text="You don't have rights to invite users....")
             return False
 
@@ -427,7 +429,7 @@ async def get_invitelink(c: Gojo, m: Message):
 async def setgtitle(_, m: Message):
     user = await m.chat.get_member(m.from_user.id)
 
-    if not user.can_change_info and user.status != "creator":
+    if not user.can_change_info and user.status != CMS.OWNER:
         await m.reply_text(
             "You don't have enough permission to use this command!",
         )
@@ -450,7 +452,7 @@ async def setgtitle(_, m: Message):
 async def setgdes(_, m: Message):
 
     user = await m.chat.get_member(m.from_user.id)
-    if not user.can_change_info and user.status != "creator":
+    if not user.can_change_info and user.status != CMS.OWNER:
         await m.reply_text(
             "You don't have enough permission to use this command!",
         )
@@ -473,7 +475,7 @@ async def setgdes(_, m: Message):
 async def set_user_title(c: Gojo, m: Message):
 
     user = await m.chat.get_member(m.from_user.id)
-    if not user.can_promote_members and user.status != "creator":
+    if not user.can_promote_members and user.status != CMS.OWNER:
         await m.reply_text(
             "You don't have enough permission to use this command!",
         )
@@ -516,7 +518,7 @@ async def set_user_title(c: Gojo, m: Message):
 @Gojo.on_message(command("setgpic") & admin_filter)
 async def setgpic(c: Gojo, m: Message):
     user = await m.chat.get_member(m.from_user.id)
-    if not user.can_change_info and user.status != "creator":
+    if not user.can_change_info and user.status != CMS.OWNER:
         await m.reply_text(
             "You don't have enough permission to use this command!",
         )

Powers/plugins/approve.py

@@ -7,6 +7,7 @@ from Powers.database.approve_db import Approve
 from pyrogram.types import Message, CallbackQuery
 from Powers.utils.extract_user import extract_user
 from pyrogram.errors import RPCError, PeerIdInvalid, UserNotParticipant
+from pyrogram.enums import ChatMemberStatus as CMS
 from Powers.utils.custom_filters import command, admin_filter, owner_filter
 
 
@@ -37,7 +38,7 @@ async def approve_user(c: Gojo, m: Message):
             f"<b>Error</b>: <code>{ef}</code>\nReport it to @{SUPPORT_GROUP}",
         )
         return
-    if member.status in ("administrator", "creator"):
+    if member.status in (CMS.ADMINISTRATOR, CMS.OWNER):
         await m.reply_text(
             "User is already admin - blacklists and locks already don't apply to them.",
         )
@@ -97,7 +98,7 @@ async def disapprove_user(c: Gojo, m: Message):
         )
         return
 
-    if member.status in ("administrator", "creator"):
+    if member.status in (CMS.OWNER, CMS.ADMINISTRATOR):
         await m.reply_text("This user is an admin, they can't be disapproved.")
         return
 
@@ -202,7 +203,7 @@ async def unapproveall_callback(_, q: CallbackQuery):
     db = Approve(q.message.chat.id)
     approved_people = db.list_approved()
     user_status = (await q.message.chat.get_member(user_id)).status
-    if user_status not in {"creator", "administrator"}:
+    if user_status not in {CMS.OWNER, CMS.ADMINISTRATOR}:
         await q.answer(
             "You're not even an admin, don't try this explosive shit!",
             show_alert=True,

Powers/plugins/blacklist.py

@@ -4,6 +4,7 @@ from pyrogram import filters
 from Powers.bot_class import Gojo
 from Powers.utils.kbhelpers import ikb
 from pyrogram.types import Message, CallbackQuery
+from pyrogram.enums import ChatMemberStatus as CMS
 from Powers.database.blacklist_db import Blacklist
 from Powers.utils.custom_filters import command, owner_filter, restrict_filter
 
@@ -55,7 +56,8 @@ async def add_blacklist(_, m: Message):
             ", ".join([f"<code>{i}</code>" for i in bl_words])
             + " already added in blacklist, skipped them!"
         )
-    LOGGER.info(f"{m.from_user.id} added new blacklists ({bl_words}) in {m.chat.id}")
+    LOGGER.info(
+        f"{m.from_user.id} added new blacklists ({bl_words}) in {m.chat.id}")
     trigger = ", ".join(f"<code>{i}</code>" for i in bl_words)
     await m.reply_text(
         text=f"Added <code>{trigger}</code> in blacklist words!"
@@ -110,10 +112,12 @@ async def rm_blacklist(_, m: Message):
 
     if non_found_words:
         rep_text = (
-            "Could not find " + ", ".join(f"<code>{i}</code>" for i in non_found_words)
+            "Could not find " +
+            ", ".join(f"<code>{i}</code>" for i in non_found_words)
         ) + " in blcklisted words, skipped them."
 
-    LOGGER.info(f"{m.from_user.id} removed blacklists ({bl_words}) in {m.chat.id}")
+    LOGGER.info(
+        f"{m.from_user.id} removed blacklists ({bl_words}) in {m.chat.id}")
     bl_words = ", ".join(f"<code>{i}</code>" for i in bl_words)
     await m.reply_text(
         text=f"Removed <b>{bl_words}</b> from blacklist words!"
@@ -148,7 +152,8 @@ async def set_bl_action(_, m: Message):
         await m.reply_text(text=f"Set action for blacklist for this to <b>{action}</b>")
     elif len(m.text.split()) == 1:
         action = db.get_action()
-        LOGGER.info(f"{m.from_user.id} checking blacklist action in {m.chat.id}")
+        LOGGER.info(
+            f"{m.from_user.id} checking blacklist action in {m.chat.id}")
         await m.reply_text(
             text=f"""The current action for blacklists in this chat is <i><b>{action}</b></i>
       All blacklist modes delete the message containing blacklist word."""
@@ -184,13 +189,13 @@ async def rm_allbl_callback(_, q: CallbackQuery):
     user_id = q.from_user.id
     db = Blacklist(q.message.chat.id)
     user_status = (await q.message.chat.get_member(user_id)).status
-    if user_status not in {"creator", "administrator"}:
+    if user_status not in {CMS.ADMINISTRATOR, CMS.OWNER}:
         await q.answer(
             "You're not even an admin, don't try this explosive shit!",
             show_alert=True,
         )
         return
-    if user_status != "creator":
+    if user_status != CMS.OWNER:
         await q.answer(
             "You're just an admin, not owner\nStay in your limits!",
             show_alert=True,

Powers/plugins/disable.py

@@ -1,5 +1,6 @@
 from html import escape
 from pyrogram import filters
+from pyrogram.enums import ChatMemberStatus as CMS
 from Powers.bot_class import Gojo
 from Powers import LOGGER, HELP_COMMANDS
 from Powers.database.disable_db import Disabling
@@ -116,7 +117,8 @@ async def rm_alldisbl(_, m: Message):
                         "Confirm",
                         callback_data="enableallcmds",
                     ),
-                    InlineKeyboardButton("Cancel", callback_data="close_admin"),
+                    InlineKeyboardButton(
+                        "Cancel", callback_data="close_admin"),
                 ],
             ],
         ),
@@ -128,13 +130,13 @@ async def rm_alldisbl(_, m: Message):
 async def enablealll(_, q: CallbackQuery):
     user_id = q.from_user.id
     user_status = (await q.message.chat.get_member(user_id)).status
-    if user_status not in {"creator", "administrator"}:
+    if user_status not in {CMS.OWNER, CMS.ADMINISTRATOR}:
         await q.answer(
             "You're not even an admin, don't try this explosive shit!",
             show_alert=True,
         )
         return
-    if user_status != "creator":
+    if user_status != CMS.OWNER:
         await q.answer(
             "You're just an admin, not owner\nStay in your limits!",
             show_alert=True,

Powers/plugins/filters.py

@@ -10,6 +10,7 @@ from Powers.database.filters_db import Filters
 from Powers.utils.regex_utils import regex_searcher
 from Powers.utils.msg_types import Types, get_filter_type
 from pyrogram.types import Message, CallbackQuery, InlineKeyboardMarkup
+from pyrogram.enums import ChatMemberStatus as CMS
 from Powers.utils.custom_filters import command, admin_filter, owner_filter
 from Powers.utils.string import (
     parse_button, split_quotes, build_keyboard,
@@ -95,7 +96,8 @@ async def add_filter(_, m: Message):
         )
 
     add = db.save_filter(m.chat.id, keyword, teks, msgtype, file_id)
-    LOGGER.info(f"{m.from_user.id} added new filter ({keyword}) in {m.chat.id}")
+    LOGGER.info(
+        f"{m.from_user.id} added new filter ({keyword}) in {m.chat.id}")
     if add:
         await m.reply_text(
             f"Saved filter for '<code>{', '.join(keyword.split('|'))}</code>' in <b>{m.chat.title}</b>!",
@@ -119,7 +121,8 @@ async def stop_filter(_, m: Message):
     for keyword in act_filters:
         if keyword == m.text.split(None, 1)[1].lower():
             db.rm_filter(m.chat.id, m.text.split(None, 1)[1].lower())
-            LOGGER.info(f"{m.from_user.id} removed filter ({keyword}) in {m.chat.id}")
+            LOGGER.info(
+                f"{m.from_user.id} removed filter ({keyword}) in {m.chat.id}")
             await m.reply_text(
                 f"Okay, I'll stop replying to that filter and it's aliases in <b>{m.chat.title}</b>.",
             )
@@ -154,13 +157,13 @@ async def rm_allfilters(_, m: Message):
 async def rm_allfilters_callback(_, q: CallbackQuery):
     user_id = q.from_user.id
     user_status = (await q.message.chat.get_member(user_id)).status
-    if user_status not in {"creator", "administrator"}:
+    if user_status not in {CMS.OWNER, CMS.ADMINISTRATOR}:
         await q.answer(
             "You're not even an admin, don't try this explosive shit!",
             show_alert=True,
         )
         return
-    if user_status != "creator":
+    if user_status != CMS.OWNER:
         await q.answer(
             "You're just an admin, not owner\nStay in your limits!",
             show_alert=True,
@@ -280,7 +283,8 @@ async def filters_watcher(c: Gojo, m: Message):
         if match:
             try:
                 msgtype = await send_filter_reply(c, m, trigger)
-                LOGGER.info(f"Replied with {msgtype} to {trigger} in {m.chat.id}")
+                LOGGER.info(
+                    f"Replied with {msgtype} to {trigger} in {m.chat.id}")
             except Exception as ef:
                 await m.reply_text(f"Error: {ef}")
                 LOGGER.error(ef)

Powers/plugins/greetings.py

@@ -9,6 +9,7 @@ from Powers.database.antispam_db import GBan
 from Powers.database.greetings_db import Greetings
 from Powers.utils.msg_types import Types, get_wlcm_type
 from pyrogram.errors import RPCError, ChatAdminRequired
+from pyrogram.enums import ChatMemberStatus as CMS
 from Powers.utils.parser import mention_html, escape_markdown
 from pyrogram.types import Message, ChatMemberUpdated, InlineKeyboardMarkup
 from Powers.utils.custom_filters import command, admin_filter, bot_admin_filter
@@ -238,7 +239,7 @@ async def member_has_joined(c: Gojo, member: ChatMemberUpdated):
 
     if (
         member.new_chat_member
-        and member.new_chat_member.status not in {"banned", "left", "restricted"}
+        and member.new_chat_member.status not in {CMS.BANNED, CMS.LEFT, CMS.RESTRICTED}
         and not member.old_chat_member
     ):
         pass
@@ -321,7 +322,7 @@ async def member_has_left(c: Gojo, member: ChatMemberUpdated):
 
     if (
         not member.new_chat_member
-        and member.old_chat_member.status not in {"banned", "restricted"}
+        and member.old_chat_member.status not in {CMS.BANNED, CMS.RESTRICTED}
         and member.old_chat_member
     ):
         pass

Powers/plugins/notes.py

@@ -5,6 +5,7 @@ from traceback import format_exc
 from Powers.bot_class import Gojo
 from pyrogram import enums, filters
 from pyrogram.errors import RPCError
+from pyrogram.enums import ChatMemberStatus as CMS
 from Powers.utils.kbhelpers import ikb
 from Powers.utils.cmd_senders import send_cmd
 from Powers.utils.msg_types import Types, get_note_type
@@ -73,7 +74,8 @@ async def get_note_func(c: Gojo, m: Message, note_name, priv_notes_status):
 
     if priv_notes_status:
 
-        note_hash = next(i[1] for i in db.get_all_notes(m.chat.id) if i[0] == note_name)
+        note_hash = next(i[1] for i in db.get_all_notes(
+            m.chat.id) if i[0] == note_name)
         await reply_text(
             f"Click on the button to get the note <code>{note_name}</code>",
             reply_markup=ikb(
@@ -299,11 +301,13 @@ async def priv_notes(_, m: Message):
         option = (m.text.split())[1]
         if option in ("on", "yes"):
             db_settings.set_privatenotes(chat_id, True)
-            LOGGER.info(f"{m.from_user.id} enabled privatenotes in {m.chat.id}")
+            LOGGER.info(
+                f"{m.from_user.id} enabled privatenotes in {m.chat.id}")
             msg = "Set private notes to On"
         elif option in ("off", "no"):
             db_settings.set_privatenotes(chat_id, False)
-            LOGGER.info(f"{m.from_user.id} disabled privatenotes in {m.chat.id}")
+            LOGGER.info(
+                f"{m.from_user.id} disabled privatenotes in {m.chat.id}")
             msg = "Set private notes to Off"
         else:
             msg = "Enter correct option"
@@ -311,7 +315,8 @@ async def priv_notes(_, m: Message):
     elif len(m.text.split()) == 1:
         curr_pref = db_settings.get_privatenotes(m.chat.id)
         msg = msg = f"Private Notes: {curr_pref}"
-        LOGGER.info(f"{m.from_user.id} fetched privatenotes preference in {m.chat.id}")
+        LOGGER.info(
+            f"{m.from_user.id} fetched privatenotes preference in {m.chat.id}")
         await m.reply_text(msg)
     else:
         await m.replt_text("Check help on how to use this command!")
@@ -399,13 +404,13 @@ async def clear_allnote(_, m: Message):
 async def clearallnotes_callback(_, q: CallbackQuery):
     user_id = q.from_user.id
     user_status = (await q.message.chat.get_member(user_id)).status
-    if user_status not in {"creator", "administrator"}:
+    if user_status not in {CMS.OWNER, CMS.ADMINISTRATOR}:
         await q.answer(
             "You're not even an admin, don't try this explosive shit!",
             show_alert=True,
         )
         return
-    if user_status != "creator":
+    if user_status != CMS.OWNER:
         await q.answer(
             "You're just an admin, not owner\nStay in your limits!",
             show_alert=True,

Powers/plugins/pin.py

@@ -5,6 +5,7 @@ from html import escape as escape_html
 from Powers import LOGGER, SUPPORT_GROUP
 from Powers.database.pins_db import Pins
 from pyrogram.types import Message, CallbackQuery
+from pyrogram.enums import ChatMemberStatus as CMS
 from Powers.utils.string import parse_button, build_keyboard
 from Powers.utils.custom_filters import command, admin_filter
 from pyrogram.errors import RPCError, RightForbidden, ChatAdminRequired
@@ -101,13 +102,13 @@ async def unpinall_message(_, m: Message):
 async def unpinall_calllback(c: Gojo, q: CallbackQuery):
     user_id = q.from_user.id
     user_status = (await q.message.chat.get_member(user_id)).status
-    if user_status not in {"creator", "administrator"}:
+    if user_status not in {CMS.OWNER, CMS.ADMINISTRATOR}:
         await q.answer(
             "You're not even an admin, don't try this explosive shit!",
             show_alert=True,
         )
         return
-    if user_status != "creator":
+    if user_status != CMS.OWNER:
         await q.answer(
             "You're just an admin, not owner\nStay in your limits!",
             show_alert=True,

Powers/plugins/start.py

@@ -9,6 +9,7 @@ from Powers.utils.chat_type import chattype
 from Powers.utils.custom_filters import command
 from pyrogram.types import Message, CallbackQuery
 from pyrogram.errors import UserIsBlocked, QueryIdInvalid, MessageNotModified
+from pyrogram.enums import ChatMemberStatus as CMS
 from Powers.utils.start_utils import (
     gen_cmds_kb, gen_start_kb, get_help_msg, get_private_note,
     get_private_rules)
@@ -37,13 +38,13 @@ You can donate by contacting my owner: [Captain Ezio](http://t.me/iamgojoof6eyes
 async def close_admin_callback(_, q: CallbackQuery):
     user_id = q.from_user.id
     user_status = (await q.message.chat.get_member(user_id)).status
-    if user_status not in {"creator", "administrator"}:
+    if user_status not in {CMS.OWNER, CMS.ADMINISTRATOR}:
         await q.answer(
             "You're not even an admin, don't try this explosive shit!",
             show_alert=True,
         )
         return
-    if user_status != "creator":
+    if user_status != CMS.OWNER:
         await q.answer(
             "You're just an admin, not owner\nStay in your limits!",
             show_alert=True,
@@ -69,7 +70,8 @@ async def start(c: Gojo, m: Message):
                 await get_private_note(c, m, help_option)
                 return
             if help_option.startswith("rules"):
-                LOGGER.info(f"{m.from_user.id} fetched privaterules in {m.chat.id}")
+                LOGGER.info(
+                    f"{m.from_user.id} fetched privaterules in {m.chat.id}")
                 await get_private_rules(c, m, help_option)
                 return
 
@@ -181,7 +183,8 @@ async def help_menu(_, m: Message):
         help_msg, help_kb = await get_help_msg(m, help_option)
 
         if not help_msg:
-            LOGGER.error(f"No help_msg found for help_option - {help_option}!!")
+            LOGGER.error(
+                f"No help_msg found for help_option - {help_option}!!")
             return
 
         LOGGER.info(

Powers/utils/admin_check.py

@@ -1,5 +1,6 @@
 from traceback import format_exc
 from pyrogram.types import Message, CallbackQuery
+from pyrogram.enums import ChatMemberStatus as CMS
 from Powers import LOGGER, OWNER_ID, DEV_USERS, SUDO_USERS
 
 
@@ -21,7 +22,7 @@ async def admin_check(m: Message or CallbackQuery) -> bool:
         LOGGER.error(format_exc())
 
     user = await m.chat.get_member(user_id)
-    admin_strings = ("creator", "administrator")
+    admin_strings = (CMS.OWNER, CMS.ADMINISTRATOR)
 
     if user.status not in admin_strings:
         reply = "Nigga, you're not admin, don't try this explosive shit."
@@ -48,9 +49,9 @@ async def check_rights(m: Message or CallbackQuery, rights) -> bool:
         app = m.message._client
 
     user = await app.get_chat_member(chat_id, user_id)
-    if user.status == "member":
+    if user.status == CMS.MEMBER:
         return False
-    admin_strings = ("creator", "administrator")
+    admin_strings = (CMS.OWNER, CMS.ADMINISTRATOR)
     if user.status in admin_strings:
         return bool(getattr(user, rights, None))
     return False
@@ -73,8 +74,8 @@ async def owner_check(m: Message or CallbackQuery) -> bool:
 
     user = await m.chat.get_member(user_id)
 
-    if user.status != "creator":
-        if user.status == "administrator":
+    if user.status != CMS.OWNER:
+        if user.status == CMS.ADMINISTRATOR:
             reply = "Stay in your limits, or lose adminship too."
         else:
             reply = "You ain't even admin, what are you trying to do?"

Powers/utils/custom_filters.py

@@ -3,6 +3,7 @@ from pyrogram import enums
 from Powers.vars import Config
 from typing import List, Union
 from pyrogram.filters import create
+from pyrogram.enums import ChatMemberStatus as CMS
 from Powers.utils.chat_type import chattype
 from re import escape, compile as compile_re
 from Powers.database.disable_db import Disabling
@@ -73,14 +74,14 @@ def command(
                     user_status = (await m.chat.get_member(m.from_user.id)).status
                 except UserNotParticipant:
                     # i.e anon admin
-                    user_status = "administrator"
+                    user_status = CMS.ADMINISTRATOR
                 except ValueError:
                     # i.e. PM
-                    user_status = "creator"
+                    user_status = CMS.OWNER
                 ddb = Disabling(m.chat.id)
                 if str(matches.group(1)) in ddb.get_disabled() and user_status not in (
-                    "creator",
-                    "administrator",
+                    CMS.OWNER,
+                    CMS.ADMINISTRATOR,
                 ):
                     if bool(ddb.get_action() == "del"):
                         try:
@@ -194,11 +195,11 @@ async def owner_check_func(_, __, m: Message or CallbackQuery):
 
     user = await m.chat.get_member(m.from_user.id)
 
-    if user.status == "creator":
+    if user.status == CMS.OWNER:
         status = True
     else:
         status = False
-        if user.status == "administrator":
+        if user.status == CMS.ADMINISTRATOR:
             msg = "You're an admin only, stay in your limits!"
         else:
             msg = "Do you think that you can execute owner commands?"
@@ -222,7 +223,7 @@ async def restrict_check_func(_, __, m: Message or CallbackQuery):
 
     user = await m.chat.get_member(m.from_user.id)
 
-    if user.can_restrict_members or user.status == "creator":
+    if user.can_restrict_members or user.status == CMS.OWNER:
         status = True
     else:
         status = False
@@ -245,7 +246,7 @@ async def promote_check_func(_, __, m):
 
     user = await m.chat.get_member(m.from_user.id)
 
-    if user.can_promote_members or user.status == "creator":
+    if user.can_promote_members or user.status == CMS.OWNER:
         status = True
     else:
         status = False
@@ -274,7 +275,7 @@ async def changeinfo_check_func(_, __, m):
 
     user = await m.chat.get_member(m.from_user.id)
 
-    if user.can_change_info or user.status == "creator":
+    if user.can_change_info or user.status == CMS.OWNER:
         status = True
     else:
         status = False
@@ -302,7 +303,7 @@ async def can_pin_message_func(_, __, m):
 
     user = await m.chat.get_member(m.from_user.id)
 
-    if user.can_pin_messages or user.status == "creator":
+    if user.can_pin_messages or user.status == CMS.OWNER:
         status = True
     else:
         status = False