功能：新增业务基础资料接口

shierfeng-fastapi-backend/module_business/service/master_service.py

@@ -0,0 +1,375 @@
+from typing import Any, Literal
+
+from pydantic import BaseModel
+from sqlalchemy.exc import IntegrityError
+from sqlalchemy.ext.asyncio import AsyncSession
+
+from common.vo import CrudResponseModel, PageModel
+from exceptions.exception import ServiceException, ServiceWarning
+from module_business.dao.master_dao import MasterDao
+from module_business.entity.vo.master_vo import (
+    CustomerModel,
+    CustomerPageQueryModel,
+    DeleteCustomerModel,
+    DeleteLineModel,
+    DeleteSiteModel,
+    DeleteStrainModel,
+    DeleteWarehouseModel,
+    LineModel,
+    LinePageQueryModel,
+    SiteModel,
+    SitePageQueryModel,
+    StrainModel,
+    StrainPageQueryModel,
+    WarehouseModel,
+    WarehousePageQueryModel,
+)
+
+MasterResource = Literal['strain', 'customer', 'site', 'line', 'warehouse']
+
+
+_RESOURCE_CONFIG: dict[MasterResource, dict[str, Any]] = {
+    'strain': {
+        'label': '菌种',
+        'id_field': 'strain_id',
+        'code_field': 'strain_code',
+        'ids_field': 'strain_ids',
+        'vo': StrainModel,
+    },
+    'customer': {
+        'label': '客户',
+        'id_field': 'customer_id',
+        'code_field': 'customer_code',
+        'ids_field': 'customer_ids',
+        'vo': CustomerModel,
+    },
+    'site': {
+        'label': '客户现场',
+        'id_field': 'site_id',
+        'code_field': 'site_code',
+        'ids_field': 'site_ids',
+        'vo': SiteModel,
+    },
+    'line': {
+        'label': '生产线',
+        'id_field': 'line_id',
+        'code_field': 'line_code',
+        'ids_field': 'line_ids',
+        'vo': LineModel,
+    },
+    'warehouse': {
+        'label': '仓库',
+        'id_field': 'warehouse_id',
+        'code_field': 'warehouse_code',
+        'ids_field': 'warehouse_ids',
+        'vo': WarehouseModel,
+    },
+}
+
+
+class MasterService:
+    """菌种、客户、现场、产线和仓库主数据服务层。"""
+
+    @staticmethod
+    async def _validate_parent(db: AsyncSession, resource: MasterResource, page_object: BaseModel) -> None:
+        if resource == 'site':
+            customer = await MasterDao.get_customer(db, page_object.customer_id)
+            if customer is None:
+                raise ServiceException(message='目标客户不存在')
+            if customer.status != '0':
+                raise ServiceException(message='目标客户已停用')
+        elif resource == 'line':
+            site = await MasterDao.get_site(db, page_object.site_id)
+            if site is None:
+                raise ServiceException(message='目标客户现场不存在')
+            if site.status != '0':
+                raise ServiceException(message='目标客户现场已停用')
+
+    @staticmethod
+    async def _find_duplicate(
+        db: AsyncSession,
+        resource: MasterResource,
+        page_object: BaseModel,
+    ) -> Any | None:
+        if resource == 'site':
+            return await MasterDao.find_site_by_code(
+                db,
+                page_object.site_code,
+                page_object.customer_id,
+            )
+        if resource == 'line':
+            return await MasterDao.find_line_by_code(
+                db,
+                page_object.line_code,
+                page_object.site_id,
+            )
+        code_field = _RESOURCE_CONFIG[resource]['code_field']
+        return await getattr(MasterDao, f'find_{resource}_by_code')(
+            db,
+            getattr(page_object, code_field),
+        )
+
+    @classmethod
+    async def _validate_unique(
+        cls,
+        db: AsyncSession,
+        resource: MasterResource,
+        page_object: BaseModel,
+        current_id: int | None = None,
+    ) -> None:
+        duplicate = await cls._find_duplicate(db, resource, page_object)
+        if duplicate is None:
+            return
+
+        config = _RESOURCE_CONFIG[resource]
+        duplicate_id = getattr(duplicate, config['id_field'], None)
+        if current_id is None or duplicate_id != current_id:
+            code = getattr(page_object, config['code_field'])
+            raise ServiceException(message=f'{config["label"]}编码{code}已存在')
+
+    @staticmethod
+    def _extract_ids(resource: MasterResource, value: int | BaseModel) -> list[int]:
+        ids = [value] if isinstance(value, int) else list(getattr(value, _RESOURCE_CONFIG[resource]['ids_field']))
+        if not ids or any(resource_id <= 0 for resource_id in ids):
+            raise ServiceException(message='主数据ID必须为正整数')
+        if len(ids) != len(set(ids)):
+            raise ServiceException(message='主数据ID不能重复')
+        return ids
+
+    @classmethod
+    async def _list(
+        cls,
+        db: AsyncSession,
+        resource: MasterResource,
+        query_object: BaseModel,
+        is_page: bool,
+    ) -> PageModel | list[dict[str, Any]]:
+        return await getattr(MasterDao, f'list_{resource}')(db, query_object, is_page)
+
+    @classmethod
+    async def _detail(cls, db: AsyncSession, resource: MasterResource, resource_id: int) -> BaseModel:
+        config = _RESOURCE_CONFIG[resource]
+        record = await getattr(MasterDao, f'get_{resource}')(db, resource_id)
+        if record is None:
+            raise ServiceException(message=f'{config["label"]}不存在')
+        return config['vo'].model_validate(record)
+
+    @classmethod
+    async def _add(
+        cls,
+        db: AsyncSession,
+        resource: MasterResource,
+        page_object: BaseModel,
+    ) -> CrudResponseModel:
+        config = _RESOURCE_CONFIG[resource]
+        try:
+            await cls._validate_parent(db, resource, page_object)
+            await cls._validate_unique(db, resource, page_object)
+            safe_object = page_object.model_copy(update={config['id_field']: None})
+            await getattr(MasterDao, f'add_{resource}')(db, safe_object)
+            await db.commit()
+            return CrudResponseModel(is_success=True, message='新增成功')
+        except IntegrityError:
+            await db.rollback()
+            raise ServiceException(message=f'新增{config["label"]}失败，编码已存在或关联数据无效') from None
+        except Exception:
+            await db.rollback()
+            raise
+
+    @classmethod
+    async def _edit(
+        cls,
+        db: AsyncSession,
+        resource: MasterResource,
+        page_object: BaseModel,
+    ) -> CrudResponseModel:
+        config = _RESOURCE_CONFIG[resource]
+        try:
+            resource_id = getattr(page_object, config['id_field'])
+            if resource_id is None:
+                raise ServiceException(message=f'{config["label"]}ID不能为空')
+            current = await getattr(MasterDao, f'get_{resource}')(db, resource_id)
+            if current is None:
+                raise ServiceException(message=f'{config["label"]}不存在')
+            await cls._validate_parent(db, resource, page_object)
+            await cls._validate_unique(db, resource, page_object, resource_id)
+            await getattr(MasterDao, f'update_{resource}')(db, page_object)
+            await db.commit()
+            return CrudResponseModel(is_success=True, message='更新成功')
+        except IntegrityError:
+            await db.rollback()
+            raise ServiceException(message=f'修改{config["label"]}失败，编码已存在或关联数据无效') from None
+        except Exception:
+            await db.rollback()
+            raise
+
+    @classmethod
+    async def _remove(
+        cls,
+        db: AsyncSession,
+        resource: MasterResource,
+        value: int | BaseModel,
+    ) -> CrudResponseModel:
+        config = _RESOURCE_CONFIG[resource]
+        try:
+            resource_ids = cls._extract_ids(resource, value)
+            for resource_id in resource_ids:
+                if await MasterDao.count_references(db, resource, resource_id) > 0:
+                    raise ServiceWarning(message=f'{config["label"]}已被引用，只能停用，不能删除')
+                if await getattr(MasterDao, f'get_{resource}')(db, resource_id) is None:
+                    raise ServiceException(message=f'{config["label"]}不存在')
+            for resource_id in resource_ids:
+                await getattr(MasterDao, f'delete_{resource}')(db, resource_id)
+            await db.commit()
+            return CrudResponseModel(is_success=True, message='删除成功')
+        except IntegrityError:
+            await db.rollback()
+            raise ServiceException(message=f'删除{config["label"]}失败，该数据仍被引用') from None
+        except Exception:
+            await db.rollback()
+            raise
+
+    @classmethod
+    async def list_strain(
+        cls,
+        db: AsyncSession,
+        query_object: StrainPageQueryModel,
+        is_page: bool = True,
+    ) -> PageModel | list[dict[str, Any]]:
+        return await cls._list(db, 'strain', query_object, is_page)
+
+    @classmethod
+    async def detail_strain(cls, db: AsyncSession, strain_id: int) -> StrainModel:
+        return await cls._detail(db, 'strain', strain_id)
+
+    @classmethod
+    async def add_strain(cls, db: AsyncSession, strain: StrainModel) -> CrudResponseModel:
+        return await cls._add(db, 'strain', strain)
+
+    @classmethod
+    async def edit_strain(cls, db: AsyncSession, strain: StrainModel) -> CrudResponseModel:
+        return await cls._edit(db, 'strain', strain)
+
+    @classmethod
+    async def remove_strain(
+        cls,
+        db: AsyncSession,
+        value: int | DeleteStrainModel,
+    ) -> CrudResponseModel:
+        return await cls._remove(db, 'strain', value)
+
+    @classmethod
+    async def list_customer(
+        cls,
+        db: AsyncSession,
+        query_object: CustomerPageQueryModel,
+        is_page: bool = True,
+    ) -> PageModel | list[dict[str, Any]]:
+        return await cls._list(db, 'customer', query_object, is_page)
+
+    @classmethod
+    async def detail_customer(cls, db: AsyncSession, customer_id: int) -> CustomerModel:
+        return await cls._detail(db, 'customer', customer_id)
+
+    @classmethod
+    async def add_customer(cls, db: AsyncSession, customer: CustomerModel) -> CrudResponseModel:
+        return await cls._add(db, 'customer', customer)
+
+    @classmethod
+    async def edit_customer(cls, db: AsyncSession, customer: CustomerModel) -> CrudResponseModel:
+        return await cls._edit(db, 'customer', customer)
+
+    @classmethod
+    async def remove_customer(
+        cls,
+        db: AsyncSession,
+        value: int | DeleteCustomerModel,
+    ) -> CrudResponseModel:
+        return await cls._remove(db, 'customer', value)
+
+    @classmethod
+    async def list_site(
+        cls,
+        db: AsyncSession,
+        query_object: SitePageQueryModel,
+        is_page: bool = True,
+    ) -> PageModel | list[dict[str, Any]]:
+        return await cls._list(db, 'site', query_object, is_page)
+
+    @classmethod
+    async def detail_site(cls, db: AsyncSession, site_id: int) -> SiteModel:
+        return await cls._detail(db, 'site', site_id)
+
+    @classmethod
+    async def add_site(cls, db: AsyncSession, site: SiteModel) -> CrudResponseModel:
+        return await cls._add(db, 'site', site)
+
+    @classmethod
+    async def edit_site(cls, db: AsyncSession, site: SiteModel) -> CrudResponseModel:
+        return await cls._edit(db, 'site', site)
+
+    @classmethod
+    async def remove_site(
+        cls,
+        db: AsyncSession,
+        value: int | DeleteSiteModel,
+    ) -> CrudResponseModel:
+        return await cls._remove(db, 'site', value)
+
+    @classmethod
+    async def list_line(
+        cls,
+        db: AsyncSession,
+        query_object: LinePageQueryModel,
+        is_page: bool = True,
+    ) -> PageModel | list[dict[str, Any]]:
+        return await cls._list(db, 'line', query_object, is_page)
+
+    @classmethod
+    async def detail_line(cls, db: AsyncSession, line_id: int) -> LineModel:
+        return await cls._detail(db, 'line', line_id)
+
+    @classmethod
+    async def add_line(cls, db: AsyncSession, line: LineModel) -> CrudResponseModel:
+        return await cls._add(db, 'line', line)
+
+    @classmethod
+    async def edit_line(cls, db: AsyncSession, line: LineModel) -> CrudResponseModel:
+        return await cls._edit(db, 'line', line)
+
+    @classmethod
+    async def remove_line(
+        cls,
+        db: AsyncSession,
+        value: int | DeleteLineModel,
+    ) -> CrudResponseModel:
+        return await cls._remove(db, 'line', value)
+
+    @classmethod
+    async def list_warehouse(
+        cls,
+        db: AsyncSession,
+        query_object: WarehousePageQueryModel,
+        is_page: bool = True,
+    ) -> PageModel | list[dict[str, Any]]:
+        return await cls._list(db, 'warehouse', query_object, is_page)
+
+    @classmethod
+    async def detail_warehouse(cls, db: AsyncSession, warehouse_id: int) -> WarehouseModel:
+        return await cls._detail(db, 'warehouse', warehouse_id)
+
+    @classmethod
+    async def add_warehouse(cls, db: AsyncSession, warehouse: WarehouseModel) -> CrudResponseModel:
+        return await cls._add(db, 'warehouse', warehouse)
+
+    @classmethod
+    async def edit_warehouse(cls, db: AsyncSession, warehouse: WarehouseModel) -> CrudResponseModel:
+        return await cls._edit(db, 'warehouse', warehouse)
+
+    @classmethod
+    async def remove_warehouse(
+        cls,
+        db: AsyncSession,
+        value: int | DeleteWarehouseModel,
+    ) -> CrudResponseModel:
+        return await cls._remove(db, 'warehouse', value)