Now using black for formatting

classes/enums/submission_types.py

@@ -5,4 +5,4 @@ class SubmissionType(Enum):
     DOCUMENT = "document"
     VIDEO = "video"
     ANIMATION = "animation"
-    PHOTO = "photo"
+    PHOTO = "photo"

classes/exceptions.py

@@ -4,13 +4,19 @@ from typing import Any
 class SubmissionUnavailableError(Exception):
     pass
 
+
 class SubmissionUploadError(Exception):
     def __init__(self, file_path: str, status_code: int, content: Any) -> None:
         self.status_code = status_code
         self.content = content
-        super().__init__(f"Could not upload photo '{file_path}' due to HTTP {self.status_code}: {self.content}")
+        super().__init__(
+            f"Could not upload photo '{file_path}' due to HTTP {self.status_code}: {self.content}"
+        )
+
 
 class SubmissionDuplicatesError(Exception):
     def __init__(self, file_path: str, duplicates: list) -> None:
         self.duplicates = duplicates
-        super().__init__(f"Found duplicates of a photo '{file_path}': {self.duplicates}")
+        super().__init__(
+            f"Found duplicates of a photo '{file_path}': {self.duplicates}"
+        )

classes/poster_client.py

@@ -11,15 +11,14 @@ from modules.logger import logWrite
 
 from modules.utils import configGet
 
-class PosterClient(Client):
 
-    def __init__(self, name: str, **kwargs): # type: ignore
+class PosterClient(Client):
+    def __init__(self, name: str, **kwargs):  # type: ignore
         super().__init__(name, **kwargs)
         self.owner = configGet("owner")
-        self.admins = configGet("admins")+[configGet("owner")]
+        self.admins = configGet("admins") + [configGet("owner")]
 
     async def submit_photo(self, id: str) -> Union[Message, None]:
-        
         db_entry = col_submitted.find_one({"_id": ObjectId(id)})
         submission = None
 
@@ -27,35 +26,66 @@ class PosterClient(Client):
             raise SubmissionUnavailableError()
         else:
             if db_entry["temp"]["uuid"] is not None:
-                if not path.exists(path.join(configGet("data", "locations"), "submissions", db_entry["temp"]["uuid"], db_entry["temp"]["file"])):
+                if not path.exists(
+                    path.join(
+                        configGet("data", "locations"),
+                        "submissions",
+                        db_entry["temp"]["uuid"],
+                        db_entry["temp"]["file"],
+                    )
+                ):
                     raise SubmissionUnavailableError()
                 else:
-                    filepath = path.join(configGet("data", "locations"), "submissions", db_entry["temp"]["uuid"], db_entry["temp"]["file"])
+                    filepath = path.join(
+                        configGet("data", "locations"),
+                        "submissions",
+                        db_entry["temp"]["uuid"],
+                        db_entry["temp"]["file"],
+                    )
                     try:
-                        submission = await self.get_messages(db_entry["user"], db_entry["telegram"]["msg_id"])
+                        submission = await self.get_messages(
+                            db_entry["user"], db_entry["telegram"]["msg_id"]
+                        )
                     except:
                         pass
             else:
                 try:
-                    submission = await self.get_messages(db_entry["user"], db_entry["telegram"]["msg_id"])
-                    filepath = await self.download_media(submission, file_name=configGet("tmp", "locations")+sep)
+                    submission = await self.get_messages(
+                        db_entry["user"], db_entry["telegram"]["msg_id"]
+                    )
+                    filepath = await self.download_media(
+                        submission, file_name=configGet("tmp", "locations") + sep
+                    )
                 except:
                     raise SubmissionUnavailableError()
-                
-        response = await upload_pic(str(filepath), allow_duplicates=configGet("allow_duplicates", "submission"))
+
+        response = await upload_pic(
+            str(filepath), allow_duplicates=configGet("allow_duplicates", "submission")
+        )
 
         if len(response[1]) > 0:
             raise SubmissionDuplicatesError(str(filepath), response[1])
-        
-        col_submitted.find_one_and_update({"_id": ObjectId(id)}, {"$set": {"done": True}})
+
+        col_submitted.find_one_and_update(
+            {"_id": ObjectId(id)}, {"$set": {"done": True}}
+        )
 
         try:
             if db_entry["temp"]["uuid"] is not None:
-                rmtree(path.join(configGet("data", "locations"), "submissions", db_entry["temp"]["uuid"]), ignore_errors=True)
+                rmtree(
+                    path.join(
+                        configGet("data", "locations"),
+                        "submissions",
+                        db_entry["temp"]["uuid"],
+                    ),
+                    ignore_errors=True,
+                )
             else:
                 remove(str(filepath))
         except (FileNotFoundError, NotADirectoryError):
-            logWrite(f"Could not delete '{filepath}' on submission accepted", debug=True)
+            logWrite(
+                f"Could not delete '{filepath}' on submission accepted", debug=True
+            )
 
         return submission
 
@@ -63,4 +93,4 @@ class PosterClient(Client):
         pass
 
     async def unban_user(self, id: int) -> None:
-        pass
+        pass

classes/user.py

@@ -3,8 +3,8 @@ from datetime import datetime
 from modules.database import col_banned, col_users
 from modules.utils import configGet
 
-class PosterUser():
 
+class PosterUser:
     def __init__(self, id: int):
         self.id = id
 
@@ -13,16 +13,16 @@ class PosterUser():
 
         ### 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."""        
+        """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."""        
+        """Allow user to use command and submit posts again."""
         col_banned.find_one_and_delete({"user": self.id})
 
     def is_limited(self) -> bool:
@@ -30,16 +30,26 @@ class PosterUser():
 
         ### Returns:
             `bool`: Must be `True` if on the cooldown and `False` if not
-        """        
+        """
         if self.id in app.admins:
             return False
         else:
             db_record = col_users.find_one({"user": self.id})
             if db_record is None:
                 return False
-            return True if (datetime.now() - db_record["cooldown"]).total_seconds() < configGet("timeout", "submission") else False
-        
+            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()})
+        """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()})