Improved linting and removed unused imports

classes/poster_client.py

@@ -2,9 +2,7 @@ from os import path, remove, sep
 from shutil import rmtree
 from typing import Union
 from pyrogram.client import Client
-from pyrogram.types import Message, CallbackQuery
-from pyrogram.enums.parse_mode import ParseMode
-from pyrogram.session.session import Session
+from pyrogram.types import Message
 from classes.exceptions import SubmissionDuplicatesError, SubmissionUnavailableError
 from modules.api_client import upload_pic
 from modules.database import col_submitted

classes/user.py

@@ -9,16 +9,28 @@ class PosterUser():
         self.id = id
 
     def is_blocked(self) -> bool:
+        """Check if user is banned from submitting content.
+
+        ### Returns:
+            `bool`: Must be `True` if banned and `False` if not
+        """  
         return False if col_banned.find_one({"user": self.id}) is None else True
     
     def block(self) -> None:
+        """Ban user from using command and submitting content."""        
         if col_banned.find_one({"user": self.id}) is None:
             col_banned.insert_one({"user": self.id, "date": datetime.now()})
 
     def unblock(self) -> None:
+        """Allow user to use command and submit posts again."""        
         col_banned.find_one_and_delete({"user": self.id})
 
     def is_limited(self) -> bool:
+        """Check if user is on a cooldown after submitting something.
+
+        ### Returns:
+            `bool`: Must be `True` if on the cooldown and `False` if not
+        """        
         if self.id in app.admins:
             return False
         else:
@@ -28,5 +40,6 @@ class PosterUser():
             return True if (datetime.now() - db_record["cooldown"]).total_seconds() < configGet("timeout", "submission") else False
         
     def limit(self) -> None:
+        """Restart user's cooldown. Used after post has been submitted."""        
         if col_users.find_one_and_update({"user": self.id}, {"$set": {"cooldown": datetime.now()}}) is None:
             col_users.insert_one({"user": self.id, "cooldown": datetime.now()})