42 lines
1.1 KiB
Python
42 lines
1.1 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
|