学习目录

安装Mysql数据库

安装mysql数据库的连接工具

安装pymysql

导入pymysql库连接数据库并操作

关闭数据库

配合with语句实现上下文管理器，数据库语句执行后，自动退出连接

mysql数据库在软件开发或测试环节被大量使用，本章节主要介绍使用python的pymysql库操作mysql数据库，包括查询、创建，更新，删除数据等功能，同时介绍使用with语句实现数据库的上下文管理器功能。

安装Mysql数据库

安装windows操作系统的mysql数据库，在官网下载安装https://downloads.mysql.com/archives/installer/，注意保存好创建时填写的root密码或者其他用户名和密码，后续连接时需要使用。

或者大家在https://sqlpub.com/#/网站上申请一个免费的数据库空间，申请只需要注册下就可以使用，以下举例的数据库信息就是从该网站上申请的，平常大家也可以用来练手。

安装pymysql

通过pip命令安装pymysql

pip install pymysql -i https://mirrors.aliyun.com/pypi/simple/

导入pymysql库连接数据库

import pymysql

连接数据库，输入参数为用户名，密码，数据库地址，端口，数据库名
db =pymysql.connect(user='testone',
                    password='RQiUbA66G4G',
                    host='mysql.sqlpub.com',
                    port=3306,
                    database='testsqldata')
#设置数据库游标，数据库操作都是通过游标。
cursor = db.cursor()

通过execute函数执行sql语句

1. 创建表

sql = """CREATE TABLE userinfo (
           NAME CHAR(20) NOT NULL, 
           AGE INT, 
           SEX CHAR(1),
           INCOME FLOAT )"""
#执行sql
cursor.execute(sql)

2. 插入数据

#往表中插入10条数据
for i in range(10):
  age = 20 + i
  sql = """INSERT INTO userinfo(
           NAME, AGE, SEX, INCOME)
           VALUES ('Lihan', %d, 'M', 2000)""" % age
  cursor.execute(sql)

3.查询表数据

sql = """SELECT * FROM userinfo"""
cursor.execute(sql)

• fetchall函数获取所有查询的结果（返回元组）

print(cursor.fetchall())
(('Lihan', 20, 'M', 2000.0), ('Lihan', 21, 'M', 2000.0), 
 ('Lihan', 22, 'M', 2000.0), ('Lihan', 23, 'M', 2000.0),
 ('Lihan', 24, 'M', 2000.0), ('Lihan', 25, 'M', 2000.0),
 ('Lihan', 26, 'M', 2000.0), ('Lihan', 27, 'M', 2000.0), 
 ('Lihan', 28, 'M', 2000.0), ('Lihan', 29, 'M', 2000.0))

• fetchone函数每次返回1条数据

比如执行两次打印

print(cursor.fetchone())
print(cursor.fetchone())

打印的结果就是前2条数据

('Lihan', 20, 'M', 2000.0)
('Lihan', 21, 'M', 2000.0)

• fetchmany函数自定义返回多少条的数据

通过传入size参数，表示返回几条数据

print(cursor.fetchmany(size=5))

传入size=5，打印前5条数据

(('Lihan', 20, 'M', 2000.0), ('Lihan', 20, 'M', 2000.0), 
 ('Lihan', 21, 'M', 2000.0), ('Lihan', 22, 'M', 2000.0), 
 ('Lihan', 23, 'M', 2000.0))

关闭数据库

执行完数据库操作后，大家不要忘了关闭数据库连接。

# 关闭数据库连接
db.close()

配合with语句实现上下文管理器，数据库语句执行后，自动退出连接。

为了防止我们写代码时会忘记关闭数据库连接，我们可以配合with语句实现上下文管理器，而pymysql的connect类支持__enter__() 和__exit__()魔法方法，就可以支持上下文管理器。如果不熟悉上下文管理器的可以参考之前的文章（python实现上下文管理器）。

比如将上面插入的数据删掉，写成with语句如下：

db = pymysql.connect(user='testone',
                    password='RQiUb7Na6G4G',
                    host='mysql.sqlpub.com',
                    port=3306,
                    database='testsqldata')
with db as f:
  cursor = f.cursor()
  sql = "DELETE FROM userinfo"
  cursor.execute(sql)
  print('程序已执行完成')

为了方便执行后看结果，我们在源代码中的__exit__()函数加一句priint打印：

程序执行后打印如下：此时可以实现自动关闭数据库连接了

程序已执行完成
程序执行完成后，关闭数据库连接

共勉： 东汉·班固《汉书·枚乘传》：“泰山之管穿石，单极之绠断干。水非石之钻，索非木之锯，渐靡使之然也。”

-----指水滴不断地滴，可以滴穿石头；

-----比喻坚持不懈，集细微的力量也能成就难能的功劳。

----感谢读者的阅读和学习，谢谢大家。