diff --git a/backend/plugin/code_generator/service/column_service.py b/backend/plugin/code_generator/service/column_service.py index 8b9b4df82..46707b749 100644 --- a/backend/plugin/code_generator/service/column_service.py +++ b/backend/plugin/code_generator/service/column_service.py @@ -3,8 +3,9 @@ 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 @@ async def get(*, db: AsyncSession, pk: int) -> GenColumn: @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