From ef640b0b5c1fd854392285fd7afc1a2f9a92bed8 Mon Sep 17 00:00:00 2001 From: Wu Clan Date: Tue, 25 Nov 2025 18:07:09 +0800 Subject: [PATCH] Fix get column types in code generation (#935) --- .../plugin/code_generator/service/column_service.py | 10 +++++++--- 1 file changed, 7 insertions(+), 3 deletions(-) diff --git a/backend/plugin/code_generator/service/column_service.py b/backend/plugin/code_generator/service/column_service.py index 8b9b4df8..46707b74 100644 --- a/backend/plugin/code_generator/service/column_service.py +++ b/backend/plugin/code_generator/service/column_service.py @@ -3,8 +3,9 @@ from collections.abc import Sequence from sqlalchemy.ext.asyncio import AsyncSession from backend.common.exception import errors +from backend.core.conf import settings from backend.plugin.code_generator.crud.crud_column import gen_column_dao -from backend.plugin.code_generator.enums import GenMySQLColumnType +from backend.plugin.code_generator.enums import GenMySQLColumnType, GenPostgreSQLColumnType from backend.plugin.code_generator.model import GenColumn from backend.plugin.code_generator.schema.column import CreateGenColumnParam, UpdateGenColumnParam from backend.plugin.code_generator.utils.type_conversion import sql_type_to_pydantic @@ -30,8 +31,11 @@ class GenColumnService: @staticmethod async def get_types() -> list[str]: - """获取所有 MySQL 列类型""" - types = GenMySQLColumnType.get_member_keys() + """获取所有列类型""" + if settings.DATABASE_TYPE == 'mysql': + types = GenMySQLColumnType.get_member_keys() + else: + types = GenPostgreSQLColumnType.get_member_keys() types.sort() return types