AcidMap/app/api/cd_flux_removal.py

"""
Cd通量移除计算API接口
@description: 提供籽粒移除和秸秆移除的Cd通量计算功能
"""

from fastapi import APIRouter, HTTPException, Query, Path
from pydantic import BaseModel, Field
from typing import Dict, Any, Optional
import logging
from ..services.cd_flux_removal_service import CdFluxRemovalService

router = APIRouter()

# =============================================================================
# 数据模型定义
# =============================================================================

class CdFluxRemovalResponse(BaseModel):
    """
    Cd通量移除计算响应模型
    
    @description: 标准化的API响应格式
    """
    success: bool = Field(..., description="是否成功")
    message: str = Field(..., description="响应消息")
    data: Optional[Dict[str, Any]] = Field(None, description="计算结果数据")


# 设置日志
logger = logging.getLogger(__name__)

# =============================================================================
# Cd通量移除计算接口
# =============================================================================

@router.get("/grain-removal",
           summary="计算籽粒移除Cd通量",
           description="根据指定地区计算籽粒移除Cd通量，公式：EXP(LnCropCd)*F11*0.5*15/1000",
           response_model=CdFluxRemovalResponse)
async def calculate_grain_removal(
    area: str = Query(..., description="地区名称，如：韶关")
) -> Dict[str, Any]:
    """
    计算籽粒移除Cd通量
    
    @param area: 地区名称
    @returns: 籽粒移除Cd通量计算结果
    
    计算公式：籽粒移除(g/ha/a) = EXP(LnCropCd) * F11 * 0.5 * 15 / 1000
    数据来源：
    - LnCropCd: CropCd_output_data表
    - F11: Parameters表（作物亩产量）
    """
    try:
        service = CdFluxRemovalService()
        result = service.calculate_grain_removal_by_area(area)
        
        if not result["success"]:
            raise HTTPException(
                status_code=404, 
                detail=result["message"]
            )
        
        return result
        
    except HTTPException:
        raise
    except Exception as e:
        logger.error(f"计算地区 '{area}' 的籽粒移除Cd通量失败: {str(e)}")
        raise HTTPException(
            status_code=500, 
            detail=f"计算失败: {str(e)}"
        )


@router.get("/straw-removal",
           summary="计算秸秆移除Cd通量",
           description="根据指定地区计算秸秆移除Cd通量，公式：[EXP(LnCropCd)/(EXP(LnCropCd)*0.76-0.0034)]*F11*0.5*15/1000",
           response_model=CdFluxRemovalResponse)
async def calculate_straw_removal(
    area: str = Query(..., description="地区名称，如：韶关")
) -> Dict[str, Any]:
    """
    计算秸秆移除Cd通量
    
    @param area: 地区名称
    @returns: 秸秆移除Cd通量计算结果
    
    计算公式：秸秆移除(g/ha/a) = [EXP(LnCropCd)/(EXP(LnCropCd)*0.76-0.0034)] * F11 * 0.5 * 15 / 1000
    数据来源：
    - LnCropCd: CropCd_output_data表
    - F11: Parameters表（作物亩产量）
    """
    try:
        service = CdFluxRemovalService()
        result = service.calculate_straw_removal_by_area(area)
        
        if not result["success"]:
            raise HTTPException(
                status_code=404, 
                detail=result["message"]
            )
        
        return result
        
    except HTTPException:
        raise
    except Exception as e:
        logger.error(f"计算地区 '{area}' 的秸秆移除Cd通量失败: {str(e)}")
        raise HTTPException(
            status_code=500, 
            detail=f"计算失败: {str(e)}"
        )