From bf048b8d7c709035f23cbb6bd20bc772bbb0d766 Mon Sep 17 00:00:00 2001 From: -LAN- Date: Fri, 1 Nov 2024 16:10:55 +0800 Subject: [PATCH] refactor(migration/model): update column types for workflow schema (#10160) --- ...0623-09a8d1878d9b_update_workflows_graph_features_and_.py | 4 +--- api/models/workflow.py | 5 +++-- 2 files changed, 4 insertions(+), 5 deletions(-) diff --git a/api/migrations/versions/2024_11_01_0623-09a8d1878d9b_update_workflows_graph_features_and_.py b/api/migrations/versions/2024_11_01_0623-09a8d1878d9b_update_workflows_graph_features_and_.py index 0c6b986738..117a7351cd 100644 --- a/api/migrations/versions/2024_11_01_0623-09a8d1878d9b_update_workflows_graph_features_and_.py +++ b/api/migrations/versions/2024_11_01_0623-09a8d1878d9b_update_workflows_graph_features_and_.py @@ -39,7 +39,6 @@ def upgrade(): nullable=False) batch_op.alter_column('features', existing_type=sa.TEXT(), - type_=sa.String(), nullable=False) batch_op.alter_column('updated_at', existing_type=postgresql.TIMESTAMP(), @@ -55,8 +54,7 @@ def downgrade(): existing_type=postgresql.TIMESTAMP(), nullable=True) batch_op.alter_column('features', - existing_type=sa.String(), - type_=sa.TEXT(), + existing_type=sa.TEXT(), nullable=True) batch_op.alter_column('graph', existing_type=sa.TEXT(), diff --git a/api/models/workflow.py b/api/models/workflow.py index 75c33f4d27..24dd10fbc5 100644 --- a/api/models/workflow.py +++ b/api/models/workflow.py @@ -4,6 +4,7 @@ from datetime import datetime from enum import Enum from typing import Any, Optional, Union +import sqlalchemy as sa from sqlalchemy import func from sqlalchemy.orm import Mapped, mapped_column @@ -99,8 +100,8 @@ class Workflow(db.Model): app_id: Mapped[str] = mapped_column(StringUUID, nullable=False) type: Mapped[str] = mapped_column(db.String(255), nullable=False) version: Mapped[str] = mapped_column(db.String(255), nullable=False) - graph: Mapped[str] = mapped_column(db.Text) - _features: Mapped[str] = mapped_column("features") + graph: Mapped[str] = mapped_column(sa.Text) + _features: Mapped[str] = mapped_column("features", sa.TEXT) created_by: Mapped[str] = mapped_column(StringUUID, nullable=False) created_at: Mapped[datetime] = mapped_column( db.DateTime, nullable=False, server_default=db.text("CURRENT_TIMESTAMP(0)")