def onExport(self, evt):
dialog = wx.FileDialog(self, u"选择要导出的文件位置", os.getcwd(), style=wx.FD_SAVE | wx.FD_OVERWRITE_PROMPT,
wildcard="*.csv")
if dialog.ShowModal() == wx.ID_OK:
filePath = dialog.GetPath()
if filePath:
try:
personalWorkDayDict = self.exportData[0]
dailyAttendenceList = self.exportData[1]
personalWorkDayLengthList = self.exportData[2]
startDate = self.exportData[3]
firstLine = u'员工名,班组名'
totalWeekNum = int(len(dailyAttendenceList) / 7)
for totalWeekNum in range(1, totalWeekNum + 1):
dateStr = u',"第 ' + str(totalWeekNum) + u' 周\n'
endDate = TimeUtil.getFormatedDate(startDate, 6)
dateStr += (startDate + ' -- ' + endDate + '"')
firstLine += dateStr
startDate = TimeUtil.getFormatedDate(endDate, 1)
lines = [firstLine]
for item in personalWorkDayDict:
wokerName = item[0]
groupName = item[1]
workDayList = item[2]
lines.append(wokerName + ',' + groupName + u',' + u','.join(
list(map(lambda week: ' '.join(list(map(str, week))), workDayList))))
lines.append('\n')
lines.append(u'员工名,班组名,总天数,总工时（小时）')
# 工时
for i in range(len(personalWorkDayLengthList)):
lines.append(','.join(personalWorkDayLengthList[i]))
lines.append('\n')
lines.append(u'日期,出勤员工')
groupInfo = ',' + ''.join(list(map(lambda x: x[1].groupName + ',' * int(x[1].workLoad), dailyAttendenceList[0][1])))
innerLines = list()
for dailyAttendence in dailyAttendenceList:
date = dailyAttendence[0]
groupAttendenceList = dailyAttendence[1]
attendenceStr = date
for groupAttendence in groupAttendenceList:
attendence = groupAttendence[0]
attendenceStr += ',' + attendence
innerLines.append(attendenceStr)
lines.append(groupInfo)
lines.extend(innerLines)
FileUtil.writeAll(filePath, lines)
wx.MessageBox(u'成功导出到文件', filePath)
except Exception as e:
wx.MessageBox(u'导出失败，原因为: ' + str(e))
dialog.Destroy()
def displayScheduleResult(self, resultList, startDate):
personalWorkDayDict = list()
personalWorkDayLengthList = list()
dailyAttendenceDict = dict()
for result in resultList:
calendar = result[0]
workers = result[1]
group = result[2]
try:
workdayDict = Scheduler.getWorkDayForEachWorker(calendar, workers)
for (workIndex, workDayList) in workdayDict.items():
workDayList = list(map(int, workDayList))
workDayList.sort()
weeklyWorkDayList = list()
week = 1
startIndex = 0
for dateIndex in range(len(workDayList)):
if int(workDayList[dateIndex]) > week * 7:
weeklyWorkDayList.append(
list(
map(lambda date: date % 7 if date % 7 != 0 else 7,
workDayList[startIndex: dateIndex])))
startIndex = dateIndex
week += 1
if startIndex < len(workDayList):
weeklyWorkDayList.append(
list(map(lambda date: date % 7 if date % 7 != 0 else 7,
workDayList[startIndex: len(workDayList)])))
personalWorkDayDict.append([workers[workIndex], group.groupName, weeklyWorkDayList])
personalTotalWorkDay = Scheduler.calculateWorkDayPerWorker(calendar)
for i in range(len(workers)):
personalWorkDayLengthList.append(
[workers[i], group.groupName, str(personalTotalWorkDay.get(i, 0)),
str(personalTotalWorkDay.get(i, 0) * int(group.workHour))])
# # 添加平均数行
# avgWorkDayNum = math.ceil(float(len(calendar.keys())) * int(
# group.workLoad) / len(
# workers))
# avgWorkHour = avgWorkDayNum * int(group.workHour)
# workDayData.append([u'平均出勤天数', str(avgWorkDayNum), str(avgWorkHour)])
for keyValuePair in sorted(calendar.items(), key=lambda d: int(d[0])):
currentDate = TimeUtil.getFormatedDate(startDate, int(keyValuePair[0]) - 1)
currentDateArrange = dailyAttendenceDict.get(currentDate, list())
currentDateArrange.append(
[", ".join(map(str, map(lambda index: workers[index], keyValuePair[1]))), group])
dailyAttendenceDict[currentDate] = currentDateArrange
except Exception as e:
wx.MessageBox(u'出错啦: ' + str(e))
# 按日期排序
dailyAttendenceList = sorted(dailyAttendenceDict.items(), key=lambda d: d[0])
self.exportData = [personalWorkDayDict, dailyAttendenceList, personalWorkDayLengthList, startDate]
self.updateGrid(personalWorkDayLengthList)
self.updateGrid1(personalWorkDayDict, startDate)
self.updateGrid2(dailyAttendenceList)
def updateGrid1(self, rows, startDate):
self.grid1.ClearGrid()
currentRowNum = self.grid1.GetNumberRows()
if currentRowNum < len(rows):
self.grid1.AppendRows(numRows=(len(rows) - currentRowNum))
elif currentRowNum > len(rows):
self.grid1.DeleteRows(numRows=(currentRowNum - len(rows)))
totalWeekNum = len(rows[0][2]) if len(rows) > 0 else 0
currentColNum = self.grid1.GetNumberCols() - 2
if currentColNum < totalWeekNum:
self.grid1.AppendCols(numCols=(totalWeekNum - currentColNum))
elif currentColNum > totalWeekNum:
self.grid1.DeleteCols(numCols=(currentColNum - totalWeekNum))
for totalWeekNum in range(1, totalWeekNum + 1):
dateStr = u'第 ' + str(totalWeekNum) + u' 周\n'
endDate = TimeUtil.getFormatedDate(startDate, 6)
dateStr += (startDate + ' -- ' + endDate)
self.grid1.SetColLabelValue(totalWeekNum + 1, dateStr)
startDate = TimeUtil.getFormatedDate(endDate, 1)
rowNum = 0
previousGroupName = ''
colorCnt = 0
for item in rows:
wokerName = item[0]
groupName = item[1]
workDayList = item[2]
if groupName != previousGroupName:
colorCnt += 1
previousGroupName = groupName
self.grid1.SetCellValue(rowNum, 0, wokerName)
self.grid1.SetCellValue(rowNum, 1, groupName)
self.grid1.SetCellBackgroundColour(rowNum, 0, self.colorList[colorCnt % 2])
self.grid1.SetCellBackgroundColour(rowNum, 1, self.colorList[colorCnt % 2])
for weekNum in range(len(workDayList)):
self.grid1.SetCellValue(rowNum, weekNum + 2, ', '.join(list(map(str, workDayList[weekNum]))))
self.grid1.SetCellBackgroundColour(rowNum, weekNum + 2, self.colorList[colorCnt % 2])
rowNum += 1
self.grid1.AutoSize()
self.vBox.Layout()
def setupDateInput(self):
optionSizer = wx.GridBagSizer(4, 4)
optionOuterSizer = wx.StaticBoxSizer(wx.HORIZONTAL, self, label=u'选项')
userGroupText = wx.StaticText(self, label=u'排班班组')
optionSizer.Add(userGroupText, pos=(0, 0), flag=wx.EXPAND | wx.TOP | wx.LEFT, border=15)
self.userGroupDropDown = wx.ListBox(self, style=wx.LB_MULTIPLE, choices=self.loadGroupList(), size=(100, 50))
self.userGroupDropDown.Bind(wx.EVT_LISTBOX, self.onSelection)
optionSizer.Add(self.userGroupDropDown, pos=(0, 1), flag=wx.EXPAND | wx.TOP | wx.LEFT, border=15)
minWorkDays = wx.StaticText(self, label=u'最大连续休息天数')
optionSizer.Add(minWorkDays, pos=(1, 0), flag=wx.EXPAND | wx.TOP | wx.LEFT, border=15)
self.maxRestDaysInput = wx.TextCtrl(self, value='3', style=wx.TE_PROCESS_ENTER)
optionSizer.Add(self.maxRestDaysInput, pos=(1, 1),
flag=wx.TOP | wx.LEFT, border=12)
maxWorkDaysText = wx.StaticText(self, label=u'最大连续工作天数')
optionSizer.Add(maxWorkDaysText, pos=(1, 2), flag=wx.EXPAND | wx.TOP | wx.LEFT, border=15)
self.maxWorkDaysInput = wx.TextCtrl(self, value='6', style=wx.TE_PROCESS_ENTER)
optionSizer.Add(self.maxWorkDaysInput, pos=(1, 3),
flag=wx.TOP | wx.LEFT, border=12)
startDate = wx.StaticText(self, label=u'开始日期')
optionSizer.Add(startDate, pos=(2, 0), flag=wx.EXPAND | wx.TOP | wx.LEFT, border=15)
self.startDateInput = wx.TextCtrl(self, value=TimeUtil.getMonday(TimeUtil.getToday()),
style=wx.TE_PROCESS_ENTER)
self.Bind(wx.EVT_TEXT, self.OnEnterDate, self.startDateInput)
optionSizer.Add(self.startDateInput, pos=(2, 1),
flag=wx.TOP | wx.LEFT, border=12)
endDate = wx.StaticText(self, label=u'结束日期')
optionSizer.Add(endDate, pos=(2, 2), flag=wx.EXPAND | wx.TOP | wx.LEFT, border=15)
self.endDateInput = wx.TextCtrl(self,
value=TimeUtil.getSunday(
TimeUtil.getFormatedDate(self.startDateInput.GetValue(), 55)),
style=wx.TE_PROCESS_ENTER)
self.Bind(wx.EVT_TEXT, self.OnEnterDate, self.endDateInput)
optionSizer.Add(self.endDateInput, pos=(2, 3),
flag=wx.TOP | wx.LEFT, border=12)
blankMsg = wx.StaticText(self, label=u' ')
optionSizer.Add(blankMsg, pos=(3, 0), flag=wx.TOP | wx.LEFT, border=15)
self.warnMsg = wx.StaticText(self, label=u'非法日期，请重新输入')
self.warnMsg.SetForegroundColour('red')
optionSizer.Add(self.warnMsg, pos=(3, 1), span=(1, 3), flag=wx.TOP | wx.LEFT, border=15)
self.warnMsg.Hide()
optionOuterSizer.Add(optionSizer)
fontBtn = wx.Font(13, wx.FONTFAMILY_MODERN, wx.NORMAL, wx.FONTWEIGHT_BOLD)
btnOuterSizer = wx.StaticBoxSizer(wx.VERTICAL, self, u'操作')
btnSizer = wx.GridBagSizer(4, 4)
self.scheduleBtn = wx.Button(self, label=u'开始\n排班', size=(80, 66))
self.scheduleBtn.SetFont(fontBtn)
btnSizer.Add(self.scheduleBtn, pos=(0, 0), flag=wx.LEFT | wx.RIGHT, border=12)
self.scheduleBtn.Enable(False)
self.Bind(wx.EVT_BUTTON, self.onSchedule, self.scheduleBtn)
self.exportBtn = wx.Button(self, label=u'导出\n排班', size=(80, 66))
self.exportBtn.SetFont(fontBtn)
btnSizer.Add(self.exportBtn, pos=(1, 0), flag=wx.BOTTOM | wx.LEFT | wx.RIGHT, border=12)
self.exportBtn.Enable(False)
self.Bind(wx.EVT_BUTTON, self.onExport, self.exportBtn)
self.convertDisplayBtn = wx.Button(self, label=u'切换\n排班\n格式', size=(80, 66))
self.convertDisplayBtn.SetFont(fontBtn)
btnSizer.Add(self.convertDisplayBtn, pos=(0, 2), span=(0, 1), flag=wx.BOTTOM | wx.LEFT | wx.RIGHT, border=12)
self.Bind(wx.EVT_BUTTON, self.onConvertDisplay, self.convertDisplayBtn)
self.saveBtn = wx.Button(self, label=u'保存\n排班', size=(80, 66))
self.saveBtn.SetFont(fontBtn)
btnSizer.Add(self.saveBtn, pos=(0, 1), flag=wx.BOTTOM | wx.LEFT | wx.RIGHT, border=12)
self.saveBtn.Enable(False)
self.Bind(wx.EVT_BUTTON, self.onSaveCalendar, self.saveBtn)
self.loadBtn = wx.Button(self, label=u'加载\n排班', size=(80, 66))
self.loadBtn.SetFont(fontBtn)
btnSizer.Add(self.loadBtn, pos=(1, 1), flag=wx.BOTTOM | wx.LEFT | wx.RIGHT, border=12)
self.Bind(wx.EVT_BUTTON, self.onLoadCalendar, self.loadBtn)
self.deleteBtn = wx.Button(self, label=u'删除\n排班', size=(80, 66))
self.deleteBtn.SetFont(fontBtn)
btnSizer.Add(self.deleteBtn, pos=(1, 2), flag=wx.BOTTOM | wx.LEFT | wx.RIGHT, border=12)
self.Bind(wx.EVT_BUTTON, self.onDeleteCalendar, self.deleteBtn)
btnOuterSizer.Add(btnSizer)
outerSizer = wx.BoxSizer(orient=wx.HORIZONTAL)
outerSizer.Add(optionOuterSizer)
outerSizer.AddSpacer(15)
outerSizer.Add(btnOuterSizer)
self.vBox.Add(outerSizer, wx.ALIGN_TOP | wx.ALIGN_LEFT, 10)
