Pythonfrom PyQt4 import QtGui, QtCore, QtSqlimport os.path class mainform(QtGui.QWidget): def __init__(self): QtGui.QWidget.__init__(self) self.resize(600, 700) self.con = QtSql.QSqlDatabase.addDatabase("QSQLITE", "Base") self.con.setDatabaseName("./SQLiteBase/PyQt4DB.s3db") self.OpenOrCreateDB() self.mainLayout = QtGui.QVBoxLayout() self.setLayout(self.mainLayout) self.btnFill = QtGui.QPushButton("Заполнить БД некоторыми данными") self.mainLayout.addWidget(self.btnFill) self.btnFill.clicked.connect(self.Fill_View_With_Some_Values) #реляционная модель данных self.model = QtSql.QSqlRelationalTableModel(None, self.con) #создание self.model.setTable('thirst') #выбор таблицы для модели self.typeindex = self.model.fieldIndex("type") #объект содержит индекс столбца по его имени self.model.setRelation(self.typeindex, QtSql.QSqlRelation("second", "type", "text")) #указание связи self.model.select() #заполнение модели данными из БД # столбец 2(type) будет заполнен соответствующими значениями поля text из таблицы second #создание виджета таблицы (вида, отображения) self.view1 = QtGui.QTableView() # сам вид self.mainLayout.addWidget(self.view1) # добавляем его к контейнеру self.view1.setModel(self.model) # указываем какую модель использовать таблице self.view1.setItemDelegate(QtSql.QSqlRelationalDelegate(self)) # создаём делегат #эта таблица служит лишь для того, чтобы показать, что разные таблицы могут использовать одну модель self.view2 = QtGui.QTableView() self.mainLayout.addWidget(self.view2) self.view2.setModel(self.model) #текстбокс для ввода товара self.txtBoxTovar = QtGui.QLineEdit() self.mainLayout.addWidget(self.txtBoxTovar) #Комбобокс для ввода типа товара self.comboBoxType = QtGui.QComboBox() self.mainLayout.addWidget(self.comboBoxType) #кнопка для добавления строки в БД self.btnNew = QtGui.QPushButton("Новая запись") self.btnNew.clicked.connect(self.Create_New_Record) self.mainLayout.addWidget(self.btnNew) # создаём модель для комбобокса self.relModel = self.model.relationModel(self.typeindex) # модель для комбобокса self.comboBoxType.setModel(self.relModel) self.comboBoxType.setModelColumn(self.relModel.fieldIndex("text")) # создаём маппер (Он нужен для связывания полей таблицы БД и контролов) self.mapper = QtGui.QDataWidgetMapper() # сам маппер self.mapper.setModel(self.model) self.mapper.setItemDelegate(QtSql.QSqlRelationalDelegate(self)) #взял из примера, не понимаю на кой оно надо, #но без него не будет соответствия данных в комбобоксе и таблице! self.mapper.addMapping(self.comboBoxType, self.typeindex) self.mapper.addMapping(self.txtBoxTovar, 1) self.mapper.toFirst() def Create_New_Record(self): addRecord = QtSql.QSqlRecord(self.model.record()) addRecord.setValue("id", 0) self.model.insertRecord(-1, addRecord) self.model.submitAll() #self.model.insertRow(self.model.rowCount()) #self.model.setData() self.mapper.toLast() def Fill_View_With_Some_Values(self): query = QtSql.QSqlQuery(self.con) query.exec("INSERT INTO thirst (tovar, type) VALUES ('Potato', 1)") query.exec("INSERT INTO thirst (tovar, type) VALUES ('Apple', 2)") query.exec("INSERT INTO thirst (tovar, type) VALUES ('Juice', 3)") query.exec("INSERT INTO thirst (tovar, type) VALUES ('Nuts', 4)") self.model.select() def OpenOrCreateDB(self): ThisIsNewBase = False if not os.path.exists("./SQLiteBase/PyQt4DB.s3db"): ThisIsNewBase = True if not self.con.open(): print ("БД не открылась!") else: print ("БД открыта") if ThisIsNewBase == True: self.Create_Structure_of_DB() def Create_Structure_of_DB(self): query = QtSql.QSqlQuery(self.con) query.exec("CREATE TABLE thirst (" "id INTEGER PRIMARY KEY AUTOINCREMENT, " "tovar VARCHAR(20) NOT NULL, " "type INTEGER NOT NULL" ")") query.exec("CREATE TABLE second (" "type INTEGER, " "text VARCHAR(50)" ")") query.exec("INSERT INTO second VALUES(1, 'Бяка')") query.exec("INSERT INTO second VALUES(2, 'Кака')") query.exec("INSERT INTO second VALUES(3, 'Добро')") query.exec("INSERT INTO second VALUES(4, 'Благо')") if __name__=="__main__": import sys app = QtGui.QApplication(sys.argv) form1 = mainform() form1.setWindowTitle("Работа с БД при помощи PyQt4") form1.show() sys.exit(app.exec_())
PythonaddRecord = QtSql.QSqlRecord(self.model.record())addRecord.setValue("id", QtCore.QVariant)self.model.insertRecord(-1, addRecord)self.model.submitAll()
