0.0.7 fixes and graph

2023-04-22 04:21:24 +03:00
parent cbe9db4932
commit 297445fd50
8 changed files with 140 additions and 21 deletions
--- a/tgbot/profile.py
+++ b/tgbot/profile.py
@@ -19,10 +19,14 @@ class Profile:
        return s

    def save(self, s):
-        self.storage.set(f'usr-{member_id}', json.dumps(s))
+        self.storage.set(f'usr-{s["id"]}', json.dumps(s))

    def get(self, member_id):
-        return json.loads(self.storage.get(f'usr-{member_id}')) or self.create_session(member_id)
+        data = self.storage.get(f'usr-{member_id}')
+        if data is None:
+            return self.create(member_id)
+        else:
+            return json.loads(data)

    def leaving(self, s):
        if len(s['parents']) == 0: