|
|
# -*- coding:utf-8 -*-
|
|
|
"""
|
|
|
@Created on : 2023/6/30 14:29
|
|
|
@Author: hxl
|
|
|
@Des:
|
|
|
"""
|
|
|
from datetime import datetime, timedelta
|
|
|
from fastapi import APIRouter, Depends, Request
|
|
|
from fastapi.encoders import jsonable_encoder
|
|
|
from pydantic import BaseModel
|
|
|
from tortoise.queryset import QuerySet
|
|
|
|
|
|
from helper import login_required, respond_to
|
|
|
from models import Log, DrugUseLog, User
|
|
|
from models.drug_use_log import DrugUseStateEnum
|
|
|
from helper.drug import milligram_to_gram
|
|
|
from models.cabinet import Cabinet
|
|
|
|
|
|
|
|
|
router = APIRouter(prefix='/logs', dependencies=[Depends(login_required)])
|
|
|
|
|
|
class DrugUseLogSearch(BaseModel):
|
|
|
user_id: str | None
|
|
|
state: int | None
|
|
|
stime: str | None
|
|
|
etime: str | None
|
|
|
drug_id: str | None
|
|
|
cabinet_id: str | None
|
|
|
page_no: int = 1
|
|
|
page_size: int = 20
|
|
|
|
|
|
@router.get('', summary='平台日志列表')
|
|
|
async def index(page_no: int = 1, page_size: int = 20, begin_date: str = '', end_date: str = '', keyword: str = None):
|
|
|
"""
|
|
|
获取平台日志列表
|
|
|
:return:
|
|
|
"""
|
|
|
kwargs = {}
|
|
|
if keyword:
|
|
|
kwargs = {"comment__contains": keyword.strip()}
|
|
|
|
|
|
query = Log.filter(**kwargs)
|
|
|
if begin_date:
|
|
|
query = query.filter(created_at__gte=f"{begin_date} 00:00:00")
|
|
|
if end_date:
|
|
|
temp_time = datetime.strptime(end_date, "%Y-%m-%d").replace(hour=0, minute=0, second=0)
|
|
|
end_time = temp_time + timedelta(days=1)
|
|
|
query = query.filter(created_at__lt=end_time)
|
|
|
offset = (page_no - 1) * page_size
|
|
|
count = await query.count()
|
|
|
log_objs = await query.limit(page_size).offset(offset).order_by('-created_at')
|
|
|
|
|
|
return respond_to(data=dict(count=count, logs=log_objs))
|
|
|
|
|
|
# 流转日志
|
|
|
async def parse_archive_cabinet_ids(archive_id: str, client_ids: list):
|
|
|
"""
|
|
|
根据大类ID获取柜体
|
|
|
如果传递客户端ids参数，则返回用户有权限的大类柜体列表
|
|
|
:param archive_id:
|
|
|
:param client_ids: 用户有权限的柜体id，如果不传则代表全部柜体
|
|
|
:return:
|
|
|
"""
|
|
|
query = QuerySet(Cabinet).filter()
|
|
|
if archive_id:
|
|
|
query = query.filter(archive_id=archive_id)
|
|
|
if client_ids:
|
|
|
query = query.filter(id__in=client_ids)
|
|
|
# 用户有权限的柜体
|
|
|
cabinets_ids = await query.values_list("id", flat=True)
|
|
|
return cabinets_ids
|
|
|
@router.post('/drug', summary='流转日志列表')
|
|
|
async def index(request: Request, keyword: DrugUseLogSearch):
|
|
|
"""
|
|
|
获取平台流转日志列表
|
|
|
:return:
|
|
|
"""
|
|
|
page_no = keyword.page_no
|
|
|
page_size = keyword.page_size
|
|
|
offset = (page_no - 1) * page_size
|
|
|
query = QuerySet(DrugUseLog)
|
|
|
if keyword.user_id:
|
|
|
query = query.filter(user_id=keyword.user_id)
|
|
|
else:
|
|
|
user = await User.filter(id=request.state.current_user.id).get_or_none()
|
|
|
if user.role_id != 1 and user.role_id != 100:
|
|
|
query = query.filter(user_id=request.state.current_user.id)
|
|
|
if keyword.state in [0, 1, 2, 3, 4]:
|
|
|
query = query.filter(state=keyword.state)
|
|
|
if keyword.stime:
|
|
|
query = query.filter(created_at__gte=f"{keyword.stime} 00:00:00")
|
|
|
if keyword.etime:
|
|
|
query = query.filter(created_at__lte=f"{keyword.etime} 23:59:59")
|
|
|
if keyword.drug_id:
|
|
|
query = query.filter(drug_id=keyword.drug_id)
|
|
|
archive_id = request.state.archive_id
|
|
|
if keyword.cabinet_id:
|
|
|
query = query.filter(cabinet_id=keyword.cabinet_id)
|
|
|
else:
|
|
|
if archive_id:
|
|
|
cabinets_ids = await parse_archive_cabinet_ids(archive_id, [])
|
|
|
if cabinets_ids:
|
|
|
query = query.filter(cabinet_id__in=cabinets_ids)
|
|
|
|
|
|
count = await query.count()
|
|
|
logs = await query.prefetch_related('drug').limit(page_size).offset(offset).order_by('-created_at')
|
|
|
result = list()
|
|
|
for log in logs:
|
|
|
if not log.drug:
|
|
|
continue
|
|
|
if log.state in [1, 3]:
|
|
|
use_weight = log.use_weight
|
|
|
use_weight = use_weight
|
|
|
finally_use_weight = "" if not use_weight and log.state != DrugUseStateEnum.PUT else use_weight
|
|
|
else:
|
|
|
finally_use_weight = ''
|
|
|
# weight > 1000 转换为g 保留一位小数否则返回mg
|
|
|
weight = float(log.weight) if log.weight else 0
|
|
|
if weight > 1000:
|
|
|
weight_str = f"{milligram_to_gram(log.weight)}g"
|
|
|
else:
|
|
|
weight_str = f"{log.weight}mg" if log.weight is not None else "-"
|
|
|
finally_use_weight_str = log.parse_use_weight(finally_use_weight)
|
|
|
print("finally_use_weight_str", finally_use_weight_str)
|
|
|
result.append({
|
|
|
**jsonable_encoder(log),
|
|
|
"drug_info": await log.drug.attribute_drug_info(),
|
|
|
"weight": weight_str,
|
|
|
"use_weight": finally_use_weight_str,
|
|
|
"created_at": log.created_at.strftime("%Y-%m-%d %H:%M:%S"),
|
|
|
})
|
|
|
return respond_to(data=dict(count=count, data=result))
|
