pymysql的使用

pymysql是从Python连接到MySQL数据库服务器的接口，其官方文档为：https://pymysql.readthedocs.io/en/latest/

安装：pip install pymysql

对于数据库的操作，我们一般是这样的操作：
步骤1：连接数据库
步骤2：数据库的操作（增删改查）
步骤3：关闭数据库连接

以下是pymysql经常被用到的一些方法

方法	描述
pymysql.connect()	connect()方法返回要给数据库连接对象，参数可以传入很多，常用的参数有：host、port、user、password、database、charset，connect()创建了连接对象，执行完sql操作后，必须使用close()关闭
close()	数据库连接对象的一个方法，用于关闭数据库连接
cursor()	数据库连接对象的一个方法，用于获取游标对象，游标对象的execute(sql语句)方法可以执行sql语句
execute(sql)	游标对象的一个方法，可以执行sql语句
commit()	提交到数据库，数据库连接对象的一个方法，如果对表数据有修改的时候，就需要将修改提交到数据库，否则修改没有生效
rollback()	回滚已提交的内容，，数据库连接对象的一个方法，依据事务的原子性，提交要么全部生效，要么全不生效，如果遇到异常，需要对已提交的内容进行回滚

数据库连接

方法	描述
pymysql.connect()	connect()方法返回要给数据库连接对象，参数可以传入很多，常用的参数有：host、port、user、password、database、charset，connect()创建了连接对象，执行完sql操作后，必须使用close()关闭
close()	数据库连接对象的一个方法，用于关闭数据库连接

host：mysql服务器地址
port：mysql服务器连接端口，默认值为3306
user：用户名
password：密码
database：数据库名称
charset：编码方式，推荐使用"utf8mb4"

简单示例：

import pymysql# 建立连接
conn = pymysql.connect(host="localhost",port=3306,user="root",password="wen",database="mysql",charset="utf8mb4")# 关闭连接
conn.close()

数据库操作

CRUD操作

mysql的操作一般归纳为数据库、表、表数据的增删改查

功能	描述
创建表	create table 表名 (column_name column_type);
删除表	drop table 表名;
查询表数据	select 字段名1,字段名2,…,字段名n from 表名 where xxx [limit n][offset m];
插入表数据	insert into 表名(字段名1,字段名2,…,字段名n) values(值1,值2,…,值n),(值21,值22,…,值2n)… ;
更新表数据	update 表名 set 字段名1=新值1,字段名2=新值2,…,字段名n=新值n where xxx;
删除表数据	delete from 表名 where xxx;

示例：

import pymysql# 建立连接
conn = pymysql.connect(host="localhost",port=3306,user="root",password="wen",database="test",charset="utf8mb4")
# 执行操作（先获取游标对象，再执行sql语句）
cursor = conn.cursor()  # 获取游标对象cursor.execute("show tables;")
print(f"当前test库的表有(创建表前）：{cursor.fetchall()}")
# 创建表
creat_table_sql = """create table person(`id` int unsigned auto_increment primary key,`name` varchar(40) not null,`birthday` date);
"""
cursor.execute(creat_table_sql)
cursor.execute("show tables;")
print(f"当前test库的表有（创建表后）：{cursor.fetchall()}\n")# 插入表数据
cursor.execute("insert into person(name, birthday) values('wen', '2001-09-23'), ('xiaoba', '1998-04-17');")
conn.commit()   # 提交记录# 查询表数据
cursor.execute("select * from person")
print(f"person插入数据后表内容为：{cursor.fetchall()}\n")# 更新表数据
cursor.execute("update person set birthday='2001-12-17' where name='xiaoba';")
conn.commit()
cursor.execute("select * from person")
print(f"person更新数据后表内容为：{cursor.fetchall()}\n")# 删除表数据
cursor.execute("delete from person where name='wen'")
conn.commit()
print(f"person删除数据后表内容为：{cursor.fetchall()}\n")# 关闭连接
conn.close()

执行结果为：

当前test库的表有(创建表前）：()
当前test库的表有（创建表后）：(('person',),)person插入数据后表内容为：((1, 'wen', datetime.date(2001, 9, 23)), (2, 'xiaoba', datetime.date(1998, 4, 17)))person更新数据后表内容为：((1, 'wen', datetime.date(2001, 9, 23)), (2, 'xiaoba', datetime.date(2001, 12, 17)))person删除数据后表内容为：()

查询操作

游标对象提供了3种获取查询记录的方法

方法	描述
fetchone()	获取单条记录（元组形式）
fetchmany(n)	获取n条记录（元组形式）
fetchall()	获取所有结果记录（元组形式）

示例：

import pymysql# 建立连接
conn = pymysql.connect(host="localhost",port=3306,user="root",password="wen",database="test",charset="utf8mb4")
# 执行操作（先获取游标对象，再执行sql语句）
cursor = conn.cursor()  # 获取游标对象cursor.execute("select * from person where birthday>='2000-01-01'")
result_one = cursor.fetchone()
print(type(result_one))
print(f"fetchone(）查询到的内容：{result_one}\n")cursor.execute("select * from person where birthday>='2000-01-01'")
result_3 = cursor.fetchmany(3)
print(type(result_3))
print(f"fetchmany(3）查询到的内容：{result_3}\n")cursor.execute("select * from person where birthday>='2000-01-01'")
result_all = cursor.fetchall()
print(type(result_all))
print(f"fetchall(）查询到的内容：{result_all}")# 关闭连接
conn.close()

表内容为:

执行结果为：

<class 'tuple'>
fetchone(）查询到的内容：(1, 'wen', datetime.date(2001, 9, 23))<class 'tuple'>
fetchmany(3）查询到的内容：((1, 'wen', datetime.date(2001, 9, 23)), (3, '张三', datetime.date(2012, 7, 9)), (5, 'harry', datetime.date(2013, 7, 9)))<class 'tuple'>
fetchall(）查询到的内容：((1, 'wen', datetime.date(2001, 9, 23)), (3, '张三', datetime.date(2012, 7, 9)), (5, 'harry', datetime.date(2013, 7, 9)), (6, 'heng李', datetime.date(2004, 11, 26)))

执行事务

在增加、修改、删除表数据的时候，并不是说实时更新数据库的，当前连接对数据库的操作，要么全做，要么全不做。
pymysql提供了commit()和rollback()这2个方法

功能	描述
commit()	将游标的所有更新操作进行提交
rollback()	回滚当前游标的所有操作

以下是没有执行commit()方法的示例：

import pymysql# 建立连接
conn = pymysql.connect(host="localhost",port=3306,user="root",password="wen",database="test",charset="utf8mb4")
# 执行操作（先获取游标对象，再执行sql语句）
cursor = conn.cursor()  # 获取游标对象# 插入表数据
cursor.execute("insert into person(name, birthday) values('harry', '2004-11-23'), ('heng李', '1983-04-28');")# 更新表数据
cursor.execute("update person set birthday='2001-12-17' where name='xiaoba';")# 删除表数据
cursor.execute("delete from person where name='wen'")# 查询表数据
cursor.execute("select * from person")
print(f"person表内容为：{cursor.fetchall()}\n")# 关闭连接
conn.close()

控制台打印结果为：

person表内容为：((2, 'xiaoba', datetime.date(2001, 12, 17)), (3, '张三', datetime.date(2012, 7, 9)), (9, 'harry', datetime.date(2004, 11, 23)), (10, 'heng李', datetime.date(1983, 4, 28)))

test数据库的person表在脚本执行前后如图：

我们看到，在该没有进行commit的脚本中，控制台打印的数据是更新后的数据，但是实际去mysql服务器查询的时候，脚本执行前后，数据并没有变化，可以判断出：游标对象对表数据的修改并没有在表中实际生效。

使用了commit()方法

import pymysql# 建立连接
conn = pymysql.connect(host="localhost",port=3306,user="root",password="wen",database="test",charset="utf8mb4")
# 执行操作（先获取游标对象，再执行sql语句）
cursor = conn.cursor()  # 获取游标对象# 插入表数据
cursor.execute("insert into person(name, birthday) values('harry', '2004-11-23'), ('heng李', '1983-04-28');")# 更新表数据
cursor.execute("update person set birthday='2001-12-17' where name='xiaoba';")# 删除表数据
cursor.execute("delete from person where name='wen'")
# 将修改内容提交到数据库
conn.commit()# 查询表数据
cursor.execute("select * from person")
print(f"person表内容为：{cursor.fetchall()}\n")# 关闭连接
conn.close()

执行后：
控制台打印内容为：

person表内容为：((2, 'xiaoba', datetime.date(2001, 12, 17)), (3, '张三', datetime.date(2012, 7, 9)), (11, 'harry', datetime.date(2004, 11, 23)), (12, 'heng李', datetime.date(1983, 4, 28)))

mysql查询：

对比未使用commit()的场景，mysql查询person表的时候，person表内容更新了
rollback()方法一般是用于执行表操作错误后，对已更新的内容进行回滚（如果已经commit()成功的内容，就不回滚，一般来说服，数据库操作均使用try-catch-finally去兼容异常

import pymysqltry:# 建立连接conn = pymysql.connect(host="localhost",port=3306,user="root",password="wen",database="test",charset="utf8mb4")# 执行操作（先获取游标对象，再执行sql语句）cursor = conn.cursor()  # 获取游标对象# 更新表数据cursor.execute("update person set birthdaye='2001-12-17' where name='张三';")# 提交修改conn.commit()
except Exception as e:# 回滚事务print(e)print("即将对数据进行回滚")conn.rollback()
finally:# 关闭连接conn.close()

执行结果为：

(1054, "Unknown column 'birthdaye' in 'field list'")
即将对数据进行回滚