def __init__(self): QFrame.__init__(self) self.ui = Ui_all_distributors_form() self.ui.setupUi(self) self.drawTable() QObject.connect(self.ui.add_distributor_btn, SIGNAL("clicked()"), self.showNewDistributorForm) QObject.connect(self.ui.del_distributor_btn, SIGNAL("clicked()"), self.delDistributor)
class AllDistributorsWindow(QFrame): def __init__(self): QFrame.__init__(self) self.ui = Ui_all_distributors_form() self.ui.setupUi(self) self.drawTable() QObject.connect(self.ui.add_distributor_btn, SIGNAL("clicked()"), self.showNewDistributorForm) QObject.connect(self.ui.del_distributor_btn, SIGNAL("clicked()"), self.delDistributor) def drawTable(self): self.distributors = query_session.query(Distributor).all() self.ui.distributors_table.clear() self.ui.distributors_table.setRowCount(1) self.ui.distributors_table.setColumnCount(2) self.ui.distributors_table.setHorizontalHeaderLabels([QString.fromUtf8('Номер'), QString.fromUtf8('Название')]) for distributor in self.distributors: data = [] data.append(str(distributor.id)) data.append(distributor.name) for i in range(0,2): tableitem = QTableWidgetItem() tableitem.setText(data[i]) tableitem.font = QFont("Arial", 10) tableitem.font.setBold(True) tableitem.textcolor = QColor("black") self.ui.distributors_table.setItem(self.ui.distributors_table.rowCount() - 1,i,tableitem) self.ui.distributors_table.setRowCount(self.ui.distributors_table.rowCount()+1) self.ui.distributors_table.resizeColumnsToContents() def showNewDistributorForm(self): self.new_distributor_form = NewDistributorForm(self) QObject.connect(self.new_distributor_form, SIGNAL("distributorAdded()"), self.drawTable) self.new_distributor_form.setWindowModality(2) self.new_distributor_form.show() def delDistributor(self): row = self.ui.distributors_table.currentRow() id_for_delete = int(self.ui.distributors_table.item(row, 0).text()) distributor_for_delete = query_session.query(Distributor).filter_by(id=id_for_delete).one() s = Session() s.delete(distributor_for_delete) s.commit() s.close self.drawTable()