pymysql模块

时间:2023-03-10 06:15:37
pymysql模块
一、pymysql模块
1、说明:
想在python代码中连接上mysql数据库,就需要使用pymysql模块,
pymysql是在 Python3.x 版本中用于连接 MySQL 服务器的一个库,在Python2中则使用mysqldb。
Django中也可以使用PyMySQL连接MySQL数据库。 2、 安装
在cmd窗口执行命令行:pip install pymysql # 注意:如果电脑上安装了两个版本的python解释器,则应该明确pip的版本,pip2还是pip3。
补充
pip -V --> 查看当前pip的版本
pip list --> 查看当前python解释器环境中安装的第三方包和版本(安装完成后可用此命令查看是否安装成功) 3、注意事项
应该确定你有一个MySQL数据库,并且已经启动
应该确定你有可以连接该数据库的用户名和密码
应该确定你有一个有权限操作的database 4、使用步骤
1. 导入pymysql模块
import pymysql 2. 连接database
conn = pymysql.connect(
host='127.0.0.1', # 数据库的IP地址
port=3306, # 数据库的端口
user='root', # 用户名
password='123abc' # 密码
database='test', # 具体的一个数据库(文件夹)
charset='urf8' # 编码方式,注意:charset='utf8',不要写成'utf-8'
) 3. 获取光标对象
cursor = conn.cursor() 4. 执行SQL语句
cursor.execute('select * from userinfo;') 5. 关闭
1. 关闭光标
cursor.close() 2. 关闭连接
conn.close() 6.例子
import pymysql # 获取用户输入
name = input('请输入用户名:')
pwd = input('请输入密码:') # 连接数据库,得到一个连接
conn = pymysql.connect(
host='127.0.0.1',
port=3306, # mysql默认端口3306
user='root',
password='123abc',
database='test',
charset='utf8'
) # 获取光标
cursor = conn.cursor() # sql语句
sql = "select * from userinfo where username='%s' and password='%s';" % (name, pwd)
print(sql) # 执行sql语句
ret = cursor.execute(sql)
print(ret) # ret是受影响的记录数量,若不为空,则代表查到相应的记录 # 关闭
cursor.close()
conn.close() # 结果
if ret:
print('登陆成功')
else:
print('登录失败') 结果:
请输入用户名:ming
请输入密码:112233 select * from userinfo where username='ming' and password=''
1
登陆成功 5、 SQL注入问题
1. 什么是SQL注入?
用户输入的内容有恶意的SQL语句,后端拿到用户输入的内容不做检测直接做字符串拼接,得到一个和预期不一致的SQL语句 例如:上面例子的代码完全不变,但是我这样输入:
请输入用户名:ming' --
请输入密码:12345 select * from userinfo where username='ming' -- ' and password='12345';
1
登陆成功 结果是登录成功了,为什么?
这是因为我在输入用户名的时候,输入的是ming' --
在sql语句中,--代表注释的意思,也就是说
select * from userinfo where username='ming' -- ' and password='12345'; 这句sql语句相当于
select * from userinfo where username='ming'
那肯定能登录成功啊,因为我数据库中就是有这个用户,相当于找用户名,有这个用户就成功,并没有去匹配密码,
这就是恶意注入的问题。 2. 如何解决SQL注入?
对用户输入的内容做检测,有引号怎么处理,注释怎么处理,过程很麻烦,但是,
pymysql内置了这种检测,我们可以直接使用,所以我们不应该自己拼接sql语句,而是让pymysql帮我们拼接sql语句。
cursor.execute(sql, [name, pwd]) # 让pymysql模块帮我们拼接sql语句,执行SQL语句 例子:
import pymysql # 获取用户输入
name = input('请输入用户名:')
pwd = input('请输入密码:') # 连接数据库,得到一个连接
conn = pymysql.connect(
host='127.0.0.1',
port=3306, # mysql默认端口3306
user='root',
password='123abc',
database='test',
charset='utf8'
) # 获取光标
cursor = conn.cursor() # sql语句:按照pymysql模块的写法定义好占位符,pymysql只有%s一种占位符
sql = "select * from userinfo where username=%s and password=%s;"
print(sql) # 执行sql语句
ret = cursor.execute(sql,[name,pwd]) # 让pymysql模块帮我们拼接sql语句,执行SQL语句
print(ret) # ret是受影响的记录数量,若不为空,则代表查到相应的记录 # 关闭
cursor.close()
conn.close() # 结果
if ret:
print('登陆成功')
else:
print('登录失败') 结果1:
请输入用户名:ming' --
请输入密码:12345 select * from userinfo where username=%s and password=%s;
0
登录失败 结果2:
请输入用户名:ming
请输入密码:112233 select * from userinfo where username=%s and password=%s;
1
登陆成功 6、pymysql的使用
涉及到修改数据库内容的时候,一定要提交:conn.commit() 1. 增、删、改
import pymysql # 连接数据库,得到一个连接
conn = pymysql.connect(
host='127.0.0.1',
port=3306,
user='root',
password='123abc',
database='test',
charset='utf8'
) # 获取光标
cursor = conn.cursor() # 得到SQL语句
sql1 = "insert into userinfo(username, password) values (%s,%s);" # 增
sql2 = "delete from userinfo where username=%s;" # 删
sql3 = "update userinfo set password=%s where username=%s;" # 改 # 使用光标对象执行SQL语句
cursor.execute(sql1, ['sb', '']) # 让pymysql模块帮我们拼接sql语句,执行SQL语句
# 涉及到修改数据库内容,一定要提交
conn.commit() cursor.execute(sql2, ['ming'])
conn.commit() cursor.execute(sql3, ['', 'sb'])
conn.commit() # 关闭
cursor.close()
conn.close() 2. 查
1. 返回的数据类型
1. 默认返回的元组,且每个元素也是用元组
2. 可以设置为返回的是列表,列表中的每个元素是字典
cursor = conn.cursor(cursor=pymysql.cursors.DictCursor)
2. 常用的方法
1. fetchall() 返回所有查询到的记录
2. fetchone() 返回一条查询到的记录
3. fetchmany(size) 返回size条查询到的记录
4.cursor.scroll(1, mode="absolute") 光标按绝对位置移动
5.cursor.scroll(1, mode="relative") 光标按照相对位置(当前位置)移动 例子1:
import pymysql # 连接数据库,得到一个连接
conn = pymysql.connect(
host='127.0.0.1',
port=3306,
user='root',
password='123abc',
database='test',
charset='utf8'
) # 获取光标
cursor = conn.cursor() # 得到SQL语句
sql = "select * from userinfo;" # 使用光标对象执行SQL语句
cursor.execute(sql) # 没有参数则不用拼接,直接执行 ret = cursor.fetchall()
print(ret)
# 关闭
cursor.close()
conn.close() 结果1:
(('hong', 123), ('sb', 123), ('ming', 456), ('dong', 789)) 例子2:
import pymysql # 连接数据库,得到一个连接
conn = pymysql.connect(
host='127.0.0.1',
port=3306,
user='root',
password='123abc',
database='test',
charset='utf8'
) # 获取光标,设置查询结果为列表且元素为字典
cursor = conn.cursor(cursor=pymysql.cursors.DictCursor) # 得到SQL语句
sql = "select * from userinfo;" # 使用光标对象执行SQL语句
cursor.execute(sql) # 没有参数则不用拼接,直接执行 # 查询所有
# ret = cursor.fetchall()
# print(ret) # 查询单条记录
# ret = cursor.fetchone()
# print(ret) # ret = cursor.fetchone()
# print(ret) # 查询指定数量的数据
ret = cursor.fetchmany(2)
print(ret) # [{'username': 'hong', 'password': 123}, {'username': 'sb', 'password': 123}] print(cursor.fetchone()) # {'username': 'ming', 'password': 456}
print(cursor.fetchone()) # {'username': 'dong', 'password': 789} # 移动光标
cursor.scroll(-1, mode='relative') # 相对位置,基于光标当前位置移动
print(cursor.fetchone()) # {'username': 'dong', 'password': 789} cursor.scroll(0, mode='absolute') # 绝对位置,你让光标移动到哪里就到哪里
print(cursor.fetchone()) # {'username': 'hong', 'password': 123} # 关闭
cursor.close()
conn.close() 3.批量增(插入) import pymysql # 连接数据库,得到一个连接
conn = pymysql.connect(
host='127.0.0.1',
port=3306,
user='root',
password='123abc',
database='test',
charset='utf8'
) # 获取光标
cursor = conn.cursor() # 得到SQL语句
sql = "insert into userinfo(username, password) values (%s,%s);" # 增 # 数据
data = [("a", 18), ("b", 19), ("c", 20)] # 使用光标对象执行SQL语句
cursor.executemany(sql, data) # 批量增使用executemany # 涉及到修改数据库内容,一定要提交
conn.commit() # 关闭
cursor.close()
conn.close() 5、
conn.rollback():在执行增删改操作时,如果不想提交前面的操作,可以使用 rollback() 回滚取消操作。 cursor.lastrowid:获取插入数据的ID(关联操作时会用到) # 导入pymysql模块
import pymysql # 连接数据库,得到一个连接
conn = pymysql.connect(
host='127.0.0.1',
port=3306,
user='root',
password='123abc',
database='test',
charset='utf8'
) # 得到一个可以执行SQL语句的光标对象
cursor = conn.cursor() # sql语句
sql = "insert into userinfo(username, password) values (%s,%s);" try:
# 执行SQL语句
cursor.execute(sql, ['xiazi', 58]) # 提交
conn.commit()
# 提交之后,获取刚插入的数据的ID
last_id = cursor.lastrowid except Exception as e:
# 有异常,回滚取消操作
conn.rollback() cursor.close()
conn.close()