aho/welcomecenterbot
2
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

dokku-comp

Browse Source
This commit is contained in:
Untone
2023-09-11 23:04:53 +03:00
parent 4be2dc3a45
commit b9a98f1e56
25 changed files with 646 additions and 460 deletions
Download Patch File Download Diff File Expand all files Collapse all files

29
bot/storage/profile.py
View File

@@ -2,38 +2,33 @@ 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": []
}
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 "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']:
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))
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}')
data = self.storage.get(f"usr-{member_id}")
if data is None:
r = self.create(member_id, msg)
else: