日常我们写程序或进行数据分析时,都需要跟本地数据库进行大量交互,涉及数据的查询、提取、创建、修改等等。pymysql是python的第三方库,作为常用的数据库交互模块,pymysql能够简单易懂得将python和Mysql联系起来。
安装:pip install pymysql
接下来开始介绍pymysql的常规用法。
一、pymysql的运行逻辑
- pymysql通过固定的写法,创建一个数据库对象,继而创建游标(cursor);
- 游标类似于此次创建出来的小工具人,它负责往返于脚本和数据库进行一系列操作。如提交sql语句、接收语句操作数据库后的返回结果等。
二、连接数据库
obj.connect(host, database, user, password, charset)
参数 | 说明 | 举例 |
---|---|---|
host | IP地址 | 127.0.0.1 |
database | 数据库名 | myKU |
user | 数据库的用户名 | admin |
password | 数据库的密码 | 123456 |
charset | 指定读写的编码 | utf-8 |
import pymysql as pm
# 创建数据库对象
db = pm.connect(host='127.0.0.1',
database='mydatabase',
user='zhangsanfeng',
password='taiqiquan123',
charset='utf-8'
)
# 为本次的数据库对象 db 创建游标对象 cursor
cursor = db.cursor()
这样就连接成功了。接下来可以无限次使用游标cursor对数据库进行操作。
三、事务查询:提交sql查询,接收查询结果
方法 | 说明 |
---|---|
cursor.execute('sql语句') | 提交sql |
cursor.fetchone() | 一条条返回结果 |
cursor.fetchall() | 一次返回所有结果 |
fetchone()方法:一条条取数据
import pymysql as pm
# 创建数据库对象
db = pm.connect(host='127.0.0.1',
database='mydatabase',
user='zhangsanfeng',
password='taiqiquan123',
charset='utf-8'
)
# 为本次的数据库对象 db 创建游标对象 cursor
cursor = db.cursor()
# 定义 sql
sql = 'SELECT * FROM namelist WHERE age>18;'
# 提交 sql
cursor.execute(sql)
result = cursor.fetchone() # 从返回的结果中取出第一条数据
num = 1
while result:
print(f'本次查询的第{num}条结果:\n{result}')
print('- ' * 20)
num += 1
result = cursor.fetchone() # 取出第二条数据
print('本次查询结束。')
fetchall()方法:一次取所有数据
allResult = cursor.fetchall() # 一次获取所有的数据
print(f'本次共查询到{len(allResult)}条结果。')
num = 1
for msg in allResult: # 遍历每条数据
print(f'第{num}条结果: \n{msg}')
num += 1
四、事务提交:提交sql事务
- 事务查询:使用sql指令查询数据,整个过程并未对数据库进行任何改动操作。
- 事务提交:使用sql指令操作改动数据库(增加、修改、删除等操作)。
使用obj.commit()方法进行事务提交。
注意,由于是对数据库操作改动,commit()方法不再针对游标对象cursor,而是针对前面创建的数据库对象db。
# 定义 sql
sql = 'DELETE FROM namelist WHERE age<18;'
# 提交 sql
cursor.execute(sql)
# 提交事务
db.commit()
print(f'删除成功。')
五、移动游标
可能你也发现了,游标就像我们电脑屏幕上闪动的光标,负责定位本次查询结果当前位置的数据,并通过指令显示出来。
针对每次的查询结果,初始游标都在首位。前面的例子中,我们使用fetchone()一条条取数据,每取完一次游标都会向后移动一位。
我们也可以通过scroll()方法操作游标的移动。这里简单介绍一下,真正用到的场景并不太多。
方法 | 说明 |
---|---|
cursor.scroll(2, mode='absolute') | 绝对位置:以初始位置向后移动2位 |
cursor.scroll(-2, mode='relative') | 相对位置:以当前位置向前移动2位 |
六、关闭游标、关闭数据库连接
cursor.close():关闭游标
obj.close():关闭数据库
切记最后不要忘了关闭连接。
import pymysql as pm
# 创建数据库对象
db = pm.connect(host='127.0.0.1',
database='mydatabase',
user='zhangsanfeng',
password='taiqiquan123',
charset='utf-8'
)
# 为本次的数据库对象 db 创建游标对象 cursor
cursor = db.cursor()
# 定义 sql
sql = 'DELETE FROM namelist WHERE age<18;'
# 提交 sql
cursor.execute(sql)
# 提交事务
db.commit()
print(f'删除成功。')
cursor.close()
db.close()