Fix get column types in code generation (#935)

Author: wu-clan

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