# -*- coding: utf-8 -*-
# @Time    : 2023/8/25 15:42
# @Author  : tx
# @File    : cabinet.py
# @Description :

from datetime import datetime

from helper.utils import timezone_now
from models.drug import Drug, DrugStateEnum
from models.drug_use_log import DrugUseLog, DrugUseStateEnum
from models.terminal import Terminal
from models.cabinet import Cabinet
from conf.setting import setting


async def calculate_today_data(cabinet_id: str):
    """
    计算每个柜子/每个终端的今日统计数据
    :param cabinet:
    :return:
    """

    terminal_obj = await Terminal.get(id=setting.TERMINAL_ID)
    stime = datetime.combine(timezone_now(), datetime.min.time())
    etime = datetime.combine(timezone_now(), datetime.max.time())

    q1 = {"storage_at__range": [stime, etime]}
    # q2 = {"state": DrugStateEnum.IN, "expired_at__lt": timezone_now().strftime('%Y-%m-%d %H:%M:%S')}
    q3 = {"state": DrugUseStateEnum.TAKE, "created_at__range": [stime, etime]}
    q4 = {"state": DrugUseStateEnum.PUT, "created_at__range": [stime, etime]}

    if not cabinet_id:
        if setting.TERMINAL_ID:
            cabinets_ids = await Cabinet.filter(terminal_id=setting.TERMINAL_ID).values_list("id", flat=True)
            q1.update({"cabinet_id__in": cabinets_ids})
            # q2.update({"cabinet_id__in": cabinets_ids})
        q3.update({"terminal_id": terminal_obj.id})
        q4.update({"terminal_id": terminal_obj.id})
    else:
        q1.update({"cabinet_id": cabinet_id})
        # q2.update({"cabinet_id": cabinet_id})
        q3.update({"cabinet_id": cabinet_id})
        q4.update({"cabinet_id": cabinet_id})

    # 今日入库
    storage_count = await Drug.filter(**q1).count()
    # 截止目前在库已过期
    # expire_count = await Drug.filter(**q2).count()
    # todo：逾期未归还
    overdue_count = 0
    # 今日领用
    receive_count = await DrugUseLog.filter(**q3).count()
    # 今日归还
    return_count = await DrugUseLog.filter(**q4).count()

    result = dict(storage_count=storage_count, overdue_count=overdue_count, receive_count=receive_count,
                  return_count=return_count)
    return result