MySQL数据库篇之pymysql模块的使用

主要内容：

　　一、pymysql模块的使用

　　二、pymysq模块增删改查

1️⃣  pymsql模块的使用

　　1、前言：之前我们都是通过MySQL自带的命令行客户端工具mysql来操作数据库，

　　那如何在python程序中操作数据库呢？这就用到了pymysql模块，该模块本质就是一个套接字

　　客户端软件，使用前需要事先安装。

pip3 install pymysql

　　2、实例：

#！/user/bin/env python3
# -*- coding:utf-8-*-
# write by congcong

import pymysql

user = input('user>>:').strip()
pwd = input('password>>:').strip()
# 建立链接
conn=pymysql.connect(
    host='127.0.0.1',
    port=3306,
    user='root',
    password='0514',
    db='db6',
    charset='utf8'
)
# 拿到游标
cursor = conn.cursor() # 执行完毕后返回的结果集默认以元祖显示
# 执行sql语句
sql = 'select * from userinfo where name="%s" and pwd="%s"' %(user,pwd)
print(sql)
rows = cursor.execute(sql) # 受影响的行数，execute表示执行

cursor.close() # 关闭游标
conn.close() # 关闭链接

if rows:
    print('登陆成功！')
else:
    print('失败...')
 

注意：

　　这种方式存在很大的隐患，即sql注入问题，可绕过密码登录，如：cc1" -- hello 

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

 user>>:cc1" -- hello    # “ 对应左边的左引号，”--“ 在sql语句中表示注释后面的语句，注意中间有空格
    password>>:
    select * from userinfo where name="cc1" -- hello" and pwd=""
    登陆成功！

# 甚至可同时绕过用户名和密码，如：xxxx" or 1=1 -- hello
'''
user>>:xxx" or 1=1 -- hello     # or表示或，用户名满足任意一条件即可
password>>:
select * from userinfo where name="xxx" or 1=1 -- hello" and pwd=""
登陆成功！
'''

　　上述程序的改进版如下：

#！/user/bin/env python3
# -*- coding:utf-8-*-
# write by congcong

import pymysql
# 改进版如下：
user = input('用户名>>:').strip()
pwd = input('密码>>:').strip()
# 建立链接
conn = pymysql.connect(
    host='localhost',
    port= 3306,
    user = 'root',
    password = '0514',
    db='db6',
    charset = 'utf8'
)
# 拿到游标
cursor = conn.cursor()
# 执行
sql = 'select * from userinfo where name=%s and pwd=%s'
# print(sql)
rows = cursor.execute(sql,(user,pwd))

cursor.close()
conn.close()
if rows:
    print('welcome login!')
else:
    print('failed...')

2️⃣  pymysq模块增删改查

　　1、插入数据

#！/user/bin/env python3
# -*- coding:utf-8-*-
# write by congcong
import pymysql

#1、建链接
conn = pymysql.connect(
    host = '127.0.0.1',  # 本地主机
    port = 3306, # mysql默认端口
    user = 'root',  # MySQL用户
    password = '0514', # 用户密码
    db = 'db6',   # 数据库
    charset = 'utf8'  # 字符编码
)

# 2、取游标
cursor = conn.cursor()

# 3、执行sql语句
'''
# 插入数据前
    mysql> select * from userinfo;
    +----+------+-----+
    | id | name | pwd |
    +----+------+-----+
    |  1 | cc1  | 123 |
    |  2 | hyt  | 456 |
    +----+------+-----+
    2 rows in set (0.00 sec)
    
# 插入（增）
sql = 'insert into userinfo(name,pwd) values(%s,%s)'
# rows = cursor.execute(sql,('cc2','666')) # 插入单行语句
rows = cursor.executemany(sql,[('hy1','514'),('hyt2','0514'),('cc2','666888')]) # 插入多条数据
print(rows)

# 插入数据后
    mysql> select * from userinfo;
    +----+------+--------+
    | id | name | pwd    |
    +----+------+--------+
    |  1 | cc1  |    123 |
    |  2 | hyt  |    456 |
    |  3 | cc2  |    666 |
    |  4 | hy1  |    514 |
    |  5 | hyt2 |    514 |
    |  6 | cc2  | 666888 |
    +----+------+--------+
    6 rows in set (0.00 sec)
'''

sql = 'insert into userinfo(name,pwd) values (%s,%s)'
rows = cursor.executemany(sql,[('sc1','111'),('sc2','222')])
print(rows) # 2
# 查询表中最后一条数据的id,必须在插入后查询
print(cursor.lastrowid) # 7,返回的数据是现在插入的数据起始id
'''
    mysql> select * from userinfo;
    +----+------+-----+
    | id | name | pwd |
    +----+------+-----+
    |  1 | 0    | 123 |
    |  2 | hyt  | 456 |
    |  3 | cc2  | 666 |
    |  7 | sc1  | 111 |
    |  8 | sc2  | 222 |
    +----+------+-----+
    5 rows in set (0.00 sec)
'''

conn.commit() # 更新数据表，更新后才能在查询插入后的结果

cursor.close() # 关闭游标
conn.close() # 关闭链接

　　2、删除数据

import pymysql

#1、建链接
conn = pymysql.connect(
    host = '127.0.0.1',  # 本地主机
    port = 3306, # mysql默认端口
    user = 'root',  # MySQL用户
    password = '0514', # 用户密码
    db = 'db6',   # 数据库
    charset = 'utf8'  # 字符编码
)

# 2、取游标
cursor = conn.cursor()

# 3、执行sql语句
# 删除
sql = 'delete  from userinfo where id=%s'
# rows=cursor.execute(sql,6) # 删除一条
rows=cursor.executemany(sql,[4,5,6]) # 删除多条

    mysql> select * from userinfo;
    +----+------+-----+
    | id | name | pwd |
    +----+------+-----+
    |  1 | cc1  | 123 |
    |  2 | hyt  | 456 |
    |  3 | cc2  | 666 |
    +----+------+-----+
    3 rows in set (0.00 sec)

conn.commit() # 更新数据表

cursor.close() # 关闭游标
conn.close() # 关闭链接

　　3、修改数据

import pymysql

#1、建链接
conn = pymysql.connect(
    host = '127.0.0.1',  # 本地主机
    port = 3306, # mysql默认端口
    user = 'root',  # MySQL用户
    password = '0514', # 用户密码
    db = 'db6',   # 数据库
    charset = 'utf8'  # 字符编码
)

# 2、取游标
cursor = conn.cursor()

# 3、执行sql语句
# 修改
sql = 'update userinfo set name="cc" where id=%s '
cursor.execute(sql,1)

    mysql> select * from userinfo;
    +----+------+-----+
    | id | name | pwd |
    +----+------+-----+
    |  1 | cc   | 123 |
    |  2 | hyt  | 456 |
    |  3 | cc2  | 666 |
    +----+------+-----+
    3 rows in set (0.00 sec)

conn.commit() # 更新数据表

cursor.close() # 关闭游标
conn.close() # 关闭链接

　　4、查询数据

import pymysql

#1、建链接
conn = pymysql.connect(
    host = '127.0.0.1',  # 本地主机
    port = 3306, # mysql默认端口
    user = 'root',  # MySQL用户
    password = '0514', # 用户密码
    db = 'db6',   # 数据库
    charset = 'utf8'  # 字符编码
)

# 2、取游标
cursor = conn.cursor()

# 3、执行sql语句
查询数据

cursor = conn.cursor(pymysql.cursors.DictCursor) # 取游标，数据以字典形式取出
sql = 'select * from userinfo'
rows = cursor.execute(sql) # 执行sql语句，返回sql影响成功的行数rows,将结果放入一个集合，等待被查询
# rows = cursor.execute('select * from userinfo;')
# fetchone 一次取一条数据
# print(cursor.fetchone()) # {'id': 1, 'name': '0', 'pwd': 123}
# print(cursor.fetchone())
# print(cursor.fetchone())
# print(cursor.fetchone()) # 不会报错，取完后返回None

# fetchmany 一次取多条数据,超过数据条数大小时不报错
#print(cursor.fetchmany(2))  # [{'id': 1, 'name': '0', 'pwd': 123}, {'id': 2, 'name': 'hyt', 'pwd': 456}]

# fetchall 一次取出全部数据
# print(cursor.fetchall())

# cursor.scroll(1,mode='absolute') # 相对绝对位置移动,从头向后移动一条数据
# print(cursor.fetchone()) # {'id': 2, 'name': 'hyt', 'pwd': 456}
print(cursor.fetchone())
cursor.scroll(1,mode='relative') # 相对当前位置移动,从上一条查询数据向后移动一条数据
print(cursor.fetchone())
'''
conn.commit() # 更新数据表

cursor.close() # 关闭游标
conn.close() # 关闭链接