Python之pymysql详解

PyMySQL

PyMySQL 是在 Python3.x 版本中用于连接 MySQL 服务器的一个库，Python2中则使用mysqldb。

安装pymysql

PyMySQL ：这是一个使Python连接到MySQL的库，它是一个纯Python库。

PyMySQL是一个开源项目：https://github.com/PyMySQL/PyMySQL

通过下面的命令来进行安装pymysql模块：

pip3 install PyMySQL

连接数据库

安装好pymysql后，在python程序中可以用 import pymysql 来导入模块。

在连接数据库前，首先要确定数据库服务端已启动。在数据库中有已经创建好的数据库，这里连接数据库‘mydb’，使用的用户名为‘root’，密码为‘mysql’，你可以可以自己设定。

比较常用的参数包括:

host:数据库主机名.默认是用本地主机
user:数据库登陆名.默认是当前用户
passwd:数据库登陆的秘密.默认为空
db:要使用的数据库名.没有默认值
port:MySQL服务使用的TCP端口.默认是3306
charset:数据库编码

导入pymysql
import pymysql
 
打开数据库连接
db = pymysql.connect(
    host = '127.0.0.1',   服务端地址
    port = 3306,    服务端端口
    user = 'root',   用户名		
    password = 'mysql',   密码
    database = 'mydb',   要连接的数据库
    charset = 'utf8'    设置数据库编码
)

使用 cursor() 方法创建一个游标对象 cursor
cursor = db.cursor()

关闭数据库连接
db.close()

创建表

导入pymysql
import pymysql
 
打开数据库连接
db = pymysql.connect(
    host = '127.0.0.1',
    port = 3306, 
    user = 'root', 		
    password = 'mysql', 
    database = 'mydb', 
    charset = 'utf8' 
)
cursor = db.cursor()

使用预处理语句创建表，要确定数据库中不存在该表，如果存在会报错InternalError，无法进行创建
sql = """CREATE TABLE EMPLOYEE (
		ID  Int NOT NULL AUTO_INCREMENT,
         NAME  CHAR(20) NOT NULL,
         PASSWORD  CHAR(20),
         AGE INT,  
         SEX CHAR(1),
         PRIMARY KEY (ID))"""
使用 execute()  方法执行 SQL语句
cursor.execute(sql)

db.close()

插入数据

这个连接对象也提供了对事务操作的支持,标准的方法:

commit() 提交数据
rollback() 发生错误后回滚

commit()方法游标的所有更新操作，rollback（）方法回滚当前游标的所有操作

import pymysql

db = pymysql.connect(
    host = '127.0.0.1', 
    port = 3306,
    user = 'root',
    password = 'mysql',  
    database = 'mydb',  
    charset = 'utf8' 
)

cursor = db.cursor()
sql = """INSERT INTO EMPLOYEE(NAME,
         PASSWORD, AGE, SEX)
         VALUES ('LinWoW', 'mysql', 20, 'M')"""
try：
	cursor.execute(sql)
	确认数据无误之后 commit之后才会将数据真正修改到数据库，如果不执行，程序运行正常，但是数据实际未插入成功
	db.commit()
except:
    print('插入数据失败！')
    发生错误时回滚
    db.rollback()
db.close()

为了操作方便，可以在打开数据库连接的参数中添加下面的语句，修改完数据后就不需要手动确认数据执行。

autocommit = True

查询数据

Python查询Mysql使用 fetchone() 方法获取单条数据, 使用fetchall() 方法获取多条数据。

fetchone(): 该方法获取下一个查询结果集。结果集是一个对象
etchall(): 接收全部的返回结果行.
rowcount: 这是一个只读属性，并返回执行execute()方法后影响的行数。

import pymysql
db = pymysql.connect(
    host = '127.0.0.1', 
    port = 3306,
    user = 'root',
    password = 'mysql',  
    database = 'mydb',  
    charset = 'utf8',
    autocommit = True
)

cursor = db.cursor(pymysql.cursors.DictCursor)  设置为字典的形式返回数据（默认是元组）
使用 execute()  方法执行 SQL 查询 ，这里返回的是查询到数据的总条数
cursor.execute("SELECT * from employee")    
 
使用 fetchone() 方法获取查询到的单条数据，可以利用该函数实现一条一条数据的读取
data1 = cursor.fetchone()
print (data1)
'''
不设置数据返回类型打印的结果
(2, 'LinWoW', 'mysql', 20, 'M')
设置数据返回类型为字典后打印的结果
{'ID': 2, 'NAME': 'LinWoW', 'PASSWORD': 'mysql', 'AGE': 20, 'SEX': 'M'}
'''

使用 fetchall() 方法获取所有查询到的所有数据
data2 = cursor.fetchall()
print (data2)
'''
[{'ID': 3, 'NAME': 'LinWoW', 'PASSWORD': 'mysql', 'AGE': 20, 'SEX': 'M'}, {'ID': 4, 'NAME': 'LinWoW', 'PASSWORD': 'mysql', 'AGE': 20, 'SEX': 'M'}]
'''
scroll()类似文件操作中的seek(),第一个参数表示偏移量，第二个参数表示偏移的起始位置

绝对移动，参照开始的位置
cursor.scroll(1,'absolute')

相对移动，参照当前光标的位置
cursor.scroll(1,'relative')
 
db.close()

更新数据

import pymysql

db = pymysql.connect(
    host = '127.0.0.1', 
    port = 3306,
    user = 'root',
    password = 'mysql',  
    database = 'mydb',  
    charset = 'utf8' 
)

cursor = db.cursor()
sql = "UPDATE EMPLOYEE SET AGE = 18 WHERE ID = %s" % (2)
try：
	cursor.execute(sql)
	db.commit()
except:
    print('修改数据失败！')
    db.rollback()
db.close()

删除数据

import pymysql

db = pymysql.connect(
    host = '127.0.0.1', 
    port = 3306,
    user = 'root',
    password = 'mysql',  
    database = 'mydb',  
    charset = 'utf8' 
)

cursor = db.cursor()
sql = "DELETE FROM EMPLOYEE WHERE AGE > %s" % (20)
try：
	cursor.execute(sql)
	db.commit()
except:
    print('删除数据失败！')
    db.rollback()
db.close()

查询优化

为了防止SQL注入，千万不要手动拼接(关键参数)查询条件，通过下面的示例来实现对查询数据的优化。

import pymysql
cone = pymysql.connect(
    host = '127.0.0.1',
    port = 3306,
    user = 'root',
    password = 'mysql',
    database = 'day41',
    charset = 'utf8',
    autocommit = True
)
ursor = cone.cursor(pymysql.cursors.DictCursor)
username = input('username>>>:')
password = input('password>>>:')
sql = "select * from employee  where name=%s and password=%s"

res = cursor.execute(sql,(username,password))   利用方法实现字符的自动拼接
if res:
    print('输入正确！')
else:
    print('用户名或密码错误！')

cone.close()
'''
username>>>:LinWoW
password>>>:mysql
输入正确！

username>>>:"qwwqw" or 1=1 --
password>>>:
用户名或密码错误！
'''