# Create your tasks here
from __future__ import absolute_import, unicode_literals
from apps.utils.tasks import CustomTask
from celery import shared_task
from apps.utils.sql import DbConnection
from server.settings import get_sysconfig
from django.core.cache import cache
from apps.wpm.models import SfLog
from apps.mtm.models import Mgroup
import datetime
from django.db.models import Sum
from apps.wpm.services import make_sflogs


@shared_task(base=CustomTask)
def make_sflogs_simple(days=2):
    """
    根据班次规则生成今明两天的排班记录
    """
    start_date = datetime.datetime.today()
    end_date = start_date + datetime.timedelta(days=days)
    mgroups = Mgroup.objects.filter(cate='section').all()
    for mgroup in mgroups:
            make_sflogs(mgroup, start_date, end_date)