fix-graph3

2023-09-18 14:45:08 +03:00
parent 00ac859312
commit bf31e8aa51
2 changed files with 20 additions and 3 deletions
--- a/bot/storage/profile.py
+++ b/bot/storage/profile.py
@@ -34,3 +34,14 @@ class Profile:
        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))