Initialize GovOPlaN calendar module

src/govoplan_calendar/backend/db/__init__.py

@@ -0,0 +1 @@
+"""Calendar database models."""

src/govoplan_calendar/backend/db/models.py

@@ -0,0 +1,97 @@
+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 CalendarCollection(Base, TimestampMixin):
+    __tablename__ = "calendar_collections"
+    __table_args__ = (
+        Index(
+            "uq_calendar_collections_active_slug",
+            "tenant_id",
+            "slug",
+            unique=True,
+            sqlite_where=text("deleted_at IS NULL"),
+            postgresql_where=text("deleted_at IS NULL"),
+        ),
+        Index("ix_calendar_collections_owner", "tenant_id", "owner_type", "owner_id"),
+    )
+
+    id: Mapped[str] = mapped_column(String(36), primary_key=True, default=new_uuid)
+    tenant_id: Mapped[str] = mapped_column(ForeignKey("tenants.id", ondelete="CASCADE"), nullable=False, index=True)
+    slug: Mapped[str] = mapped_column(String(100), nullable=False)
+    name: Mapped[str] = mapped_column(String(255), nullable=False)
+    description: Mapped[str | None] = mapped_column(Text)
+    timezone: Mapped[str] = mapped_column(String(100), default="UTC", nullable=False)
+    color: Mapped[str | None] = mapped_column(String(20))
+    owner_type: Mapped[str] = mapped_column(String(20), default="tenant", nullable=False, index=True)
+    owner_id: Mapped[str | None] = mapped_column(String(36), nullable=True, index=True)
+    visibility: Mapped[str] = mapped_column(String(20), default="tenant", nullable=False, index=True)
+    is_default: Mapped[bool] = mapped_column(Boolean, default=False, nullable=False, index=True)
+    created_by_user_id: Mapped[str | None] = mapped_column(ForeignKey("users.id", ondelete="SET NULL"), 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)
+
+    events: Mapped[list["CalendarEvent"]] = relationship(back_populates="calendar", cascade="all, delete-orphan")
+
+
+class CalendarEvent(Base, TimestampMixin):
+    __tablename__ = "calendar_events"
+    __table_args__ = (
+        UniqueConstraint("calendar_id", "uid", "recurrence_id", name="uq_calendar_events_calendar_uid_recurrence"),
+        Index("ix_calendar_events_range", "tenant_id", "calendar_id", "start_at", "end_at"),
+        Index("ix_calendar_events_tenant_uid", "tenant_id", "uid"),
+        Index("ix_calendar_events_tenant_status", "tenant_id", "status"),
+    )
+
+    id: Mapped[str] = mapped_column(String(36), primary_key=True, default=new_uuid)
+    tenant_id: Mapped[str] = mapped_column(ForeignKey("tenants.id", ondelete="CASCADE"), nullable=False, index=True)
+    calendar_id: Mapped[str] = mapped_column(ForeignKey("calendar_collections.id", ondelete="CASCADE"), nullable=False, index=True)
+    uid: Mapped[str] = mapped_column(String(255), nullable=False, index=True)
+    recurrence_id: Mapped[str | None] = mapped_column(String(255), nullable=True, index=True)
+    sequence: Mapped[int] = mapped_column(Integer, default=0, nullable=False)
+    summary: Mapped[str] = mapped_column(String(500), nullable=False)
+    description: Mapped[str | None] = mapped_column(Text)
+    location: Mapped[str | None] = mapped_column(String(500))
+    status: Mapped[str] = mapped_column(String(40), default="CONFIRMED", nullable=False, index=True)
+    transparency: Mapped[str] = mapped_column(String(20), default="OPAQUE", nullable=False)
+    classification: Mapped[str] = mapped_column(String(20), default="PUBLIC", nullable=False)
+    start_at: Mapped[datetime] = mapped_column(DateTime(timezone=True), nullable=False, index=True)
+    end_at: Mapped[datetime | None] = mapped_column(DateTime(timezone=True), nullable=True, index=True)
+    duration_seconds: Mapped[int | None] = mapped_column(Integer)
+    all_day: Mapped[bool] = mapped_column(Boolean, default=False, nullable=False, index=True)
+    timezone: Mapped[str | None] = mapped_column(String(100))
+    organizer: Mapped[dict[str, Any] | None] = mapped_column(JSON, nullable=True)
+    attendees: Mapped[list[dict[str, Any]]] = mapped_column(JSON, default=list, nullable=False)
+    categories: Mapped[list[str]] = mapped_column(JSON, default=list, nullable=False)
+    rrule: Mapped[dict[str, Any] | None] = mapped_column(JSON, nullable=True)
+    rdate: Mapped[list[dict[str, Any]]] = mapped_column(JSON, default=list, nullable=False)
+    exdate: Mapped[list[dict[str, Any]]] = mapped_column(JSON, default=list, nullable=False)
+    reminders: Mapped[list[dict[str, Any]]] = mapped_column(JSON, default=list, nullable=False)
+    attachments: Mapped[list[dict[str, Any]]] = mapped_column(JSON, default=list, nullable=False)
+    related_to: Mapped[list[dict[str, Any]]] = mapped_column(JSON, default=list, nullable=False)
+    source_kind: Mapped[str] = mapped_column(String(30), default="local", nullable=False, index=True)
+    source_href: Mapped[str | None] = mapped_column(String(1000))
+    etag: Mapped[str | None] = mapped_column(String(255), index=True)
+    icalendar: Mapped[dict[str, Any]] = mapped_column(JSON, default=dict, nullable=False)
+    raw_ics: Mapped[str | None] = mapped_column(Text)
+    created_by_user_id: Mapped[str | None] = mapped_column(ForeignKey("users.id", ondelete="SET NULL"), nullable=True, index=True)
+    updated_by_user_id: Mapped[str | None] = mapped_column(ForeignKey("users.id", ondelete="SET NULL"), 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)
+
+    calendar: Mapped[CalendarCollection] = relationship(back_populates="events")
+
+
+__all__ = ["CalendarCollection", "CalendarEvent", "new_uuid"]