welcomecenterbot/bot/storage/profile.py

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