From beb454392bff8fc56e85d911dde23e4b002ac6ca Mon Sep 17 00:00:00 2001 From: Leo dev Date: Thu, 5 Sep 2024 15:47:16 +0200 Subject: [PATCH] cooldown --- aura.py | 10 ++++++---- 1 file changed, 6 insertions(+), 4 deletions(-) diff --git a/aura.py b/aura.py index 842abff..33fec5b 100644 --- a/aura.py +++ b/aura.py @@ -5,9 +5,10 @@ from config import REDIS_URL, REDIS_USER, REDIS_PSW class User: user_id: str last_earned: float = 0.0 - def __init__(self, user_id: str, balance: int) -> None: + def __init__(self, user_id: str, balance: int, last_earned: float = 0.0) -> None: self.balance = balance self.user_id = str(user_id) + self.last_earned = float(last_earned) def set_balance(self, balance) -> int: self.balance = balance @@ -24,14 +25,14 @@ class User: return f"**- balance: {self.balance}**" class Aura: - data: dict[str, User] = {} + last_earns: dict[str, float] def __init__(self) -> None: redis_host = REDIS_URL.split(':') self.r = redis.Redis(host=redis_host[0], port=int(redis_host[1]), username=REDIS_USER, password=REDIS_PSW, decode_responses=True) def get(self, user_id) -> User | None: try: - return User(user_id, int(self.r.get(str(user_id)))) + return User(user_id, int(self.r.get(str(user_id))), self.last_earns[user_id] if user_id in self.last_earns else 0.0) except: return None @@ -41,4 +42,5 @@ class Aura: return u def update(self, u: User): - self.r.set(u.user_id, u.balance) \ No newline at end of file + self.r.set(u.user_id, u.balance) + self.last_earns[str(u.user_id)] = u.last_earned \ No newline at end of file