# -*- coding: utf-8 -*-
# !/usr/bin/env python

from celery.utils.log import get_task_logger

from apilib.utils_datetime import timestamp_to_dt
from apps.web.ad.models import AdRecord
from apps.web.core.utils import generate_excel_report

logger = get_task_logger(__name__)

def generate_ad_excel_report(filepath, queryAttrs):
    #: 此任务IO耗时大，所以将其作为异步任务
    queryAttrs['dateTimeAdded__lte'] = timestamp_to_dt(queryAttrs['dateTimeAdded__lte'])
    queryAttrs['dateTimeAdded__gte'] = timestamp_to_dt(queryAttrs['dateTimeAdded__gte'])

    records = [_.to_dict_in_cn() for _ in AdRecord.objects(**queryAttrs)]

    generate_excel_report(filepath, records)