48 lines
1.5 KiB
Python
48 lines
1.5 KiB
Python
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))
|