Prototype: Files import

plugins/handlers/submission.py

@@ -4,17 +4,17 @@ from traceback import format_exc
 from uuid import uuid4
 
 from pyrogram import filters
-from pyrogram.types import InlineKeyboardButton, InlineKeyboardMarkup, Message
 from pyrogram.enums.chat_action import ChatAction
+from pyrogram.types import InlineKeyboardButton, InlineKeyboardMarkup, Message
+
+from classes.enums.submission_types import SubmissionType
 from classes.exceptions import SubmissionDuplicatesError
 from classes.poster_client import PosterClient
 from classes.user import PosterUser
-
-from modules.app import app
+from modules.app import app, users_with_context
 from modules.database import col_banned, col_submitted
 from modules.logger import logWrite
 from modules.utils import configGet, locale
-from classes.enums.submission_types import SubmissionType
 
 
 @app.on_message(
@@ -24,6 +24,9 @@ from classes.enums.submission_types import SubmissionType
     | filters.document
 )
 async def get_submission(app: PosterClient, msg: Message):
+    global users_with_context
+    if msg.from_user.id in users_with_context:
+        return
     try:
         if col_banned.find_one({"user": msg.from_user.id}) is not None:
             return