python/fastapi_best_architecture
1
0
Fork 0
mirror of https://github.com/fastapi-practices/fastapi_best_architecture.git synced 2026-03-13 09:31:31 +08:00
Code Issues Packages Projects Releases Wiki Activity

Fix get column types in code generation (#935)

Browse Source
This commit is contained in:
Wu Clan
2025-11-25 18:07:09 +08:00
committed by GitHub
parent 4d43c6a3ea
commit ef640b0b5c
1 changed files with 7 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

10
backend/plugin/code_generator/service/column_service.py
View File

@@ -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