From 53c2f5c366ff6922f88df3862ebf124739a2559c Mon Sep 17 00:00:00 2001
From: Olivier Desenfans <desenfans.olivier@gmail.com>
Date: Thu, 11 May 2023 17:32:01 +0200
Subject: [PATCH] Fix: use epoch format for all messages on websocket

Problem: only aggregates were using the epoch format for
the `message.time` field.

Solution: apply the JSON encoder on the base message API schema.
---
 src/aleph/schemas/api/messages.py | 5 ++---
 1 file changed, 2 insertions(+), 3 deletions(-)

diff --git a/src/aleph/schemas/api/messages.py b/src/aleph/schemas/api/messages.py
index 5bba74961..d8dc71c29 100644
--- a/src/aleph/schemas/api/messages.py
+++ b/src/aleph/schemas/api/messages.py
@@ -37,6 +37,7 @@ class BaseMessage(GenericModel, Generic[MType, ContentType]):
     class Config:
         orm_mode = True
         json_loads = aleph_json.loads
+        json_encoders = {dt.datetime: lambda d: d.timestamp()}
 
     sender: str
     chain: Chain
@@ -55,9 +56,7 @@ class Config:
 class AggregateMessage(
     BaseMessage[Literal[MessageType.aggregate], AggregateContent]  # type: ignore
 ):
-    class Config:
-        orm_mode = True
-        json_encoders = {dt.datetime: lambda d: d.timestamp()}
+    ...
 
 
 class ForgetMessage(