zcdsj
/
factory
5
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

refector: 生成排班记录

This commit is contained in:
caoqianming 2023-06-30 10:24:15 +08:00
parent a6167294f7
commit f620dc0bcb
2 changed files with 19 additions and 20 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

33
apps/wpm/services.py
View File

@ -2,19 +2,20 @@ from apps.wpm.models import SfLog
from apps.mtm.models import Shift, Mgroup
import datetime
def make_sflogs(mgroup: Mgroup, shift: Shift, start_date: datetime.date, end_date: datetime.date):
start_time_o = shift.start_time_o
end_time_o = shift.end_time_o
current_date = start_date
while current_date <= end_date:
start_time = datetime.datetime.combine(current_date, start_time_o)
end_time = datetime.datetime.combine(current_date, end_time_o)
if start_time > end_time:
start_time -= datetime.timedelta(days=1)
SfLog.objects.get_or_create(mgroup=mgroup, shift=shift, start_time=start_time, defaults={
"mgroup": mgroup,
"shift": shift,
"start_time": start_time,
"end_time": end_time,
})
current_date = current_date + datetime.timedelta(days=1)
def make_sflogs(mgroup: Mgroup, start_date: datetime.date, end_date: datetime.date):
for shift in Shift.objects.all():
start_time_o = shift.start_time_o
end_time_o = shift.end_time_o
current_date = start_date
while current_date <= end_date:
start_time = datetime.datetime.combine(current_date, start_time_o)
end_time = datetime.datetime.combine(current_date, end_time_o)
if start_time > end_time:
start_time -= datetime.timedelta(days=1)
SfLog.objects.get_or_create(mgroup=mgroup, shift=shift, start_time=start_time, defaults={
"mgroup": mgroup,
"shift": shift,
"start_time": start_time,
"end_time": end_time,
})
current_date = current_date + datetime.timedelta(days=1)

6
apps/wpm/tasks.py
View File

@ -6,7 +6,7 @@ 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, Shift
from apps.mtm.models import Mgroup
import datetime
from django.db.models import Sum
from apps.wpm.services import make_sflogs
@ -20,7 +20,5 @@ 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()
shifts = Shift.objects.all()
for mgroup in mgroups:
for shift in shifts:
make_sflogs(mgroup, shift, start_date, end_date)
make_sflogs(mgroup, start_date, end_date)