discours.io/inbox
5
0
Fork 0
Code Pull Requests Actions Wiki Activity

precommit-setup

Browse Source
This commit is contained in:
Untone
2024-01-25 12:25:52 +03:00
parent 8435c8e6b5
commit 152d5a4e99
14 changed files with 320 additions and 309 deletions
Download Patch File Download Diff File Expand all files Collapse all files

109
resolvers/chats.py
View File

@@ -1,7 +1,7 @@
import json
import logging
import time
import uuid
import logging
from models.chat import Chat, ChatUpdate
from services.auth import login_required
@@ -9,11 +9,12 @@ from services.presence import notify_chat
from services.rediscache import redis
from services.schema import mutation
logger = logging.getLogger("[resolvers.chats] ")
logger = logging.getLogger('[resolvers.chats] ')
logger.setLevel(logging.DEBUG)
@mutation.field("update_chat")
@mutation.field('update_chat')
@login_required
async def update_chat(_, info, chat_new: ChatUpdate):
"""
@@ -24,38 +25,38 @@ async def update_chat(_, info, chat_new: ChatUpdate):
:param chat_new: dict with chat data
:return: Result { error chat }
"""
logger.info("update_chat")
author_id = info.context["author_id"]
chat_id = chat_new["id"]
chat_str = await redis.execute("GET", f"chats/{chat_id}")
logger.info('update_chat')
author_id = info.context['author_id']
chat_id = chat_new['id']
chat_str = await redis.execute('GET', f'chats/{chat_id}')
if not chat_str:
return {"error": "chat not exist"}
return {'error': 'chat not exist'}
elif isinstance(chat_str, str):
chat: Chat = json.loads(chat_str)
if author_id in chat["admins"]:
if author_id in chat['admins']:
chat.update(
{
"title": chat_new.get("title", chat["title"]),
"description": chat_new.get("description", chat["description"]),
"updated_at": int(time.time()),
"admins": chat_new.get("admins", chat.get("admins") or []),
"members": chat_new.get("members", chat["members"]),
'title': chat_new.get('title', chat['title']),
'description': chat_new.get('description', chat['description']),
'updated_at': int(time.time()),
'admins': chat_new.get('admins', chat.get('admins') or []),
'members': chat_new.get('members', chat['members']),
}
)
await redis.execute("SET", f"chats/{chat['id']}", json.dumps(chat))
for member_id in chat["members"]:
await notify_chat(chat, member_id, "update")
await redis.execute('SET', f"chats/{chat['id']}", json.dumps(chat))
for member_id in chat['members']:
await notify_chat(chat, member_id, 'update')
return {"error": None, "chat": chat}
return {'error': None, 'chat': chat}
@mutation.field("create_chat")
@mutation.field('create_chat')
@login_required
async def create_chat(_, info, title="", members=None):
logger.info("create_chat")
async def create_chat(_, info, title='', members=None):
logger.info('create_chat')
members = members or []
author_id = info.context["author_id"]
author_id = info.context['author_id']
chat: Chat
if author_id:
if author_id not in members:
@@ -63,51 +64,51 @@ async def create_chat(_, info, title="", members=None):
# NOTE: private chats has no title
# reuse private chat created before if exists
if len(members) == 2 and title == "":
chatset1 = await redis.execute("SMEMBERS", f"chats_by_author/{members[0]}")
chatset2 = await redis.execute("SMEMBERS", f"chats_by_author/{members[1]}")
if len(members) == 2 and title == '':
chatset1 = await redis.execute('SMEMBERS', f'chats_by_author/{members[0]}')
chatset2 = await redis.execute('SMEMBERS', f'chats_by_author/{members[1]}')
for c in chatset1.intersection(chatset2):
chat = await redis.execute("GET", f"chats/{c}")
if chat["title"] == "":
logger.info("[inbox] createChat found old chat")
return {"chat": chat, "error": "existed"}
chat = await redis.execute('GET', f'chats/{c}')
if chat['title'] == '':
logger.info('[inbox] createChat found old chat')
return {'chat': chat, 'error': 'existed'}
chat_id = str(uuid.uuid4())
chat: Chat = {
"id": chat_id,
"members": members,
"title": title,
"description": "",
"created_by": author_id,
"created_at": int(time.time()),
"updated_at": int(time.time()),
"admins": members if (len(members) == 2 and title == "") else [],
'id': chat_id,
'members': members,
'title': title,
'description': '',
'created_by': author_id,
'created_at': int(time.time()),
'updated_at': int(time.time()),
'admins': members if (len(members) == 2 and title == '') else [],
}
for member_id in members:
await redis.execute("SADD", f"chats_by_author/{member_id}", chat_id)
await notify_chat(chat, member_id, "create")
await redis.execute('SADD', f'chats_by_author/{member_id}', chat_id)
await notify_chat(chat, member_id, 'create')
print(f"\n\n[resolvers.chats] creating: {chat}\n\n")
print(f'\n\n[resolvers.chats] creating: {chat}\n\n')
await redis.execute("SET", f"chats/{chat_id}", json.dumps(chat))
await redis.execute("SET", f"chats/{chat_id}/next_message_id", str(0))
await redis.execute('SET', f'chats/{chat_id}', json.dumps(chat))
await redis.execute('SET', f'chats/{chat_id}/next_message_id', str(0))
return {"error": None, "chat": chat}
return {"error": "no chat was created"}
return {'error': None, 'chat': chat}
return {'error': 'no chat was created'}
@mutation.field("delete_chat")
@mutation.field('delete_chat')
@login_required
async def delete_chat(_, info, chat_id: str):
logger.info("delete_chat")
author_id = info.context["author_id"]
chat_str = await redis.execute("GET", f"chats/{chat_id}")
logger.info('delete_chat')
author_id = info.context['author_id']
chat_str = await redis.execute('GET', f'chats/{chat_id}')
if isinstance(chat_str, str):
chat: Chat = json.loads(chat_str)
if author_id in chat["admins"]:
await redis.execute("DEL", f"chats/{chat_id}")
await redis.execute("SREM", f"chats_by_author/{author_id}", chat_id)
for member_id in chat["members"]:
await notify_chat(chat, member_id, "delete")
return {"error": "chat not exist"}
if author_id in chat['admins']:
await redis.execute('DEL', f'chats/{chat_id}')
await redis.execute('SREM', f'chats_by_author/{author_id}', chat_id)
for member_id in chat['members']:
await notify_chat(chat, member_id, 'delete')
return {'error': 'chat not exist'}