Python程序接入MySQL数据库

在 Python3 中，我们可以使用mysqlclient或者pymysql三方库来接入 MySQL 数据库并实现数据持久化操作。二者的用法完全相同，只是导入的模块名不一样。我们推荐大家使用纯 Python 的三方库pymysql，因为它更容易安装成功。下面我们仍然以之前创建的名为hrs的数据库为例，为大家演示如何通过 Python 程序操作 MySQL 数据库实现数据持久化操作。

建库建表

  1. -- 创建名为hrs的数据库并指定默认的字符集
  2. create database `hrs` default character set utf8mb4;
  4. -- 切换到hrs数据库
  5. use `hrs`;
  7. -- 创建部门表
  8. create table `tb_dept`
  9. (
  10. `dno` int not null comment '编号',
  11. `dname` varchar(10) not null comment '名称',
  12. `dloc` varchar(20) not null comment '所在地',
  13. primary key (`dno`)
  14. );
  16. -- 插入4个部门
  17. insert into `tb_dept` values 
  18.     (10, '会计部', '北京'),
  19.     (20, '研发部', '成都'),
  20.     (30, '销售部', '重庆'),
  21.     (40, '运维部', '深圳');
  23. -- 创建员工表
  24. create table `tb_emp`
  25. (
  26. `eno` int not null comment '员工编号',
  27. `ename` varchar(20) not null comment '员工姓名',
  28. `job` varchar(20) not null comment '员工职位',
  29. `mgr` int comment '主管编号',
  30. `sal` int not null comment '员工月薪',
  31. `comm` int comment '每月补贴',
  32. `dno` int not null comment '所在部门编号',
  33. primary key (`eno`),
  34. constraint `fk_emp_mgr` foreign key (`mgr`) references tb_emp (`eno`),
  35. constraint `fk_emp_dno` foreign key (`dno`) references tb_dept (`dno`)
  36. );
  38. -- 插入14个员工
  39. insert into `tb_emp` values 
  40.     (7800, '张三丰', '总裁', null, 9000, 1200, 20),
  41.     (2056, '乔峰', '分析师', 7800, 5000, 1500, 20),
  42.     (3088, '李莫愁', '设计师', 2056, 3500, 800, 20),
  43.     (3211, '张无忌', '程序员', 2056, 3200, null, 20),
  44.     (3233, '丘处机', '程序员', 2056, 3400, null, 20),
  45.     (3251, '张翠山', '程序员', 2056, 4000, null, 20),
  46.     (5566, '宋远桥', '会计师', 7800, 4000, 1000, 10),
  47.     (5234, '郭靖', '出纳', 5566, 2000, null, 10),
  48.     (3344, '黄蓉', '销售主管', 7800, 3000, 800, 30),
  49.     (1359, '胡一刀', '销售员', 3344, 1800, 200, 30),
  50.     (4466, '苗人凤', '销售员', 3344, 2500, null, 30),
  51.     (3244, '欧阳锋', '程序员', 3088, 3200, null, 20),
  52.     (3577, '杨过', '会计', 5566, 2200, null, 10),
  53.     (3588, '朱九真', '会计', 5566, 2500, null, 10);

接入MySQL

首先，我们可以在命令行或者 PyCharm 的终端中通过下面的命令安装pymysql，如果需要接入 MySQL 8，还需要安装一个名为cryptography的三方库来支持 MySQL 8 的密码认证方式。

  1. pip install pymysql cryptography

使用pymysql操作 MySQL 的步骤如下所示：

  1. 创建连接。MySQL 服务器启动后，提供了基于 TCP （传输控制协议）的网络服务。我们可以通过pymysql模块的connect函数连接 MySQL 服务器。在调用connect函数时，需要指定主机（host）、端口（port）、用户名（user）、口令（password）、数据库（database）、字符集（charset）等参数，该函数会返回一个Connection对象。
  2. 获取游标。连接 MySQL 服务器成功后，接下来要做的就是向数据库服务器发送 SQL 语句，MySQL 会执行接收到的 SQL 并将执行结果通过网络返回。要实现这项操作，需要先通过连接对象的cursor方法获取游标（Cursor）对象。
  3. 发出 SQL。通过游标对象的execute方法，我们可以向数据库发出 SQL 语句。
  4. 如果执行insert、delete或update操作，需要根据实际情况提交或回滚事务。因为创建连接时，默认开启了事务环境，在操作完成后，需要使用连接对象的commit或rollback方法，实现事务的提交或回滚，rollback方法通常会放在异常捕获代码块except中。如果执行select操作，需要通过游标对象抓取查询的结果，对应的方法有三个，分别是：fetchone、fetchmany和fetchall。其中fetchone方法会抓取到一条记录，并以元组或字典的方式返回；fetchmany和fetchall方法会抓取到多条记录，以嵌套元组或列表装字典的方式返回。
  5. 关闭连接。在完成持久化操作后，请不要忘记关闭连接，释放外部资源。我们通常会在finally代码块中使用连接对象的close方法来关闭连接。

代码实操

下面，我们通过代码实操的方式为大家演示上面说的五个步骤。

插入数据

  1. import pymysql
  3. no = int(input('部门编号: '))
  4. name = input('部门名称: ')
  5. location = input('部门所在地: ')
  7. # 1. 创建连接（Connection）
  8. conn = pymysql.connect(host='127.0.0.1', port=3306,
  9.                        user='guest', password='Guest.618',
  10.                        database='hrs', charset='utf8mb4')
  11. try:
  12.     # 2. 获取游标对象（Cursor）
  13.     with conn.cursor() as cursor:
  14.         # 3. 通过游标对象向数据库服务器发出SQL语句
  15.         affected_rows = cursor.execute(
  16.             'insert into `tb_dept` values (%s, %s, %s)',
  17.             (no, name, location)
  18.         )
  19.         if affected_rows == 1:
  20.             print('新增部门成功!!!')
  21.     # 4. 提交事务（transaction）
  22.     conn.commit()
  23. except pymysql.MySQLError as err:
  24.     # 4. 回滚事务
  25.     conn.rollback()
  26.     print(type(err), err)
  27. finally:
  28.     # 5. 关闭连接释放资源
  29.     conn.close()

说明：上面的127.0.0.1称为回环地址，它代表的是本机。下面的guest是我提前创建好的用户，该用户拥有对hrs数据库的insert、delete、update和select权限。我们不建议大家在项目中直接使用root超级管理员账号访问数据库，这样做实在是太危险了。我们可以使用下面的命令创建名为guest的用户并为其授权。

  1. create user 'guest'@'%' identified by 'Guest.618';
  2. grant insert, delete, update, select on `hrs`.* to 'guest'@'%';

如果要插入大量数据，建议使用游标对象的executemany方法做批处理（一个insert操作后面跟上多组数据），大家可以尝试向一张表插入10000条记录，然后看看不使用批处理一条条的插入和使用批处理有什么差别。游标对象的executemany方法第一个参数仍然是 SQL 语句，第二个参数可以是包含多组数据的列表或元组。

删除数据

  1. import pymysql
  3. no = int(input('部门编号: '))
  5. # 1. 创建连接（Connection）
  6. conn = pymysql.connect(host='127.0.0.1', port=3306,
  7.                        user='guest', password='Guest.618',
  8.                        database='hrs', charset='utf8mb4',
  9.                        autocommit=True)
  10. try:
  11.     # 2. 获取游标对象（Cursor）
  12.     with conn.cursor() as cursor:
  13.         # 3. 通过游标对象向数据库服务器发出SQL语句
  14.         affected_rows = cursor.execute(
  15.             'delete from `tb_dept` where `dno`=%s',
  16.             (no, )
  17.         )
  18.         if affected_rows == 1:
  19.             print('删除部门成功!!!')
  20. finally:
  21.     # 5. 关闭连接释放资源
  22.     conn.close()

说明：如果不希望每次 SQL 操作之后手动提交或回滚事务，可以connect函数中加一个名为autocommit的参数并将它的值设置为True，表示每次执行 SQL 成功后自动提交。但是我们建议大家手动提交或回滚，这样可以根据实际业务需要来构造事务环境。如果不愿意捕获异常并进行处理，可以在try代码块后直接跟finally块，省略except意味着发生异常时，代码会直接崩溃并将异常栈显示在终端中。

更新数据

  1. import pymysql
  3. no = int(input('部门编号: '))
  4. name = input('部门名称: ')
  5. location = input('部门所在地: ')
  7. # 1. 创建连接（Connection）
  8. conn = pymysql.connect(host='127.0.0.1', port=3306,
  9.                        user='guest', password='Guest.618',
  10.                        database='hrs', charset='utf8mb4')
  11. try:
  12.     # 2. 获取游标对象（Cursor）
  13.     with conn.cursor() as cursor:
  14.         # 3. 通过游标对象向数据库服务器发出SQL语句
  15.         affected_rows = cursor.execute(
  16.             'update `tb_dept` set `dname`=%s, `dloc`=%s where `dno`=%s',
  17.             (name, location, no)
  18.         )
  19.         if affected_rows == 1:
  20.             print('更新部门信息成功!!!')
  21.     # 4. 提交事务
  22.     conn.commit()
  23. except pymysql.MySQLError as err:
  24.     # 4. 回滚事务
  25.     conn.rollback()
  26.     print(type(err), err)
  27. finally:
  28.     # 5. 关闭连接释放资源
  29.     conn.close()

查询数据

  1. 查询部门表的数据。
  1. import pymysql
  3. # 1. 创建连接（Connection）
  4. conn = pymysql.connect(host='127.0.0.1', port=3306,
  5.                        user='guest', password='Guest.618',
  6.                        database='hrs', charset='utf8mb4')
  7. try:
  8.     # 2. 获取游标对象（Cursor）
  9.     with conn.cursor() as cursor:
  10.         # 3. 通过游标对象向数据库服务器发出SQL语句
  11.         cursor.execute('select `dno`, `dname`, `dloc` from `tb_dept`')
  12.         # 4. 通过游标对象抓取数据
  13.         row = cursor.fetchone()
  14.         while row:
  15.             print(row)
  16.             row = cursor.fetchone()
  17. except pymysql.MySQLError as err:
  18.     print(type(err), err)
  19. finally:
  20.     # 5. 关闭连接释放资源
  21.     conn.close()

说明：上面的代码中，我们通过构造一个while循环实现了逐行抓取查询结果的操作。这种方式特别适合查询结果有非常多行的场景。因为如果使用fetchall一次性将所有记录抓取到一个嵌套元组中，会造成非常大的内存开销，这在很多场景下并不是一个好主意。如果不愿意使用while循环，还可以考虑使用iter函数构造一个迭代器来逐行抓取数据，有兴趣的读者可以自行研究。

  1. 分页查询员工表的数据。
  1. import pymysql
  3. page = int(input('页码: '))
  4. size = int(input('大小: '))
  6. # 1. 创建连接（Connection）
  7. con = pymysql.connect(host='127.0.0.1', port=3306,
  8.                       user='guest', password='Guest.618',
  9.                       database='hrs', charset='utf8')
  10. try:
  11.     # 2. 获取游标对象（Cursor）
  12.     with con.cursor(pymysql.cursors.DictCursor) as cursor:
  13.         # 3. 通过游标对象向数据库服务器发出SQL语句
  14.         cursor.execute(
  15.             'select `eno`, `ename`, `job`, `sal` from `tb_emp` order by `sal` desc limit %s,%s',
  16.             ((page - 1) * size, size)
  17.         )
  18.         # 4. 通过游标对象抓取数据
  19.         for emp_dict in cursor.fetchall():
  20.             print(emp_dict)
  21. finally:
  22.     # 5. 关闭连接释放资源
  23.     con.close()

案例讲解

下面我们为大家讲解一个将数据库表数据导出到 Excel 文件的例子，我们需要先安装openpyxl三方库，命令如下所示。

  1. pip install openpyxl

接下来，我们通过下面的代码实现了将数据库hrs中所有员工的编号、姓名、职位、月薪、补贴和部门名称导出到一个 Excel 文件中。

  1. import openpyxl
  2. import pymysql
  4. # 创建工作簿对象
  5. workbook = openpyxl.Workbook()
  6. # 获得默认的工作表
  7. sheet = workbook.active
  8. # 修改工作表的标题
  9. sheet.title = '员工基本信息'
  10. # 给工作表添加表头
  11. sheet.append(('工号', '姓名', '职位', '月薪', '补贴', '部门'))
  12. # 创建连接（Connection）
  13. conn = pymysql.connect(host='127.0.0.1', port=3306,
  14.                        user='guest', password='Guest.618',
  15.                        database='hrs', charset='utf8mb4')
  16. try:
  17.     # 获取游标对象（Cursor）
  18.     with conn.cursor() as cursor:
  19.         # 通过游标对象执行SQL语句
  20.         cursor.execute(
  21.             'select `eno`, `ename`, `job`, `sal`, coalesce(`comm`, 0), `dname` '
  22.             'from `tb_emp` natural join `tb_dept`'
  23.         )
  24.         # 通过游标抓取数据
  25.         row = cursor.fetchone()
  26.         while row:
  27.             # 将数据逐行写入工作表中
  28.             sheet.append(row)
  29.             row = cursor.fetchone()
  30.     # 保存工作簿
  31.     workbook.save('hrs.xlsx')
  32. except pymysql.MySQLError as err:
  33.     print(err)
  34. finally:
  35.     # 关闭连接释放资源
  36.     conn.close()

大家可以参考上面的例子，试一试把 Excel 文件的数据导入到指定数据库的指定表中，看看是否可以成功。