welcomecenterbot/tgbot/profile.py

import json


class Profile:

    def __init__(self, storage):
        self.storage = storage

    def create(self, member_id):
        s = {
            "id": member_id,
            "name": "newcomer",
            "mention": "",
            "welcome_id": 0,
            "parents": [],
            "children": []
        }
        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):
        data = self.storage.get(f'usr-{member_id}')
        if data is None:
            return self.create(member_id)
        else:
            return json.loads(data)

    def leaving(self, s):
        if len(s['parents']) == 0:
            self.storage.delete(f'usr-{s["id"]}')