Context handling implemented

plugins/commands/set_offset.py

@@ -6,12 +6,13 @@ from pyrogram import filters
 from pyrogram.types import ForceReply, Message, ReplyKeyboardRemove
 
 from classes.pyroclient import PyroClient
+from modules import custom_filters
 
 logger = logging.getLogger(__name__)
 
 
 @PyroClient.on_message(
-    ~filters.scheduled & filters.private & filters.command(["set_offset"], prefixes=["/"])  # type: ignore
+    ~filters.scheduled & filters.private & filters.command(["set_offset"], prefixes=["/"]) & ~custom_filters.context  # type: ignore
 )
 async def command_set_offset(app: PyroClient, message: Message):
     user = await app.find_user(message.from_user)
@@ -24,7 +25,9 @@ async def command_set_offset(app: PyroClient, message: Message):
     )
 
     while True:
+        app.contexts.append(message.from_user.id)
         answer = await listen_message(app, message.chat.id, 300)
+        app.contexts.remove(message.from_user.id)
 
         if answer is None or answer.text == "/cancel":
             await message.reply_text(