From 728f5930610efc033935a052dfc8e25ca32523f9 Mon Sep 17 00:00:00 2001 From: Klemek Date: Tue, 6 Apr 2021 23:38:42 +0200 Subject: [PATCH] remove non serializable from dicts --- src/logs/channel_logs.py | 5 ++++- src/logs/message_log.py | 1 + 2 files changed, 5 insertions(+), 1 deletion(-) diff --git a/src/logs/channel_logs.py b/src/logs/channel_logs.py index 615fcd1..65290a0 100644 --- a/src/logs/channel_logs.py +++ b/src/logs/channel_logs.py @@ -28,7 +28,9 @@ class ChannelLogs: if channel["last_message_id"] is not None else None ) - self.messages = [MessageLog(message, self) for message in channel["messages"]] + self.messages = [ + MessageLog(message, self) for message in channel["messages"] + ] def is_format(self): return self.format == FORMAT @@ -78,5 +80,6 @@ class ChannelLogs: def dict(self) -> dict: channel = dict(self.__dict__) channel.pop("channel", None) + channel.pop("guild", None) channel["messages"] = [message.dict() for message in self.messages] return channel diff --git a/src/logs/message_log.py b/src/logs/message_log.py index b77ea49..f534155 100644 --- a/src/logs/message_log.py +++ b/src/logs/message_log.py @@ -79,6 +79,7 @@ class MessageLog: def dict(self) -> dict: message = dict(self.__dict__) + message.pop("channel", None) message["created_at"] = self.created_at.isoformat() message["edited_at"] = ( self.edited_at.isoformat() if self.edited_at is not None else None