File size: 1,056 Bytes
4cd2145 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 |
from cbh.api.chat.dto import Paging
from cbh.api.message import message_router
from cbh.api.message.db_requests import create_message_obj, get_all_chat_messages_obj
from cbh.api.message.model import MessageModel
from cbh.api.message.schemas import CreateMessageRequest, AllMessagesResponse
from cbh.core.wrappers import CbhResponseWrapper
@message_router.get(
'/{chatId}/all', response_model_by_alias=False, response_model=CbhResponseWrapper[AllMessagesResponse]
)
async def get_all_chat_messages(chatId: str):
messages, _ = await get_all_chat_messages_obj(chatId)
response = AllMessagesResponse(
paging=Paging(pageSize=len(messages), pageIndex=0, totalCount=len(messages)),
data=messages
)
return CbhResponseWrapper(data=response)
@message_router.post('/{chatId}', response_model_by_alias=False, response_model=CbhResponseWrapper[MessageModel])
async def create_message(chatId: str, message: CreateMessageRequest):
message = await create_message_obj(chatId, message)
return CbhResponseWrapper(data=message)
|