diff --git a/apps/wpm/tasks.py b/apps/wpm/tasks.py index 2a7c3955..087fed1d 100644 --- a/apps/wpm/tasks.py +++ b/apps/wpm/tasks.py @@ -15,12 +15,16 @@ from django.utils.timezone import localtime from django.db.models import F @shared_task(base=CustomTask) -def make_sflogs_simple(days=2): +def make_sflogs_simple(days, state_date:str, end_date:str): """ 根据班次规则生成今明两天的排班记录 """ - start_date = datetime.datetime.today() - end_date = start_date + datetime.timedelta(days=days) + if days: + start_date = datetime.datetime.today() + end_date = start_date + datetime.timedelta(days=days) + else: + start_date = datetime.datetime.strptime(state_date, "%Y-%m-%d").date() + end_date = datetime.datetime.strptime(end_date, "%Y-%m-%d").date() mgroups = Mgroup.objects.filter(cate='section').all() for mgroup in mgroups: make_sflogs(mgroup, start_date, end_date)