class Students(QtGui.QDialog):
def __init__(self):
QtGui.QDialog.__init__(self)
# setup
self.dialog = Ui_studentsDialog()
self.dialog.setupUi(self)
self.sid = None
self.adding = False
header = ['Sid', 'Apelido', 'First Name', 'Last Name']
self.dialog.studentTableWidget.setHorizontalHeaderLabels(header)
self.connect(self.dialog.studentTableWidget, \
QtCore.SIGNAL("cellClicked(int, int)"), \
self._on_cell_click)
self.connect(self.dialog.updateButton, QtCore.SIGNAL("clicked()"), \
self._on_update_click)
self.connect(self.dialog.newButton, QtCore.SIGNAL("clicked()"), \
self._on_new_click)
self.db = AttendDB()
self._update_list()
def _update_list(self):
"""Refresh the student list."""
#for i in range(self.dialog.studentTableWidget.rowCount()):
# self.dialog.studentTableWidget.removeRow(i)
self.dialog.studentTableWidget.clear()
rows = self.dialog.studentTableWidget.rowCount()
students = self.db.get_students()
while (rows < len(students)):
self.dialog.studentTableWidget.insertRow(rows)
rows += 1
for i, student in enumerate(students):
#self.dialog.studentTableWidget.insertRow(i)
for j, item in enumerate(student):
if item == None:
item = ''
newitem = QtGui.QTableWidgetItem(str(item))
self.dialog.studentTableWidget.setItem(i, j, newitem)
if self.sid != None and str(student[0]) == str(self.sid):
self.dialog.studentTableWidget.setCurrentItem(newitem)
self.dialog.studentTableWidget.sortItems(1)
def _on_cell_click(self, row, column):
"""Fill out the entries when a cell is clicked."""
# Get selected row
items = self.dialog.studentTableWidget.selectedItems()
# Set the input fields
self.dialog.nickNameInput.setText(items[1].text())
self.dialog.firstNameInput.setText(items[2].text())
self.dialog.lastNameInput.setText(items[3].text())
def _on_update_click(self):
"""Update current student or add new student."""
# Get filled in information
apelido = str(self.dialog.nickNameInput.text())
fn = str(self.dialog.firstNameInput.text())
ln = str(self.dialog.lastNameInput.text())
if self.adding:
self.sid = self.db.add_student(apelido, fn, ln)
self.dialog.studentTableWidget.setEnabled(True)
else:
items = self.dialog.studentTableWidget.selectedItems()
self.sid = str(items[0].text())
self.db.update_student(self.sid, apelido, fn, ln)
self.adding = False
self._update_list()
def _on_new_click(self):
self.adding = True
self.dialog.studentTableWidget.setEnabled(False)
class Main(QtGui.QMainWindow):
def __init__(self):
QtGui.QMainWindow.__init__(self)
# Setup the main window
self.ui = Ui_MainWindow()
self.ui.setupUi(self)
# Set date to today
self.date = QtCore.QDate.currentDate()
self.ui.calendarWidget.setSelectedDate(self.date)
# Date format for SQLite
self.date_format = "yyyy-MM-dd"
# Used to remember the Students Id
self.student_dict = {}
# Create Attendance DB Connection Object
self.db = AttendDB()
# Create list models
self.availModel = QtGui.QStandardItemModel(0, 2)
self.attendModel = QtGui.QStandardItemModel(0, 2)
self.ui.availListView.setModel(self.availModel)
self.ui.attendListView.setModel(self.attendModel)
# Connect signals to handlers
self.connect(self.ui.addButton, QtCore.SIGNAL("clicked()"), \
self.on_add_clicked)
self.connect(self.ui.removeButton, QtCore.SIGNAL("clicked()"), \
self.on_remove_clicked)
self.connect(self.ui.calendarWidget, \
QtCore.SIGNAL("selectionChanged()"), \
self.on_date_change)
self.connect(self.ui.actionEdit_Students, \
QtCore.SIGNAL("triggered()"), \
self.on_edit_students_select)
# Update the lists to reflect the current date
self.update_views()
def on_edit_students_select(self):
"""Open the editing student window."""
edit_window = Students()
edit_window.exec_()
def on_date_change(self):
"""Update views when the date is changed."""
self.date = self.ui.calendarWidget.selectedDate()
self.update_views()
def update_views(self):
"""Refresh the students available and attending for the given date."""
# Get correct date format
self.date_string = self.date.toString(self.date_format)
# Clear Models
self.availModel.clear()
self.attendModel.clear()
for student in self.db.get_attendance_for_date(self.date_string):
if student[1] != None and student[1] != '':
name = str(student[1])
else:
name = str(student[2]) + ' ' + str(student[3])
# Add student to the attended list view
itemlist = [QtGui.QStandardItem(name), \
QtGui.QStandardItem(str(student[0]))]
self.attendModel.appendRow(itemlist)
for student in self.db.get_students():
# Don't add them to available list if they attended given date
if student[1] != None and student[1] != '':
name = str(student[1])
else:
name = str(student[2]) + ' ' + str(student[3])
# Add student to available list view
itemlist = [QtGui.QStandardItem(name), \
QtGui.QStandardItem(str(student[0]))]
if self.attendModel.findItems(name) == []:
self.availModel.appendRow(itemlist)
self.availModel.sort(0)
self.attendModel.sort(0)
def on_add_clicked(self):
"""Move student from available to attended list."""
selected_indexes = self.ui.availListView.selectedIndexes()
for index in selected_indexes:
row = self.availModel.itemFromIndex(index).row()
#rowList = self.availModel.takeRow(row)
student = self.availModel.item(row, 0).text()
sid = self.availModel.item(row, 1).text()
try:
# Actually add the student for the date into the database
self.db.student_attend(sid, self.date_string)
except KeyError:
# Display error window if student missing
err_msg = QtGui.QErrorMessage()
err_msg.showMessage("Sid not found for student %s" % student)
self.update_views()
def on_remove_clicked(self):
"""Move student from attended to available list."""
selected_indexes = self.ui.attendListView.selectedIndexes()
for index in selected_indexes:
row = self.attendModel.itemFromIndex(index).row()
student = self.attendModel.item(row, 0).text()
sid = self.attendModel.item(row, 1).text()
try:
# Actually add the student for the date into the database
self.db.student_deattend(sid, self.date_string)
except KeyError:
# Display error window if student missing
err_msg = QtGui.QErrorMessage()
err_msg.showMessage("Sid not found for student %s" % student)
self.update_views()
class Main(QtGui.QMainWindow):
def __init__(self):
QtGui.QMainWindow.__init__(self)
# Setup the main window
self.ui=Ui_MainWindow()
self.ui.setupUi(self)
# Set date to today
self.date = QtCore.QDate.currentDate()
self.ui.calendarWidget.setSelectedDate(self.date)
# Date format for SQLite
self.date_format = "yyyy-MM-dd"
# Used to remember the Students Id
self.student_dict = {}
# Create Attendance DB Connection Object
self.db = AttendDB()
# Create list models
self.availModel = QtGui.QStandardItemModel(0, 2)
self.attendModel = QtGui.QStandardItemModel(0, 2)
self.ui.availListView.setModel(self.availModel)
self.ui.attendListView.setModel(self.attendModel)
# Connect signals to handlers
self.connect(self.ui.addButton, QtCore.SIGNAL("clicked()"), \
self.on_add_clicked)
self.connect(self.ui.removeButton, QtCore.SIGNAL("clicked()"), \
self.on_remove_clicked)
self.connect(self.ui.calendarWidget, \
QtCore.SIGNAL("selectionChanged()"), \
self.on_date_change)
self.connect(self.ui.actionEdit_Students, \
QtCore.SIGNAL("triggered()"), \
self.on_edit_students_select)
# Update the lists to reflect the current date
self.update_views()
def on_edit_students_select(self):
"""Open the editing student window."""
edit_window = Students()
edit_window.exec_()
def on_date_change(self):
"""Update views when the date is changed."""
self.date = self.ui.calendarWidget.selectedDate()
self.update_views()
def update_views(self):
"""Refresh the students available and attending for the given date."""
# Get correct date format
self.date_string = self.date.toString(self.date_format)
# Clear Models
self.availModel.clear()
self.attendModel.clear()
for student in self.db.get_attendance_for_date(self.date_string):
if student[1] != None and student[1] != '':
name = str(student[1])
else:
name = str(student[2]) + ' ' + str(student[3])
# Add student to the attended list view
itemlist = [QtGui.QStandardItem(name), \
QtGui.QStandardItem(str(student[0]))]
self.attendModel.appendRow(itemlist)
for student in self.db.get_students():
# Don't add them to available list if they attended given date
if student[1] != None and student[1] != '':
name = str(student[1])
else:
name = str(student[2]) + ' ' + str(student[3])
# Add student to available list view
itemlist = [QtGui.QStandardItem(name), \
QtGui.QStandardItem(str(student[0]))]
if self.attendModel.findItems(name) == []:
self.availModel.appendRow(itemlist)
self.availModel.sort(0)
self.attendModel.sort(0)
def on_add_clicked(self):
"""Move student from available to attended list."""
selected_indexes = self.ui.availListView.selectedIndexes()
for index in selected_indexes:
row = self.availModel.itemFromIndex(index).row()
#rowList = self.availModel.takeRow(row)
student = self.availModel.item(row, 0).text()
sid = self.availModel.item(row, 1).text()
try:
# Actually add the student for the date into the database
self.db.student_attend(sid, self.date_string)
except KeyError:
# Display error window if student missing
err_msg = QtGui.QErrorMessage()
err_msg.showMessage("Sid not found for student %s" % student)
self.update_views()
def on_remove_clicked(self):
"""Move student from attended to available list."""
selected_indexes = self.ui.attendListView.selectedIndexes()
for index in selected_indexes:
row = self.attendModel.itemFromIndex(index).row()
student = self.attendModel.item(row, 0).text()
sid = self.attendModel.item(row, 1).text()
try:
# Actually add the student for the date into the database
self.db.student_deattend(sid, self.date_string)
except KeyError:
# Display error window if student missing
err_msg = QtGui.QErrorMessage()
err_msg.showMessage("Sid not found for student %s" % student)
self.update_views()
class Students(QtGui.QDialog):
def __init__(self):
QtGui.QDialog.__init__(self)
# setup
self.dialog = Ui_studentsDialog()
self.dialog.setupUi(self)
self.sid = None
self.adding = False
header = ['Sid','Apelido', 'First Name', 'Last Name']
self.dialog.studentTableWidget.setHorizontalHeaderLabels(header)
self.connect(self.dialog.studentTableWidget, \
QtCore.SIGNAL("cellClicked(int, int)"), \
self._on_cell_click)
self.connect(self.dialog.updateButton, QtCore.SIGNAL("clicked()"), \
self._on_update_click)
self.connect(self.dialog.newButton, QtCore.SIGNAL("clicked()"), \
self._on_new_click)
self.db = AttendDB()
self._update_list()
def _update_list(self):
"""Refresh the student list."""
#for i in range(self.dialog.studentTableWidget.rowCount()):
# self.dialog.studentTableWidget.removeRow(i)
self.dialog.studentTableWidget.clear()
rows = self.dialog.studentTableWidget.rowCount()
students = self.db.get_students()
while(rows < len(students)):
self.dialog.studentTableWidget.insertRow(rows)
rows += 1
for i, student in enumerate(students):
#self.dialog.studentTableWidget.insertRow(i)
for j, item in enumerate(student):
if item == None:
item = ''
newitem = QtGui.QTableWidgetItem(str(item))
self.dialog.studentTableWidget.setItem(i, j, newitem)
if self.sid != None and str(student[0]) == str(self.sid):
self.dialog.studentTableWidget.setCurrentItem(newitem)
self.dialog.studentTableWidget.sortItems(1)
def _on_cell_click(self, row, column):
"""Fill out the entries when a cell is clicked."""
# Get selected row
items = self.dialog.studentTableWidget.selectedItems()
# Set the input fields
self.dialog.nickNameInput.setText(items[1].text())
self.dialog.firstNameInput.setText(items[2].text())
self.dialog.lastNameInput.setText(items[3].text())
def _on_update_click(self):
"""Update current student or add new student."""
# Get filled in information
apelido = str(self.dialog.nickNameInput.text())
fn = str(self.dialog.firstNameInput.text())
ln = str(self.dialog.lastNameInput.text())
if self.adding:
self.sid = self.db.add_student(apelido, fn, ln)
self.dialog.studentTableWidget.setEnabled(True)
else:
items = self.dialog.studentTableWidget.selectedItems()
self.sid = str(items[0].text())
self.db.update_student(self.sid, apelido, fn, ln)
self.adding = False
self._update_list()
def _on_new_click(self):
self.adding = True
self.dialog.studentTableWidget.setEnabled(False)