Update code generator plugin naming to CodeGen prefix (#1243)

backend/plugin/code_generator/service/column_service.py

@@ -5,19 +5,19 @@ 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_business import gen_business_dao
-from backend.plugin.code_generator.crud.crud_column import gen_column_dao
+from backend.plugin.code_generator.crud.crud_business import code_gen_business_dao
+from backend.plugin.code_generator.crud.crud_column import code_gen_column_dao
 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.model import CodeGenColumn
+from backend.plugin.code_generator.schema.column import CreateCodeGenColumnParam, UpdateCodeGenColumnParam
 from backend.plugin.code_generator.utils.type_conversion import sql_type_to_pydantic
 
 
-class GenColumnService:
+class CodeGenColumnService:
     """代码生成模型列服务类"""
 
     @staticmethod
-    async def get(*, db: AsyncSession, pk: int) -> GenColumn:
+    async def get(*, db: AsyncSession, pk: int) -> CodeGenColumn:
         """
         获取指定 ID 的模型列
 
@@ -26,10 +26,10 @@ class GenColumnService:
         :return:
         """
 
-        column = await gen_column_dao.get(db, pk)
+        column = await code_gen_column_dao.get(db, pk)
         if not column:
             raise errors.NotFoundError(msg='代码生成模型列不存在')
-        if not await gen_business_dao.get(db, column.gen_business_id):
+        if not await code_gen_business_dao.get(db, column.code_gen_business_id):
             raise errors.NotFoundError(msg='代码生成业务不存在')
         return column
 
@@ -44,7 +44,7 @@ class GenColumnService:
         return types
 
     @staticmethod
-    async def get_columns(*, db: AsyncSession, business_id: int) -> Sequence[GenColumn]:
+    async def get_columns(*, db: AsyncSession, business_id: int) -> Sequence[CodeGenColumn]:
         """
         获取指定业务的所有模型列
 
@@ -53,12 +53,12 @@ class GenColumnService:
         :return:
         """
 
-        if not await gen_business_dao.get(db, business_id):
+        if not await code_gen_business_dao.get(db, business_id):
             raise errors.NotFoundError(msg='代码生成业务不存在')
-        return await gen_column_dao.get_all_by_business(db, business_id)
+        return await code_gen_column_dao.get_all_by_business(db, business_id)
 
     @staticmethod
-    async def create(*, db: AsyncSession, obj: CreateGenColumnParam) -> None:
+    async def create(*, db: AsyncSession, obj: CreateCodeGenColumnParam) -> None:
         """
         创建模型列
 
@@ -67,18 +67,18 @@ class GenColumnService:
         :return:
         """
 
-        if not await gen_business_dao.get(db, obj.gen_business_id):
+        if not await code_gen_business_dao.get(db, obj.code_gen_business_id):
             raise errors.NotFoundError(msg='代码生成业务不存在')
 
-        gen_columns = await gen_column_dao.get_all_by_business(db, obj.gen_business_id)
-        if obj.name in [gen_column.name for gen_column in gen_columns]:
+        code_gen_columns = await code_gen_column_dao.get_all_by_business(db, obj.code_gen_business_id)
+        if obj.name in [code_gen_column.name for code_gen_column in code_gen_columns]:
             raise errors.ForbiddenError(msg='模型列已存在')
 
         pd_type = sql_type_to_pydantic(obj.type)
-        await gen_column_dao.create(db, obj, pd_type=pd_type)
+        await code_gen_column_dao.create(db, obj, pd_type=pd_type)
 
     @staticmethod
-    async def update(*, db: AsyncSession, pk: int, obj: UpdateGenColumnParam) -> int:
+    async def update(*, db: AsyncSession, pk: int, obj: UpdateCodeGenColumnParam) -> int:
         """
         更新模型列
 
@@ -88,20 +88,20 @@ class GenColumnService:
         :return:
         """
 
-        column = await gen_column_dao.get(db, pk)
+        column = await code_gen_column_dao.get(db, pk)
         if not column:
             raise errors.NotFoundError(msg='代码生成模型列不存在')
-        if not await gen_business_dao.get(db, column.gen_business_id):
+        if not await code_gen_business_dao.get(db, column.code_gen_business_id):
             raise errors.NotFoundError(msg='代码生成业务不存在')
-        if not await gen_business_dao.get(db, obj.gen_business_id):
+        if not await code_gen_business_dao.get(db, obj.code_gen_business_id):
             raise errors.NotFoundError(msg='代码生成业务不存在')
         if obj.name != column.name:
-            gen_columns = await gen_column_dao.get_all_by_business(db, obj.gen_business_id)
-            if obj.name in [gen_column.name for gen_column in gen_columns]:
+            code_gen_columns = await code_gen_column_dao.get_all_by_business(db, obj.code_gen_business_id)
+            if obj.name in [code_gen_column.name for code_gen_column in code_gen_columns]:
                 raise errors.ConflictError(msg='模型列名已存在')
 
         pd_type = sql_type_to_pydantic(obj.type)
-        return await gen_column_dao.update(db, pk, obj, pd_type=pd_type)
+        return await code_gen_column_dao.update(db, pk, obj, pd_type=pd_type)
 
     @staticmethod
     async def delete(*, db: AsyncSession, pk: int) -> int:
@@ -113,12 +113,12 @@ class GenColumnService:
         :return:
         """
 
-        column = await gen_column_dao.get(db, pk)
+        column = await code_gen_column_dao.get(db, pk)
         if not column:
             raise errors.NotFoundError(msg='代码生成模型列不存在')
-        if not await gen_business_dao.get(db, column.gen_business_id):
+        if not await code_gen_business_dao.get(db, column.code_gen_business_id):
             raise errors.NotFoundError(msg='代码生成业务不存在')
-        return await gen_column_dao.delete(db, pk)
+        return await code_gen_column_dao.delete(db, pk)
 
 
-gen_column_service: GenColumnService = GenColumnService()
+code_gen_column_service: CodeGenColumnService = CodeGenColumnService()