43 lines
1.2 KiB
Python
43 lines
1.2 KiB
Python
from json import dumps, loads
|
|
import redis
|
|
from config import REDIS_URL, REDIS_USER, REDIS_PSW
|
|
|
|
class User:
|
|
user_id: str
|
|
def __init__(self, user_id: str, balance: int) -> None:
|
|
self.balance = balance
|
|
self.user_id = str(user_id)
|
|
|
|
def set_balance(self, balance) -> int:
|
|
self.balance = balance
|
|
return self.balance
|
|
|
|
def add_balance(self, balance) -> int:
|
|
self.balance += balance
|
|
return self.balance
|
|
|
|
def to_dict(self) -> dict:
|
|
return {'balance': self.balance}
|
|
|
|
def to_md(self) -> str:
|
|
return f"**- balance: {self.balance}**"
|
|
|
|
class Aura:
|
|
data: dict[str, User] = {}
|
|
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))))
|
|
except:
|
|
return None
|
|
|
|
def add(self, user_id) -> User:
|
|
u = User(user_id, 100)
|
|
self.r.set(user_id, u.balance)
|
|
return u
|
|
|
|
def update(self, u: User):
|
|
self.r.set(u.user_id, u.balance) |