class ImportArrangementDialog(QDialog):
list = []
def __init__(self):
super(ImportArrangementDialog, self).__init__()
self.init()
self.loading()
def init(self):
self.initUI()
self.initInteraction()
self.initSQL()
self.initOther()
def initUI(self):
self.ui = Ui_ImportArrangementDialog()
self.ui.setupUi(self)
def initInteraction(self):
self.ui.dialogNavClose.clicked.connect(self.dialogNavCloseClicked)
self.ui.pickFile.clicked.connect(self.pickFileClicked)
self.ui.cancel.clicked.connect(self.cancelClicked)
self.ui.comfirm.clicked.connect(self.comfirmClicked)
self.ui.periodSelection.currentTextChanged.connect(
self.periodSelectionChanged)
def initSQL(self):
self.SQLHelper = SQLHelper()
def initOther(self):
self.moveFlag = False
def loading(self):
self.loadingPeriod()
self.loadingDate()
def loadingPeriod(self):
periods = self.SQLHelper.getPeriodNotVacation()
data = []
for period in periods:
data.append(period[1])
self.ui.loadingPeriodSelection(data)
def loadingDate(self):
periodName = self.ui.periodSelection.currentText()
period = self.SQLHelper.getPeriodByName(periodName)
startDate = datetime.strptime(period[0][2], "%Y-%m-%d")
start = QDate(startDate.year, startDate.month, startDate.day)
endDate = datetime.strptime(period[0][3], "%Y-%m-%d")
end = QDate(endDate.year, endDate.month, endDate.day)
self.ui.loadingPeriodDate(start, end)
def dialogNavCloseClicked(self):
self.SQLHelper.close()
self.reject()
def cancelClicked(self):
self.SQLHelper.close()
self.reject()
def pickFileClicked(self):
fileUrls = QFileDialog.getOpenFileUrl(self, "選取工讀生資料",
QUrl(os.getcwd()),
"Excel檔案 (*.xlsx)")
fileUrl = fileUrls[0].toLocalFile()
if fileUrl != "":
excel = pandas.read_excel(fileUrl,
sheet_name=0,
header=None,
converters={
0: str,
1: str,
2: str,
3: str
})
self.list = excel.values.tolist()
self.ui.loadingList(self.list)
def comfirmClicked(self):
period = self.SQLHelper.getPeriodByName(
self.ui.periodSelection.currentText())[0]
start = self.ui.dateStart.date()
end = self.ui.dateEnd.date()
self.SQLHelper.updatePeriodStartAndEndById(
f"{start.year():04d}-{start.month():02d}-{start.day():02d}",
f"{end.year():04d}-{end.month():02d}-{end.day():02d}", period[0])
data = []
weekTrans = {
"星期一": 0,
"星期二": 1,
"星期三": 2,
"星期四": 3,
"星期五": 4,
"星期六": 5,
"星期日": 6
}
for item in self.list:
room = self.SQLHelper.getRoomByName(item[4])
shift = self.SQLHelper.getShiftByName(item[3])
data.append([
int(item[0]), weekTrans[item[2]], shift[0][0], room[0][0],
period[0]
])
self.SQLHelper.delArrangementByPeriod(period[0])
self.SQLHelper.newArrangements(data)
self.reloadTodaySchedules(period)
self.SQLHelper.close()
self.accept()
def periodSelectionChanged(self, text):
self.loadingDate()
def setPeriodSelection(self, period: str):
self.ui.periodSelection.setCurrentText(period)
def reloadTodaySchedules(self, updatePeriod):
now = datetime.now()
todayPeriod = self.SQLHelper.getPeriodByDate(now.strftime("%Y-%m-%d"))
if len(todayPeriod) != 0:
todayPeriod = todayPeriod[0]
if todayPeriod[0] == updatePeriod[0]:
dialog = MessageDialog("提示訊息", "請問是否重新載入今日所有已執行及未執行之班表")
if dialog.exec():
self.SQLHelper.delScheduleByDate(now.strftime("%Y-%m-%d"))
arrangements = self.SQLHelper.getArrangementByPeriodAndWeek(
todayPeriod[0], now.weekday())
self.SQLHelper.newSchedules(now.strftime("%Y-%m-%d"),
arrangements)
def mousePressEvent(self, event):
if event.button() == Qt.LeftButton:
self.moveFlag = True
self.movePosition = event.globalPos() - self.pos()
self.setCursor(QCursor(Qt.OpenHandCursor))
event.accept()
def mouseMoveEvent(self, event):
if Qt.LeftButton and self.moveFlag:
self.move(event.globalPos() - self.movePosition)
event.accept()
def mouseReleaseEvent(self, event):
self.moveFlag = False
self.setCursor(Qt.ArrowCursor)
class EditArrangementCellDialog(QDialog):
def __init__(self, arrangementId):
super(EditArrangementCellDialog, self).__init__()
self.init()
self.loading(arrangementId)
def init(self):
self.initUI()
self.initSQL()
self.initOther()
self.initInteraction()
def initUI(self):
self.ui = Ui_EditArrangementCellDialog()
self.ui.setupUi(self)
def initSQL(self):
self.SQLHelper = SQLHelper()
def initOther(self):
self.moveFlag = False
def initInteraction(self):
self.ui.cancel.clicked.connect(self.cancelClicked)
self.ui.dialogNavClose.clicked.connect(self.dialogNavCloseClicked)
self.ui.comfirm.clicked.connect(self.comfirmClicked)
def loading(self, arrangementId):
self.loadingWeek()
self.loadingShift()
self.loadingRoom()
self.loadingPeriod()
self.loadingStaff()
self.loadingArrangement(arrangementId)
def loadingWeek(self):
result = [["星期一", 0], ["星期二", 1], ["星期三", 2], ["星期四", 3], ["星期五", 4],
["星期六", 5], ["星期日", 6]]
self.ui.loadingWeek(result)
def loadingShift(self):
shifts = self.SQLHelper.getAllShift()
result = []
for shift in shifts:
result.append([shift[1], shift[0]])
self.ui.loadingShift(result)
def loadingRoom(self):
rooms = self.SQLHelper.getAllRoom()
result = []
for room in rooms:
result.append([room[1], room[0]])
self.ui.loadingRoom(result)
def loadingPeriod(self):
periods = self.SQLHelper.getPeriodNotVacation()
result = []
for period in periods:
result.append([period[1], period[0]])
self.ui.loadingPeriod(result)
def loadingStaff(self):
staffs = self.SQLHelper.getAllStaff()
result = []
for staff in staffs:
result.append([f'{staff[0]:03d}', staff[2], staff[0]])
self.ui.loadingStaff(result)
def loadingArrangement(self, arrangementId):
self.arrangementId = arrangementId
if not arrangementId is None:
arrangement = self.SQLHelper.getArrangementDetailById(
arrangementId)[0]
weekTrans = {
0: "星期一",
1: "星期二",
2: "星期三",
3: "星期四",
4: "星期五",
5: "星期六",
6: "星期日"
}
self.ui.weekSelection.setCurrentText(weekTrans[arrangement[3]])
self.ui.shiftSelection.setCurrentText(arrangement[5])
self.ui.roomSelection.setCurrentText(arrangement[7])
self.ui.periodSelection.setCurrentText(arrangement[9])
self.ui.staffSelection.setCurrentText(
f'{arrangement[1]:03d} {arrangement[2]}')
def cancelClicked(self):
self.SQLHelper.close()
self.reject()
def dialogNavCloseClicked(self):
self.SQLHelper.close()
self.reject()
def comfirmClicked(self):
week = self.ui.weekSelection.currentData(Qt.UserRole)
shiftId = self.ui.shiftSelection.currentData(Qt.UserRole)
roomId = self.ui.roomSelection.currentData(Qt.UserRole)
periodId = self.ui.periodSelection.currentData(Qt.UserRole)
staffId = self.ui.staffSelection.currentData(Qt.UserRole)
if self.arrangementId is None:
dialog = MessageDialog("提示訊息", "請確認是否新增該筆排班資料")
if dialog.exec():
self.SQLHelper.newArrangement(staffId, week, shiftId, roomId,
periodId)
self.accept()
else:
dialog = MessageDialog("提示訊息", "請確認是否修改該筆排班資料")
if dialog.exec():
self.SQLHelper.updateArrangementById(self.arrangementId,
staffId, week, shiftId,
roomId, periodId)
self.accept()
def mousePressEvent(self, event):
if event.button() == Qt.LeftButton:
self.moveFlag = True
self.movePosition = event.globalPos() - self.pos()
self.setCursor(QCursor(Qt.OpenHandCursor))
event.accept()
def mouseMoveEvent(self, event):
if Qt.LeftButton and self.moveFlag:
self.move(event.globalPos() - self.movePosition)
event.accept()
def mouseReleaseEvent(self, event):
self.moveFlag = False
self.setCursor(Qt.ArrowCursor)