PyQt5 - 資料庫處理

---

---

PyQt5 庫包含 **QtSql** 模組。它是一個用於與許多基於 SQL 的資料庫通訊的完善的類系統。它的 **QSqlDatabase** 透過連線物件提供訪問。以下是當前可用的 SQL 驅動程式列表：

示例

在本節中，使用靜態方法建立與 SQLite 資料庫的連線：

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

QSqlDatabase 類的其他方法如下：

**QSqlQuery** 類具有執行和操作 SQL 命令的功能。可以執行 DDL 和 DML 型別的 SQL 查詢。第一步是使用以下語句建立 SQLite 資料庫：

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

接下來，使用 **QSqlQuery()** 方法獲取查詢物件，並呼叫其最重要的方法 exec_()，該方法將包含要執行的 SQL 語句的字串作為引數。

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

以下指令碼建立一個名為 sports.db 的 SQLite 資料庫，其中包含一個名為 sportsperson 的表，並填充了五個記錄。

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 資料庫是否已建立，並在其中的 sportsmen 表中添加了上述記錄，請使用名為 **SQLiteStudio** 的 SQLite GUI 工具。

PyQt 中的 **QSqlTableModel** 類是一個高階介面，它提供可編輯的資料模型來讀取和寫入單個表中的記錄。此模型用於填充 **QTableView** 物件。它向用戶呈現一個可滾動且可編輯的檢視，該檢視可以放置在任何頂級視窗上。

QSqlTableModel 物件以以下方式宣告：

model = QtSql.QSqlTableModel()

其編輯策略可以設定為以下任何一種：

示例

在以下示例中，sportsperson 表用作模型，策略設定為：

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 視窗。add 按鈕的 Clicked() 訊號連線到 addrow()，它在模型表上執行 insertRow()。

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

與 delete 按鈕關聯的槽執行一個 lambda 函式，該函式刪除使用者選擇的一行。

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 以確認事務。