def __init__(self): QtGui.QMainWindow.__init__(self) self.ui = Ui_Search_MainWindow() self.ui.setupUi(self) self.conn() #need catch exception #TODO deal with foreign key self.ui.Class = QSqlRelationalTableModel(db=self.db) self.ui.Class.setTable("Class") self.ui.Class.setHeaderData(0, QtCore.Qt.Horizontal, "ID") self.ui.Class.setHeaderData(1, QtCore.Qt.Horizontal, "Name") self.ui.Class.setHeaderData(2, QtCore.Qt.Horizontal, "Cost") self.ui.Class.setHeaderData(3, QtCore.Qt.Horizontal, "Start Time") self.ui.Class.setHeaderData(4, QtCore.Qt.Horizontal, "End Time") self.ui.Class.setHeaderData(6, QtCore.Qt.Horizontal, "Location") self.ui.Class.setHeaderData(7, QtCore.Qt.Horizontal, "Capacity") self.ui.Class.setFilter('') self.ui.Class.select() #display window self.ui.Class_view.setModel(self.ui.Class) self.ui.Class_view.hideColumn(2) self.ui.Class_view.hideColumn(8) self.ui.Class_view.hideColumn(9) self.ui.Class_view.hideColumn(10) self.ui.Class_view.hideColumn(11) self.ui.Class_view.hideColumn(12) self.ui.Class_view.hideColumn(13) self.ui.Class_view.setEditTriggers(QAbstractItemView.NoEditTriggers) self.ui.Search_btn.clicked.connect(self.search) self.ui.Adv_search_btn.clicked.connect(self.advsearch_show) self.ui.Reset_search_btn.clicked.connect(self.reset_table) self.ui.Detail_btn.clicked.connect(self.detail_show) self.ui.Back_btn.clicked.connect(self.close) self.ui.Class_view.setSelectionMode(QAbstractItemView.SingleSelection) self.ui.Class_view.setSelectionBehavior(QAbstractItemView.SelectRows)