ynerant/plateforme-tfjm2
1
0
Fork 0
mirror of https://gitlab.com/animath/si/plateforme.git synced 2025-06-24 09:08:50 +02:00
Code Issues Releases Wiki Activity

Editing and deleting is working

Browse Source 
Signed-off-by: Emmy D'Anello <emmy.danello@animath.fr>
This commit is contained in:
Emmy D'Anello
2024-04-28 16:56:30 +02:00
parent 6385e53425
commit 756f2074b3
2 changed files with 96 additions and 9 deletions
Download Patch File Download Diff File Expand all files Collapse all files

54
chat/consumers.py
View File

@ -62,11 +62,15 @@ class ChatConsumer(AsyncJsonWebsocketConsumer):
case 'fetch_channels':
await self.fetch_channels()
case 'send_message':
await self.receive_message(content)
await self.receive_message(**content)
case 'edit_message':
await self.edit_message(**content)
case 'delete_message':
await self.delete_message(**content)
case 'fetch_messages':
await self.fetch_messages(**content)
case 'start_private_chat':
await self.start_private_chat(content['user_id'])
await self.start_private_chat(**content)
case unknown:
print("Unknown message type:", unknown)
@ -90,10 +94,10 @@ class ChatConsumer(AsyncJsonWebsocketConsumer):
}
await self.send_json(message)
async def receive_message(self, message: dict) -> None:
async def receive_message(self, channel_id: int, content: str, **kwargs) -> None:
user = self.scope['user']
channel = await Channel.objects.prefetch_related('tournament__pools__juries', 'pool', 'team', 'invited') \
.aget(id=message['channel_id'])
.aget(id=channel_id)
write_channels = await Channel.get_accessible_channels(user, 'write')
if not await write_channels.acontains(channel):
return
@ -101,7 +105,7 @@ class ChatConsumer(AsyncJsonWebsocketConsumer):
message = await Message.objects.acreate(
author=user,
channel=channel,
content=message['content'],
content=content,
)
await self.channel_layer.group_send(f'chat-{channel.id}', {
@ -114,6 +118,36 @@ class ChatConsumer(AsyncJsonWebsocketConsumer):
'content': message.content,
})
async def edit_message(self, message_id: int, content: str, **kwargs) -> None:
message = await Message.objects.aget(id=message_id)
user = self.scope['user']
if user.id != message.author_id and not user.is_superuser:
return
message.content = content
await message.asave()
await self.channel_layer.group_send(f'chat-{message.channel_id}', {
'type': 'chat.edit_message',
'id': message_id,
'channel_id': message.channel_id,
'content': content,
})
async def delete_message(self, message_id: int, **kwargs) -> None:
message = await Message.objects.aget(id=message_id)
user = self.scope['user']
if user.id != message.author_id and not user.is_superuser:
return
await message.adelete()
await self.channel_layer.group_send(f'chat-{message.channel_id}', {
'type': 'chat.delete_message',
'id': message_id,
'channel_id': message.channel_id,
})
async def fetch_messages(self, channel_id: int, offset: int = 0, limit: int = 50, **_kwargs) -> None:
channel = await Channel.objects.aget(id=channel_id)
read_channels = await Channel.get_accessible_channels(self.scope['user'], 'read')
@ -138,7 +172,7 @@ class ChatConsumer(AsyncJsonWebsocketConsumer):
]))
})
async def start_private_chat(self, user_id: int) -> None:
async def start_private_chat(self, user_id: int, **kwargs) -> None:
user = self.scope['user']
other_user = await User.objects.aget(id=user_id)
channel_qs = Channel.objects.filter(private=True).filter(invited=user).filter(invited=other_user)
@ -183,6 +217,14 @@ class ChatConsumer(AsyncJsonWebsocketConsumer):
'timestamp': message['timestamp'], 'author': message['author'],
'content': message['content']})
async def chat_edit_message(self, message) -> None:
print(message)
await self.send_json({'type': 'edit_message', 'id': message['id'], 'channel_id': message['channel_id'],
'content': message['content']})
async def chat_delete_message(self, message) -> None:
await self.send_json({'type': 'delete_message', 'id': message['id'], 'channel_id': message['channel_id']})
async def chat_start_private_chat(self, message) -> None:
await self.channel_layer.group_add(f"chat-{message['channel']['id']}", self.channel_name)
await self.send_json({'type': 'start_private_chat', 'channel': message['channel']})