# -*- coding:utf-8 -*- """ @Created on : 2023/6/30 16:17 @Author: hxl @Des: """ from datetime import datetime import json import os.path from fastapi import APIRouter, Request,Depends,Response from fastapi.encoders import jsonable_encoder from fastapi.responses import FileResponse from urllib.parse import quote from pydantic import BaseModel from pydantic.datetime_parse import timedelta from tortoise.expressions import Q from tortoise.queryset import QuerySet from helper import respond_to ,login_required from helper.report_excel import ReportData from helper.write_doc import WriteDoc from helper.log import logger_wrapper from helper.utils import get_seq_no,get_UDisk_path,mkdir from models import Cabinet, Drawer, User, Archive,Fluid,Drug router = APIRouter(prefix='/fluids',dependencies=[Depends(login_required)]) #dependencies=[Depends(login_required)] class LabelSet(BaseModel): seq_no_list :list class UpdateFuild(BaseModel): name: str # 配液名称 config_concentration:str # 配置浓度 allocation:str # 配置量 solvent:str # 溶剂量 configuration_date: str # 配置日期 validity_date:str # 有效期 configuration_record:str # 配置记录 configuration_basis:str # 配置依据 class CreateFuild(BaseModel): name: str # 配液名称 barcode_list:str # 单位编号， 组合barcode config_concentration:str # 配置浓度 allocation:str # 配置量 solvent:str # 溶剂量 configuration_date: str # 配置日期 validity_date:str # 有效期 configuration_record:str = "" # 配置记录 configuration_basis:str = "" # 配置依据 @router.get('',summary='获取配液列表') async def index (request: Request, page_no: int = 1, page_size: int = 20, state: int = 0, search: str = '',): """ 获取配液列表 :param page_no: int :param page_size: int :param state: int # 0 全部， 1 在库 2 出库 :param search: str :return: """ offset = (page_no - 1) * page_size condition_filter = Q(name__contains=search) if state: condition_filter = condition_filter & Q(state=state) query = QuerySet(Fluid).filter(condition_filter) count = await query.count() query_objs = await query.limit(page_size).offset(offset).order_by('-created_at') return respond_to(data=dict(count=count, data=query_objs)) @router.post('/add_fluid',summary='新增配液') async def add_fluid(request: Request,model:CreateFuild): """ 新增配液 :param request: Request请求头信息 :param model: 配液信息的请求体 :return: """ # 获取配液当天最大的number today = datetime.now().replace(hour=0, minute=0, second=0, microsecond=0) # 获取当前日期，截断到天 tomorrow = today + timedelta(days=1) # 当天起始时间的下一天 fluid_max_num = await Fluid.filter(created_at__gte=today, created_at__lt=tomorrow).limit(1).order_by('-number') if fluid_max_num and fluid_max_num[0].number: number = fluid_max_num[0].number+1 else: number = 1 seq_no = get_seq_no(number) user_id = request.state.current_user.id username = request.state.current_user.username fluid_obj = await Fluid.create(**model.dict(),user_id=user_id,user_name=username,seq_no=seq_no) return respond_to(data=fluid_obj) @router.get('/get_fluid/{id}',summary='获取单条记录') async def get_fluid(id:str): """ 获取单条记录 :param id: str :return: """ fluid_obj = await Fluid.get(id=id) return respond_to(data=fluid_obj) @router.put('/update_fluid/{id}',summary='编辑配液信息') async def update_fluid(id:str,post:UpdateFuild): """ 编辑配液信息 :param id: str :param post: 修改配液信息的请求体 :return: """ fluid_obj = await Fluid.get(id=id) data = post.dict() await fluid_obj.update_from_dict(data) await fluid_obj.save() return respond_to(desc=f"修改[{fluid_obj.name}]信息成功") @router.get('/get_drug/{barcode}',summary='查询单个药剂') async def get_drug(barcode:str): """ 查询单个药剂 :param barcode: str :return: """ # drug_obj = await Drug.get(barcode = barcode) drug_obj = await Drug.get_or_none(Q(barcode=barcode) | Q(rfid=barcode)) # drug_obj = await Fluid.get(seq_no = barcode) # 测试数据 return respond_to(data=drug_obj) @router.post('/print_fluid_label',summary='打印配液标签') async def print_fluid_label(model:LabelSet): """ 打印配液标签 :param model: list :return: """ # 预留调用标签打印机服务 print('LabelSet:::',model) return respond_to(data={'data':model}) @router.delete('/delete_fluid/{id_list}',summary='删除配液') async def delete_fluid(id_list:str): """ 删除配液 :param id_list: str :return: """ id_list = id_list.split(',') if id_list: for id in id_list: fluid_obj = await Fluid.get_or_none(id=id) await fluid_obj.delete() return respond_to(desc='删除成功') return respond_to(code=400, desc='请传入配液id') @router.get('/download_fluid_report',summary='下载配液报表') async def download_fluid_report(visit_type: int = 1 , state: int = 0 , search: str=''): """ 下载配液报表excel :param visit_type: int # 1 u盘下载， 2 网页下载 ：param state : int # 0 全部 ，1 在库 ，2 出库 :param search : str :return: """ condition_filter = Q(name__contains=search) if state: condition_filter = condition_filter & Q(state=state) query = QuerySet(Fluid).filter(condition_filter) query_obj = query.order_by('-created_at') data = await query_obj.all() file_name = '配液报表' obj_report = ReportData() obj = obj_report.build_file(**{ "file_name": file_name, "data_list": data, "key_list": ["name", "seq_no", "config_concentration", "allocation", "solvent", "state", "configuration_date", 'validity_date', 'user_name', 'configuration_record', 'configuration_basis'], "finds_name": ["配液名称", "储备液编号", "配置浓度", "配置量", "溶剂", "配液状态", "配置日期", "有效期","配置人", "配置记录", "配置依据", ], }) res_save_file = f"{file_name}-{datetime.now().strftime('%Y%m%d%H%M%S')}" # # visit_type: 1 , 默认导出到U盘， 2：导出到前端 if visit_type==1: file_path= get_UDisk_path() if not file_path: return respond_to(code=400,desc = '未检测到U盘' ) obj.save(os.path.join(file_path,res_save_file)) return respond_to( desc ='导出成功') else: # 远程访问 try: file_path = "Resource/" mkdir(file_path) obj.save(os.path.join(file_path, res_save_file)) return FileResponse(os.path.join(os.getcwd(),file_path,res_save_file+'.xlsx'),filename=res_save_file+'.xlsx',media_type="application/vnd.openxmlformats-officedocument.spreadsheetml.sheet") except Exception as e : return respond_to(code=400, desc= '导出失败'+str(e)) @router.get('/download_fluid_report1',summary='下载配液报表word') async def download_fluid_report1(visit_type: int = 1 , state: int = 0 , search: str=''): """ 下载配液报表word :param visit_type: int # 1 u盘下载， 2 网页下载 ：param state : int # 0 全部 ，1 在库 ，2 出库 :param search : str :return: """ condition_filter = Q(name__contains=search) if state: condition_filter = condition_filter & Q(state=state) query = QuerySet(Fluid).filter(condition_filter) query_obj = query.order_by('-created_at') data = await query_obj.all() file_name = '配液报表' obj = WriteDoc(data,'').add_table() rsp_file_name = f"{file_name}{datetime.now().strftime('%Y%m%d%H%M%S')}" if visit_type==1: file_path= get_UDisk_path() if not file_path: return respond_to(code=400,desc = '未检测到U盘' ) obj.save(os.path.join(file_path, rsp_file_name)+'.docx') return respond_to( desc ='导出成功') else: try: file_path = "Resource" mkdir(file_path) obj.save(os.path.join(file_path, rsp_file_name)+'.docx') file_name = rsp_file_name+".docx" response = Response(WriteDoc.download_doc(os.path.join(file_path, file_name))) response.headers["Content-Disposition"] = \ "attachment;" \ "filename*=UTF-8''{utf_filename}".format( utf_filename=quote(file_name) ) return response except Exception as e : return respond_to(code=400, desc= '导出失败'+str(e))