Sith/com/schemas.py

from datetime import datetime

from ninja import FilterSchema, ModelSchema
from ninja_extra import service_resolver
from ninja_extra.controllers import RouteContext
from pydantic import Field

from club.schemas import ClubProfileSchema
from com.models import News, NewsDate
from core.markdown import markdown


class NewsDateFilterSchema(FilterSchema):
    before: datetime | None = Field(None, q="end_date__lt")
    after: datetime | None = Field(None, q="start_date__gt")
    club_id: int | None = Field(None, q="news__club_id")
    news_id: int | None = None
    is_moderated: bool | None = Field(None, q="news__is_moderated")
    title: str | None = Field(None, q="news__title__icontains")


class NewsSchema(ModelSchema):
    class Meta:
        model = News
        fields = ["id", "title", "summary", "is_moderated"]

    club: ClubProfileSchema
    url: str

    @staticmethod
    def resolve_summary(obj: News) -> str:
        # if this is returned from a route that allows the
        # user to choose the text format (md or html)
        # and the user chose "html", convert the markdown to html
        context: RouteContext = service_resolver(RouteContext)
        if context.kwargs.get("text_format", "") == "html":
            return markdown(obj.summary)
        return obj.summary

    @staticmethod
    def resolve_url(obj: News) -> str:
        return obj.get_absolute_url()


class NewsDateSchema(ModelSchema):
    """Basic infos about an event occurrence.

    Warning:
        This uses [NewsSchema][], which itself
        uses [ClubProfileSchema][club.schemas.ClubProfileSchema].
        Don't forget the appropriated `select_related`.
    """

    class Meta:
        model = NewsDate
        fields = ["id", "start_date", "end_date"]

    news: NewsSchema