"""Data classes."""

from dataclasses import dataclass

from src.utils.constants import StatementAttachmentFailureReason
from src.utils.constants import StatementAttachmentStatus


@dataclass
class StatementAttachmentPayload(dict):
    """Statement Attachment Payload dataclass."""

    file_location: str | None = None
    statement_attachment_status: StatementAttachmentStatus | None = None
    failure_reason: StatementAttachmentFailureReason | None = None