class MainWindow(QWidget):
"""The main window of the program"""
def __init__(self, testing=False):
# Initialize object using ui_mainwindow
super(MainWindow, self).__init__()
self.window = QMainWindow()
self.ui = Ui_MainWindow()
self.ui.setupUi(self.window)
self.connectSlots()
self.testing = testing
self.scheduler = Scheduler()
def connectSlots(self):
"""connect the various ui signals to their slots"""
self.ui.addSoloParticipantBtn.clicked.connect(self.addSoloParticipantBtn_clicked)
self.ui.addGroupParticipantBtn.clicked.connect(self.addGroupParticipantBtn_clicked)
self.ui.addTeacherBtn.clicked.connect(self.addTeacherBtn_clicked)
self.ui.addEntryBtn.clicked.connect(self.addEntryBtn_clicked)
self.ui.makeScheduleBtn.clicked.connect(self.makeScheduleBtn_clicked)
self.ui.loadScheduleBtn.clicked.connect(self.loadScheduleBtn_clicked)
self.ui.editParticipantBtn.clicked.connect(self.editParticipantBtn_clicked)
self.ui.editTeacherBtn.clicked.connect(self.editTeacherBtn_clicked)
self.ui.editEntryBtn.clicked.connect(self.editEntryBtn_clicked)
self.ui.deleteParticipantBtn.clicked.connect(self.deleteParticipantBtn_clicked)
self.ui.deleteTeacherBtn.clicked.connect(self.deleteTeacherBtn_clicked)
self.ui.deleteEntryBtn.clicked.connect(self.deleteEntryBtn_clicked)
self.ui.backupDbBtn.clicked.connect(self.backupDbBtn_clicked)
self.ui.restoreDbBtn.clicked.connect(self.restoreDbBtn_clicked)
self.ui.createNewDbBtn.clicked.connect(self.createNewDbBtn_clicked)
self.ui.entriesByDisciplineBtn.clicked.connect(self.entriesByDisciplineBtn_clicked)
self.ui.entriesByTeacherBtn.clicked.connect(self.entriesByTeacherBtn_clicked)
# self.ui.entriesByGroupBtn.clicked.connect(self.entriesByGroupBtn_clicked)
self.ui.dumpBtn.clicked.connect(self.dumpBtn_clicked)
###### Slots ######
def addSoloParticipantBtn_clicked(self):
dialog = AddSoloParticipantDialog(testing=self.testing)
# For Modal dialog
dialog.exec_()
def addGroupParticipantBtn_clicked(self):
dialog = AddGroupParticipantDialog(testing=self.testing)
# For Modal dialog
dialog.exec_()
def addTeacherBtn_clicked(self):
dialog = AddTeacherDialog(testing=self.testing)
# For Modal dialog
dialog.exec_()
def addEntryBtn_clicked(self):
dialog = AddEntryDialog(testing=self.testing)
# For Modal dialog
dialog.exec_()
def makeScheduleBtn_clicked(self):
### Test Code ###
# s1Start = datetime.datetime(2014, 4, 7, 9)
# s1End = datetime.datetime(2014, 4, 7, 12)
# s2Start = datetime.datetime(2014, 4, 7, 13)
# s2End = datetime.datetime(2014, 4, 7, 17)
# s3Start = datetime.datetime(2014, 4, 7, 18)
# s3End = datetime.datetime(2014, 4, 7, 21)
# sessionDatetimes = [(s1Start, s1End), (s2Start, s2End), (s3Start, s3End)]
####
scheduleOptionsDialog = ScheduleOptionsDialog()
result = scheduleOptionsDialog.exec_()
if result == True:
entries = dbInteractionInstance.getAllEntriesInDiscipline(settingsInteractionInstance.loadDiscipline())
solution = self.scheduler.process(entries, settingsInteractionInstance.loadSessionDatetimes())
print solution
if solution is None:
QMessageBox.warning(self, 'No schedule found', 'No schedule was found for the specified parameters. Try increasing tolerance for overtime or making the sessions longer.',
QMessageBox.Ok)
return
dialog = ScheduleDialog(schedule=solution)
result = dialog.exec_()
if result == True:
pass
def loadScheduleBtn_clicked(self):
"""Loads a schedule from file"""
schedule = Schedule()
filename = QFileDialog.getOpenFileName(self, "Load Schedule", exportsPath, "Schedule Files (*.sched)")
if filename is not None and filename != "":
try:
schedule.load(filename)
dialog = ScheduleDialog(schedule=schedule)
result = dialog.exec_()
if result == True:
pass
except Exception:
QMessageBox.critical(self, "Error", "Failed to load schedule.", QMessageBox.Ok)
def editParticipantBtn_clicked(self):
"""Opens chooseParticipantDialog then dialog for editing"""
dialog = ChooseParticipantDialog()
# For Modal dialog
result = dialog.exec_()
if result == True:
participantId = dialog.getParticipantId()
# Open appropriate edit dialog with participant
if participantId[0] == 's':
dialog = EditSoloParticipantDialog(participantId=participantId)
dialog.exec_()
elif participantId[0] == 'g':
dialog = EditGroupParticipantDialog(participantId=participantId)
dialog.exec_()
else:
QMessageBox.critical(self, "Error", "Unrecognized Participant", QMessageBox.Ok)
def editTeacherBtn_clicked(self):
"""Opens chooseTeacherDialog then dialog for editing"""
dialog = ChooseTeacherDialog()
# For Modal dialog
result = dialog.exec_()
if result == True:
teacherId = dialog.getTeacherId()
# Open edit dialog with teacher
dialog = EditTeacherDialog(teacherId=teacherId)
dialog.exec_()
def editEntryBtn_clicked(self):
"""Opens chooseEntryDialog then dialog for editing"""
dialog = ChooseEntryDialog()
# For Modal dialog
result = dialog.exec_()
if result == True:
entryId = dialog.entryId
# Open edit dialog with entry
dialog = EditEntryDialog(entryId=entryId)
dialog.exec_()
def editParticipantBtn_clicked(self):
"""Opens chooseParticipantDialog then dialog for editing"""
dialog = ChooseParticipantDialog()
# For Modal dialog
result = dialog.exec_()
if result == True:
participantId = dialog.getParticipantId()
# Open appropriate edit dialog with participant
if participantId[0] == 's':
dialog = EditSoloParticipantDialog(participantId=participantId)
dialog.exec_()
elif participantId[0] == 'g':
dialog = EditGroupParticipantDialog(participantId=participantId)
dialog.exec_()
else:
QMessageBox.critical(self, "Error", "Unrecognized Participant", QMessageBox.Ok)
def deleteTeacherBtn_clicked(self):
"""Opens chooseTeacherDialog for deleting"""
dialog = ChooseTeacherDialog()
# For Modal dialog
result = dialog.exec_()
if result == True:
teacherId = dialog.getTeacherId()
if QMessageBox.question(self, "Cannot be undone!", "Warning! This action cannot be undone. \nAre you sure you want to delete this teacher/contact?", QMessageBox.Yes|QMessageBox.No) == QMessageBox.Yes:
# Delete the teacher
dbInteractionInstance.deleteTeacherFromId(teacherId)
def deleteEntryBtn_clicked(self):
"""Opens chooseEntryDialog for deleting"""
dialog = ChooseEntryDialog()
# For Modal dialog
result = dialog.exec_()
if result == True:
entryId = dialog.entryId
if QMessageBox.question(self, "Cannot be undone!", "Warning! This action cannot be undone. \nAre you sure you want to delete this entry?", QMessageBox.Yes|QMessageBox.No) == QMessageBox.Yes:
# Delete the entry
dbInteractionInstance.deleteEntryFromId(entryId)
def deleteParticipantBtn_clicked(self):
"""Opens chooseParticipantDialog for deleting"""
dialog = ChooseParticipantDialog()
# For Modal dialog
result = dialog.exec_()
if result == True:
participantId = dialog.getParticipantId()
if QMessageBox.question(self, "Cannot be undone!", "Warning! This action cannot be undone. \nAre you sure you want to delete this participant?", QMessageBox.Yes|QMessageBox.No) == QMessageBox.Yes:
# Call appropriate delete
if participantId[0] == 's':
dbInteractionInstance.deleteSoloParticipantFromId(participantId[1:])
elif participantId[0] == 'g':
dbInteractionInstance.deleteGroupParticipantFromId(participantId[1:])
else:
QMessageBox.critical(self, "Error", "Unrecognized Participant", QMessageBox.Ok)
def backupDbBtn_clicked(self):
"""Copies the current db file to AFS-YYYY-MM-DD-HH-MM-SS.bak"""
result = dbInteractionInstance.backupDb()
if result == "":
QMessageBox.information(self, "Success", "Database successfully backed up.", QMessageBox.Ok)
else:
QMessageBox.critical(self, "Failed", "Could not create backup. Aborted with error:\n{0}".format(result))
def restoreDbBtn_clicked(self):
"""Grabs the most recent backup, backs up the current db, copies recent backup to current db"""
result = dbInteractionInstance.restoreDb()
if result == "":
QMessageBox.information(self, "Success", "Database successfully restored.", QMessageBox.Ok)
else:
QMessageBox.critical(self, "Failed", "Could not restore from backup. Aborted with error:\n{0}".format(result))
def createNewDbBtn_clicked(self):
"""Backs up current db, then drops and recreates all the tables"""
# TODO low priority, probably don't need it until after the festival
pass
def entriesByDisciplineBtn_clicked(self):
"""Saves a csv of all the entries sorted by Discipline"""
filename = QFileDialog.getSaveFileName(self, "Report Entries by Discipline", exportsPath, "CSV Files (*.csv)")
if filename is not None and filename != "":
if filename[-4:] != ".csv":
filename += ".csv"
entries = dbInteractionInstance.getAllEntries()
# Get participant and teacher names for entries
# Note super bad hack where I replace the ID with a name
for entry in entries:
participant = dbInteractionInstance.getParticipantFromId(entry.participantID)
try:
entry.participantID = "{last}, {first}".format(last=participant.last, first=participant.first)
except Exception:
entry.participantID = "{groupName}".format(groupName=participant.groupName)
if entry.teacherID != "":
teacher = dbInteractionInstance.getTeacherFromId(entry.teacherID)
entry.teacherID = "{last}, {first}".format(last=teacher.last, first=teacher.first)
entries.sort(key=lambda x: (x.discipline, x.classNumber, x.participantID))
fout = open(filename, 'w')
fout.write(Entry.reportByDisciplineHeader())
for entry in entries:
entry.reportByDiscipline(fout)
fout.close()
QMessageBox.information(self, 'Report Entries by Discipline', 'Report saved to ' + filename, QMessageBox.Ok)
def entriesByTeacherBtn_clicked(self):
"""Saves a csv of all the entries sorted by Teacher"""
filename = QFileDialog.getSaveFileName(self, "Report Entries by Teacher", exportsPath, "CSV Files (*.csv)")
if filename is not None and filename != "":
if filename[-4:] != ".csv":
filename += ".csv"
entries = dbInteractionInstance.getAllEntries()
# Get participant and teacher names for entries
# Note super bad hack where I replace the ID with a name
for entry in entries:
participant = dbInteractionInstance.getParticipantFromId(entry.participantID)
try:
entry.participantID = "{last}, {first}".format(last=participant.last, first=participant.first)
except Exception:
entry.participantID = "{groupName}".format(groupName=participant.groupName)
if entry.teacherID != "":
teacher = dbInteractionInstance.getTeacherFromId(entry.teacherID)
entry.teacherID = "{last}, {first}".format(last=teacher.last, first=teacher.first)
entries.sort(key=lambda x: (x.teacherID, x.participantID, x.discipline, x.classNumber))
fout = open(filename, 'w')
fout.write(Entry.reportByTeacherHeader())
for entry in entries:
entry.reportByTeacher(fout)
fout.close()
QMessageBox.information(self, 'Report Entries by Teacher', 'Report saved to ' + filename, QMessageBox.Ok)
def entriesByGroupBtn_clicked(self):
"""Saves a csv of all the entries sorted by Group"""
filename = QFileDialog.getSaveFileName(self, "Report Entries by School/Group", exportsPath, "CSV Files (*.csv)")
if filename is not None and filename != "":
if filename[-4:] != ".csv":
filename += ".csv"
entries = dbInteractionInstance.getAllEntries()
# TODO sort by school then group name (for group participants only)
# entries.sort(key=lambda x: (x.discipline, x.classNumber))
for entry in entries:
print entry
# TODO write csv
QMessageBox.information(self, 'Report Entries by School/Group', 'Report saved to ' + filename, QMessageBox.Ok)
def dumpBtn_clicked(self):
"""Saves a csv of all the data so they can do what they want with it"""
filename = QFileDialog.getSaveFileName(self, "Database Dump", exportsPath, "CSV Files (*.csv)")
if filename is not None and filename != "":
if filename[-4:] != ".csv":
filename += ".csv"
entries = dbInteractionInstance.getAllEntries()
fout = open(filename, 'w')
fout.write(Entry.dumpHeader())
for entry in entries:
entry.dump(fout)
fout.close()
QMessageBox.information(self, 'Database Dump', 'Data saved to ' + filename, QMessageBox.Ok)
##########
def run(self):
# Show the form
self.window.show()
# Run the application
app.exec_()
class MainWindow(QWidget):
"""The main window of the program"""
def __init__(self, testing=False):
# Initialize object using ui_mainwindow
super(MainWindow, self).__init__()
self.window = QMainWindow()
self.ui = Ui_MainWindow()
self.ui.setupUi(self.window)
self.connectSlots()
self.testing = testing
self.scheduler = Scheduler()
def connectSlots(self):
"""connect the various ui signals to their slots"""
self.ui.addParticipantBtn.clicked.connect(self.addParticipantBtn_clicked)
self.ui.addTeacherBtn.clicked.connect(self.addTeacherBtn_clicked)
self.ui.addEntryBtn.clicked.connect(self.addEntryBtn_clicked)
self.ui.runSchedulerAction.triggered.connect(self.runSchedulerAction_triggered)
self.ui.loadScheduleAction.triggered.connect(self.loadScheduleAction_triggered)
self.ui.editParticipantBtn.clicked.connect(self.editParticipantBtn_clicked)
self.ui.editTeacherBtn.clicked.connect(self.editTeacherBtn_clicked)
self.ui.editEntryBtn.clicked.connect(self.editEntryBtn_clicked)
self.ui.deleteParticipantBtn.clicked.connect(self.deleteParticipantBtn_clicked)
self.ui.deleteTeacherBtn.clicked.connect(self.deleteTeacherBtn_clicked)
self.ui.deleteEntryBtn.clicked.connect(self.deleteEntryBtn_clicked)
self.ui.createBackupAction.triggered.connect(self.createBackupAction_triggered)
# self.ui.restoreDbBtn.clicked.connect(self.restoreDbBtn_clicked)
# self.ui.createNewDbBtn.clicked.connect(self.createNewDbBtn_clicked)
self.ui.createNewDatabaseAction.triggered.connect(self.createNewDatabaseAction_triggered)
self.ui.exportAction.triggered.connect(self.exportAction_triggered)
self.ui.participantTableView.setModel(dbInteractionInstance.participantModel)
self.ui.participantTableView.model().sort(8, Qt.DescendingOrder)
self.ui.participantTableView.setSelectionModel(QItemSelectionModel(dbInteractionInstance.participantModel))
self.ui.participantTableView.setSelectionBehavior(QAbstractItemView.SelectRows)
self.ui.participantTableView.setEditTriggers(QAbstractItemView.NoEditTriggers)
self.ui.participantTableView.hideColumn(8) # id
self.ui.teacherTableView.setModel(dbInteractionInstance.teacherModel)
self.ui.teacherTableView.model().sort(0, Qt.DescendingOrder)
self.ui.teacherTableView.setSelectionModel(QItemSelectionModel(dbInteractionInstance.teacherModel))
self.ui.teacherTableView.setSelectionBehavior(QAbstractItemView.SelectRows)
self.ui.teacherTableView.setEditTriggers(QAbstractItemView.NoEditTriggers)
self.ui.teacherTableView.hideColumn(0) # id
self.ui.entryTableView.setModel(dbInteractionInstance.entryModel)
self.ui.entryTableView.model().sort(0, Qt.DescendingOrder)
self.ui.entryTableView.setSelectionModel(QItemSelectionModel(dbInteractionInstance.entryModel))
self.ui.entryTableView.setSelectionBehavior(QAbstractItemView.SelectRows)
self.ui.entryTableView.setEditTriggers(QAbstractItemView.NoEditTriggers)
self.ui.entryTableView.hideColumn(0) # id
self.ui.pieceTableView.setModel(dbInteractionInstance.pieceModel)
self.ui.pieceTableView.model().sort(0, Qt.DescendingOrder)
self.ui.pieceTableView.setSelectionModel(QItemSelectionModel(dbInteractionInstance.pieceModel))
self.ui.pieceTableView.setSelectionBehavior(QAbstractItemView.SelectRows)
self.ui.pieceTableView.setEditTriggers(QAbstractItemView.NoEditTriggers)
self.ui.pieceTableView.hideColumn(0) # id
self.ui.pieceTableView.hideColumn(4) # entryId
def getSelectedId(self, model, view):
"""Get the id for the selected row given the model and view"""
try:
# Get which row in view is selected
indexList = view.selectionModel().selectedIndexes() # gets all selected cells
row = indexList[0].row()
# Get column id is in
column = model.fieldIndex("id")
# Get QModelIndex of id
modelIndex = [i for i in indexList if i.row() == row and i.column() == column][0]
# Get id of selected object
data = model.data(modelIndex)
return str(data.toString())
except Exception:
print traceback.format_exc()
QMessageBox.critical(self, 'No Row Selected', 'Please select a row.', QMessageBox.Ok)
###### Slots ######
def addParticipantBtn_clicked(self):
dialog = AddParticipantDialog(testing=self.testing)
# For Modal dialog
dialog.exec_()
def addTeacherBtn_clicked(self):
dialog = AddTeacherDialog(testing=self.testing)
# For Modal dialog
dialog.exec_()
def addEntryBtn_clicked(self):
dialog = AddEntryDialog(testing=self.testing)
# For Modal dialog
dialog.exec_()
def runSchedulerAction_triggered(self):
### Test Code ###
# s1Start = datetime.datetime(2014, 4, 7, 9)
# s1End = datetime.datetime(2014, 4, 7, 12)
# s2Start = datetime.datetime(2014, 4, 7, 13)
# s2End = datetime.datetime(2014, 4, 7, 17)
# s3Start = datetime.datetime(2014, 4, 7, 18)
# s3End = datetime.datetime(2014, 4, 7, 21)
# sessionDatetimes = [(s1Start, s1End), (s2Start, s2End), (s3Start, s3End)]
####
scheduleOptionsDialog = ScheduleOptionsDialog()
result = scheduleOptionsDialog.exec_()
if result == True:
entries = dbInteractionInstance.getAllEntriesInDiscipline(settingsInteractionInstance.loadDiscipline())
solution = self.scheduler.process(entries, settingsInteractionInstance.loadSessionDatetimes())
print solution
if solution is None:
QMessageBox.warning(self, 'No schedule found', 'No schedule was found for the specified parameters. Try increasing tolerance for overtime or making the sessions longer.',
QMessageBox.Ok)
return
dialog = ScheduleDialog(schedule=solution)
result = dialog.exec_()
if result == True:
pass
def loadScheduleAction_triggered(self):
"""Loads a schedule from file"""
schedule = Schedule()
filename = QFileDialog.getOpenFileName(self, "Load Schedule", exportsPath, "Schedule Files (*.sched)")
if filename is not None and filename != "":
try:
schedule.load(filename)
dialog = ScheduleDialog(schedule=schedule)
result = dialog.exec_()
if result == True:
pass
except Exception:
QMessageBox.critical(self, "Error", "Failed to load schedule.", QMessageBox.Ok)
def editParticipantBtn_clicked(self):
"""Opens the dialog for editing"""
participantId = self.getSelectedId(model=dbInteractionInstance.participantModel, view=self.ui.participantTableView)
if participantId:
dialog = EditParticipantDialog(participantId=participantId)
# For Modal dialog
dialog.exec_()
def editTeacherBtn_clicked(self):
"""Opens dialog for editing"""
teacherId = self.getSelectedId(model=dbInteractionInstance.teacherModel, view=self.ui.teacherTableView)
if teacherId:
dialog = EditTeacherDialog(teacherId=teacherId)
# For Modal dialog
dialog.exec_()
def editEntryBtn_clicked(self):
"""Opens dialog for editing"""
entryId = self.getSelectedId(model=dbInteractionInstance.entryModel, view=self.ui.entryTableView)
if entryId:
dialog = EditEntryDialog(entryId=entryId)
# For Modal dialog
dialog.exec_()
def deleteTeacherBtn_clicked(self):
"""Deletes selected teacher"""
teacherId = self.getSelectedId(model=dbInteractionInstance.teacherModel, view=self.ui.teacherTableView)
if QMessageBox.question(self, "Cannot be undone!", "Warning! This action cannot be undone. \nAre you sure you want to delete this teacher/contact?", QMessageBox.Yes|QMessageBox.No) == QMessageBox.Yes:
# Delete the teacher
dbInteractionInstance.deleteTeacherFromId(teacherId)
def deleteEntryBtn_clicked(self):
"""Deletes selected entry"""
entryId = self.getSelectedId(model=dbInteractionInstance.entryModel, view=self.ui.entryTableView)
if QMessageBox.question(self, "Cannot be undone!", "Warning! This action cannot be undone. \nAre you sure you want to delete this entry?", QMessageBox.Yes|QMessageBox.No) == QMessageBox.Yes:
# Delete the entry
dbInteractionInstance.deleteEntryFromId(entryId)
def deleteParticipantBtn_clicked(self):
"""Deletes selected participant"""
participantId = self.getSelectedId(model=dbInteractionInstance.participantModel, view=self.ui.participantTableView)
if QMessageBox.question(self, "Cannot be undone!", "Warning! This action cannot be undone. \nAre you sure you want to delete this participant?", QMessageBox.Yes|QMessageBox.No) == QMessageBox.Yes:
# Delete the participant
dbInteractionInstance.deleteParticipantFromId(participantId)
def createBackupAction_triggered(self):
"""Copies the current db file to AFS-YYYY-MM-DD-HH-MM-SS.bak"""
result = dbInteractionInstance.backupDb()
if result == "":
QMessageBox.information(self, "Success", "Database successfully backed up.", QMessageBox.Ok)
else:
QMessageBox.critical(self, "Failed", "Could not create backup. Aborted with error:\n{0}".format(result))
def restoreDbBtn_clicked(self):
"""Grabs the most recent backup, backs up the current db, copies recent backup to current db"""
result = dbInteractionInstance.restoreDb()
if result == "":
QMessageBox.information(self, "Success", "Database successfully restored.", QMessageBox.Ok)
else:
QMessageBox.critical(self, "Failed", "Could not restore from backup. Aborted with error:\n{0}".format(result))
# def createNewDbBtn_clicked(self):
# """Backs up current db, then drops and recreates all the tables"""
# # TODO low priority, probably don't need it until after the festival
# pass
def createNewDatabaseAction_triggered(self):
"""Backs up current db, then drops all records"""
if QMessageBox.question(self, "Create new database", "This will create a backup and then clear the current database. \nAre you sure you want to continue?", QMessageBox.Yes|QMessageBox.No) == QMessageBox.Yes:
self.createBackupAction_triggered()
result = dbInteractionInstance.cleanDb()
if result == "":
QMessageBox.information(self, "Success", "New Database created.", QMessageBox.Ok)
else:
QMessageBox.critical(self, "Failed", "Could not create new database. Aborted with error:\n{0}".format(result))
def entriesByDisciplineBtn_clicked(self):
"""Saves a csv of all the entries sorted by Discipline"""
filename = QFileDialog.getSaveFileName(self, "Report Entries by Discipline", exportsPath, "CSV Files (*.csv)")
if filename is not None and filename != "":
if filename[-4:] != ".csv":
filename += ".csv"
entries = dbInteractionInstance.getAllEntries()
# Get participant and teacher names for entries
# Note super bad hack where I replace the ID with a name
for entry in entries:
participant = dbInteractionInstance.getParticipantFromId(entry.participantID)
try:
entry.participantID = "{last}, {first}".format(last=participant.last, first=participant.first)
except Exception:
entry.participantID = "{groupName}".format(groupName=participant.groupName)
if entry.teacherID != "":
teacher = dbInteractionInstance.getTeacherFromId(entry.teacherID)
entry.teacherID = "{last}, {first}".format(last=teacher.last, first=teacher.first)
entries.sort(key=lambda x: (x.discipline, x.classNumber, x.participantID))
fout = open(filename, 'w')
fout.write(Entry.reportByDisciplineHeader())
for entry in entries:
entry.reportByDiscipline(fout)
fout.close()
QMessageBox.information(self, 'Report Entries by Discipline', 'Report saved to ' + filename, QMessageBox.Ok)
def entriesByTeacherBtn_clicked(self):
"""Saves a csv of all the entries sorted by Teacher"""
filename = QFileDialog.getSaveFileName(self, "Report Entries by Teacher", exportsPath, "CSV Files (*.csv)")
if filename is not None and filename != "":
if filename[-4:] != ".csv":
filename += ".csv"
entries = dbInteractionInstance.getAllEntries()
# Get participant and teacher names for entries
# Note super bad hack where I replace the ID with a name
for entry in entries:
participant = dbInteractionInstance.getParticipantFromId(entry.participantID)
try:
entry.participantID = "{last}, {first}".format(last=participant.last, first=participant.first)
except Exception:
entry.participantID = "{groupName}".format(groupName=participant.groupName)
if entry.teacherID != "":
teacher = dbInteractionInstance.getTeacherFromId(entry.teacherID)
entry.teacherID = "{last}, {first}".format(last=teacher.last, first=teacher.first)
entries.sort(key=lambda x: (x.teacherID, x.participantID, x.discipline, x.classNumber))
fout = open(filename, 'w')
fout.write(Entry.reportByTeacherHeader())
for entry in entries:
entry.reportByTeacher(fout)
fout.close()
QMessageBox.information(self, 'Report Entries by Teacher', 'Report saved to ' + filename, QMessageBox.Ok)
def entriesByGroupBtn_clicked(self):
"""Saves a csv of all the entries sorted by Group"""
filename = QFileDialog.getSaveFileName(self, "Report Entries by School/Group", exportsPath, "CSV Files (*.csv)")
if filename is not None and filename != "":
if filename[-4:] != ".csv":
filename += ".csv"
entries = dbInteractionInstance.getAllEntries()
# TODO sort by school then group name (for group participants only)
# entries.sort(key=lambda x: (x.discipline, x.classNumber))
for entry in entries:
print entry
# TODO write csv
QMessageBox.information(self, 'Report Entries by School/Group', 'Report saved to ' + filename, QMessageBox.Ok)
def exportAction_triggered(self):
"""Saves a csv of all the data so they can do what they want with it"""
filename = QFileDialog.getSaveFileName(self, "Database Dump", exportsPath, "CSV Files (*.csv)")
if filename is not None and filename != "":
if filename[-4:] != ".csv":
filename += ".csv"
entries = dbInteractionInstance.getAllEntries()
fout = open(filename, 'w')
fout.write(Entry.dumpHeader())
for entry in entries:
entry.dump(fout)
fout.close()
QMessageBox.information(self, 'Database Dump', 'Data saved to ' + filename, QMessageBox.Ok)
##########
def run(self):
# Show the form
self.window.show()
# Run the application
app.exec_()
