diff --git a/src/snek/model/channel_member.py b/src/snek/model/channel_member.py index d199498..65ba3e4 100644 --- a/src/snek/model/channel_member.py +++ b/src/snek/model/channel_member.py @@ -13,3 +13,4 @@ class ChannelMemberModel(BaseModel): ) is_muted = ModelField(name="is_muted", required=True, kind=bool, value=False) is_banned = ModelField(name="is_banned", required=True, kind=bool, value=False) + new_count = ModelField(name="new_count", required=False, kind=int, value=0) diff --git a/src/snek/service/notification.py b/src/snek/service/notification.py index 6db762c..ad0acbe 100644 --- a/src/snek/service/notification.py +++ b/src/snek/service/notification.py @@ -36,6 +36,8 @@ class NotificationService(BaseService): is_muted=False, deleted_at=None, ): + channel_member['new_count'] += 1 + await self.services.channel_member.save(channel_member) model = await self.new() model["object_uid"] = channel_message_uid model["object_type"] = "channel_message"