Replaced user block logic

This commit is contained in:
Profitroll
2023-02-17 21:54:52 +01:00
parent 68ea087963
commit bd9917fb17
2 changed files with 24 additions and 16 deletions

17
classes/user.py Normal file
View File

@@ -0,0 +1,17 @@
from datetime import datetime
from modules.database import col_banned
class PosterUser():
def __init__(self, id: int):
self.id = id
def is_blocked(self) -> bool:
return False if col_banned.find_one({"user": self.id}) is None else True
def block(self) -> None:
if col_banned.find_one({"user": self.id}) is None:
col_banned.insert_one({"user": self.id, "date": datetime.now()})
def unblock(self) -> None:
col_banned.find_one_and_delete({"user": self.id})