Improved typing and linting

modules/commands/application.py

@@ -2,7 +2,9 @@ from datetime import datetime
 from app import app, isAnAdmin
 from pyrogram import filters
 from pyrogram.enums.parse_mode import ParseMode
+from pyrogram.types import Message
 from pyrogram.errors import bad_request_400
+from pyrogram.client import Client
 from classes.holo_user import HoloUser, UserNotFoundError
 from modules.utils import logWrite, locale, should_quote
 from dateutil.relativedelta import relativedelta
@@ -10,7 +12,7 @@ from modules.database import col_applications
 
 # Applications command =========================================================================================================
 @app.on_message(~ filters.scheduled & filters.command(["application"], prefixes=["/"]))
-async def cmd_application(app, msg):
+async def cmd_application(app: Client, msg: Message):
 
     if await isAnAdmin(msg.from_user.id) is True:
 

modules/commands/applications.py

@@ -2,6 +2,8 @@ from os import sep, makedirs, remove
 from uuid import uuid1
 from app import app, isAnAdmin
 from pyrogram import filters
+from pyrogram.types import Message
+from pyrogram.client import Client
 from pyrogram.enums.chat_action import ChatAction
 from modules.logging import logWrite
 from modules.utils import should_quote, jsonSave
@@ -9,7 +11,7 @@ from modules.database import col_applications
 
 # Applications command =========================================================================================================
 @app.on_message(~ filters.scheduled & filters.command(["applications"], prefixes=["/"]))
-async def cmd_applications(app, msg):
+async def cmd_applications(app: Client, msg: Message):
 
     if await isAnAdmin(msg.from_user.id) is True:
         logWrite(f"Admin {msg.from_user.id} requested export of a database")

modules/commands/cancel.py

@@ -1,7 +1,9 @@
 from app import app
 from pyrogram import filters
+from pyrogram.types import Message
+from pyrogram.client import Client
 from modules.utils import should_quote
 
 @app.on_message(~ filters.scheduled & filters.command("cancel", prefixes=["/"]))
-async def command_cancel(app, msg):
+async def command_cancel(app: Client, msg: Message):
     await msg.reply_text("Command exists.", quote=should_quote(msg))

modules/commands/label.py

@@ -1,10 +1,12 @@
 from app import app, isAnAdmin
 from pyrogram import filters
+from pyrogram.types import Message
+from pyrogram.client import Client
 from modules.utils import locale, should_quote, find_user
 from classes.holo_user import HoloUser, LabelTooLongError
 
 @app.on_message(~ filters.scheduled & filters.private & filters.command(["label"], prefixes=["/"]))
-async def cmd_label(app, msg):
+async def cmd_label(app: Client, msg: Message):
 
     if await isAnAdmin(msg.from_user.id) is True:
 

modules/commands/message.py

@@ -1,11 +1,13 @@
 from app import app, isAnAdmin
 from pyrogram import filters
+from pyrogram.types import Message
+from pyrogram.client import Client
 from classes.holo_user import HoloUser
 from modules.utils import logWrite, locale, should_quote
 
 # Message command ==============================================================================================================
 @app.on_message(~ filters.scheduled & filters.command(["message"], prefixes=["/"]))
-async def cmd_message(app, msg):
+async def cmd_message(app: Client, msg: Message):
 
     if await isAnAdmin(msg.from_user.id) is True:
 

modules/commands/nearby.py

@@ -1,11 +1,13 @@
 from app import app, isAnAdmin
 from pyrogram import filters
+from pyrogram.types import Message
+from pyrogram.client import Client
 from modules.utils import configGet, should_quote
 from modules.database import col_applications
 
 # Nearby command ===============================================================================================================
 @app.on_message(~ filters.scheduled & (filters.private | (filters.chat(configGet("admin_group")) | filters.chat(configGet("destination_group")))) & filters.command(["nearby"], prefixes=["/"]))
-async def cmd_nearby(app, msg):
+async def cmd_nearby(app: Client, msg: Message):
     if (await isAnAdmin(msg) is True) or (col_applications.find_one({"user": msg.from_user.id}) is not None):
         await msg.reply_text("Yes, I exist.", quote=should_quote(msg))
     # if not path.exists(f"{configGet('data', 'locations')}{sep}sponsors{sep}{msg.from_user.id}.json"):

modules/commands/reapply.py

@@ -1,6 +1,7 @@
 from app import app
 from pyrogram import filters
-from pyrogram.types import InlineKeyboardMarkup, InlineKeyboardButton
+from pyrogram.types import InlineKeyboardMarkup, InlineKeyboardButton, Message
+from pyrogram.client import Client
 from classes.holo_user import HoloUser
 from modules.utils import configGet, locale, should_quote
 from modules.handlers.welcome import welcome_pass
@@ -8,7 +9,7 @@ from modules.database import col_tmp
 
 # Reapply command ==============================================================================================================
 @app.on_message(~ filters.scheduled & filters.private & filters.command(["reapply"], prefixes=["/"]))
-async def cmd_reapply(app, msg):
+async def cmd_reapply(app: Client, msg: Message):
 
     holo_user = HoloUser(msg.from_user)
 

modules/commands/reboot.py

@@ -2,6 +2,8 @@ from app import app, isAnAdmin
 from os import getpid
 from sys import exit
 from pyrogram import filters
+from pyrogram.types import Message
+from pyrogram.client import Client
 from modules.utils import locale, logWrite, should_quote
 from modules.scheduled import scheduler
 
@@ -9,7 +11,7 @@ pid = getpid()
 
 # Shutdown command =============================================================================================================
 @app.on_message(~ filters.scheduled & filters.private & filters.command(["kill", "die", "reboot"], prefixes=["/"]))
-async def cmd_kill(app, msg):
+async def cmd_kill(app: Client, msg: Message):
 
     if await isAnAdmin(msg.from_user.id) is True:
         logWrite(f"Shutting down bot with pid {pid}")

modules/commands/rules.py

@@ -1,7 +1,8 @@
 from typing import Union
 from app import app
 from pyrogram import filters
-from pyrogram.types import InlineKeyboardMarkup, InlineKeyboardButton, User
+from pyrogram.types import InlineKeyboardMarkup, InlineKeyboardButton, User, Message
+from pyrogram.client import Client
 from modules.utils import locale
 from classes.holo_user import HoloUser
 
@@ -35,6 +36,6 @@ class DefaultRulesMarkup(list):
 
 # Rules command =============================================================================================================
 @app.on_message(~ filters.scheduled & filters.private & filters.command(["rules"], prefixes=["/"]))
-async def cmd_rules(app, msg):
+async def cmd_rules(app: Client, msg: Message):
     await msg.reply_text(locale("rules_msg", locale=msg.from_user), disable_web_page_preview=True, reply_markup=DefaultRulesMarkup(msg.from_user).keyboard)
 # ==============================================================================================================================

modules/commands/sponsorship.py

@@ -1,13 +1,14 @@
 from app import app, isAnAdmin
 from pyrogram import filters
-from pyrogram.types import InlineKeyboardMarkup, InlineKeyboardButton
+from pyrogram.types import InlineKeyboardMarkup, InlineKeyboardButton, Message
+from pyrogram.client import Client
 from classes.holo_user import HoloUser
 from modules.utils import locale, should_quote
 from modules.database import col_applications
 
 # Sponsorship command ==========================================================================================================
 @app.on_message(~ filters.scheduled & filters.command(["sponsorship"], prefixes=["/"]))
-async def cmd_sponsorship(app, msg):
+async def cmd_sponsorship(app: Client, msg: Message):
     if (await isAnAdmin(msg) is True) or (col_applications.find_one({"user": msg.from_user.id}) is not None):
         if HoloUser(msg.from_user).application_state()[0] == "fill":
             await msg.reply_text(locale("finish_application", "message"), quote=should_quote(msg))

modules/commands/start.py

@@ -1,12 +1,13 @@
 from app import app
 from pyrogram import filters
-from pyrogram.types import ReplyKeyboardMarkup
+from pyrogram.types import ReplyKeyboardMarkup, Message
+from pyrogram.client import Client
 from modules.utils import locale, logWrite
 from modules.database import col_users
 
 # Start command ================================================================================================================
 @app.on_message(~ filters.scheduled & filters.private & filters.command(["start"], prefixes=["/"]))
-async def cmd_start(app, msg):
+async def cmd_start(app: Client, msg: Message):
 
     user = col_users.find_one({"user": msg.from_user.id})
 

modules/commands/warn.py

@@ -1,12 +1,14 @@
 from datetime import datetime
 from app import app, isAnAdmin
 from pyrogram import filters
+from pyrogram.types import Message
+from pyrogram.client import Client
 from modules.utils import configGet, locale
 from modules.database import col_warnings
 
 # Warn command =================================================================================================================
 @app.on_message(~ filters.scheduled & filters.command(["warn"], prefixes=["/"]))
-async def cmd_warn(app, msg):
+async def cmd_warn(app: Client, msg: Message):
 
     if msg.chat.id == configGet("destination_group"):
         if msg.reply_to_message_id != None:

modules/commands/warnings.py

@@ -1,12 +1,14 @@
 from app import app, isAnAdmin
 from pyrogram import filters
+from pyrogram.types import Message
+from pyrogram.client import Client
 from pyrogram.enums.chat_members_filter import ChatMembersFilter
 from modules.utils import configGet, locale, should_quote
 from modules.database import col_users, col_warnings
 
 # Warnings command =============================================================================================================
 @app.on_message(~ filters.scheduled & filters.command(["warnings"], prefixes=["/"]))
-async def cmd_warnings(app, msg):
+async def cmd_warnings(app: Client, msg: Message):
 
     if await isAnAdmin(msg.from_user.id) is True: