from fastapi import APIRouter
from pydantic import BaseModel

INDEX = "供水及污水处理能力（万吨/日）"

router = APIRouter()


class Params(BaseModel):
    供水及污水处理能力: float


# 计算接口
@router.post("/{}".format(INDEX))
def func(p: Params):
    try:
        result = p.供水及污水处理能力 / 10000
        return round(result, 2)

    except Exception:
        return "计算错误"