129 lines
4.0 KiB
Python
129 lines
4.0 KiB
Python
"""add moments + junction tables + moment embeddings
|
|
|
|
Revision ID: 0041
|
|
Revises: 0040
|
|
Create Date: 2026-04-25
|
|
|
|
People, Places, Tasks, and Notes all live in the `notes` table (distinguished
|
|
by note_type and is_task). The four junction tables FK to notes(id) but stay
|
|
separate so per-link-kind queries don't require a discriminator filter.
|
|
"""
|
|
from alembic import op
|
|
import sqlalchemy as sa
|
|
from sqlalchemy.dialects.postgresql import ARRAY, JSONB
|
|
|
|
|
|
revision = "0041"
|
|
down_revision = "0040"
|
|
branch_labels = None
|
|
depends_on = None
|
|
|
|
|
|
def upgrade() -> None:
|
|
op.create_table(
|
|
"moments",
|
|
sa.Column("id", sa.Integer, primary_key=True),
|
|
sa.Column(
|
|
"user_id",
|
|
sa.Integer,
|
|
sa.ForeignKey("users.id", ondelete="CASCADE"),
|
|
nullable=False,
|
|
),
|
|
sa.Column(
|
|
"conversation_id",
|
|
sa.Integer,
|
|
sa.ForeignKey("conversations.id", ondelete="SET NULL"),
|
|
nullable=True,
|
|
),
|
|
sa.Column(
|
|
"source_message_id",
|
|
sa.Integer,
|
|
sa.ForeignKey("messages.id", ondelete="SET NULL"),
|
|
nullable=True,
|
|
),
|
|
sa.Column("day_date", sa.Date, nullable=False),
|
|
sa.Column("occurred_at", sa.DateTime(timezone=True), nullable=False),
|
|
sa.Column(
|
|
"recorded_at",
|
|
sa.DateTime(timezone=True),
|
|
nullable=False,
|
|
server_default=sa.text("now()"),
|
|
),
|
|
sa.Column("content", sa.Text, nullable=False),
|
|
sa.Column("raw_excerpt", sa.Text, nullable=True),
|
|
sa.Column(
|
|
"tags",
|
|
ARRAY(sa.Text),
|
|
nullable=False,
|
|
server_default=sa.text("'{}'::text[]"),
|
|
),
|
|
sa.Column(
|
|
"pinned",
|
|
sa.Boolean,
|
|
nullable=False,
|
|
server_default=sa.text("false"),
|
|
),
|
|
)
|
|
op.create_index("ix_moments_user_day", "moments", ["user_id", "day_date"])
|
|
op.create_index("ix_moments_user_occurred", "moments", ["user_id", "occurred_at"])
|
|
|
|
# Four junction tables, all FK to notes(id). Separate (vs. one merged
|
|
# table with a discriminator) so per-kind queries don't need a filter.
|
|
for table_name, fk_col in [
|
|
("moment_people", "person_id"),
|
|
("moment_places", "place_id"),
|
|
("moment_tasks", "task_id"),
|
|
("moment_notes", "note_id"),
|
|
]:
|
|
op.create_table(
|
|
table_name,
|
|
sa.Column(
|
|
"moment_id",
|
|
sa.Integer,
|
|
sa.ForeignKey("moments.id", ondelete="CASCADE"),
|
|
primary_key=True,
|
|
),
|
|
sa.Column(
|
|
fk_col,
|
|
sa.Integer,
|
|
sa.ForeignKey("notes.id", ondelete="CASCADE"),
|
|
primary_key=True,
|
|
),
|
|
)
|
|
op.create_index(f"ix_{table_name}_{fk_col}", table_name, [fk_col])
|
|
|
|
op.create_table(
|
|
"moment_embeddings",
|
|
sa.Column(
|
|
"moment_id",
|
|
sa.Integer,
|
|
sa.ForeignKey("moments.id", ondelete="CASCADE"),
|
|
primary_key=True,
|
|
),
|
|
sa.Column("user_id", sa.Integer, nullable=False),
|
|
sa.Column("embedding", JSONB, nullable=False),
|
|
sa.Column(
|
|
"updated_at",
|
|
sa.DateTime(timezone=True),
|
|
nullable=False,
|
|
server_default=sa.text("now()"),
|
|
),
|
|
)
|
|
op.create_index("ix_moment_embeddings_user", "moment_embeddings", ["user_id"])
|
|
|
|
|
|
def downgrade() -> None:
|
|
op.drop_index("ix_moment_embeddings_user", table_name="moment_embeddings")
|
|
op.drop_table("moment_embeddings")
|
|
for table_name, fk_col in [
|
|
("moment_notes", "note_id"),
|
|
("moment_tasks", "task_id"),
|
|
("moment_places", "place_id"),
|
|
("moment_people", "person_id"),
|
|
]:
|
|
op.drop_index(f"ix_{table_name}_{fk_col}", table_name=table_name)
|
|
op.drop_table(table_name)
|
|
op.drop_index("ix_moments_user_occurred", table_name="moments")
|
|
op.drop_index("ix_moments_user_day", table_name="moments")
|
|
op.drop_table("moments")
|