|
|
#!/usr/bin/env python
|
|
|
# encoding: utf-8
|
|
|
"""
|
|
|
@author: tx
|
|
|
@file: drug.py
|
|
|
@time: 2023/5/11 11:04
|
|
|
@desc: 药剂领用归还
|
|
|
"""
|
|
|
from decimal import Decimal
|
|
|
|
|
|
from fastapi import APIRouter, Request, Depends
|
|
|
from fastapi.encoders import jsonable_encoder
|
|
|
from pydantic import BaseModel
|
|
|
from tortoise.queryset import Q
|
|
|
|
|
|
from conf import setting
|
|
|
from helper.drug import drugs_except_info, drugs_list_info, gram_to_milligram, milligram_to_gram, open_update_expired_at
|
|
|
from helper import respond_to, login_required
|
|
|
from models import Dictionary
|
|
|
from models.drug import Drug, DrugStateEnum
|
|
|
from models.terminal import Terminal
|
|
|
from models.drawers import Drawer
|
|
|
from models.cabinet import Cabinet
|
|
|
from transfer.standard import Standard
|
|
|
from transfer.code_standard import CodeStandard
|
|
|
|
|
|
router = APIRouter(prefix='/drugs', dependencies=[Depends(login_required)])
|
|
|
|
|
|
|
|
|
class DrugUpdateRequest(BaseModel):
|
|
|
open_date: str = "" # 开封日期
|
|
|
weight: Decimal = 0.0 # 余量
|
|
|
use_weight: Decimal = 0.0 # 用量
|
|
|
rfid: str = ""
|
|
|
id: str = ""
|
|
|
|
|
|
|
|
|
class DrugUpdateExpired(BaseModel):
|
|
|
expired_at: str
|
|
|
rfid: str = ""
|
|
|
id: str = ""
|
|
|
|
|
|
|
|
|
class DrugRfids(BaseModel):
|
|
|
rfids: list
|
|
|
|
|
|
|
|
|
class DrugTakeOut(BaseModel):
|
|
|
terminal_id: str | None
|
|
|
page_no: int = 1
|
|
|
page_size: int = 20
|
|
|
|
|
|
|
|
|
class UpdateDictionary(BaseModel):
|
|
|
lack_stock_count: int = 0
|
|
|
expiration_alert: int = 0
|
|
|
|
|
|
|
|
|
class EmptyBottleDisposal(BaseModel):
|
|
|
rfid: str
|
|
|
weight: int | None
|
|
|
open_date: str | None
|
|
|
|
|
|
|
|
|
class ReturnForm(BaseModel):
|
|
|
# barcode: str
|
|
|
weight: Decimal = 0.0 # 余量
|
|
|
use_weight: Decimal = 0.0 # 用量
|
|
|
drawer_id: str = '' # 层id
|
|
|
|
|
|
|
|
|
@router.post('/empty', summary='试剂置为空瓶')
|
|
|
async def empty_bottle_disposal(request: Request, post: EmptyBottleDisposal):
|
|
|
"""
|
|
|
试剂置为空瓶
|
|
|
:param request: Request请求头信息
|
|
|
:param post: 空瓶信息的请求体
|
|
|
:return:
|
|
|
"""
|
|
|
users = request.state.users
|
|
|
standard = Standard(users)
|
|
|
rfid = post.rfid
|
|
|
# user_weight = post.weight
|
|
|
open_date = post.open_date
|
|
|
if open_date:
|
|
|
drug = await Drug.get(rfid=rfid)
|
|
|
drug.open_date = open_date
|
|
|
await drug.save()
|
|
|
await standard.empty(rfid)
|
|
|
return respond_to()
|
|
|
|
|
|
|
|
|
@router.post('/weight', summary="药剂用量或余量更新")
|
|
|
async def update(keyword: DrugUpdateRequest):
|
|
|
"""
|
|
|
药剂用量或余量更新
|
|
|
接收id或rfid都可以更新重量
|
|
|
余量与用量二选一，只有一个有值
|
|
|
:param keyword: 包含更新信息的请求体
|
|
|
:return:
|
|
|
"""
|
|
|
print("open_date", keyword.open_date)
|
|
|
rfid = keyword.rfid if keyword.rfid else ""
|
|
|
query_param = {"id": keyword.id} if keyword.id else {"rfid": rfid}
|
|
|
drug_obj = await Drug.get(**query_param).prefetch_related("template")
|
|
|
if keyword.weight:
|
|
|
mill_weight = keyword.weight
|
|
|
await drug_obj.update_last_weight(weight=mill_weight)
|
|
|
elif keyword.use_weight:
|
|
|
mill_weight = keyword.use_weight
|
|
|
await drug_obj.update_last_use_weight(weight=mill_weight)
|
|
|
if keyword.open_date:
|
|
|
if not drug_obj.open_date:
|
|
|
drug_obj.open_date = keyword.open_date
|
|
|
template_obj = await drug_obj.template
|
|
|
new_expired_at = open_update_expired_at(drug_obj, template_obj, keyword.open_date)
|
|
|
if new_expired_at:
|
|
|
drug_obj.expired_at = new_expired_at
|
|
|
await drug_obj.save()
|
|
|
return respond_to(code=200, desc="重量更新成功")
|
|
|
|
|
|
|
|
|
@router.post('/except', summary='RFID药剂异常信息')
|
|
|
async def index(request: DrugRfids):
|
|
|
"""
|
|
|
获取RFID药剂异常信息
|
|
|
:return:
|
|
|
"""
|
|
|
result = list()
|
|
|
drugs = await Drug.filter(rfid__in=request.rfids).prefetch_related('dictionary', 'template').all()
|
|
|
terminal_obj = await Terminal.get(id=setting.TERMINAL_ID)
|
|
|
for drug in drugs:
|
|
|
data = await drugs_except_info(drug, terminal_obj)
|
|
|
data["rfid"] = drug.rfid
|
|
|
result.append(data)
|
|
|
# 传入rfid与data所有rfid比较，差集就是非法标签
|
|
|
post_rfid = set(request.rfids)
|
|
|
effective_rfid = set([i["rfid"] for i in result])
|
|
|
diff_rfid = post_rfid - effective_rfid
|
|
|
if diff_rfid:
|
|
|
result.extend([{
|
|
|
"rfid": rfid,
|
|
|
"is_rfid_except": True,
|
|
|
"sound": True,
|
|
|
"message": "非法标签",
|
|
|
"state": 0,
|
|
|
} for rfid in diff_rfid])
|
|
|
return respond_to(code=200, desc="获取RFID药剂信息成功", data=result)
|
|
|
|
|
|
|
|
|
@router.post('', summary='RFID药剂详情')
|
|
|
async def index(request: DrugRfids):
|
|
|
"""
|
|
|
获取RFID药剂信息 药剂信息详情
|
|
|
:return:
|
|
|
"""
|
|
|
drugs = list()
|
|
|
for rfid in request.rfids:
|
|
|
drug_obj = await Drug.get_or_none(rfid=rfid).prefetch_related('dictionary', 'template')
|
|
|
if drug_obj is None:
|
|
|
continue
|
|
|
result = jsonable_encoder(drug_obj)
|
|
|
template_obj = await drug_obj.template
|
|
|
terminal_obj = await Terminal.get(id=setting.TERMINAL_ID)
|
|
|
data = await drugs_except_info(drug_obj, terminal_obj)
|
|
|
|
|
|
drug_info = await drug_obj.attribute_drug_info()
|
|
|
parse_fill_json_content = await drug_obj.parse_fill_json_content()
|
|
|
parse_fill_json_content.update({f'余重': f'{milligram_to_gram(drug_obj.remain_gross_weight)}'})
|
|
|
|
|
|
return_require_weigh = template_obj.archive.params.get("return_require_weigh", '')
|
|
|
result.update({
|
|
|
"drugs_except_info": data,
|
|
|
"fill_json_content": parse_fill_json_content,
|
|
|
"rfid_drug": {drug_obj.rfid: f"[{drug_obj.rfid}]" + ",".join(list(map(lambda x:str(x), drug_info.values()))) + "," +
|
|
|
f"{drug_obj.remain_gross_weight}"},
|
|
|
"return_require_weigh": return_require_weigh
|
|
|
})
|
|
|
drugs.append(result)
|
|
|
return respond_to(code=200, desc="获取RFID药剂信息成功", data=drugs)
|
|
|
|
|
|
@router.get('', summary='药剂列表查询')
|
|
|
async def index(request: Request, drug_name: str = '', page_no: int = 1, page_size: int = 20):
|
|
|
"""
|
|
|
货架药剂列表查询
|
|
|
:return:
|
|
|
"""
|
|
|
query = Drug.filter()
|
|
|
cabinets_ids = await Cabinet.filter(type=3).values_list("id", flat=True)
|
|
|
if cabinets_ids:
|
|
|
query = query.filter(cabinet_id__in=cabinets_ids)
|
|
|
|
|
|
if drug_name:
|
|
|
query = query.filter(Q(dictionary__k1__contains=drug_name) | Q(dictionary__k2__contains=drug_name))
|
|
|
|
|
|
total_count = await query.count()
|
|
|
offset = (page_no - 1) * page_size
|
|
|
drugs = await query.offset(offset).limit(page_size).all()
|
|
|
|
|
|
result = await drugs_list_info(drugs)
|
|
|
return respond_to(200, data=dict(count=total_count, drugs=result))
|
|
|
|
|
|
@router.get('/take_out', summary='药剂领用查询')
|
|
|
async def index(request: Request, drug_name: str = '', take_out_type: int = 0, page_no: int = 1, page_size: int = 20):
|
|
|
"""
|
|
|
药剂领用列表查询
|
|
|
本终端的排前面
|
|
|
:return:
|
|
|
"""
|
|
|
terminal = await Terminal.get(id=setting.TERMINAL_ID)
|
|
|
|
|
|
query = Drug.filter(state=DrugStateEnum.IN).order_by('-drawer_board_id')
|
|
|
if not take_out_type:
|
|
|
query = query.filter(drawer_board__cabinet__terminal=terminal)
|
|
|
|
|
|
if drug_name:
|
|
|
query = query.filter(Q(dictionary__k1__contains=drug_name) | Q(dictionary__k2__contains=drug_name))
|
|
|
|
|
|
total_count = await query.count()
|
|
|
offset = (page_no - 1) * page_size
|
|
|
drugs = await query.offset(offset).limit(page_size).all()
|
|
|
|
|
|
result = await drugs_list_info(drugs)
|
|
|
return respond_to(200, data=dict(count=total_count, drugs=result))
|
|
|
|
|
|
|
|
|
@router.get('/put_in', summary='获取当前用户待归还清单')
|
|
|
async def index(request: Request):
|
|
|
"""
|
|
|
获取当前用户壁挂终端待归还清单
|
|
|
:return:
|
|
|
"""
|
|
|
current_user = request.state.current_user
|
|
|
|
|
|
terminal = await Terminal.get(id=setting.TERMINAL_ID)
|
|
|
# 当前柜体待归还清单
|
|
|
query = Drug.filter(last_receive_id=current_user.id, state=DrugStateEnum.OUT).filter(cabinet__terminal=terminal)
|
|
|
drugs = await query.all()
|
|
|
result = await drugs_list_info(drugs)
|
|
|
return respond_to(data=result)
|
|
|
|
|
|
|
|
|
@router.put('/expired_at', summary='药剂信息修改')
|
|
|
async def update(keyword: DrugUpdateExpired):
|
|
|
"""
|
|
|
药剂信息过期时间修改
|
|
|
:param id: 用户药剂模板条目id
|
|
|
:param keyword:
|
|
|
:return:
|
|
|
"""
|
|
|
rfid = keyword.rfid if keyword.rfid else ""
|
|
|
query_param = {"id": keyword.id} if keyword.id else {"rfid": rfid}
|
|
|
drug_obj = await Drug.get(**query_param)
|
|
|
drug_obj.expired_at = keyword.expired_at
|
|
|
await drug_obj.save()
|
|
|
return respond_to(code=200, desc="更新成功")
|
|
|
|
|
|
|
|
|
@router.get('/dictionaries', summary='获取药剂字典列表')
|
|
|
async def index(keyword: str = '', page_no: int = 1, page_size: int = 10):
|
|
|
"""
|
|
|
获取药剂字典列表
|
|
|
:param keyword: 按药剂信息搜索
|
|
|
:param page_no: 分页页码，默认为1
|
|
|
:param page_size: 分页大小，默认为10
|
|
|
:return:
|
|
|
"""
|
|
|
dictionaries = await Dictionary.all().offset((page_no - 1) * page_size).limit(page_size).values()
|
|
|
result_list = []
|
|
|
for dictionary in dictionaries:
|
|
|
drug_info = ",".join([value for key, value in dictionary.items() if key.startswith('k') and value is not None])
|
|
|
if keyword in drug_info:
|
|
|
dictionary['drug_info'] = drug_info
|
|
|
for k, v in dictionary['params'].items():
|
|
|
dictionary[k] = v
|
|
|
result_list.append(dictionary)
|
|
|
return respond_to(data={'data': result_list, 'count': len(result_list)})
|
|
|
|
|
|
|
|
|
@router.put('/dictionary/{id}', summary='编辑药剂字典')
|
|
|
async def update(id: str, post: UpdateDictionary):
|
|
|
"""
|
|
|
编辑药剂字典
|
|
|
:param id: id
|
|
|
:param post:
|
|
|
:return:
|
|
|
"""
|
|
|
dictionary = await Dictionary.get(id=id)
|
|
|
dictionary.params = post.dict()
|
|
|
await dictionary.save()
|
|
|
return respond_to()
|
|
|
|
|
|
|
|
|
@router.put('/receive/{bar_code}', summary='药剂领用')
|
|
|
async def update(request: Request, bar_code: str):
|
|
|
"""
|
|
|
药剂领用
|
|
|
每瓶药剂扫码即领用
|
|
|
:param request: request
|
|
|
:param bar_code: 药剂条码
|
|
|
:return:
|
|
|
"""
|
|
|
drug_obj = await Drug.get_or_none(Q(barcode=bar_code) | Q(rfid=bar_code)).prefetch_related('template', 'dictionary')
|
|
|
if not drug_obj:
|
|
|
return respond_to(code=404, desc="药剂未找到")
|
|
|
if drug_obj.state in [DrugStateEnum.OUT, DrugStateEnum.EMPTY]:
|
|
|
desc = "药剂不在库" if DrugStateEnum.OUT else "药剂已空瓶"
|
|
|
return respond_to(code=400, desc=desc)
|
|
|
users = request.state.users
|
|
|
standard = CodeStandard(users, drug_obj)
|
|
|
transfer_results = await standard.take_out(barcode=bar_code)
|
|
|
print('领用结果', transfer_results)
|
|
|
return respond_to(desc="领用完成", data=transfer_results)
|
|
|
|
|
|
|
|
|
@router.put('/return/{bar_code}', summary='药剂归还')
|
|
|
async def update(request: Request, bar_code: str, keyword: ReturnForm):
|
|
|
"""
|
|
|
药剂归还
|
|
|
药剂扫码后填入用量/余量，归还成功
|
|
|
:param bar_code: 药剂条码
|
|
|
:param post:
|
|
|
:return:
|
|
|
"""
|
|
|
# 如果调用保存已经保存过用量数据，则不再进行保存
|
|
|
drug_obj = await Drug.get_or_none(Q(barcode=bar_code) | Q(rfid=bar_code)).prefetch_related('template', 'dictionary')
|
|
|
if not drug_obj:
|
|
|
return respond_to(code=404, desc="药剂未找到")
|
|
|
if drug_obj.state == DrugStateEnum.IN:
|
|
|
return respond_to(code=401, desc="药剂已在库，请勿再次归还！")
|
|
|
|
|
|
if keyword.weight:
|
|
|
# 余量转用量，更新用量
|
|
|
mill_weight = drug_obj.remain_gross_weight - keyword.weight
|
|
|
await drug_obj.update_last_use_weight(weight=mill_weight)
|
|
|
elif keyword.use_weight:
|
|
|
mill_weight = keyword.use_weight
|
|
|
await drug_obj.update_last_use_weight(weight=mill_weight)
|
|
|
|
|
|
# 位置更新
|
|
|
if keyword.drawer_id:
|
|
|
drawer_obj = await Drawer.get(id=keyword.drawer_id).prefetch_related("cabinet")
|
|
|
cabinet = drawer_obj.cabinet
|
|
|
position_str = f"{cabinet.label}-{drawer_obj.label}"
|
|
|
drug_obj.drawer = drawer_obj
|
|
|
drug_obj.cabinet = cabinet
|
|
|
drug_obj.position = position_str
|
|
|
await drug_obj.save()
|
|
|
|
|
|
users = request.state.users
|
|
|
standard = CodeStandard(users, drug_obj)
|
|
|
transfer_results = await standard.put_in(barcode=bar_code)
|
|
|
print('归还结果', transfer_results)
|
|
|
return respond_to(desc="归还完成", data=transfer_results)
|
