kms-backend/endpoints/web/cabinet.py

# -*- coding:utf-8 -*-
"""
@Created on : 2023/6/30 16:17
@Author: hxl
@Des:
"""
import httpx

from fastapi import APIRouter, Request, Depends
from pydantic import BaseModel

from helper import respond_to, login_required
from helper.log import logger_wrapper, logger
from models import Cabinet, Drawer, User, Archive

router = APIRouter(prefix='/cabinets',  dependencies=[Depends(login_required)])


class UpdateCabinet(BaseModel):
    label: str = None
    type: int = 1
    location: str = None
    director: str = None
    phone: str = None
    model: str = None
    archive_id: str
    door: bool

class DeleteCabinetRequest(BaseModel):
    cabinet_ids: list


class CreateCabinet(BaseModel):
    label: str
    type: int = 2 #1=智能柜 2=普通柜
    location: str = None
    director: str = None
    phone: str = None
    model: str = None
    archive_id: str


@router.post('', summary='新增普通柜体')
async def create_cabinet(cabinet: CreateCabinet):
    """
    新增普通柜体
    :return:
    """

    if not cabinet.label:
        return respond_to(code=400, desc='柜体名称必填')

    # 验证 archive 外键是否存在
    archive_exists = await Archive.filter(id=cabinet.archive_id).exists()
    if not archive_exists:
        return respond_to(code=404, desc='试剂大类不存在')

    cabinet_obj = await Cabinet.create(
        label=cabinet.label,
        location=cabinet.location,
        director=cabinet.director,
        phone=cabinet.phone,
        model=cabinet.model,
        type=cabinet.type,
        archive_id=cabinet.archive_id if cabinet.archive_id else None,  # 如果archive为空，则设置为None
    )

    for i in range(1, 10):
        await Drawer.create(label=f'抽屉-{i}', line_no=0, cabinet_id = cabinet_obj.id)

    return respond_to(desc=f'新增{cabinet.label}成功')


@router.get('/index', summary='柜体下拉列表')
async def index(request: Request):
    """
    获取柜体下拉列表
    :return:
    """
    query = Cabinet.filter()
    if request.state and request.state.archive_id:
        query = query.filter(archive_id=request.state.archive_id)
    cabinets = await query.all().values('id', 'label')
    data = [{"label": cabinet['label'], "id": cabinet['id']} for cabinet in cabinets]
    return respond_to(data=data)
@router.get('/all', summary='全部柜体下拉列表')
async def index(request: Request):
    """
    获取柜体下拉列表
    :return:
    """
    cabinets = await Cabinet.filter().all().values('id', 'label')
    data = [{"label": cabinet['label'], "id": cabinet['id']} for cabinet in cabinets]
    return respond_to(data=data)

@router.get('', summary='柜体管理列表')
async def index(page_no: int = None, page_size: int = None, label: str = None, archive_id: str = None, type: str = None):
    """
    柜体管理列表
    :param page_no: int
    :param page_size: int
    :param label: str
    :param archive_id: int
    :param type: int 1=智能柜 2=普通柜
    :return:
    """
    query = Cabinet.all().prefetch_related('archive')
    if label:
        query = query.filter(label__contains=label)
    if archive_id:
        query = query.filter(archive=archive_id)
    if type:
        query = query.filter(type=type)
    total_count = await query.count()
    if page_no is not None and page_size is not None:
        offset = (page_no - 1) * page_size
        cabinets = await query.offset(offset).limit(page_size).order_by("-created_at")
    else:
        cabinets = await query.limit(total_count).order_by("-created_at")
    cabinet_list = [
        {
            'id': cabinet.id,
            'label': cabinet.label,
            'model': cabinet.model,
            'type': '智能柜' if cabinet.type == 1 else '普通柜',
            'archive_id': cabinet.archive.id if cabinet.archive else None,
            'archive_name': cabinet.archive.name if cabinet.archive else None,
            'location': cabinet.location,
            'director': cabinet.director,
            'phone': cabinet.phone,
        }
        for cabinet in cabinets
    ]
    return respond_to(data={"total_count": total_count, "cabinets": cabinet_list})


@router.put('/{cabinet_id}', summary='柜体编辑信息')
@logger_wrapper
async def update(cabinet_id: str, cabinet_data: UpdateCabinet):
    """
    柜体信息编辑
    :param cabinet_id: str
    :param cabinet_data: ProfileEdit
    :return:
    """
    cabinet_obj = await Cabinet.get_or_none(id=cabinet_id)
    if not cabinet_obj:
        return respond_to(code=404, desc="柜体不存在")
    archive_exists = await Archive.filter(id=cabinet_data.archive_id).exists()
    if not archive_exists:
        return respond_to(code=404, desc='试剂大类不存在')
    # 更新柜体信息
    # update_fields = {k: v for k, v in cabinet_data.dict().items() if v}
    # await cabinet_obj.update_from_dict(update_fields).save()

    return respond_to(200) # , desc=f"{cabinet_data.label}修改成功"


@router.delete('/delete', summary='删除普通柜体')
async def delete_cabinet(body: DeleteCabinetRequest):
    """
    删除普通柜体
    :param cabinet_id: str
    :return:
    """

    for cabinet_id in body.cabinet_ids:
        cabinet = await Cabinet.get_or_none(id=cabinet_id)
        if not cabinet:
            return respond_to(code=400, desc='柜体不存在')
        if cabinet.type == 1:
            return respond_to(code=400, desc='智能柜无法删除')

    await Cabinet.filter(id__in=body.cabinet_ids).delete()
    return respond_to(desc='删除成功')


@router.get('/{cabinet_id}/{user_id}', summary='抽屉列表')
async def index(request: Request, cabinet_id: str, user_id: str):
    """
    柜体管理列表
    :param request: Request
    :param cabinet_id: str
    :return:
    """
    cabinet = await Cabinet.get(id=cabinet_id)
    # if not cabinet.matrix:
        # return respond_to(400, '请先设置布局信息')
    user = await User.get(id=user_id)
    query = Drawer.filter(cabinet_id=cabinet.id)
    drawers = await query.order_by("-rank", "created_at").values("id", "cabinet_id", "line_no", "label", "tags", "rank")
    result = []
    for drawer in drawers:
        drawer['is_permission'] = True if str(drawer['id']) in user.drawers else False
        result.append(drawer)
    return respond_to(data=dict(drawers=result))


@router.patch('/{cabinet_id}', summary='柜体参数编辑')
async def update(cabinet_id: str, params: dict):
    """
    柜体参数编辑
    :param cabinet_id: str
    :param params: dict
    :return:
    """
    cabinet = await Cabinet.get(id=cabinet_id)
    url = f'http://{cabinet.ip}/api/cabinet/v1/setting'
    async with httpx.AsyncClient() as client:
        try:
            response = await client.post(url, json=params)
        except Exception as e:
            logger.error(f"下位机通讯失败：{str(e)}")
            return respond_to(code=400, desc='下位机通讯失败')

    if response.json()['code'] != 200:
        return respond_to(code=405, desc='参数有误')
    # cabinet.params = params
    # await cabinet.save()
    return respond_to()