Moved cooldowns to PosterUser class

plugins/handlers/submission.py

@@ -1,4 +1,4 @@
-from datetime import datetime, timezone
+from datetime import datetime
 from os import makedirs, path, sep
 from traceback import format_exc
 from uuid import uuid4
@@ -8,11 +8,11 @@ from pyrogram.types import InlineKeyboardButton, InlineKeyboardMarkup, Message
 from pyrogram.enums.chat_action import ChatAction
 from classes.exceptions import SubmissionDuplicatesError
 from classes.poster_client import PosterClient
+from classes.user import PosterUser
 
 from modules.app import app
 from modules.database import col_banned, col_submitted
 from modules.logger import logWrite
-from modules.submissions import subLimit, subLimited
 from modules.utils import configGet, locale
 from classes.enums.submission_types import SubmissionType
 
@@ -31,7 +31,7 @@ async def get_submission(app: PosterClient, msg: Message):
         save_tmp = True
         contents = None
 
-        if subLimited(msg.from_user):
+        if PosterUser(msg.from_user.id).is_limited():
             await msg.reply_text(locale("sub_cooldown", "message", locale=user_locale).format(str(configGet("timeout", "submission"))))
             return
 
@@ -180,7 +180,7 @@ async def get_submission(app: PosterClient, msg: Message):
                 # ]
             ]
 
-            subLimit(msg.from_user)
+            PosterUser(msg.from_user.id).limit()
 
         if msg.from_user.id != app.owner:
             await msg.reply_text(locale("sub_sent", "message", locale=user_locale), quote=True)