Add initial polling backend logic

2026-07-12 16:26:45 +02:00
parent ddf1fcc217
commit d0e68f5aa6
12 changed files with 1325 additions and 0 deletions
@@ -0,0 +1,101 @@
+from __future__ import annotations
+
+import uuid
+from datetime import datetime
+from typing import Any
+
+from sqlalchemy import Boolean, DateTime, ForeignKey, Index, Integer, JSON, String, Text, UniqueConstraint, text
+from sqlalchemy.orm import Mapped, mapped_column, relationship
+
+from govoplan_core.db.base import Base, TimestampMixin
+
+
+def new_uuid() -> str:
+    return str(uuid.uuid4())
+
+
+class Poll(Base, TimestampMixin):
+    __tablename__ = "poll_polls"
+    __table_args__ = (
+        Index(
+            "uq_poll_polls_active_slug",
+            "tenant_id",
+            "slug",
+            unique=True,
+            sqlite_where=text("deleted_at IS NULL"),
+            postgresql_where=text("deleted_at IS NULL"),
+        ),
+        Index("ix_poll_polls_tenant_status", "tenant_id", "status"),
+        Index("ix_poll_polls_tenant_kind", "tenant_id", "kind"),
+        Index("ix_poll_polls_window", "tenant_id", "opens_at", "closes_at"),
+    )
+
+    id: Mapped[str] = mapped_column(String(36), primary_key=True, default=new_uuid)
+    tenant_id: Mapped[str] = mapped_column(String(36), nullable=False, index=True)
+    slug: Mapped[str] = mapped_column(String(120), nullable=False)
+    title: Mapped[str] = mapped_column(String(500), nullable=False)
+    description: Mapped[str | None] = mapped_column(Text)
+    kind: Mapped[str] = mapped_column(String(40), nullable=False, index=True)
+    status: Mapped[str] = mapped_column(String(30), default="draft", nullable=False, index=True)
+    visibility: Mapped[str] = mapped_column(String(30), default="private", nullable=False, index=True)
+    result_visibility: Mapped[str] = mapped_column(String(30), default="after_close", nullable=False)
+    allow_anonymous: Mapped[bool] = mapped_column(Boolean, default=False, nullable=False)
+    allow_response_update: Mapped[bool] = mapped_column(Boolean, default=True, nullable=False)
+    min_choices: Mapped[int] = mapped_column(Integer, default=1, nullable=False)
+    max_choices: Mapped[int | None] = mapped_column(Integer)
+    opens_at: Mapped[datetime | None] = mapped_column(DateTime(timezone=True), nullable=True, index=True)
+    closes_at: Mapped[datetime | None] = mapped_column(DateTime(timezone=True), nullable=True, index=True)
+    closed_at: Mapped[datetime | None] = mapped_column(DateTime(timezone=True), nullable=True)
+    decided_at: Mapped[datetime | None] = mapped_column(DateTime(timezone=True), nullable=True)
+    decided_option_id: Mapped[str | None] = mapped_column(String(36), nullable=True, index=True)
+    created_by_user_id: Mapped[str | None] = mapped_column(String(36), nullable=True, index=True)
+    deleted_at: Mapped[datetime | None] = mapped_column(DateTime(timezone=True), nullable=True, index=True)
+    metadata_: Mapped[dict[str, Any] | None] = mapped_column("metadata", JSON, nullable=True)
+
+    options: Mapped[list["PollOption"]] = relationship(back_populates="poll", cascade="all, delete-orphan", order_by="PollOption.position")
+    responses: Mapped[list["PollResponse"]] = relationship(back_populates="poll", cascade="all, delete-orphan")
+
+
+class PollOption(Base, TimestampMixin):
+    __tablename__ = "poll_options"
+    __table_args__ = (
+        UniqueConstraint("poll_id", "key", name="uq_poll_options_poll_key"),
+        Index("ix_poll_options_poll_position", "poll_id", "position"),
+    )
+
+    id: Mapped[str] = mapped_column(String(36), primary_key=True, default=new_uuid)
+    tenant_id: Mapped[str] = mapped_column(String(36), nullable=False, index=True)
+    poll_id: Mapped[str] = mapped_column(ForeignKey("poll_polls.id", ondelete="CASCADE"), nullable=False, index=True)
+    key: Mapped[str] = mapped_column(String(120), nullable=False)
+    label: Mapped[str] = mapped_column(String(500), nullable=False)
+    description: Mapped[str | None] = mapped_column(Text)
+    position: Mapped[int] = mapped_column(Integer, default=0, nullable=False)
+    value: Mapped[dict[str, Any] | None] = mapped_column(JSON, nullable=True)
+    metadata_: Mapped[dict[str, Any] | None] = mapped_column("metadata", JSON, nullable=True)
+    deleted_at: Mapped[datetime | None] = mapped_column(DateTime(timezone=True), nullable=True, index=True)
+
+    poll: Mapped[Poll] = relationship(back_populates="options")
+
+
+class PollResponse(Base, TimestampMixin):
+    __tablename__ = "poll_responses"
+    __table_args__ = (
+        Index("ix_poll_responses_poll_submitted", "poll_id", "submitted_at"),
+        Index("ix_poll_responses_poll_respondent", "poll_id", "respondent_id"),
+        Index("ix_poll_responses_tenant_poll", "tenant_id", "poll_id"),
+    )
+
+    id: Mapped[str] = mapped_column(String(36), primary_key=True, default=new_uuid)
+    tenant_id: Mapped[str] = mapped_column(String(36), nullable=False, index=True)
+    poll_id: Mapped[str] = mapped_column(ForeignKey("poll_polls.id", ondelete="CASCADE"), nullable=False, index=True)
+    respondent_id: Mapped[str | None] = mapped_column(String(255), nullable=True, index=True)
+    respondent_label: Mapped[str | None] = mapped_column(String(500), nullable=True)
+    answers: Mapped[list[dict[str, Any]]] = mapped_column(JSON, default=list, nullable=False)
+    submitted_at: Mapped[datetime] = mapped_column(DateTime(timezone=True), nullable=False, index=True)
+    deleted_at: Mapped[datetime | None] = mapped_column(DateTime(timezone=True), nullable=True, index=True)
+    metadata_: Mapped[dict[str, Any] | None] = mapped_column("metadata", JSON, nullable=True)
+
+    poll: Mapped[Poll] = relationship(back_populates="responses")
+
+
+__all__ = ["Poll", "PollOption", "PollResponse", "new_uuid"]