mirror of
https://github.com/ae-utbm/sith.git
synced 2025-02-27 09:57:09 +00:00
59 lines
1.7 KiB
Python
59 lines
1.7 KiB
Python
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
|