mirror of
https://github.com/MODSetter/SurfSense.git
synced 2025-09-02 02:29:08 +00:00
add migration file
This commit is contained in:
parent
d92b7ee538
commit
442417b808
2 changed files with 65 additions and 0 deletions
|
@ -0,0 +1,63 @@
|
||||||
|
"""Add ClickUp connector enums
|
||||||
|
|
||||||
|
Revision ID: 15_add_clickup_connector_enums
|
||||||
|
Revises: 14_add_confluence_connector_enums
|
||||||
|
Create Date: 2025-07-29 12:00:00.000000
|
||||||
|
|
||||||
|
"""
|
||||||
|
|
||||||
|
from typing import Sequence, Union
|
||||||
|
|
||||||
|
from alembic import op
|
||||||
|
|
||||||
|
# revision identifiers, used by Alembic.
|
||||||
|
revision: str = "15_add_clickup_connector_enums"
|
||||||
|
down_revision: Union[str, None] = "14_add_confluence_connector_enums"
|
||||||
|
branch_labels: Union[str, Sequence[str], None] = None
|
||||||
|
depends_on: Union[str, Sequence[str], None] = None
|
||||||
|
|
||||||
|
|
||||||
|
def upgrade() -> None:
|
||||||
|
"""Safely add 'CLICKUP_CONNECTOR' to enum types if missing."""
|
||||||
|
|
||||||
|
# Add to searchsourceconnectortype enum
|
||||||
|
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 = 'searchsourceconnectortype' AND e.enumlabel = 'CLICKUP_CONNECTOR'
|
||||||
|
) THEN
|
||||||
|
ALTER TYPE searchsourceconnectortype ADD VALUE 'CLICKUP_CONNECTOR';
|
||||||
|
END IF;
|
||||||
|
END
|
||||||
|
$$;
|
||||||
|
"""
|
||||||
|
)
|
||||||
|
|
||||||
|
# Add to documenttype enum
|
||||||
|
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 = 'CLICKUP_CONNECTOR'
|
||||||
|
) THEN
|
||||||
|
ALTER TYPE documenttype ADD VALUE 'CLICKUP_CONNECTOR';
|
||||||
|
END IF;
|
||||||
|
END
|
||||||
|
$$;
|
||||||
|
"""
|
||||||
|
)
|
||||||
|
|
||||||
|
|
||||||
|
def downgrade() -> None:
|
||||||
|
"""Remove 'CLICKUP_CONNECTOR' from enum types."""
|
||||||
|
# Note: PostgreSQL doesn't support removing enum values directly
|
||||||
|
# This would require recreating the enum type, which is complex
|
||||||
|
# For now, we'll leave the enum values in place
|
||||||
|
pass
|
|
@ -44,6 +44,7 @@ class DocumentType(str, Enum):
|
||||||
DISCORD_CONNECTOR = "DISCORD_CONNECTOR"
|
DISCORD_CONNECTOR = "DISCORD_CONNECTOR"
|
||||||
JIRA_CONNECTOR = "JIRA_CONNECTOR"
|
JIRA_CONNECTOR = "JIRA_CONNECTOR"
|
||||||
CONFLUENCE_CONNECTOR = "CONFLUENCE_CONNECTOR"
|
CONFLUENCE_CONNECTOR = "CONFLUENCE_CONNECTOR"
|
||||||
|
CLICKUP_CONNECTOR = "CLICKUP_CONNECTOR"
|
||||||
|
|
||||||
|
|
||||||
class SearchSourceConnectorType(str, Enum):
|
class SearchSourceConnectorType(str, Enum):
|
||||||
|
@ -57,6 +58,7 @@ class SearchSourceConnectorType(str, Enum):
|
||||||
DISCORD_CONNECTOR = "DISCORD_CONNECTOR"
|
DISCORD_CONNECTOR = "DISCORD_CONNECTOR"
|
||||||
JIRA_CONNECTOR = "JIRA_CONNECTOR"
|
JIRA_CONNECTOR = "JIRA_CONNECTOR"
|
||||||
CONFLUENCE_CONNECTOR = "CONFLUENCE_CONNECTOR"
|
CONFLUENCE_CONNECTOR = "CONFLUENCE_CONNECTOR"
|
||||||
|
CLICKUP_CONNECTOR = "CLICKUP_CONNECTOR"
|
||||||
|
|
||||||
|
|
||||||
class ChatType(str, Enum):
|
class ChatType(str, Enum):
|
||||||
|
|
Loading…
Add table
Reference in a new issue