Joey Yakimowich-Payne
6038fc25f5- add startup/shutdown scripts that render a Caddyfile from JSON config and run health checks - add Python utilities and a sample sites.json for declarative multi-site configuration - document the workflow and ignore generated Caddy state artifacts - normalize double-quote style across challenge workflow controllers, nodes, and tests
113 lines
4.7 KiB
Python
113 lines
4.7 KiB
Python
from __future__ import annotations
|
|
|
|
from datetime import datetime
|
|
|
|
import sqlalchemy as sa
|
|
from sqlalchemy.dialects.postgresql import JSONB
|
|
from sqlalchemy.orm import Mapped, mapped_column
|
|
|
|
from .base import Base
|
|
from .types import StringUUID
|
|
|
|
|
|
class RedBlueChallenge(Base):
|
|
__tablename__ = "red_blue_challenges"
|
|
__table_args__ = (
|
|
sa.PrimaryKeyConstraint("id", name="red_blue_challenges_pkey"),
|
|
sa.Index("red_blue_challenges_tenant_id_idx", "tenant_id"),
|
|
sa.Index("red_blue_challenges_app_id_idx", "app_id"),
|
|
)
|
|
|
|
id: Mapped[str] = mapped_column(StringUUID, server_default=sa.text("uuid_generate_v4()"))
|
|
tenant_id: Mapped[str] = mapped_column(StringUUID, nullable=False)
|
|
app_id: Mapped[str] = mapped_column(StringUUID, nullable=False)
|
|
workflow_id: Mapped[str | None] = mapped_column(StringUUID, nullable=True)
|
|
|
|
name: Mapped[str] = mapped_column(sa.Text, nullable=False)
|
|
description: Mapped[str | None] = mapped_column(sa.Text, nullable=True)
|
|
|
|
judge_suite = mapped_column(JSONB, nullable=False)
|
|
defense_selection_policy: Mapped[str] = mapped_column(
|
|
sa.String(64), nullable=False, server_default=sa.text("'latest_best'")
|
|
)
|
|
attack_selection_policy: Mapped[str] = mapped_column(
|
|
sa.String(64), nullable=False, server_default=sa.text("'latest_best'")
|
|
)
|
|
scoring_strategy: Mapped[str] = mapped_column(
|
|
sa.String(64), nullable=False, server_default=sa.text("'red_blue_ratio'")
|
|
)
|
|
|
|
theme = mapped_column(JSONB, nullable=True)
|
|
instructions_md: Mapped[str | None] = mapped_column(sa.Text, nullable=True)
|
|
|
|
is_active: Mapped[bool] = mapped_column(sa.Boolean, nullable=False, server_default=sa.text("true"))
|
|
|
|
created_by: Mapped[str | None] = mapped_column(StringUUID, nullable=True)
|
|
created_at: Mapped[datetime] = mapped_column(
|
|
sa.DateTime,
|
|
nullable=False,
|
|
server_default=sa.func.current_timestamp(),
|
|
)
|
|
updated_by: Mapped[str | None] = mapped_column(StringUUID, nullable=True)
|
|
updated_at: Mapped[datetime] = mapped_column(
|
|
sa.DateTime,
|
|
nullable=False,
|
|
server_default=sa.func.current_timestamp(),
|
|
)
|
|
|
|
|
|
class TeamSubmission(Base):
|
|
__tablename__ = "team_submissions"
|
|
__table_args__ = (
|
|
sa.PrimaryKeyConstraint("id", name="team_submissions_pkey"),
|
|
sa.Index("team_submissions_challenge_id_idx", "red_blue_challenge_id"),
|
|
sa.Index("team_submissions_tenant_id_idx", "tenant_id"),
|
|
)
|
|
|
|
id: Mapped[str] = mapped_column(StringUUID, server_default=sa.text("uuid_generate_v4()"))
|
|
red_blue_challenge_id: Mapped[str] = mapped_column(StringUUID, nullable=False)
|
|
tenant_id: Mapped[str] = mapped_column(StringUUID, nullable=False)
|
|
account_id: Mapped[str | None] = mapped_column(StringUUID, nullable=True)
|
|
end_user_id: Mapped[str | None] = mapped_column(StringUUID, nullable=True)
|
|
|
|
team: Mapped[str] = mapped_column(sa.String(16), nullable=False) # 'red' | 'blue'
|
|
prompt: Mapped[str] = mapped_column(sa.Text, nullable=False)
|
|
active: Mapped[bool] = mapped_column(sa.Boolean, nullable=False, server_default=sa.text("true"))
|
|
|
|
created_at: Mapped[datetime] = mapped_column(
|
|
sa.DateTime,
|
|
nullable=False,
|
|
server_default=sa.func.current_timestamp(),
|
|
)
|
|
|
|
|
|
class TeamPairing(Base):
|
|
__tablename__ = "team_pairings"
|
|
__table_args__ = (
|
|
sa.PrimaryKeyConstraint("id", name="team_pairings_pkey"),
|
|
sa.Index("team_pairings_challenge_id_idx", "red_blue_challenge_id"),
|
|
sa.Index("team_pairings_tenant_id_idx", "tenant_id"),
|
|
)
|
|
|
|
id: Mapped[str] = mapped_column(StringUUID, server_default=sa.text("uuid_generate_v4()"))
|
|
red_blue_challenge_id: Mapped[str] = mapped_column(StringUUID, nullable=False)
|
|
tenant_id: Mapped[str] = mapped_column(StringUUID, nullable=False)
|
|
|
|
attack_submission_id: Mapped[str | None] = mapped_column(StringUUID, nullable=True)
|
|
defense_submission_id: Mapped[str | None] = mapped_column(StringUUID, nullable=True)
|
|
|
|
judge_output_raw = mapped_column(JSONB, nullable=True)
|
|
categories = mapped_column(JSONB, nullable=True)
|
|
judge_rating: Mapped[int | None] = mapped_column(sa.Integer, nullable=True)
|
|
judge_feedback: Mapped[str | None] = mapped_column(sa.Text, nullable=True)
|
|
red_points: Mapped[float] = mapped_column(sa.Float, nullable=False, server_default=sa.text("0"))
|
|
blue_points: Mapped[float] = mapped_column(sa.Float, nullable=False, server_default=sa.text("0"))
|
|
|
|
tokens_total: Mapped[int | None] = mapped_column(sa.Integer, nullable=True)
|
|
elapsed_ms: Mapped[int | None] = mapped_column(sa.Integer, nullable=True)
|
|
|
|
created_at: Mapped[datetime] = mapped_column(
|
|
sa.DateTime,
|
|
nullable=False,
|
|
server_default=sa.func.current_timestamp(),
|
|
)
|