dragonion/dragonion-server
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Time set by server

Browse Source
This commit is contained in:
BarsTiger
2023-07-26 16:54:46 +03:00
parent 333cc8ddf2
commit 80242a809e
2 changed files with 5 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

3
dragonion_server/modules/server/handlers/managers/connection.py
View File

@@ -1,6 +1,7 @@
from attrs import define from attrs import define
from fastapi import WebSocket from fastapi import WebSocket
from dragonion_core.proto.web.webmessage import ( from dragonion_core.proto.web.webmessage import (
set_time,
webmessages_union, webmessages_union,
webmessage_error_message_literal, webmessage_error_message_literal,
WebErrorMessage WebErrorMessage
@@ -20,7 +21,7 @@ class Connection(object):
:param obj: Should be some type of WebMessage :param obj: Should be some type of WebMessage
:return: :return:
""" """
await self.ws.send_text(obj.to_json()) await self.ws.send_text(set_time(obj).to_json())
async def send_error( async def send_error(
self, self,

5
dragonion_server/modules/server/handlers/managers/room.py
View File

@@ -8,6 +8,7 @@ from json.decoder import JSONDecodeError
from dragonion_core.proto.web.webmessage import ( from dragonion_core.proto.web.webmessage import (
webmessages_union, webmessages_union,
set_time,
WebMessageMessage, WebMessageMessage,
WebBroadcastableMessage, WebBroadcastableMessage,
WebNotificationMessage, WebNotificationMessage,
@@ -37,9 +38,9 @@ class Room(object):
await ws.receive_text() await ws.receive_text()
) )
except JSONDecodeError: except JSONDecodeError:
await ws.send_text(WebErrorMessage( await ws.send_text(set_time(WebErrorMessage(
'invalid_webmessage' 'invalid_webmessage'
).to_json()) )).to_json())
await ws.close(reason='invalid_webmessage') await ws.close(reason='invalid_webmessage')
return return