mirror of
https://github.com/ae-utbm/sith.git
synced 2024-11-15 02:33:22 +00:00
76 lines
1.9 KiB
Python
76 lines
1.9 KiB
Python
from datetime import datetime
|
|
from pathlib import Path
|
|
|
|
from django.urls import reverse
|
|
from ninja import FilterSchema, ModelSchema, Schema
|
|
from pydantic import Field, NonNegativeInt
|
|
|
|
from core.schemas import SimpleUserSchema, UserProfileSchema
|
|
from sas.models import Album, Picture, PictureModerationRequest
|
|
|
|
|
|
class AlbumSchema(ModelSchema):
|
|
class Meta:
|
|
model = Album
|
|
fields = ["id", "name"]
|
|
|
|
path: str
|
|
|
|
@staticmethod
|
|
def resolve_path(obj: Album) -> str:
|
|
return str(Path(obj.get_parent_path()) / obj.name)
|
|
|
|
|
|
class PictureFilterSchema(FilterSchema):
|
|
before_date: datetime | None = Field(None, q="date__lte")
|
|
after_date: datetime | None = Field(None, q="date__gte")
|
|
users_identified: set[int] | None = Field(None, q="people__user_id__in")
|
|
album_id: int | None = Field(None, q="parent_id")
|
|
|
|
|
|
class PictureSchema(ModelSchema):
|
|
class Meta:
|
|
model = Picture
|
|
fields = ["id", "name", "date", "size", "is_moderated", "asked_for_removal"]
|
|
|
|
owner: UserProfileSchema
|
|
sas_url: str
|
|
full_size_url: str
|
|
compressed_url: str
|
|
thumb_url: str
|
|
album: str
|
|
|
|
@staticmethod
|
|
def resolve_sas_url(obj: Picture) -> str:
|
|
return reverse("sas:picture", kwargs={"picture_id": obj.id})
|
|
|
|
@staticmethod
|
|
def resolve_full_size_url(obj: Picture) -> str:
|
|
return obj.get_download_url()
|
|
|
|
@staticmethod
|
|
def resolve_compressed_url(obj: Picture) -> str:
|
|
return obj.get_download_compressed_url()
|
|
|
|
@staticmethod
|
|
def resolve_thumb_url(obj: Picture) -> str:
|
|
return obj.get_download_thumb_url()
|
|
|
|
|
|
class PictureRelationCreationSchema(Schema):
|
|
picture: NonNegativeInt
|
|
users: list[NonNegativeInt]
|
|
|
|
|
|
class IdentifiedUserSchema(Schema):
|
|
id: int
|
|
user: UserProfileSchema
|
|
|
|
|
|
class ModerationRequestSchema(ModelSchema):
|
|
author: SimpleUserSchema
|
|
|
|
class Meta:
|
|
model = PictureModerationRequest
|
|
fields = ["id", "created_at", "reason"]
|