Files
webprewviewapi/submission/schemas.py
yuetsh c7834530fe Add flag to API schemas and submission filter
Adds flag field to SubmissionOut, SubmissionFilter, and new FlagIn schema.

Co-Authored-By: Claude Opus 4.6 <noreply@anthropic.com>
2026-03-09 17:23:10 +08:00

90 lines
2.6 KiB
Python

from typing import Optional, Literal
from ninja import Schema
from uuid import UUID
class SubmissionIn(Schema):
task_id: int
html: Optional[str] = None
css: Optional[str] = None
js: Optional[str] = None
conversation_id: Optional[UUID] = None
class SubmissionOut(Schema):
id: UUID
userid: int
username: str
task_id: int
task_display: int
task_title: str
task_type: Literal["tutorial", "challenge"]
score: float
my_score: int
html: Optional[str] = None
css: Optional[str] = None
js: Optional[str] = None
conversation_id: Optional[UUID] = None
flag: Optional[str] = None
created: str
modified: str
@staticmethod
def list(submission, rating_dict):
return {
"id": submission.id,
"userid": submission.user.id,
"username": submission.user.username,
"task_id": submission.task.id,
"task_display": submission.task.display,
"task_title": submission.task.title,
"task_type": submission.task.task_type,
"score": submission.score,
"my_score": rating_dict.get(submission.id, 0),
"conversation_id": submission.conversation_id,
"flag": submission.flag,
"created": submission.created.isoformat(),
"modified": submission.modified.isoformat(),
}
@staticmethod
def get(submission, rating):
return {
"id": submission.id,
"userid": submission.user.id,
"username": submission.user.username,
"task_id": submission.task.id,
"task_display": submission.task.display,
"task_title": submission.task.title,
"task_type": submission.task.task_type,
"score": submission.score,
"my_score": rating.score if rating else 0,
"html": submission.html,
"css": submission.css,
"js": submission.js,
"conversation_id": submission.conversation_id,
"flag": submission.flag,
"created": submission.created.isoformat(),
"modified": submission.modified.isoformat(),
}
class RatingScoreIn(Schema):
score: int
class SubmissionScoreOut(Schema):
id: UUID
score: float
class SubmissionFilter(Schema):
task_id: Optional[int] = None
task_type: Optional[Literal["tutorial", "challenge"]] = None
username: Optional[str] = None
flag: Optional[Literal["red", "blue", "green", "yellow"]] = None
class FlagIn(Schema):
flag: Optional[Literal["red", "blue", "green", "yellow"]] = None