set_time for all messages

dragonion_core/proto/web/webmessage/__init__.py

@@ -1,7 +1,8 @@
 from .webmessage import (
     webmessage_type_literal,
     webmessages_union,
-    WebMessage
+    WebMessage,
+    set_time
 )
 from .server import (
     webmessage_error_message_literal,
@@ -23,6 +24,7 @@ __all__ = [
     'webmessage_type_literal',
     'webmessages_union',
     'WebMessage',
+    'set_time',
 
     'WebMessageMessage',
     'WebBroadcastableMessage',

dragonion_core/proto/web/webmessage/message.py

@@ -29,10 +29,6 @@ class WebMessageMessage:
     def decrypt(self, identity: Identity):
         return identity.decrypt(self.message)
 
-    def set_time(self):
-        self.time = datetime.now()
-        return self
-
 
 @dataclass_json
 @dataclass

dragonion_core/proto/web/webmessage/webmessage.py

@@ -22,6 +22,7 @@ __all__ = [
     'webmessage_type_literal',
     'webmessages_union',
     'WebMessage',
+    'set_time',
 
     'WebMessageMessage',
     'WebBroadcastableMessage',
@@ -87,3 +88,8 @@ class WebMessage:
             "connect_answer": WebConnectionResultMessage.from_json,
             "broadcastable": WebBroadcastableMessage.from_json
         }[_WebAnyMessage.from_json(data).type](data)
+
+
+def set_time(webmessage: webmessages_union):
+    webmessage.time = datetime.now()
+    return webmessage