Moved cooldowns to PosterUser class

classes/user.py

@@ -1,5 +1,7 @@
+from modules.app import app
 from datetime import datetime
-from modules.database import col_banned
+from modules.database import col_banned, col_users
+from modules.utils import configGet
 
 class PosterUser():
 
@@ -14,4 +16,17 @@ class PosterUser():
             col_banned.insert_one({"user": self.id, "date": datetime.now()})
 
     def unblock(self) -> None:
-        col_banned.find_one_and_delete({"user": self.id})
+        col_banned.find_one_and_delete({"user": self.id})
+
+    def is_limited(self) -> bool:
+        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
+        
+    def limit(self) -> None:
+        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()})