13 pymysql模块
pymysql模块
pymysql是一个第三方库，如果自己的计算机上没有可以在终端使用命令进行安装。
sudo pip3 install pymysql
- pymysql使用流程
- 建立数据库连接：db = pymysql.connect(…)
- 创建游标对象：cur = db.cursor()
- 游标方法: cur.execute(“insert …”)
- 提交到数据库或者获取数据 : db.commit() / cur.fetchall()
- 关闭游标对象 ：cur.close()
- 断开数据库连接 ：db.close()
- 常用函数
db = pymysql.connect(参数列表)
功能: 链接数据库
host ：主机地址,本地 localhost
port ：端口号,默认3306
user ：用户名
password ：密码
database ：库
charset ：编码方式,推荐使用 utf8
cur = db.cursor()
功能： 创建游标
返回值：返回游标对象,用于执行具体SQL命令
cur.execute(sql,args_list)
功能： 执行SQL命令
参数： sql sql语句
args_list 列表，用于给sql语句传递参量
cur.executemany(sql命令,args_list)
功能： 多次执行SQL命令，执行次数由列表中元组数量决定
参数： sql sql语句
args_list 列表中包含元组 每个元组用于给sql语句传递参量，一般用于写操作。
db.commit() 提交到数据库执行，必须支持事务操作才有效
db.rollback() 回到原来的数据形态，必须支持事务操作才有效
cur.fetchone() 获取查询结果集的第一条数据，查找到返回一个元组否则返回None
cur.fetchmany(n) 获取前n条查找到的记录，返回结果为元组嵌套元组， ((记录1),(记录2))，查询不到内容返回空元组。
cur.fetchall() 获取所有查找到的记录，返回结果形式同上。
cur.close() 关闭游标对象
db.close() 关闭数据库连接
连接关闭数据库示例：
"""
数据库操作基本结构示例
"""
import pymysql
kwargs = {
"host": "localhost",
"port": 3306,
"user": "root",
"passwd": "123456",
"database": "stu",
"charset": "utf8"
}
# 链接数据库
db = pymysql.connect(**kwargs)
# 生成游标 游标：调用sql操作数据得到操作结果的对象
cur = db.cursor()
# 操作数据 读 / 写
# 关闭
cur.close()
db.close()
数据库读操作示例：
"""
数据库读操作示例
"""
import pymysql
kwargs = {
"host": "localhost",
"port": 3306,
"user": "root",
"passwd": "123456",
"database": "stu",
"charset": "utf8"
}
# 链接数据库
db = pymysql.connect(**kwargs)
# 生成游标 游标：调用sql操作数据得到操作结果的对象
cur = db.cursor()
# 操作数据 读
# sql = "select name,age,score from class where score>80;"
# cur.execute(sql) # 执行语句
# 给sql语句传值 使用第二个参数按照位置给sql语句%s传值
sql = "select name,age,score from class " \
"where score>%s and sex=%s;"
cur.execute(sql,[80,'m'])
# 一个查询结果
one = cur.fetchone()
print(one)
# 多个查询结果
many = cur.fetchmany(2)
print(many)
# 所有查询结果
all = cur.fetchall()
print(all)
# 迭代获取结果
# for row in cur:
# print(row)
# 关闭
cur.close()
db.close()
数据库读操作示例：在stu下创建一个数据表user 有三个字段 ：id user password
create table user(
id int primary key auto_increment,
user varchar(30),
password char(16));
其中有若干数据：
insert into user values
(1,"Tom",'abc'),
(2,"Abby","321");
编写一个函数，使用input分别输入用户名和密码
验证是否可以登录成功，如果可以则返回True
不可以返回False
"""
import pymysql
class User:
def __init__(self):
kwargs = {
"host": "localhost",
"port": 3306,
"user": "root",
"passwd": "123456",
"database": "stu",
"charset": "utf8"
}
self.db = pymysql.connect(**kwargs)
self.cur = self.db.cursor()
def close(self):
self.cur.close()
self.db.close()
def login(self,name, passwd):
sql="select user from user " \
"where user=%s and password=%s;"
self.cur.execute(sql,[name,passwd])
# (name,) or None
if self.cur.fetchone():
return True
else:
return False
if __name__ == '__main__':
user = User()
# 登录验证
name=input("用户名：")
passwd=input("密 码：")
print(user.login(name,passwd))
user.close()
文件写操作示例：
"""
数据库写操作示例
pymysql进行写操作如果数据表支持事务会自动开启事务
需要commit提交才能生效
如果数据表不支持事务则直接execute执行则生效
"""
import pymysql
kwargs = {
"host": "localhost",
"port": 3306,
"user": "root",
"passwd": "123456",
"database": "stu",
"charset": "utf8"
}
# 链接数据库
db = pymysql.connect(**kwargs)
# 生成游标 游标：调用sql操作数据得到操作结果的对象
cur = db.cursor()
# 操作数据 写 insert delete update
# try:
# sql = "update class set score=%s where name=%s;"
# cur.execute(sql,[99,"Tom"])
# db.commit() # 提交事务
# except Exception as e:
# print(e)
# db.rollback()
# 批量写操作
data = [
("zhang",'19','w',70),
("wang",'18','m',71),
("li",'20','w',72)
]
try:
sql = "insert into class " \
"(name,age,sex,score) " \
"values (%s,%s,%s,%s);"
cur.executemany(sql,data) # 多次执行语句
db.commit() # 提交事务
except Exception as e:
print(e)
db.rollback()
# 关闭
cur.close()
db.close()
写操作示例：
创建数据库 dict，数据库中创建数据表 words 字段为 ： id word mean
编写一个程序，将dict.txt单词本中的单词存储到这个数据表里
库 ： dict
create database dict charset=utf8;
表 ： words --> id word mean
create table words (
id int primary key auto_increment,
word char(30),
mean varchar(512)
);
import pymysql
import re
class Dict:
def __init__(self):
self.kwargs = {
"host": "localhost",
"port": 3306,
"user": "root",
"password": "123456",
"database": "dict",
"charset": "utf8"
}
self.connect()
# 完成数据库连接
def connect(self):
self.db = pymysql.connect(**self.kwargs)
self.cur = self.db.cursor()
# 关闭
def close(self):
self.cur.close()
self.db.close()
# 插入数据 提取文件中的单词
def insert_words(self, filename):
file = open(filename)
words = [] # [(word,mean),(word,mean)...]
for line in file:
tmp = re.findall(r"(\w+)\s+(.*)", line)
words += tmp # 大列表
file.close()
self.insert_data(words)
def insert_data(self, words):
try:
sql = "insert into words (word,mean) values (%s,%s);"
self.cur.executemany(sql, words)
self.db.commit() # 提交事务
except Exception as e:
print(e)
self.db.rollback() # 事务回滚
if __name__ == '__main__':
dict = Dict()
dict.insert_words("dict.txt") # 插入单词数据
dict.close()
- 文件存储
- 存储文件路径
- 优点：节省数据库空间，提取方便
- 缺点：文件或者数据库发生迁移会导致文件丢失
- 存储文件本身
- 优点：安全可靠，数据库在文件就在
- 缺点：占用数据库空间大，文件存取效率低
- 存储文件路径
静态文件存储示例：
"""
存取二进制数据
"""
import pymysql
kwargs = {
"host": "localhost",
"port": 3306,
"user": "root",
"password": "123456",
"database": "stu",
"charset": "utf8"
}
# 连接数据库
db = pymysql.connect(**kwargs)
# 创建游标 使用sql操作数据得到结果的对象
cur = db.cursor()
# 存储图片
# with open("left.png",'rb') as f:
# data = f.read()
# sql = "update class set image=%s where id=4;"
# cur.execute(sql,[data])
# db.commit()
# 取出图片
sql = "select image from class where id=4;"
cur.execute(sql)
data = cur.fetchone()[0]
with open("xxx.png",'wb') as f:
f.write(data)
# 关闭
cur.close()
db.close()