|
|
#!/usr/bin/env python
|
|
|
# encoding: utf-8
|
|
|
"""
|
|
|
@author: tx
|
|
|
@file: drug.py
|
|
|
@time: 2023/5/9 9:14
|
|
|
@desc:
|
|
|
"""
|
|
|
from decimal import Decimal
|
|
|
from enum import IntEnum
|
|
|
import re
|
|
|
|
|
|
from tortoise.expressions import Q
|
|
|
from tortoise import fields, models
|
|
|
from tortoise.queryset import QuerySet
|
|
|
|
|
|
from models.user import User
|
|
|
from models.drug_item import DrugItem, DrugItemStateEnum
|
|
|
from models.dictionary import Dictionary
|
|
|
from conf.setting import setting
|
|
|
|
|
|
class DrugStateEnum(IntEnum):
|
|
|
# 0 = 待入库, 1 = 在库, 2 = 出库, 3 = 空瓶, 4 = 报废
|
|
|
INITIAL = 0
|
|
|
IN = 1
|
|
|
OUT = 2
|
|
|
EMPTY = 3
|
|
|
SCRAP = 4
|
|
|
|
|
|
|
|
|
class Drug(models.Model):
|
|
|
id = fields.UUIDField(pk=True)
|
|
|
dictionary = fields.ForeignKeyField('rms.Dictionary', null=True, to_field='id')
|
|
|
rfid = fields.CharField(max_length=255, unique=True, description='电子标签')
|
|
|
# barcode = fields.CharField(null=True, max_length=255, description='标签条码')
|
|
|
state = fields.IntEnumField(DrugStateEnum, default=DrugStateEnum.INITIAL,
|
|
|
description='0=待入库,1=在库,2=出库,3=空瓶,4=报废')
|
|
|
# hole = fields.SmallIntField(null=True, description='芯片序号')
|
|
|
position = fields.CharField(null=True, max_length=255, description='位置描述')
|
|
|
# remain_gross_weight = fields.IntField(null=True, description='试剂余重')
|
|
|
# last_use_weight = fields.IntField(null=True, description='最后使用重量')
|
|
|
# total_use_weight = fields.IntField(null=True, default=0, description='总用量 耗材为总使用数量')
|
|
|
fill_json_content = fields.JSONField(null=True, description='扩展字段')
|
|
|
# last_return_odd = fields.BooleanField(default=False, description='归还是否异常')
|
|
|
bind = fields.ForeignKeyField('rms.User', related_name="bind", on_delete=fields.SET_NULL, null=True, description='绑定人')
|
|
|
bind_at = fields.DatetimeField(null=True, description='绑定时间')
|
|
|
storage = fields.ForeignKeyField('rms.User', related_name="storage", on_delete=fields.SET_NULL, null=True, description='入库人')
|
|
|
storage_at = fields.DatetimeField(null=True, description='入库时间')
|
|
|
last_receive = fields.ForeignKeyField('rms.User',related_name="last_receive", on_delete=fields.SET_NULL, null=True, description='最后领用人')
|
|
|
last_receive_at = fields.DatetimeField(null=True, description='最后领用时间')
|
|
|
last_return = fields.ForeignKeyField('rms.User',related_name="last_return", on_delete=fields.SET_NULL, null=True, description='最后归还人')
|
|
|
last_return_at = fields.DatetimeField(null=True, description='最后归还时间')
|
|
|
# expired_at = fields.DatetimeField(null=True, description='过期时间')
|
|
|
# open_date = fields.DatetimeField(null=True, description='开封日期')
|
|
|
# board = fields.ForeignKeyField('rms.Board', null=True, on_delete=fields.SET_NULL, description='定位板规格ID')
|
|
|
# drawer = fields.ForeignKeyField('rms.Drawer', null=True, on_delete=fields.SET_NULL, description='抽屉ID')
|
|
|
# drawer_board = fields.ForeignKeyField('rms.DrawerBoard', null=True, on_delete=fields.SET_NULL, description='所属柜体')
|
|
|
cabinet = fields.ForeignKeyField('rms.Cabinet', null=True, on_delete=fields.SET_NULL, description='所属柜体')
|
|
|
template = fields.ForeignKeyField('rms.Template', null=True, description='所属模版')
|
|
|
# taboo_species = fields.CharField(null=True, max_length=50, description="药剂禁忌种类")
|
|
|
# drug_type = fields.CharField(null=True, max_length=50, default='normal',description="药剂类型 普通药剂normal 盒装药剂box")
|
|
|
|
|
|
# files = fields.JSONField(null=True, default="[]")
|
|
|
# near = fields.BooleanField(default=False, description='是否临期')
|
|
|
created_at = fields.DatetimeField(auto_now_add=True)
|
|
|
updated_at = fields.DatetimeField(auto_now=True)
|
|
|
|
|
|
class Meta:
|
|
|
table = 'drugs'
|
|
|
table_description = '药剂表'
|
|
|
|
|
|
@classmethod
|
|
|
async def conflict(cls, rfid) -> bool:
|
|
|
'''药剂标签是否已存在'''
|
|
|
drug = await Drug.get_or_none(rfid=rfid).values_list('id')
|
|
|
return True if drug else False
|
|
|
|
|
|
async def update_last_weight(self, weight: Decimal):
|
|
|
"""
|
|
|
称重更新重量
|
|
|
:param weight: 重量值
|
|
|
:return:
|
|
|
"""
|
|
|
self.last_weight = weight
|
|
|
await self.save()
|
|
|
|
|
|
async def update_last_use_weight(self, weight: Decimal):
|
|
|
"""
|
|
|
更新用量
|
|
|
:param weight: 用量值
|
|
|
:return:
|
|
|
"""
|
|
|
self.last_use_weight = weight
|
|
|
await self.save()
|
|
|
|
|
|
def attribute(self, key, default=None):
|
|
|
return self.fill_json_content.get(key, default)
|
|
|
|
|
|
async def parse_fill_json_content(self, pop_count: bool = True):
|
|
|
"""
|
|
|
解析药剂信息
|
|
|
:return: {"药剂名称": "xx", "药剂规格": "xx"}
|
|
|
"""
|
|
|
fill_json_content = self.fill_json_content
|
|
|
template_obj = await self.template
|
|
|
xlsx_transfer = template_obj.parse_xlsx_transfer().get("xlsx_transfer")
|
|
|
if pop_count and 'import_count' in fill_json_content:
|
|
|
fill_json_content.pop('import_count')
|
|
|
if fill_json_content.get("gross_weight") is not None:
|
|
|
fill_json_content.pop('gross_weight')
|
|
|
result = {
|
|
|
xlsx_transfer.get(key): fill_json_content.get(key)
|
|
|
for key in fill_json_content.keys() if xlsx_transfer.get(key)}
|
|
|
return result
|
|
|
|
|
|
async def attribute_cabinet_id(self):
|
|
|
# drawer_board_obj = await DrawerBoard.get(id = self.drawer_board_id).prefetch_related("cabinet")
|
|
|
if self.drawer_board_id:
|
|
|
drawer_board_obj = await self.drawer_board.prefetch_related("cabinet")
|
|
|
else:
|
|
|
return self.cabinet_id
|
|
|
return drawer_board_obj.cabinet.id
|
|
|
|
|
|
async def attribute_drug_info(self):
|
|
|
"""
|
|
|
full_json_content内部k1-k6数据
|
|
|
:return:
|
|
|
"""
|
|
|
template_obj = await self.template
|
|
|
xlsx = list(filter(lambda i: i.get("checked"), template_obj.xlsx))
|
|
|
xlsx_transfer = {f"{item['key']}": f"{item['text']}" for item in xlsx}
|
|
|
|
|
|
if self.drug_type == 'box':
|
|
|
box_dict = setting.BOX_TRANSFER_DICT
|
|
|
result = {box_dict.get(key): self.fill_json_content.get(key) for key in box_dict}
|
|
|
else:
|
|
|
result = {xlsx_transfer.get(f"k{i}"): self.fill_json_content.get(f"k{i}")
|
|
|
for i in range(1, 7) if self.fill_json_content.get(f"k{i}")}
|
|
|
|
|
|
return result
|
|
|
|
|
|
async def attribute_last_user(self):
|
|
|
"""
|
|
|
最后使用人
|
|
|
判断最后领用人，最后归还人
|
|
|
:return:
|
|
|
"""
|
|
|
if self.last_return_at and not self.last_receive_at:
|
|
|
last_user_id = self.last_return_id
|
|
|
elif self.last_receive_id and self.last_return_id:
|
|
|
last_user_id = self.last_receive_id if self.last_receive_at < self.last_return_at else self.last_return_id
|
|
|
else:
|
|
|
last_user_id = self.last_receive_id or self.last_return_id or self.storage_id
|
|
|
user_obj = await User.get_or_none(id=last_user_id)
|
|
|
if user_obj is None:
|
|
|
return ""
|
|
|
return user_obj.name
|
|
|
|
|
|
async def format_weight(self):
|
|
|
"""
|
|
|
余重
|
|
|
:return:
|
|
|
"""
|
|
|
remain_gross_weight = self.remain_gross_weight
|
|
|
|
|
|
if self.remain_gross_weight is None:
|
|
|
return '-'
|
|
|
|
|
|
net_weight = self.fill_json_content.get('net_weight') # 净含量
|
|
|
density = self.fill_json_content.get('density') # 密度
|
|
|
total_use_weight = self.total_use_weight # 总用量
|
|
|
|
|
|
if self.state == DrugStateEnum.EMPTY:
|
|
|
if net_weight is None or net_weight == '':
|
|
|
return '0g'
|
|
|
else:
|
|
|
unit = re.findall('.*?(mg|g|ml)$', net_weight)[0]
|
|
|
return "0{}".format(unit)
|
|
|
else:
|
|
|
if net_weight is None or net_weight == '':
|
|
|
return "%.1fg" % (remain_gross_weight / 1000) if remain_gross_weight > 1000 else "{}mg".format(
|
|
|
remain_gross_weight)
|
|
|
else:
|
|
|
if net_weight.endswith('mg'):
|
|
|
# 总用量大于净含量，则余量显示为0
|
|
|
if total_use_weight >= float(net_weight[:-2]):
|
|
|
return "{}mg".format(0)
|
|
|
return "{}mg".format(float(net_weight[:-2]) - total_use_weight)
|
|
|
|
|
|
if net_weight.endswith('g'):
|
|
|
return "%.1fg" % ((float(net_weight[:-1]) * 1000 - total_use_weight) / 1000)
|
|
|
|
|
|
# 如下的代码不是BUG
|
|
|
# 如果是液体（单位：ml），有密度，前端入参是重量，需要：容量=重量/密度，获取容量
|
|
|
# 如果是液体，没有密度，则返回余重
|
|
|
if net_weight.endswith('ml'):
|
|
|
if density:
|
|
|
density = density[:-4] if density.endswith("g/ml") else density
|
|
|
return "%.1fml" % (float(net_weight[:-2]) - total_use_weight / (float(density) * 1000))
|
|
|
else:
|
|
|
return "%.1fg" % (remain_gross_weight / 1000) if remain_gross_weight > 1000 else "{}mg".format(
|
|
|
remain_gross_weight)
|
|
|
|
|
|
async def parse_archive_params(self, key):
|
|
|
"""
|
|
|
解析获取大类参数
|
|
|
drug_obj -> template ->archive params
|
|
|
:param key:
|
|
|
:return:
|
|
|
"""
|
|
|
|
|
|
archive_obj = self.template.archive
|
|
|
archive_params = archive_obj.params.get(key)
|
|
|
return archive_params
|
|
|
|
|
|
async def box_item_generate(self, *args, **kwargs):
|
|
|
"""
|
|
|
盒装药剂明细条目生成
|
|
|
:param args:
|
|
|
:param kwargs: drug_items=[{xx}]
|
|
|
:return:
|
|
|
"""
|
|
|
drug_items = kwargs.get("drug_items")
|
|
|
if not drug_items:
|
|
|
return False
|
|
|
for drug_item in drug_items:
|
|
|
k_args = {f'k{i}': drug_item.get(f'k{i}') for i in range(1, 7) if drug_item.get(f'k{i}')}
|
|
|
k_args_with_null = {k: None for k, v in k_args.items() if v is None}
|
|
|
query = Q(**k_args) & Q(**k_args_with_null)
|
|
|
dictionary_obj = await Dictionary.filter(query).first()
|
|
|
if not dictionary_obj:
|
|
|
k_args["archive"] = self.template.archive
|
|
|
k_args["params"] = {"lack_stock_count": 10, "expiration_alert": 30}
|
|
|
dictionary_obj = await Dictionary.create(**k_args)
|
|
|
# 根据导入数量生成对应数量条目
|
|
|
if not drug_item.get("import_count"):
|
|
|
import_count = 1
|
|
|
else:
|
|
|
import_count = int(drug_item.get("import_count", 1))
|
|
|
|
|
|
drug_item["import_count"] = 1
|
|
|
data = {
|
|
|
"drug_id": self.id,
|
|
|
"name": drug_item.get("k1"),
|
|
|
"fill_json_content": drug_item,
|
|
|
"dictionary": dictionary_obj,
|
|
|
"rfid": self.rfid,
|
|
|
"barcode": self.barcode,
|
|
|
"state": DrugItemStateEnum.IN if len(self.rfid) < 10 else DrugItemStateEnum.INITIAL,
|
|
|
"bind_id": self.bind_id,
|
|
|
"bind_at": self.bind_at,
|
|
|
}
|
|
|
if drug_item.get("expire_date"):
|
|
|
data.update({
|
|
|
"expired_at": drug_item.get("expire_date")
|
|
|
})
|
|
|
|
|
|
for i in range(import_count):
|
|
|
await DrugItem.create(**data)
|
|
|
return True
|
|
|
|
|
|
async def attribute_box_item_info(self, *args, **kwargs):
|
|
|
"""
|
|
|
盒内药剂明细信息
|
|
|
:param args:
|
|
|
:param kwargs:
|
|
|
:return: {”box_count“: 10, "box_name_items": ""}
|
|
|
"""
|
|
|
if self.drug_type != "box":
|
|
|
return {"box_count": 0, "box_name_items": ""}
|
|
|
query = QuerySet(DrugItem).filter(drug_id=self.id)
|
|
|
total_count = await query.count()
|
|
|
drug_item_objs = await query.all()
|
|
|
name_list = []
|
|
|
# 相同名称药剂，只需要显示一次
|
|
|
for drug_item_obj in drug_item_objs:
|
|
|
name = drug_item_obj.fill_json_content.get("k1")
|
|
|
if name and name not in name_list:
|
|
|
name_list.append(name)
|
|
|
name_str = ",".join(name_list) if name_list else ""
|
|
|
return {"box_count": total_count, "box_name_items": name_str}
|
