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

Tidy alembic migration version scripts

Browse source
This commit is contained in:
CREDO23 2026-04-24 18:49:22 +02:00
parent d1080b1298
commit dfa6c0423d
2 changed files with 55 additions and 39 deletions
Download patch file Download diff file Expand all files Collapse all files

6
surfsense_backend/alembic/versions/117_optimize_zero_publication_column_lists.py
View file

@ -79,7 +79,11 @@ def _terminate_blocked_pids(conn, table: str) -> None:
def upgrade() -> None: def upgrade() -> None:
conn = op.get_bind() conn = op.get_bind()
# asyncpg requires LOCK TABLE inside a transaction block. Alembic already
# opened one via context.begin_transaction(), but the driver still errors
# unless we use an explicit SAVEPOINT (nested transaction) for this block.
tx = conn.begin_nested() if conn.in_transaction() else conn.begin()
with tx:
conn.execute(sa.text("SET lock_timeout = '10s'")) conn.execute(sa.text("SET lock_timeout = '10s'"))
for tbl in sorted(TABLES_WITH_FULL_IDENTITY): for tbl in sorted(TABLES_WITH_FULL_IDENTITY):

34
surfsense_backend/alembic/versions/121_add_memory_md_columns.py
View file

@ -12,8 +12,6 @@ from __future__ import annotations
from collections.abc import Sequence from collections.abc import Sequence
import sqlalchemy as sa
from alembic import op from alembic import op
revision: str = "121" revision: str = "121"
@ -23,16 +21,30 @@ depends_on: str | Sequence[str] | None = None
def upgrade() -> None: def upgrade() -> None:
op.add_column( # Idempotent: column(s) may already exist after a failed run or manual DDL.
"user", op.execute(
sa.Column("memory_md", sa.Text(), nullable=True, server_default=""), """
) DO $$
op.add_column( BEGIN
"searchspaces", IF NOT EXISTS (
sa.Column("shared_memory_md", sa.Text(), nullable=True, server_default=""), SELECT 1 FROM information_schema.columns
WHERE table_schema = 'public' AND table_name = 'user'
AND column_name = 'memory_md'
) THEN
ALTER TABLE "user" ADD COLUMN memory_md TEXT DEFAULT '';
END IF;
IF NOT EXISTS (
SELECT 1 FROM information_schema.columns
WHERE table_schema = 'public' AND table_name = 'searchspaces'
AND column_name = 'shared_memory_md'
) THEN
ALTER TABLE searchspaces ADD COLUMN shared_memory_md TEXT DEFAULT '';
END IF;
END$$;
"""
) )
def downgrade() -> None: def downgrade() -> None:
op.drop_column("searchspaces", "shared_memory_md") op.execute("ALTER TABLE searchspaces DROP COLUMN IF EXISTS shared_memory_md")
op.drop_column("user", "memory_md") op.execute('ALTER TABLE "user" DROP COLUMN IF EXISTS memory_md')