from __future__ import absolute_import, unicode_literals

from celery import shared_task
from apps.hrm.models import Employee, ClockRecord
from dateutil import tz
from datetime import datetime, timedelta
from apps.third.dahua import dhClient
from apps.third.tapis import dhapis
from apps.hrm.services import HrmService
from django.utils import timezone
from apps.utils.tasks import CustomTask
import time


@shared_task
def update_all_employee_not_atwork():
    """
    将所有员工设为非在岗状态
    """
    Employee.objects.all().update(is_atwork=False, last_check_time=None, not_work_remark=None)

@shared_task(base=CustomTask)
def correct_swip_task(start_time="", end_time=""):
    # from apps.monitor.models import DrfRequestLog
    # from apps.hrm.services import HrmService
    # lgs = DrfRequestLog.objects.filter(path='/api/third/dahua/c_swip/', requested_at__gte=start_time, requested_at__lte=end_time)
    # for i in lgs:
    #     HrmService.swipe(data=eval(i.data))
    # 从大华历史记录校正打卡记录
    now = datetime.now()
    if start_time == "":
        start_time = (now - timedelta(hours=25)).strftime("%Y-%m-%d %H:%M:%S")
    if end_time == "":
        end_time = now.strftime("%Y-%m-%d %H:%M:%S")
    data0 = {
        "pageNum":1,
        "pageSize": "20",
        "startSwingTime":start_time,
        "endSwingTime":end_time,
        "openType":"61"
    }
    _, count = dhClient.request(**dhapis['swipe_list_count'], json=data0)
    data = {
        "pageNum":1,
        "pageSize": str(count),
        "startSwingTime":start_time,
        "endSwingTime":end_time,
        "openType":"61"
    }
    _, res = dhClient.request(**dhapis['swipe_list'], json=data)
    for i in res['pageData']:
        if 'paperNumber' in i:
            detail = {
                'id': i['id'],
                'channelCode': i['channelCode']
            }
            HrmService.swipe_next(i['channelCode'], i['paperNumber'], i['swingTime'], i['enterOrExit'], detail)
    # ClockRecord.objects.filter(type=30).delete()

@shared_task
def correct_card_time():
    tzinfo = tz.gettz('Asia/Shanghai')
    s_time_f = datetime.strptime("2022-11-15 14:20:20", "%Y-%m-%d %H:%M:%S").replace(tzinfo=tzinfo)
    eps = Employee.objects.filter(update_time__lte = s_time_f)
    print(eps)
    for ep in eps:
        dh_face_card = ep.third_info.get('dh_face_card', None)
        dh_face_card_end = ep.third_info.get('dh_face_card_end', None)
        if dh_face_card and dh_face_card_end is None:
            departmentId = 1
            if ep.belong_dept:
                try:
                    departmentId = ep.belong_dept.third_info['dh_id']
                except Exception:
                    pass
            # 获取卡片时间
            _, res = dhClient.request(**dhapis['card_detail'], params={'cardNumber': dh_face_card})
            time.sleep(1)
            start_time_str, end_time_str  = res['startDate'], res['endDate']
            end_time_new = datetime.strptime(end_time_str, "%Y-%m-%d %H:%M:%S") + timedelta(hours=8)
            end_time_new_str = end_time_new.strftime("%Y-%m-%d %H:%M:%S")
            json_data = {
                "cardNumber": dh_face_card,
                "startDate": start_time_str,
                "endDate": end_time_new_str,
                "departmentId": departmentId,
            }
            _, res = dhClient.request(**dhapis['card_update'], json=json_data)
            HrmService.save(ep, {'dh_face_card_start': start_time_str, 'dh_face_card_end': end_time_new_str})
            print('已更新-' + ep.name + '-' + dh_face_card + '-' + end_time_new_str)
        time.sleep(1)