From 40ecd52fee43186eaca2c685cbb46423a7f75da4 Mon Sep 17 00:00:00 2001 From: caoqianming Date: Wed, 2 Aug 2023 15:54:09 +0800 Subject: [PATCH] =?UTF-8?q?feat:=20=E5=8F=AF=E6=A0=B9=E6=8D=AE=E6=97=B6?= =?UTF-8?q?=E9=97=B4=E6=AE=B5=E7=94=9F=E6=88=90=E6=8E=92=E7=89=88=E8=AE=B0?= =?UTF-8?q?=E5=BD=95?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- apps/wpm/tasks.py | 10 +++++++--- 1 file changed, 7 insertions(+), 3 deletions(-) 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)