PyQt5 - obsługa bazy danych

Biblioteka PyQt5 zawiera QtSqlmoduł. Jest to rozbudowany system klasy do komunikacji z wieloma bazami danych opartymi na języku SQL. JegoQSqlDatabasezapewnia dostęp za pośrednictwem obiektu Connection. Poniżej znajduje się lista aktualnie dostępnych sterowników SQL -

Sr.No.	Typ i opis sterownika
1	QDB2 IBM DB2
2	QIBASE Sterownik Borland InterBase
3	QMYSQL Sterownik MySQL
4	QOCI Sterownik interfejsu wywołań Oracle
5	QODBC Sterownik ODBC (zawiera Microsoft SQL Server)
6	QPSQL Sterownik PostgreSQL
7	QSQLITE SQLite w wersji 3 lub nowszej
8	QSQLITE2 SQLite w wersji 2

Przykład

W tym rozdziale połączenie z bazą danych SQLite jest ustanawiane metodą statyczną -

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

Inne metody klasy QSqlDatabase są następujące -

Sr.No.	Metody i opis
1	setDatabaseName() Ustawia nazwę bazy danych, z którą jest poszukiwane połączenie
2	setHostName() Ustawia nazwę hosta, na którym jest zainstalowana baza danych
3	setUserName() Określa nazwę użytkownika do połączenia
4	setPassword() Ustawia hasło obiektu połączenia, jeśli istnieje
5	commit() Zatwierdza transakcje i zwraca prawdę, jeśli się powiedzie
6	rollback() Wycofuje transakcję bazy danych
7	close() Zamyka połączenie

QSqlQueryclass posiada funkcję wykonywania i manipulowania poleceniami SQL. Można wykonywać zapytania SQL zarówno typu DDL, jak i DML. Pierwszym krokiem jest utworzenie bazy danych SQlite za pomocą następujących instrukcji -

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

Następnie uzyskaj obiekt Query za pomocą QSqlQuery() metody i wywołaj jej najważniejszą metodę exec_ (), która jako argument przyjmuje łańcuch zawierający instrukcję SQL do wykonania.

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

Poniższy skrypt tworzy bazę danych SQLite sports.db z tabelą sportowców wypełnioną pięcioma rekordami.

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()

Aby potwierdzić, że baza danych SQLite została utworzona z powyższymi rekordami dodanymi do tabeli sportowców, użyj narzędzia SQLite Gui o nazwie SQLiteStudio.

QSqlTableModelclass w PyQt to interfejs wysokiego poziomu, który zapewnia edytowalny model danych do odczytu i zapisu rekordów w pojedynczej tabeli. Ten model służy do wypełniania plikuQTableViewobiekt. Przedstawia użytkownikowi przewijalny i edytowalny widok, który można umieścić w dowolnym oknie najwyższego poziomu.

Obiekt QSqlTableModel jest deklarowany w następujący sposób -

model = QtSql.QSqlTableModel()

Jego strategię edycji można ustawić na dowolną z następujących -

QSqlTableModel.OnFieldChange	Wszystkie zmiany zostaną zastosowane natychmiast
QSqlTableModel.OnRowChange	Zmiany zostaną zastosowane, gdy użytkownik wybierze inny wiersz
QSqlTableModel.OnManualSubmit	Wszystkie zmiany będą przechowywane w pamięci podręcznej do momentu wywołania submitAll () lub revertAll ()

Przykład

W poniższym przykładzie tabela sportowców jest używana jako model, a strategia jest ustawiona jako -

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

Klasa QTableView jest częścią struktury Model / View w PyQt. Obiekt QTableView jest tworzony w następujący sposób -

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

Ten obiekt QTableView i dwa widżety QPushButton są dodawane do okna QDialog najwyższego poziomu. Kliknięty () sygnał przycisku dodawania jest połączony z addrow (), który wykonuje insertRow () na tabeli modelu.

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

Boks skojarzony z przyciskiem usuwania wykonuje funkcję lambda, która usuwa wiersz wybrany przez użytkownika.

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

Kompletny kod wygląda następująco -

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_())

Powyższy kod daje następujące dane wyjściowe -

Spróbuj dodać i usunąć kilka rekordów i wróć do SQLiteStudio, aby potwierdzić transakcje.