21 lines
950 B
Python
21 lines
950 B
Python
from apps.wpm.models import SfLog
|
|
from apps.mtm.models import Shift, Mgroup
|
|
import datetime
|
|
|
|
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) |