models.py

"""Data models and classes for the FORTIFIED Marketing Copilot."""

from __future__ import annotations

from typing import List


class StateInfo:
    """Represents a state's information including playbooks and source documents."""

    def __init__(
        self,
        slug: str,
        label: str,
        source: str,
        title: str,
        plays: List[str],
    ) -> None:
        self.slug = slug
        self.label = label
        self.source = source
        self.title = title
        self.plays = plays

    def __repr__(self) -> str:
        return (
            f"StateInfo(slug={self.slug!r}, label={self.label!r}, "
            f"source={self.source!r}, title={self.title!r}, plays={self.plays!r})"
        )