Commit 83f69770 authored by Georgios Dagkakis's avatar Georgios Dagkakis

allow to define the same shift and breaks for all operators using all keyword

parent 55c02195
from dream.plugins import plugin
from dream.plugins.TimeSupport import TimeSupportMixin
import datetime
from copy import copy
class BatchesOperatorBreaks(plugin.InputPreparationPlugin, TimeSupportMixin):
""" Output the schedule of operators in an Excel file to be downloaded
......@@ -18,6 +19,20 @@ class BatchesOperatorBreaks(plugin.InputPreparationPlugin, TimeSupportMixin):
self.initializeTimeSupport(data)
breakData=data['input']['operator_shift_spreadsheet']
PBData=copy(data['input'].get('operator_skill_spreadsheet', None))
# create a string with all operator ids ids separated by commas
allString=''
for row in PBData:
if row[0]:
allString+=row[0]
allString+=','
# if in operator shift there is ALL in a cell then give the all string
for element in breakData:
if element[1] in ['ALL','All','all']:
element[1]=allString
for row in breakData:
if row[0] in ['Date', '', None]:
continue
......@@ -26,6 +41,8 @@ class BatchesOperatorBreaks(plugin.InputPreparationPlugin, TimeSupportMixin):
# if element has spaces in beginning or in end remove them
operators=self.stripStringsOfList(operators)
# remove empty strings
operators = filter(bool, operators)
i=4
while row[i] not in ['', None]:
......
Markdown is supported
0%
or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment