#!/usr/bin/env python
# -*- encoding: utf-8 -*-
'''
@Date:2022/07/18 16:19:33
'''
from flask import jsonify, request

from models.warning_models import EntityWarning

from . import home_router


from db_logic.medicament_record import BllMedicamentRecord
from db_logic.warning import BllWarning
from common.utils import PageParam, Utils

# 获取页面主概览数据 --- 查询条件，缺少柜体id，客户id
@home_router.route('/home_number', methods=["GET", "POST"])
# def getRecordTypeDrugRecordListJson():
def get_record_type_drug_record_json():
    customerId = request.values.get('customerId')
    client_id = request.values.get('clientId')
    # 获取药剂记录今日入库 领用 归还Json数据
    typ_dic = BllMedicamentRecord().getTodayDrugRecordCount(
        customerId=customerId, clientId=client_id)
    data = {
        "putin_count": typ_dic.get(1),
        "use_count": typ_dic.get(2),
        "return_count": typ_dic.get(3),
        "not_return_count": typ_dic[2] - typ_dic[3]
    }
    return jsonify(Utils.true_return(data=data))


# 获取预警信息列表接口
@home_router.router('/home_warning_list', methods=["GET", "POST"])
def get_warning_list():
    customer_id = request.values.get('customer_id', '')
    page = request.values.get('page', 1)
    page_size = request.values.get('page_size', 15)
    page_param = PageParam(page, page_size)
    warning_list = BllWarning.getWarningList(customerId=customer_id, pageParam=page_param)
    if warning_list:
        data = {
            "total_count": page_param.totalRecords,
            "data_list": warning_list
        }
        return jsonify(Utils.true_return(data=data))
    else:
        return jsonify(Utils.false_return(msg='获取预警信息列表失败'))


# 修改预警状态，根据预警id进行
@home_router.router("/update_warning_tp", methods=["GET", "POST"])
def update_warning_type():
    warning_id = request.values.get('warning_id')
    obj = BllWarning().findEntity(EntityWarning.warning_id == warning_id)
    obj.object_type = 2
    obj.solve_date= Utils.get_str_datetime()
    obj.solve_user_id = ''
    obj.solve_user_name = ''
    BllWarning.update(obj)

    return jsonify(Utils.true_return())