Add the database primary key mode config (#953)

* Add the database primary key mode config

* Update auto to autoincrement
This commit is contained in:
Wu Clan
2025-12-05 14:52:12 +08:00
committed by GitHub
parent aad9afa9bb
commit bb5bcdf430
9 changed files with 39 additions and 27 deletions
@@ -2,6 +2,7 @@ from collections.abc import Sequence
from sqlalchemy.ext.asyncio import AsyncSession
from backend.common.enums import DataBaseType
from backend.common.exception import errors
from backend.core.conf import settings
from backend.plugin.code_generator.crud.crud_column import gen_column_dao
@@ -32,7 +33,7 @@ class GenColumnService:
@staticmethod
async def get_types() -> list[str]:
"""获取所有列类型"""
if settings.DATABASE_TYPE == 'mysql':
if DataBaseType.mysql == settings.DATABASE_TYPE:
types = GenMySQLColumnType.get_member_keys()
else:
types = GenPostgreSQLColumnType.get_member_keys()