version-0.2.0

storage/__init__.py

@@ -0,0 +1,37 @@
+from redis import Redis
+from storage.profile import Profile as ProfileObj
+from bot.config import REDIS_URL
+import json
+
+
+# сохраняет сессии, айди кнопок в чатах для удаления и пересылаемые сообщения между перезагрузками
+storage = Redis.from_url(REDIS_URL)
+
+# хранение необходимой информации о пользователях
+Profile = ProfileObj(storage)
+
+
+# достаёт из хранилища jsonы по маске и количеству
+def scan(match="usr-*", count=100):
+    cursor = 0
+    keys = []
+    r = storage
+    while True:
+        # Scan for keys starting with <match> in batches of <count>
+        cursor, batch_keys = r.scan(cursor=cursor, match=match, count=count)
+        keys += batch_keys
+        # If the cursor is 0, then we've reached the end of the keys
+        if cursor == 0:
+            break
+    # Get the values of all the keys
+    values = r.mget(keys)
+    # Parse the JSON data from each value
+    items = []
+    for value in values:
+        if value:
+            value_str = value.decode("utf-8")
+            i = json.loads(value_str)
+            items.append(i)
+    print(f"scan found {len(items)} items")
+
+    return keys, items

storage/profile.py

@@ -0,0 +1,47 @@
+import json
+
+
+class Profile:
+    def __init__(self, storage):
+        self.storage = storage
+
+    def create(self, member_id, msg=None):
+        s = {"id": member_id, "parents": [], "children": [], "chats": []}
+
+        if msg:
+            if "from" in msg:
+                sender = msg.get("from")
+                s["mention"] = sender.get("username")
+                s[
+                    "name"
+                ] = f"{sender['first_name']} {sender.get('last_name', '')}".strip()
+
+            if "chat" in msg:
+                chat_id = str(msg["chat"]["id"])
+                if chat_id not in s["chats"]:
+                    s["chats"].append(chat_id)
+
+        self.storage.set(f"usr-{member_id}", json.dumps(s))
+        return s
+
+    def save(self, s):
+        self.storage.set(f'usr-{s["id"]}', json.dumps(s))
+
+    def get(self, member_id, msg=None):
+        data = self.storage.get(f"usr-{member_id}")
+        if data is None:
+            r = self.create(member_id, msg)
+        else:
+            r = json.loads(data)
+        return r
+
+    def erase(self, member_id):
+        data = self.storage.get(f"usr-{member_id}")
+        if data:
+            member = json.loads(data)
+            for child in member["children"]:
+                child_member = self.storage.get(f"usr-{child}")
+                if child_member:
+                    child_member = json.loads(child_member)
+                    child_member["parents"].remove(member_id)
+                    self.storage.set(f"usr-{child_member['id']}", json.dumps(child_member))