Python-сообщество

#-*- coding: utf-8 -*-
from PyQt4 import QtGui, QtCore, QtSql
import 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.btnNewWithData = QtGui.QPushButton("Новая запись с данными")
        self.btnNewWithData.clicked.connect(self.Create_New_Record_with_Data)
        self.mainLayout.addWidget(self.btnNewWithData)
        
        # создаём модель для комбобокса
        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("tovar", 'Урюк')
        addRecord.setValue("type", 2)
        self.model.insertRecord(-1, addRecord)
        self.model.submitAll()
        #self.model.insertRow(self.model.rowCount())
        #self.model.setData()
        self.mapper.toLast()
    def Create_New_Record_with_Data(self):
        query = QtSql.QSqlQuery(self.con)
        query.prepare = "INSERT INTO thirst (tovar, type) VALUES (?, ?)"
        query.addBindValue('Broad')
        query.addBindValue(2)
        query.exec_()
        print(query.lastQuery())
        print(query.lastInsertId())
        print(query.lastError().text())
        #query.exec("INSERT INTO thirst (tovar, type) VALUES ('Завтрак', 2)")
        
        self.model.select()
        
    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_())

query.prepare("INSERT INTO thirst (tovar, type) VALUES (?, ?)")

addRecord = self.model.record()

 def Create_New_Record(self):
        addRecord = QtSql.QSqlRecord(self.model.record())
        
        addRecord.setValue("tovar", 'Урюк')
        addRecord.setValue("type", 2)
        self.model.insertRecord(-1, addRecord)
        self.model.submitAll()
        #self.model.insertRow(self.model.rowCount())
        #self.model.setData()
        self.mapper.toLast()

    def Create_New_Record(self):
        addRecord = self.model.record()
        addRecord.append(QtSql.QSqlField("type", QtCore.QVariant.Int))
        addRecord.setValue("tovar", 'Урюк')
        addRecord.setValue("type", 2)
        self.model.insertRecord(-1, addRecord)
        print(self.model.lastError().text())
        self.model.submitAll()
        self.mapper.toLast()