本文根据一个项目实例，记录分享一下python将数据库的内容提取显示到程序界面的过程及相关设置，探索python操作数据库的用法。主要分享内容：

• 1、显示数据库内容。

• 2、修改数据库内容。

• 3、表格控件指定滑块位置。

• 4、自定义右键菜单。

使用环境：python3.7 + Mysql5 +PyQt5

针对人群：初学者或有资料查阅需求者，资深人士勿喷，敬请提出宝贵意见，本人虚心接受。

前期准备：

1、创建软件界面

2、与数据库建立连接

1、创建软件界面：

上图为使用PyQt5相关模块创建的工程造价系统界面，使用的模块及功能：

1. 1、QMainWindow模块：绘制带菜单、工具栏、状态栏的整体框架。

1. 2、QTreeWidget模块：绘制左侧导航栏。

1. 3、QTableWidgetItem模块：绘制中间用于显示数据的表格。

1. 4、QComboBox, QPushButton模块：绘制下拉框、按钮控件。

1. 5、QSplitter模块：设置各种控件的摆放组合方式，可鼠标拖动调整窗口大小。

1. 6、QMessageBox模块：设置交互弹框。

具体界面绘制过程不是本文重点，暂不详述。

2、与数据库建立连接

要建立与数据库的连接，首先得有具体的数据库，本案例在本机用mysql建立自己的数据库，然后再与软件项目进行连接。

2.1、自建数据库

下载安装好Mysql后，建议再装一个可视化工具，我用的Navicat for Mysql，感觉不错。界面见下图：

在这个工具界面下，可以很方便的创建修改数据表。有了工具后，接下来就是关键的填入数据。很多时候，客户或者我们自己的数据是放在excel里的，我们可以通过Navicat 的导入向导，很方便的将excel数据导入到数据库：

跟着向导，只需简单几步就可完成数据导入。

2.2、连接数据库

导入pymysql库，将库的连接等功能写入自建的类。本案通过读取.cfg配置文件（有关配置文件的知识，可参见python开发项目，不得不了解的.cfg配置文件），获得连接库需用到的5个参数，在_init_初始类属性时，建立连接，代码如下：

import pymysql
from configparser import ConfigParser

class MysqlDb():
    def __init__(self):
        self.cp = ConfigParser()
        self.cp.read('source/sql_config.cfg')
        host = self.cp.get('sql_connect', 'host')
        port = int(self.cp.get('sql_connect', 'port'))
        user = self.cp.get('sql_connect', 'user')
        passwd = self.cp.get('sql_connect', 'passwd')
        db = self.cp.get('sql_connect', 'db')
        self.conn = pymysql.connect(host=host,
                                    port=port,
                                    user=user,
                                    passwd=passwd,
                                    db=db)
        self.cur = self.conn.cursor(cursor=pymysql.cursors.DictCursor)

其中，sql_config.cfg文件内容为：

若不用配置文件，直接把5个参数写在程序里也可以，代码如下，貌似更简单：

class MysqlDb():
    def __init__(self):        
        self.conn = pymysql.connect(host=‘127.0.0.1’,
                                    port=3306,
                                    user=‘root’,
                                    passwd=‘771222’,
                                    db=‘1’)
        self.cur = self.conn.cursor(cursor=pymysql.cursors.DictCursor)

一、显示数据库内容

功能：在软件界面，点击左侧导航栏，找到数据库中与展开项对应的表，将数据显示在中间表格里。若目标表不存在，提示创建。

实现步骤：

创建导航栏； → 点击导航栏，获取表名； → 查询数据库，找到表，显示内容

具体过程：

1、用树形控件创建导航栏，数据来源为数据库

数据库里导航栏数据源样式为：

用代码手动逐条输入是不可接受的，根据第一列数据长度特点，采用下面的批量创建方式，其中fl为传递的参数，意思是“路基”或者其他专业。

def set_tree(self, fl):
        qd_lj = MysqlDb().select_db('SELECT * FROM 清单' + fl)  # 通过自建的Mysqlbd类的select_db函数，得到数据库里的目标表
        root = QTreeWidgetItem(self.tree)
        root.setText(0, fl)
        root.setIcon(0, QIcon(r'source\3.png'))
        for i in range(0, len(qd_lj)):
            bm = qd_lj[i]['编码']
            mc = qd_lj[i]['名称']
            if (len(bm) == 4):
                root1 = QTreeWidgetItem(root)
                root1.setText(0, mc)
                root1.setText(1, bm)
            elif (len(bm) == 7):
                root2 = QTreeWidgetItem(root1)
                root2.setText(0, mc)
                root2.setText(1, bm)
            elif (len(bm) == 10):
                root3 = QTreeWidgetItem(root2)
                root3.setText(0, mc)
                root3.setText(1, bm)
            elif (len(bm) == 13):
                root4 = QTreeWidgetItem(root3)
                root4.setText(0, mc)
                root4.setText(1, bm)
            elif (len(bm) == 16):
                root5 = QTreeWidgetItem(root4)
                root5.setText(0, mc)
                root5.setText(1, bm)
            elif (len(bm) == 19):
                root6 = QTreeWidgetItem(root5)
                root6.setText(0, mc)
                root6.setText(1, bm)
            elif (len(bm) == 22):
                root7 = QTreeWidgetItem(root6)
                root7.setText(0, mc)
                root7.setText(1, bm)
            else:
                pass
        self.tree.expandAll()  # 设置树形构件全部展开
        self.item = root  # 自定义设置初始选中根条目

2、设置点击树形控件条目的信号槽函数，将数据库表内容提取后显示在表格控件。

self.tree.itemClicked.connect(self.showtreesql)

设置左键单击导航栏时，触发函数showtreesql()。具体获取数据功能在函数里设置。

def showtreesql(self, item):  # 单击树形节点，将数据库内容显示在表里
				self.inputtable.clearContents()  #清空中间的表格控件
        self.inputtable.setRowCount(0)  # 初始表格控件
        self.item = item  # 单击导航栏时，信号槽自动带当前点击节点信息参数item
        if not item.child(0):  # 判断如果该节点没有子节点，即单击的为最底层节点，非子节点无对应表格
            self.decxcomb3.setDisabled(False)
            self.select_item = "n" + item.text(1).replace('-', '')  # 根据该节点编码，得到对应表格名称
            if self.decxcomb3.findText(self.select_item) == -1:
                self.decxcomb3.addItem(self.select_item)
            self.decxcomb3.setCurrentText(self.select_item)  # 将表格名称显示在下拉框控件里
            if self.table_exists(MysqlDb(),
                                 self.select_item):  # 查询数据库，判断是否存在对应的表，
                self.inputtable.setRowCount(1)  
                self.sql_to_input(self.select_item, self.inputtable)  #如果存在表，查询数据库，将数据读取到表格
        else:
            # self.decxcomb3.isEnabled()
            self.decxcomb3.setDisabled(True)

3、读取数据库内容到表格的功能函数代码：

def sql_to_input(self, sql_tabel, inputtable):
        data = MysqlDb().select_db('SELECT * FROM ' + sql_tabel)
        if data:
            inputtable.setRowCount(len(data))
            for i in range(len(data)):
                inputtable.setItem(i, 0, QTableWidgetItem(data[i]['定额编号']))
                inputtable.setItem(i, 1, QTableWidgetItem(data[i]['定额名称']))
                inputtable.setItem(i, 2, QTableWidgetItem(data[i]['单位']))
                inputtable.setItem(i, 3, QTableWidgetItem(data[i]['数量']))
          #  print('数据库写入已完成”')
        else:
          #  print('源表为空表，不执行写入操作')
          pass

判断数据库是否存在某表的函数块：

def table_exists(self, sql, table_name):  # 这个函数用来判断数据库是否含某表
        tables = sql.select_db('SHOW TABLES')  # 得到数据库里的所有表名，
        tabels_list = []
        for i in tables:
            n = i['Tables_in_1']
            tabels_list.append(n)
        if table_name in tabels_list:
            return 1
        else:
            return 0

4、左键双击导航栏时，触发信号槽，当不存在目标表时，提示创建新表：

def edittreesql(self, item):  # 双击树形节点，将新建数据库表
        if not item.child(0):  # 判断该节点没有子节点
            self.select_item = "n" + item.text(1).replace('-', '')  # 记录该节点编码
            self.decxcomb3.addItem(self.select_item)
            self.decxcomb3.setCurrentText(self.select_item)
            if self.table_exists(MysqlDb(), self.select_item):  # 判断是否存在本节点为名的表
                # 判断结果存在，将内容显示在表格控件里
                self.inputtable.clearContents()
                self.sql_to_input(self.select_item, self.inputtable)
            else:
                if QMessageBox.information(self, "创建定额表",
                                           "即将为本清单创建定额输入表：" + self.select_item,
                                           QMessageBox.Yes | QMessageBox.No) == QMessageBox.Yes:
                    MysqlDb().biuldtabel(self.select_item)  # 若没有则在数据库新建一个表，
                    self.inputtable.clearContents()
                    self.inputtable.setRowCount(1)
                    print('已创建表:', self.select_item)

二、修改数据库内容

1、设置保存按钮

self.saveinput = QPushButton('保存到数据库')
self.saveinput.clicked.connect(self.input_to_sql)

2、编写保存按钮点击函数块

def input_to_sql(self):
        if self.decxcomb3.currentText():
            MysqlDb().execute_db('DELETE FROM ' +
                                 self.decxcomb3.currentText())  # 写入数据库前先清空原内容
            for i in range(self.inputtable.rowCount()):
                v = []
                for j in range(4):
                    if self.inputtable.item(i, j):
                        v.append(self.inputtable.item(i, j).text())
                    else:
                        v.append('')
                insert_sql = 'INSERT INTO ' + self.decxcomb3.currentText(
                ) + " (定额编号, 定额名称, 单位, 数量) VALUES('" + v[0] + "', '" + v[
                    1] + "', '" + v[2] + "', '" + v[3] + "')"
                MysqlDb().execute_db(insert_sql)

三、表格控件指定滑块位置

设置下拉框选择项目后，根据项目名让中间下部的表格上下滑块移动至将当前项目显示在第一位：

设置信号槽：self.decxcomb2.currentTextChanged[str].connect(self.set_table2)

设置功能块代码：

def set_table2(self, fl):
        for i in range(self.table2.rowCount()):
            if self.table2.item(i, 0).text() == fl:
                break
        self.table2.verticalScrollBar().setValue(i)

四、自定义右键菜单

表格输入时，有时候难免存在需要插入行、删除行的要求，最常见的就是下面的右键菜单处理了。

实现过程：

1）设置表格控件能响应右键

self.inputtable.setContextMenuPolicy(Qt.CustomContextMenu)

2）设置右键信号槽函数

self.inputtable.customContextMenuRequested.connect(self.input_rightmenu)
def input_rightmenu(self):  # 指定定额输入表控件右键菜单
        try:
            self.contextMenu = QMenu()
            self.actionA = self.contextMenu.addAction(u'删除')
            self.actionB = self.contextMenu.addAction(u'插入')
            self.actionA.setIcon(QIcon(r"source\4.png"))
            self.contextMenu.popup(QCursor.pos())  # 菜单显示的位置
            self.actionA.triggered.connect(self.deletcurrow)
            self.actionB.triggered.connect(self.insertcurrow)
            self.contextMenu.show()
        except Exception as e:
            print(e)

3）函数功能块设置

def deletcurrow(self):
        self.inputtable.removeRow(self.inputtable.currentRow())# 删除当前行

def insertcurrow(self):
        self.inputtable.insertRow(self.inputtable.currentRow()) #在当前行插入一行