apunkt/SurfSense
1
0
Fork 0
mirror of https://github.com/MODSetter/SurfSense.git synced 2026-07-22 23:31:12 +02:00
Code Issues Projects Releases Packages Wiki Activity

feat: add ELASTICSEARCH_CONNECTOR enum values to searchsourceconnectortype and documenttype

Browse source
This commit is contained in:
Anish Sarkar 2025-10-06 21:57:35 +05:30
parent 2ad1c43059
commit 66327bf77f
1 changed files with 39 additions and 14 deletions
Download patch file Download diff file Expand all files Collapse all files

53
surfsense_backend/alembic/versions/22_add_elasticsearch_connector_enum.py
View file

@ -1,30 +1,55 @@
"""add elasticsearch connector enum """Add ElasticSearch connector enums
Revision ID: 22 Revision ID: 22
Revises: 21 Revises: 21
Create Date: 2025-09-30 12:00:00.000000 Create Date: 2025-10-4 12:00:00.000000
""" """
# import sqlalchemy as sa
from collections.abc import Sequence
from alembic import op from alembic import op
# revision identifiers # revision identifiers
revision = "22_add_elasticsearch_connector_enums" revision: str = "22"
down_revision = "21_add_luma_connector_enums" down_revision: str | None = "21"
branch_labels = None branch_labels: str | Sequence[str] | None = None
depends_on = None depends_on: str | Sequence[str] | None = None
def upgrade() -> None: def upgrade() -> None:
# Add ELASTICSEARCH to SearchSourceType enum # Add enum values
op.execute("ALTER TYPE searchsourcetype ADD VALUE 'ELASTICSEARCH'") op.execute(
"""
# Add ELASTICSEARCH to DocumentType enum DO $$
op.execute("ALTER TYPE documenttype ADD VALUE 'ELASTICSEARCH'") BEGIN
IF NOT EXISTS (
SELECT 1 FROM pg_type t
JOIN pg_enum e ON t.oid = e.enumtypid
WHERE t.typname = 'searchsourceconnectortype' AND e.enumlabel = 'ELASTICSEARCH_CONNECTOR'
) THEN
ALTER TYPE searchsourceconnectortype ADD VALUE 'ELASTICSEARCH_CONNECTOR';
END IF;
END
$$;
"""
)
op.execute(
"""
DO $$
BEGIN
IF NOT EXISTS (
SELECT 1 FROM pg_type t
JOIN pg_enum e ON t.oid = e.enumtypid
WHERE t.typname = 'documenttype' AND e.enumlabel = 'ELASTICSEARCH_CONNECTOR'
) THEN
ALTER TYPE documenttype ADD VALUE 'ELASTICSEARCH_CONNECTOR';
END IF;
END
$$;
"""
)
def downgrade() -> None: def downgrade() -> None:
# Note: PostgreSQL doesn't support removing enum values
# This would require recreating the enum types
pass pass