yy_rms_39zhiyao_duizhao/apps/drug_template/views.py

#!/usr/bin/env python
# -*- encoding: utf-8 -*-
'''
@Date:2022/07/25 09:32:18
'''
import sys
sys.path.append(".")

import datetime
import json
from flask import jsonify, request, g
from flask import Blueprint
from common.read_excel import ReadExcel

from common.utils import Utils, PageParam
from db_logic.client import BllClient
from db_logic.medicament import BllMedicament
from db_logic.user import BllUser
from db_logic.medicament_template import BllMedicamentTemplate
from db_logic.meidcament_variety import BllMedicamentVariety
from models.client_models import EntityClient
from models.medicament_models import EntityMedicament, EntityMedicamentTemplate



tmp_router = Blueprint("drug_tmplate", __name__)



# 试剂入库-模板展示
@tmp_router.route("/show_tmp", methods=["GET", "POST"])
def show_template():
    client_id = request.values.get("client_id")
    page = int(request.values.get("page", 1))
    page_size = int(request.values.get("page_size", 10))
    page_param = PageParam(page, page_size)
    data = BllMedicamentTemplate().getAllTemplateList(client_id=client_id, page_param=page_param)
    data = Utils.msyql_table_model(data)
    return jsonify(Utils.true_return(data={"total_count":page_param.totalRecords, "data_list":data}))





# 选择药剂入库
@tmp_router.route("/select_drug", methods=["GET", "POST"])
def select_drug_info():
    seach_word =request.values.get("seach_word")
    page = int(request.values.get("page", 1))
    page_size = int(request.values.get("page_size", 10))
    page_param = PageParam(page, page_size)
    data_list = BllMedicament().get_drug_info_distinct_list(
        seach_word=seach_word, page_param=page_param)
    data_list = Utils.msyql_table_model(data_list)
    return jsonify(Utils.true_return(data={"total_count": page_param.totalRecords, "data_list": data_list}))

# 绑定入库
@tmp_router.route("/bind_tmp_enter_warehouse", methods=["GET", "POST"])
def bind_tmp_input_db():
    tmp_content = request.values.get("tmplate_content")
    customer_id = request.values.get("customer_id", '')
    # client_id = request.values.get("client_id")
    client_id = '1c39cb24-07f8-11ed-abd4-f47b094925e1'
    bar_code = request.values.get("bar_code")
    # user_id = request.values.get("user_id")
    if not bar_code:
        return jsonify(Utils.false_return(msg="条码不能为空"))
    drug_obj = BllMedicament().findEntity(EntityMedicament.bar_code==bar_code)
    if drug_obj:
        return jsonify(Utils.false_return(msg="该药剂已被锁定"))
    else:
        drug_info = eval(tmp_content)
        user_info = BllUser().findEntity('4cea74ee-0d8b-11ed-943e-f47b094925e1')
        # user_info = g.current_user

        drug_variety = BllMedicamentVariety().createDrugVariety(
            customer_id=customer_id,
            drug_info=drug_info, user_info=user_info
        )
        finds = [
            "name", "cas_number", "english_name", "purity", "manufacturer", "distributor",
            "production_date", "shelf_life", "net_weight_unit", "net_weight", "tp", "price",
            "place", "expiration_date"
        ]
        key_dict = {}
        for i in finds:
            key_dict[i] = drug_info.get(i)
        new_dict = {
            "medicament_id": Utils.UUID(),
            "variety_id": drug_variety.variety_id,
            "bar_code": bar_code,
            "customer_id": customer_id,
            "client_id": client_id,
            # "expiration_date": datetime.datetime.strptime(drug_info.get('production_date'), "%Y-%m-%d") + datetime.timedelta(
            #     days=drug_info.get("shelf_life")).strftime("%Y-%m-%d %H:%M:%S"),
            "inventory_warning_value":10,
            "shelf_life_warning_value":10,
            "use_days_warning_value":10,
            "is_supervise":0,
            "remain": drug_info.get("net_weight"),
            "put_in_date": Utils.get_str_datetime(),
            "put_in_user_id": user_info.user_id,
            "put_in_user_name": user_info.real_name,
            "status":5
        }
        new_dict.update(**key_dict)
        drug_entity = EntityMedicament(**new_dict)
        objs = BllMedicament().drugPutIn(
            drug_entity, 
            BllClient().findEntity(EntityClient.client_id==client_id),
            user_info
            )
        if objs:
            return jsonify(Utils.true_return(msg="药剂入库成功"))
        else:
            return jsonify(Utils.false_return(msg="数据异常，入库失败"))



# 新增试剂模板
@tmp_router.route("/add_tmp", methods=["GET", "POST"])
def add_tmp_info():
    tmp_name = request.values.get("template_name")
    # client_id = request.values.get("client_id")
    client_id = '1c39cb24-07f8-11ed-abd4-f47b094925e1'
    tmp_content = request.values.get("tmplate_content")
    user_id = '4cea74ee-0d8b-11ed-943e-f47b094925e1'
    user = BllUser().findEntity(user_id)
    client = BllClient().findEntity(client_id)
    sql_all = """
    select bar_code_count, start_bar_code from rms_medicament_template where start_bar_code=(select max(start_bar_code) max_start from rms_medicament_template)
    """
    # 获取最大的条形码和条形码数量
    tmp_obj = BllMedicamentTemplate().execute(sql_all).fetchone()
    if tmp_obj:
        max_barcode = tmp_obj.start_bar_code
        bar_code_count = tmp_obj.bar_code_count
        max_barcode = int(max_barcode) + int(bar_code_count)
    else:
        max_barcode = 100001
    start_bar_code = max_barcode

    try:
        if isinstance(tmp_content, str):
            tmp_content = eval(tmp_content)
        total_count = 0
        for i in tmp_content:
            total_count += int(i['export_count'])
        
        obj = EntityMedicamentTemplate(
            customer_id=client.customer_id,
            client_id=client.client_id,
            client_name=client.client_name,
            template_name=tmp_name,
            template_content=json.dumps(tmp_content),
            is_wait_export=1,
            start_bar_code=start_bar_code,
            bar_code_count=bar_code_count,
            create_date=Utils.get_str_datetime(),
            create_user_id=user.user_id,
            create_user_name=user.real_name,

        )
        state = BllMedicamentTemplate().insert(obj)
        if state:
            return jsonify(Utils.true_return(msg="添加模板成功"))
        else:
            return jsonify(Utils.false_return(msg="导入失败"))
    except Exception as error:
        return jsonify(Utils.false_return(msg=f"请求有误:{error}"))




# # 编辑试剂模板
# @tmp_router.route("/update_tmp", methods=["GET", "POST"])
# def modify_tmp():
#     tmp_id = request.values.get("template_id")

# 导入入库模板
@tmp_router.route("/put_in_tmp", methods=["GET", "POST"])
def import_tmplate():
    try:
        file_path = Utils.getUDiskPath()
        if not file_path:
            return jsonify(msg="请插入U盘")
        else:
            client_id = request.values.get("cliet_id", '')
            template_name = request.values.get("template_name", '')
            user_id = request.values.get("user_id")
            read_excel = ReadExcel(file_path + "/" + template_name)
            sql_all = """
            select bar_code_count, start_bar_code from rms_medicament_template where start_bar_code=(select max(start_bar_code) max_start from rms_medicament_template)
            """
            tmp_obj = BllMedicamentTemplate().execute(sql_all).fetchone()
            if tmp_obj:
                max_barcode = tmp_obj.start_bar_code
                bar_code_count = tmp_obj.bar_code_count
                max_barcode = int(max_barcode) + int(bar_code_count)
            else:
                max_barcode = 100001
            start_bar_code = max_barcode

            result_list = read_excel.read()
            client_obj = BllClient().findEntity(client_id)
            user_obj = BllUser().findEntity(user_id)
            if result_list:
                inster_obj_list = []
                for value in result_list:
                    total_count = 0
                    template_content_list = eval(value)

                    for template_content in template_content_list:
                        total_count += int(template_content['export_cout'])
                    obj = EntityMedicamentTemplate(
                        customer_id=client_obj.customer_id,
                        client_id=client_obj.client_id,
                        client_name=client_obj.client_name,
                        template_name=f'入库模板_{Utils.getFileName()}',
                        template_content=value,
                        is_wait_export = 1,
                        start_bar_code=str(start_bar_code),
                        bar_code_count=total_count,
                        create_date=Utils.get_str_datetime(),
                        create_user_id=user_obj.user_id,
                        create_user_name=user_obj.real_name

                    )
                    inster_obj_list.append(obj)
                status = BllMedicamentTemplate().insert_many(inster_obj_list)
                if status:
                    return jsonify(Utils.true_return(msg='导入成功'))
                else:
                    return jsonify(Utils.false_return(msg="入库单异常，导入失败"))
            else:
                return jsonify(Utils.false_return(msg="入库单格式不正确,请检查之后重新导入"))
    except FileNotFoundError as e:
        return jsonify(Utils.false_return(msg='当前U盘下暂无reagentTemplate.xlsx文件'))
    except Exception as error:
        return jsonify(Utils.false_return(msg=f"模板错误：{error}"))




# 删除模板
@tmp_router.route("/del_tmp", methods=["GET", "POST"])
def remove_tmp():
    tmp_id = request.values.get("template_id")
    if tmp_id:
        BllMedicamentTemplate().delete(EntityMedicamentTemplate.template_id==tmp_id)
        return Utils.true_return(msg='删除药剂模板成功')
    return Utils.false_return(msg='删除药剂模板失败')