34 lines
794 B
Python
34 lines
794 B
Python
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"]}')
|