用Sqlite和PyQT5制作TreeView

import sys
from PyQt5.QtWidgets import QApplication, QMainWindow, QTreeView
from PyQt5.QtSql import QSqlDatabase, QSqlQuery, QSqlQueryModel

db = QSqlDatabase.addDatabase('QSQLITE')
db.setDatabaseName('database.db')  # 数据库文件名
if not db.open():
    print('无法建立数据库连接')
    sys.exit(1)

query = QSqlQuery()
query.exec_("SELECT * FROM table")  # 执行SQL查询语句

model = QSqlQueryModel()
model.setQuery(query)

treeView = QTreeView()
treeView.setModel(model)

class MainWindow(QMainWindow):
    def __init__(self):
        super().__init__()
        self.setWindowTitle('TreeView示例')
        self.setCentralWidget(treeView)

app = QApplication(sys.argv)
window = MainWindow()
window.show()
sys.exit(app.exec_())