Source code for redditwarp.models.message_SYNC


from __future__ import annotations
from typing import TYPE_CHECKING
if TYPE_CHECKING:
    from ..client_SYNC import Client

from dataclasses import dataclass

from .message import (
    MailboxMessage as BaseMailboxMessage,
    ComposedMessage as BaseComposedMessage,
    CommentMessage as BaseCommentMessage,
)

[docs]@dataclass(repr=False, eq=False)
class MailboxMessage(BaseMailboxMessage):
    client: Client

[docs]@dataclass(repr=False, eq=False)
class ComposedMessage(MailboxMessage, BaseComposedMessage):
[docs]    def reply(self, body: str) -> ComposedMessage:
        return self.client.p.message.reply(self.id, body)

[docs]    def mark_read(self) -> None:
        self.client.p.message.mark_read(self.id)

[docs]    def mark_unread(self) -> None:
        self.client.p.message.mark_unread(self.id)

[docs]@dataclass(repr=False, eq=False)
class CommentMessage(MailboxMessage, BaseCommentMessage):
[docs]    def mark_read(self) -> None:
        self.client.p.message.mark_comment_read(self.comment.id)

[docs]    def mark_unread(self) -> None:
        self.client.p.message.mark_comment_unread(self.comment.id)