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 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 SchedulingRequest(Base, TimestampMixin): __tablename__ = "scheduling_requests" __table_args__ = ( Index("ix_scheduling_requests_tenant_status", "tenant_id", "status"), Index("ix_scheduling_requests_tenant_poll", "tenant_id", "poll_id"), Index("ix_scheduling_requests_deadline", "tenant_id", "deadline_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) title: Mapped[str] = mapped_column(String(500), nullable=False) description: Mapped[str | None] = mapped_column(Text) location: Mapped[str | None] = mapped_column(String(500)) timezone: Mapped[str] = mapped_column(String(100), default="UTC", nullable=False) status: Mapped[str] = mapped_column(String(40), default="draft", nullable=False, index=True) poll_id: Mapped[str | None] = mapped_column(String(36), nullable=True, index=True) selected_slot_id: Mapped[str | None] = mapped_column(String(36), nullable=True, index=True) organizer_user_id: Mapped[str | None] = mapped_column(String(36), nullable=True, index=True) deadline_at: Mapped[datetime | None] = mapped_column(DateTime(timezone=True), nullable=True, index=True) allow_external_participants: Mapped[bool] = mapped_column(Boolean, default=True, nullable=False) allow_participant_updates: Mapped[bool] = mapped_column(Boolean, default=True, nullable=False) result_visibility: Mapped[str] = mapped_column(String(30), default="after_close", nullable=False) participant_visibility: Mapped[str] = mapped_column(String(32), default="aggregates_only", nullable=False) notify_on_answers: Mapped[bool] = mapped_column(Boolean, default=True, nullable=False) single_choice: Mapped[bool] = mapped_column(Boolean, default=False, nullable=False) max_participants_per_option: Mapped[int | None] = mapped_column(Integer, nullable=True) allow_maybe: Mapped[bool] = mapped_column(Boolean, default=True, nullable=False) allow_comments: Mapped[bool] = mapped_column(Boolean, default=False, nullable=False) participant_email_required: Mapped[bool] = mapped_column(Boolean, default=False, nullable=False) anonymous_password_protection_enabled: Mapped[bool] = mapped_column(Boolean, default=False, nullable=False) anonymous_password_hash: Mapped[str | None] = mapped_column(String(500), nullable=True) calendar_integration_enabled: Mapped[bool] = mapped_column(Boolean, default=False, nullable=False) calendar_id: Mapped[str | None] = mapped_column(String(36), nullable=True, index=True) calendar_freebusy_enabled: Mapped[bool] = mapped_column(Boolean, default=False, nullable=False) calendar_hold_enabled: Mapped[bool] = mapped_column(Boolean, default=False, nullable=False) create_calendar_event_on_decision: Mapped[bool] = mapped_column(Boolean, default=False, nullable=False) calendar_event_id: Mapped[str | None] = mapped_column(String(36), nullable=True, index=True) handed_off_at: Mapped[datetime | None] = mapped_column(DateTime(timezone=True), nullable=True) cancelled_at: Mapped[datetime | None] = mapped_column(DateTime(timezone=True), nullable=True) cancellation_notice_until: Mapped[datetime | None] = mapped_column(DateTime(timezone=True), nullable=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) slots: Mapped[list["SchedulingCandidateSlot"]] = relationship( back_populates="request", cascade="all, delete-orphan", order_by="SchedulingCandidateSlot.position", ) participants: Mapped[list["SchedulingParticipant"]] = relationship( back_populates="request", cascade="all, delete-orphan", order_by="SchedulingParticipant.created_at", ) class SchedulingCandidateSlot(Base, TimestampMixin): __tablename__ = "scheduling_candidate_slots" __table_args__ = ( Index("ix_scheduling_candidate_slots_request_position", "request_id", "position"), Index("ix_scheduling_candidate_slots_range", "tenant_id", "start_at", "end_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) request_id: Mapped[str] = mapped_column(ForeignKey("scheduling_requests.id", ondelete="CASCADE"), nullable=False, index=True) poll_option_id: Mapped[str | None] = mapped_column(String(36), nullable=True, index=True) label: Mapped[str] = mapped_column(String(500), nullable=False) description: Mapped[str | None] = mapped_column(Text) start_at: Mapped[datetime] = mapped_column(DateTime(timezone=True), nullable=False, index=True) end_at: Mapped[datetime] = mapped_column(DateTime(timezone=True), nullable=False, index=True) timezone: Mapped[str] = mapped_column(String(100), default="UTC", nullable=False) location: Mapped[str | None] = mapped_column(String(500)) position: Mapped[int] = mapped_column(Integer, default=0, nullable=False) freebusy_checked_at: Mapped[datetime | None] = mapped_column(DateTime(timezone=True), nullable=True) freebusy_status: Mapped[str | None] = mapped_column(String(40), nullable=True, index=True) freebusy_conflicts: Mapped[list[dict[str, Any]]] = mapped_column(JSON, default=list, nullable=False) tentative_hold_event_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) request: Mapped[SchedulingRequest] = relationship(back_populates="slots") class SchedulingParticipant(Base, TimestampMixin): __tablename__ = "scheduling_participants" __table_args__ = ( Index("ix_scheduling_participants_request", "tenant_id", "request_id"), Index("ix_scheduling_participants_request_email", "request_id", "email"), Index("ix_scheduling_participants_request_respondent", "request_id", "respondent_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) request_id: Mapped[str] = mapped_column(ForeignKey("scheduling_requests.id", ondelete="CASCADE"), nullable=False, index=True) respondent_id: Mapped[str | None] = mapped_column(String(255), nullable=True, index=True) display_name: Mapped[str | None] = mapped_column(String(500), nullable=True) email: Mapped[str | None] = mapped_column(String(320), nullable=True, index=True) participant_type: Mapped[str] = mapped_column(String(30), default="external", nullable=False, index=True) required: Mapped[bool] = mapped_column(Boolean, default=True, nullable=False) status: Mapped[str] = mapped_column(String(40), default="invited", nullable=False, index=True) poll_invitation_id: Mapped[str | None] = mapped_column(String(36), nullable=True, index=True) participation_gateway: Mapped[str | None] = mapped_column(String(40), nullable=True) last_invited_at: Mapped[datetime | None] = mapped_column(DateTime(timezone=True), nullable=True) responded_at: Mapped[datetime | None] = mapped_column(DateTime(timezone=True), nullable=True) response_comment: Mapped[str | None] = mapped_column(Text, nullable=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) request: Mapped[SchedulingRequest] = relationship(back_populates="participants") class SchedulingNotification(Base, TimestampMixin): __tablename__ = "scheduling_notifications" __table_args__ = ( Index("ix_scheduling_notifications_request_status", "request_id", "status"), Index("ix_scheduling_notifications_tenant_status", "tenant_id", "status"), ) 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) request_id: Mapped[str] = mapped_column(ForeignKey("scheduling_requests.id", ondelete="CASCADE"), nullable=False, index=True) participant_id: Mapped[str | None] = mapped_column(String(36), nullable=True, index=True) event_kind: Mapped[str] = mapped_column(String(40), nullable=False, index=True) channel: Mapped[str] = mapped_column(String(40), default="mail", nullable=False, index=True) recipient: Mapped[str | None] = mapped_column(String(500), nullable=True, index=True) status: Mapped[str] = mapped_column(String(40), default="pending", nullable=False, index=True) payload: Mapped[dict[str, Any]] = mapped_column(JSON, default=dict, nullable=False) error: Mapped[str | None] = mapped_column(Text) sent_at: Mapped[datetime | None] = mapped_column(DateTime(timezone=True), nullable=True) metadata_: Mapped[dict[str, Any] | None] = mapped_column("metadata", JSON, nullable=True) __all__ = [ "SchedulingCandidateSlot", "SchedulingNotification", "SchedulingParticipant", "SchedulingRequest", "new_uuid", ]