数据库入门-pymysql模块的使用

一、pymysql模块安装

由于本人的Python版本为python3.7，所以用pymysql来连接数据库（mysqldb不支持python3.x）

方法一：

#在cmd输入
pip3 install pymysql

方法二（pycharm IDE）：

[File] >> [settings] >> [Project: study] >> [Project Interpreter] >>点击右上角“+”号，搜索框输入“pymysql”>> [Install按钮]

二、连接数据库

import pymysql
# 创建链接得到一个链接对象
conn = pymysql.Connect(host="127.0.0.1",    # 数据库服务器主机地址user="root",  # 用户名password="123456", # 密码database="test", #数据库名称port=3306, # 端口号 可选 整型charset="utf8" # 编码  可选
)

还可以使用函数的形式连接数据库：

import pymysqldef connect_mysql():db_info = {'host':'192.168.13.253','user':'zfj','password':'123','port':3306,'database':'day40','charset':'utf8'}try:db = pymysql.Connect(**db_info)print('连接成功！')except Exception as e:print('e')return dbif __name__ == '__main__':db = connect_mysql()

函数的形式

三、操作数据库

1.连接对象的常用方法

commit（）#提交稳定存储的更改
rollback（）#回滚当前事务
autocommit_mode=无 #指定的自动提交模式。无表示使用服务器默认值。

要想操作数据库必须先建立游标对象，不需要自己创建调用数据库对象下面的cursor方法就可以了

2.游标对象

创建游标：

db = pymysql.connect(config)  # 创建数据库链接对象
cus = db.cursor     # 创建游标对象
print(dir(cus))     # 查看游标的方法

游标常用方法：

cus.cursor()   #创建游标对象
cus.close()  #关闭游标对象cus.excute(query,args=None)  #执行查询参数：    query（str） - 要执行的查询。args（元组，列表或字典） - 与查询一起使用的参数。（可选的）返回：受影响的行数返回类型：INTexecutemany（查询，args ） #针对一个查询运行多个数据参数：    query - 要在服务器上执行的查询args - 序列或映射的序列。它用作参数。
此方法可提高多行INSERT和REPLACE的性能。否则它等同于使用execute（）循环遍历args。cus.fetchone()  #获取下一行
cus.fetchall()   #获取所有行
cus.fetchmany(size =None)  #获取几行

注：sql必须是字符串类型

3.示例

import pymysql
# 创建链接得到一个链接对象
conn = pymysql.Connect(host="127.0.0.1",    # 数据库服务器主机地址user="root",  # 用户名password="admin", # 密码database="day42", #数据库名称port=3306, # 端口号 可选 整型charset="utf8" # 编码  可选
)
# 获取游标对象  pymysql.cursors.DictCursor指定 返回的结果类型 为字典  默认是元祖类型
cursor = conn.cursor(pymysql.cursors.DictCursor)# # 添加数据
# res = cursor.execute("insert into emp values(100,'胡歌','男',30,1,'job',60000)")
# if res:
#     print("插入成功")
# else:
#     print("插入失败")# 提交修改   因为pymysql 模块默认是启用事务的  你的sql语句 如果不提交 相当于没有执行# conn.commit()
# res = cursor.execute("drop database day42")# res = cursor.execute("delete from t1 where id = 1")
# print(res)try:cursor.execute("update moneyTable set money = money - 50 where name = '小明'")#如果小花的账户出问题了 无法更新数据 那就需要回滚cursor.execute("update moneyTable set money = money + 50 where name = '小花'")conn.commit()
except:conn.rollback()cursor.close()
conn.close()#  小明有100块 准备给小花转50
# update moneyTable set money = money - 50 where name = "小明";
# 发生一些别错误  如果发生了错误 就执行撤销操作 rollback;
# update moneyTable set money = money + 50 where name = "小花";

View Code

四、安全问题

如何保证数据安全是近几年来火热的主题之一，这里不细讲，就说一些和pymysql相关的sql注入攻击。

#sql = "select *from user where user = '%s' and pwd = '%s';" % (input("input userName"),input("input password"))# 当用户输入的用户名为字符串 为 yy' --   时
# 最终产生的sql  select *from user where user = 'yy' -- ' and pwd = '987657890';
# -- 用于mysql注释  意思是 后面的内容忽略掉
# 从而导致 密码是否正确都能登录成功
# "select *from user where user = 'axxax' or 1=1;

那python是如何避免普通的sql注入的呢？

通过excute函数，将需要传的参数放到arg参数中，让pymysql帮你屏蔽

count = cursor.execute("select *from user where user = %s and pwd = %s;",args=(input("user"),input("pwd")))print(count)
if count:print("login success")
else:print("login error")cursor.close()
conn.close()

读者想要学习更多关于安全的内容可以关注“实验吧”学习：http://www.shiyanbar.com/

部分参考 pymysql文档：https://pymysql.readthedocs.io/en/latest/index.html

连接对象规范：https://www.python.org/dev/peps/pep-0249/#connection-objects

转载于:https://www.cnblogs.com/mangM/p/9665049.html