ok

aura.py

@@ -3,8 +3,10 @@ import redis
 from config import REDIS_URL, REDIS_USER, REDIS_PSW
 
 class User:
-    def __init__(self, balance: int) -> None:
+    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
@@ -27,22 +29,13 @@ class Aura:
         self.r = redis.Redis(host=redis_host[0], port=int(redis_host[1]), username=REDIS_USER, password=REDIS_PSW, decode_responses=True)
         self.load()
     
-    def load(self) -> None:
-        data = loads(self.r.get('data'))
-        for k in data:
-            self.data[k] = User(data[k]['balance'])
-
     def get(self, user_id) -> User | None:
-        if user_id in self.data:
-            return self.r.get(str(user_id))
+        return User(int(self.r.get(user_id)))
     
     def add(self, user_id) -> User:
-        self.data[user_id] = User(100)
-        self.r.set
-        return self.data[user_id]
-            
-    def save(self):
-        data = {}
-        for k in self.data:
-            data[k] = self.data[k].to_dict()
-        self.r.set('data', dumps(data, indent=2))
+        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)