数据库入门-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="", # 密码

    database="test", #数据库名称

    port=3306, # 端口号 可选 整型

    charset="utf8" # 编码  可选

)

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

import pymysql

def connect_mysql():

    db_info = {

        'host':'192.168.13.253',

        'user':'zfj',

        'password':'',

        'port':3306,

        'database':'day40',

        'charset':'utf8'

    }

    try:

        db = pymysql.Connect(**db_info)

        print('连接成功！')

    except Exception as e:

        print('e')

    return db

if __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（元组，列表或字典） - 与查询一起使用的参数。（可选的）

    返回：受影响的行数

    返回类型：INT

executemany（查询，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 = "小花";

四、安全问题

如何保证数据安全是近几年来火热的主题之一，这里不细讲，就说一些和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

巴特西