13-[Mysql]--pymysql模块

1、介绍

之前我们都是通过MySQL自带的命令行客户端工具mysql来操作数据库，那如何在python程序中操作数据库呢？这就用到了pymysql模块，该模块本质就是一个套接字客户端软件，使用前需要事先安装

pip3 install pymysql

import pymysql

user = input('user>>>').strip()
pwd = input('pwd>>>').strip()

# 建立连接
conn = pymysql.connect(
    host='127.0.0.1',       # localhost
    port=3306,
    user='root',
    password='root',
    db='db10',
    charset='utf8'
)

# 拿到游标
cursor=conn.cursor() #执行完毕返回的结果集默认以元组显示
#cursor=conn.cursor(cursor=pymysql.cursors.DictCursor)   # 以字典显示


# 执行sql语句
sql = "select * from userinfo where username='%s' and password='%s'" % (user, pwd)       #注意%s需要加引号
rows = cursor.execute(sql)            # 执行sql语句，返回sql查询成功的记录数目

# 关闭游标
cursor.close()

# 关闭连接
conn.close()

if rows:
    print('登录成功', rows)
else:
    print('登录失败', rows)

2、execute()之sql注入

　　（1）原理

　　注意：符号--会注释掉它之后的sql，正确的语法：--后至少有一个任意字符

　　根本原理：就根据程序的字符串拼接name='%s'，我们输入一个xxx' -- haha,用我们输入的xxx加'在程序中拼接成一个判断条件name='xxx' -- haha'

最后那一个空格，在一条sql语句中如果遇到select * from t1 where id > 3 -- and name='egon';则--之后的条件被注释掉了

#1、sql注入之：用户存在，绕过密码
egon' -- 任意字符

#2、sql注入之：用户不存在，绕过用户与密码
xxx' or 1=1 -- 任意字符

　　（2）解决方法：

# 原来是我们对sql进行字符串拼接
# sql="select * from userinfo where name='%s' and password='%s'" %(user,pwd)
# print(sql)
# res=cursor.execute(sql)

#改写为（execute帮我们做字符串拼接，我们无需且一定不能再为%s加引号了）
sql="select * from userinfo where name=%s and password=%s"          #！！！注意%s需要去掉引号，因为pymysql会自动为我们加上
res=cursor.execute(sql,(user,pwd）)                                  # pymysql模块自动帮我们解决sql注入的问题，只要我们按照pymysql的规矩来。

3、增、删、改：conn.commit()

import pymysql
#链接
conn=pymysql.connect(host='localhost',user='root',password='123',database='egon')
#游标
cursor=conn.cursor()

#part2
sql='insert into userinfo(name,password) values(%s,%s);'
res=cursor.execute(sql,("root","123456"))     # 增加一条数据
print(res)     # 执行sql语句，返回sql影响成功的行数

#part3
res=cursor.executemany(sql,[("root","123456"),("lhf","12356"),("eee","156")])     # 增加多条数据
print(res)

conn.commit()    # 提交后才发现表中插入记录成功
cursor.close()
conn.close()


# in方法的使用

triggerid = ('10011', '10010')
sql = "select * from triggers where triggerid in %s"
ret = handle.execute(sql, triggerid)
print(ret）

4、查：fetchone，fetchmany，fetchall

import pymysql
#链接
conn=pymysql.connect(host='localhost',user='root',password='123',database='egon')
#游标
cursor=conn.cursor()

#执行sql语句
sql='select * from userinfo;'
rows=cursor.execute(sql) #执行sql语句，返回sql影响成功的行数rows,将结果放入一个集合，等待被查询

res1=cursor.fetchone()      # 取出第一条data
res2=cursor.fetchone()      # 取出next数据
res3=cursor.fetchone()
res4=cursor.fetchmany(2)      # 取出2条数据
res5=cursor.fetchall()                  # 全部取出来
print(res1)
print(res2)
print(res3)
print(res4)
print(res5)
print('%s rows in set (0.00 sec)' %rows)


cursor.close()
conn.close()

5、相对位置，绝对位置，获取插入的最后一条的自增id

import pymysql


# 1.建立连接
conn = pymysql.connect(
    host='127.0.0.1',
    port=3306,
    user='root',
    password='root',
    db='db10',
    charset='utf8'
)


# 2、拿到游标
cursor = conn.cursor(pymysql.cursors.DictCursor)        # 以字典形式显示

sql = 'select * from userinfo '
rows = cursor.execute(sql)

cursor.scroll(3, mode='absolute')  # 绝对位置移动  f.seek
print(cursor.fetchmany(2))      # 指定几条数据 

# 5.关闭游标
cursor.close()

# 6、关闭连接
conn.close()

import pymysql


# 1.建立连接
conn = pymysql.connect(
    host='127.0.0.1',
    port=3306,
    user='root',
    password='root',
    db='db10',
    charset='utf8'
)


# 2、拿到游标
cursor = conn.cursor(pymysql.cursors.DictCursor)        # 以字典形式显示

sql = 'select * from userinfo '
rows = cursor.execute(sql)
print(cursor.fetchone())        # 打印第1条
cursor.scroll(2, mode='relative')    # 相对位置移动2个
print(cursor.fetchone())        # 打印第4条

# 5.关闭游标
cursor.close()

# 6、关闭连接
conn.close()