PyQt5 - Обработка базы данных

Библиотека PyQt5 содержит QtSqlмодуль. Это сложная система классов для связи со многими базами данных на основе SQL. этоQSqlDatabaseобеспечивает доступ через объект Connection. Ниже приведен список доступных в настоящее время драйверов SQL -

Sr. No.	Тип и описание драйвера
1	QDB2 IBM DB2
2	QIBASE Драйвер Borland InterBase
3	QMYSQL Драйвер MySQL
4	QOCI Драйвер интерфейса вызова Oracle
5	QODBC Драйвер ODBC (включает Microsoft SQL Server)
6	QPSQL Драйвер PostgreSQL
7	QSQLITE SQLite версии 3 или выше
8	QSQLITE2 SQLite версии 2

пример

В этой главе соединение с базой данных SQLite устанавливается с использованием статического метода -

db = QtSql.QSqlDatabase.addDatabase('QSQLITE')
db.setDatabaseName('sports.db')

Другие методы класса QSqlDatabase следующие:

Sr. No.	Методы и описание
1	setDatabaseName() Устанавливает имя базы данных, с которой ищется соединение
2	setHostName() Устанавливает имя хоста, на котором установлена база данных
3	setUserName() Задает имя пользователя для подключения
4	setPassword() Устанавливает пароль объекта подключения, если есть
5	commit() Фиксирует транзакции и возвращает истину в случае успеха
6	rollback() Откатывает транзакцию базы данных
7	close() Закрывает соединение

QSqlQueryclass имеет функции для выполнения и управления командами SQL. Могут выполняться запросы SQL как DDL, так и DML. Первый шаг - создать базу данных SQlite, используя следующие инструкции:

db = QSqlDatabase.addDatabase('QSQLITE')
db.setDatabaseName('sportsdatabase.db')

Затем получите объект Query с QSqlQuery() метод и вызвать его наиболее важный метод exec_ (), который принимает в качестве аргумента строку, содержащую SQL-оператор, который нужно выполнить.

query = QtSql.QSqlQuery()
query.exec_("create table sportsmen(id int primary key, " "firstname varchar(20), lastname varchar(20))")

Следующий сценарий создает базу данных SQLite sports.db с таблицей спортсменов, заполненной пятью записями.

import sys
from PyQt5.QtSql import *
from PyQt5.QtCore import *
from PyQt5.QtGui import *
from PyQt5.QtWidgets import *

def createDB():
   db = QSqlDatabase.addDatabase('QSQLITE')
   db.setDatabaseName('sportsdatabase.db')

   if not db.open():
      msg = QMessageBox()
      msg.setIcon(QMessageBox.Critical)
      msg.setText("Error in Database Creation")
      retval = msg.exec_()
      return False
   query = QSqlQuery()

   query.exec_("create table sportsmen(
      id int primary key, ""firstname varchar(20), lastname varchar(20))")

   query.exec_("insert into sportsmen values(101, 'Roger', 'Federer')")
   query.exec_("insert into sportsmen values(102, 'Christiano', 'Ronaldo')")
   query.exec_("insert into sportsmen values(103, 'Ussain', 'Bolt')")
   query.exec_("insert into sportsmen values(104, 'Sachin', 'Tendulkar')")
   query.exec_("insert into sportsmen values(105, 'Saina', 'Nehwal')")
   return True

if __name__ == '__main__':
   app = QApplication(sys.argv)
   createDB()

Чтобы убедиться, что база данных SQLite создана с указанными выше записями, добавленными в нее в таблице спортсменов, используйте утилиту SQLite Gui под названием SQLiteStudio.

QSqlTableModelКласс в PyQt - это интерфейс высокого уровня, который предоставляет редактируемую модель данных для чтения и записи записей в одной таблице. Эта модель используется для заполненияQTableViewобъект. Он представляет пользователю прокручиваемый и редактируемый вид, который может быть помещен в любое окно верхнего уровня.

Объект QSqlTableModel объявляется следующим образом:

model = QtSql.QSqlTableModel()

Его стратегия редактирования может быть установлена на любую из следующих:

QSqlTableModel.OnFieldChange	Все изменения вступят в силу немедленно
QSqlTableModel.OnRowChange	Изменения будут применены, когда пользователь выберет другую строку
QSqlTableModel.OnManualSubmit	Все изменения будут кэшироваться до тех пор, пока не будет вызван submitAll () или revertAll ()

пример

В следующем примере таблица спортсмена используется в качестве модели, а стратегия задана как -

model.setTable('sportsmen') 
model.setEditStrategy(QtSql.QSqlTableModel.OnFieldChange)
   model.select()

Класс QTableView является частью структуры Модель / Представление в PyQt. Объект QTableView создается следующим образом -

view = QtGui.QTableView()
view.setModel(model)
view.setWindowTitle(title)
return view

Этот объект QTableView и два виджета QPushButton добавляются в окно QDialog верхнего уровня. Сигнал Clicked () кнопки добавления связан с addrow (), который выполняет insertRow () для таблицы модели.

button.clicked.connect(addrow)
def addrow():
   print model.rowCount()
   ret = model.insertRows(model.rowCount(), 1)
   print ret

Слот, связанный с кнопкой удаления, выполняет лямбда-функцию, которая удаляет строку, выбранную пользователем.

btn1.clicked.connect(lambda: model.removeRow(view1.currentIndex().row()))

Полный код выглядит следующим образом -

import sys
from PyQt5.QtSql import *
from PyQt5.QtCore import *
from PyQt5.QtGui import *
from PyQt5.QtWidgets import *

def initializeModel(model):
   model.setTable('sportsmen')
   model.setEditStrategy(QSqlTableModel.OnFieldChange)
   model.select()
   model.setHeaderData(0, Qt.Horizontal, "ID")
   model.setHeaderData(1, Qt.Horizontal, "First name")
   model.setHeaderData(2, Qt.Horizontal, "Last name")

def createView(title, model):
   view = QTableView()
   view.setModel(model)
   view.setWindowTitle(title)
   return view

def addrow():
   print (model.rowCount())
   ret = model.insertRows(model.rowCount(), 1)
   print (ret)

def findrow(i):
   delrow = i.row()

if __name__ == '__main__':
   app = QApplication(sys.argv)
   db = QSqlDatabase.addDatabase('QSQLITE')
   db.setDatabaseName('sportsdatabase.db')
   model = QSqlTableModel()
   delrow = -1
   initializeModel(model)

   view1 = createView("Table Model (View 1)", model)
   view1.clicked.connect(findrow)

   dlg = QDialog()
   layout = QVBoxLayout()
   layout.addWidget(view1)

   button = QPushButton("Add a row")
   button.clicked.connect(addrow)
   layout.addWidget(button)

   btn1 = QPushButton("del a row")
   btn1.clicked.connect(lambda: model.removeRow(view1.currentIndex().row()))
   layout.addWidget(btn1)

   dlg.setLayout(layout)
   dlg.setWindowTitle("Database Demo")
   dlg.show()
   sys.exit(app.exec_())

Приведенный выше код дает следующий вывод -

Попробуйте добавить и удалить несколько записей и вернуться в SQLiteStudio, чтобы подтвердить транзакции.