apunkt/SurfSense
1
0
Fork 0
mirror of https://github.com/MODSetter/SurfSense.git synced 2026-06-08 20:25:19 +02:00
Code Issues Projects Releases Packages Wiki Activity

fix (migration 43): Add an 'if not exists' guard

Browse source
This commit is contained in:
CREDO23 2025-12-02 10:46:47 +02:00
parent 1d8cfbb143
commit 509ef211fa
1 changed files with 46 additions and 39 deletions
Download patch file Download diff file Expand all files Collapse all files

85
surfsense_backend/alembic/versions/43_add_blocknote_fields_to_documents.py
View file

@ -16,6 +16,7 @@ import sqlalchemy as sa
from sqlalchemy.dialects import postgresql
from alembic import op
from sqlalchemy import inspect
# revision identifiers, used by Alembic.
revision: str = "43"
@ -25,51 +26,57 @@ depends_on: str | Sequence[str] | None = None
def upgrade() -> None:
"""Upgrade schema - Add BlockNote fields and trigger population task."""
"""Upgrade schema - Add BlockNote fields (idempotent)."""
# Add the columns
op.add_column(
"documents",
sa.Column(
"blocknote_document", postgresql.JSONB(astext_type=sa.Text()), nullable=True
),
)
op.add_column(
"documents",
sa.Column(
"content_needs_reindexing",
sa.Boolean(),
nullable=False,
server_default=sa.false(),
),
)
op.add_column(
"documents",
sa.Column("last_edited_at", sa.TIMESTAMP(timezone=True), nullable=True),
)
conn = op.get_bind()
inspector = inspect(conn)
existing_cols = {c["name"] for c in inspector.get_columns("documents")}
# Trigger the Celery task to populate blocknote_document for existing documents
try:
from app.tasks.celery_tasks.blocknote_migration_tasks import (
populate_blocknote_for_documents_task,
# Add blocknote_document (JSONB) if doest not exist
if "blocknote_document" not in existing_cols:
op.add_column(
"documents",
sa.Column(
"blocknote_document",
postgresql.JSONB(astext_type=sa.Text()),
nullable=True,
),
)
# Queue the task to run asynchronously
populate_blocknote_for_documents_task.apply_async()
print(
"✓ Queued Celery task to populate blocknote_document for existing documents"
# Add content_needs_reindexing (boolean) if doest not exist
if "content_needs_reindexing" not in existing_cols:
op.add_column(
"documents",
sa.Column(
"content_needs_reindexing",
sa.Boolean(),
nullable=False,
server_default=sa.false(),
),
)
except Exception as e:
# If Celery is not available or task queueing fails, log but don't fail the migration
print(f"⚠ Warning: Could not queue blocknote population task: {e}")
print(" You can manually trigger it later with:")
print(
" celery -A app.celery_app call app.tasks.celery_tasks.blocknote_migration_tasks.populate_blocknote_for_documents_task"
# Add last_edited_at (timestamp with tz) if doest not exist
if "last_edited_at" not in existing_cols:
op.add_column(
"documents",
sa.Column("last_edited_at", sa.TIMESTAMP(timezone=True), nullable=True),
)
# NOTE: We intentionally do NOT import or queue Celery tasks here.
# Running background jobs during migrations causes hard-to-debug failures.
# After running migrations, trigger the backfill task manually (instructions below).
def downgrade() -> None:
"""Downgrade schema - Remove BlockNote fields."""
op.drop_column("documents", "last_edited_at")
op.drop_column("documents", "content_needs_reindexing")
op.drop_column("documents", "blocknote_document")
"""Downgrade schema - Remove BlockNote fields (only if present)."""
conn = op.get_bind()
inspector = inspect(conn)
existing_cols = {c["name"] for c in inspector.get_columns("documents")}
if "last_edited_at" in existing_cols:
op.drop_column("documents", "last_edited_at")
if "content_needs_reindexing" in existing_cols:
op.drop_column("documents", "content_needs_reindexing")
if "blocknote_document" in existing_cols:
op.drop_column("documents", "blocknote_document")