PyMySQL介绍

pymysql介绍

PyMySQL介绍

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

Django中也可以使用PyMySQL连接MySQL数据库

PyMySQL安装

pip install pymysql

连接数据库

注意事项

在进行本文以下内容之前需要注意：

你有一个MySQL数据库，并且已经启动。

你有可以连接该数据库的用户名和密码

你有一个有权限操作的database

基本使用

# 导入pymysql模块
import pymysql
# 连接database
conn = pymysql.connect(host=“你的数据库地址”, user=“用户名”,password=“密码”,database=“数据库名”,charset=“utf8”)
# 得到一个可以执行SQL语句的光标对象
cursor = conn.cursor()
# 定义要执行的SQL语句
sql = """
CREATE TABLE USER1 (
id INT auto_increment PRIMARY KEY ,
name CHAR(10) NOT NULL UNIQUE,
age TINYINT NOT NULL
)ENGINE=innodb DEFAULT CHARSET=utf8;
"""
# 执行SQL语句
cursor.execute(sql)
# 关闭光标对象
cursor.close()
# 关闭数据库连接
conn.close()

防止SQL注入

import pymysql
name = input("用户名:>>")
pwd = input("密码:>>")
# 连接数据库
conn = pymysql.connect(host="localhost", database="s8", user="root", password="123456", charset="utf8")
# 获取光标，输入SQL语句并执行
cursor = conn.cursor()
sql = "select * from userinfo;"
# 自己拼接字符串，容易造成SQL注入问题
sql2 = "select * from userinfo where name='%s' and pwd='%s';"%(name,pwd)
# 让pymysql来拼接，防止SQL注入
sql3 = "select * from userinfo where name=%s and pwd=%s;"
ret3 = cursor.execute(sql3,[name,pwd])

ret = cursor.execute(sql)
# 关闭光标和连接
cursor.close()
conn.close()
print(ret)  # 1 一条结果

增删改查操作

增

# 导入pymysql模块
import pymysql
# 连接database
conn = pymysql.connect(host=“你的数据库地址”, user=“用户名”,password=“密码”,database=“数据库名”,charset=“utf8”)
# 得到一个可以执行SQL语句的光标对象
cursor = conn.cursor()
sql = "INSERT INTO USER1(name, age) VALUES (%s, %s);"
username = "Alex"
age = 18
# 执行SQL语句
cursor.execute(sql, [username, age])
# 提交事务
conn.commit()
cursor.close()
conn.close()

插入数据失败回滚

# 导入pymysql模块
import pymysql
# 连接database
conn = pymysql.connect(host=“你的数据库地址”, user=“用户名”,password=“密码”,database=“数据库名”,charset=“utf8”)
# 得到一个可以执行SQL语句的光标对象
cursor = conn.cursor()
sql = "INSERT INTO USER1(name, age) VALUES (%s, %s);"
username = "Alex"
age = 18
try:
    # 执行SQL语句
    cursor.execute(sql, [username, age])
    # 提交事务
    conn.commit()
except Exception as e:
    # 有异常，回滚事务
    conn.rollback()
cursor.close()
conn.close()

获取插入数据的ID(关联操作时会用到)

# 导入pymysql模块
import pymysql
# 连接database
conn = pymysql.connect(host=“你的数据库地址”, user=“用户名”,password=“密码”,database=“数据库名”,charset=“utf8”)
# 得到一个可以执行SQL语句的光标对象
cursor = conn.cursor()
sql = "INSERT INTO USER1(name, age) VALUES (%s, %s);"
username = "Alex"
age = 18
try:
    # 执行SQL语句
    cursor.execute(sql, [username, age])
    # 提交事务
    conn.commit()
    # 提交之后，获取刚插入的数据的ID
    last_id = cursor.lastrowid
except Exception as e:
    # 有异常，回滚事务
    conn.rollback()
cursor.close()
conn.close()

批量执行

import pymysql

# 连接
conn = pymysql.connect(host="localhost", user="root", password="123456", database="s8", charset="utf8")
# 获取光标
cursor = conn.cursor()
# 写SQL语句
sql = "insert into userinfo(name,pwd) values(%s,%s);"
user1 = "Egon"
pwd1 = "1234"
user2 = "Alex"
pwd2 = "1234"
# data = [("Egon","1234"),("Alex","1234")]
try:
    # 执行
    cursor.executemany(sql, ((user1, pwd1),(user2, pwd2)))
    # cursor.executemany(sql, data)
    # 把修改提交到数据库
    conn.commit()
except Exception as e:
    conn.rollback()  # 执行SQL语句有问题或提交有异常都回滚

cursor.close()
conn.close()

删

# 导入pymysql模块
import pymysql
# 连接database
conn = pymysql.connect(host=“你的数据库地址”, user=“用户名”,password=“密码”,database=“数据库名”,charset=“utf8”)
# 得到一个可以执行SQL语句的光标对象
cursor = conn.cursor()
sql = "DELETE FROM USER1 WHERE id=%s;"
try:
    cursor.execute(sql, [4])
    # 提交事务
    conn.commit()
except Exception as e:
    # 有异常，回滚事务
    conn.rollback()
cursor.close()
conn.close()

改

# 导入pymysql模块
import pymysql
# 连接database
conn = pymysql.connect(host=“你的数据库地址”, user=“用户名”,password=“密码”,database=“数据库名”,charset=“utf8”)
# 得到一个可以执行SQL语句的光标对象
cursor = conn.cursor()
# 修改数据的SQL语句
sql = "UPDATE USER1 SET age=%s WHERE name=%s;"
username = "Alex"
age = 80
try:
    # 执行SQL语句
    cursor.execute(sql, [age, username])
    # 提交事务
    conn.commit()
except Exception as e:
    # 有异常，回滚事务
    conn.rollback()
cursor.close()
conn.close()

查

查询单条数据

# 导入pymysql模块
import pymysql
# 连接database
conn = pymysql.connect(host=“你的数据库地址”, user=“用户名”,password=“密码”,database=“数据库名”,charset=“utf8”)
# 得到一个可以执行SQL语句的光标对象
cursor = conn.cursor()
# 查询数据的SQL语句
sql = "SELECT id,name,age from USER1 WHERE id=1;"
# 执行SQL语句
cursor.execute(sql)
# 获取单条查询数据
ret = cursor.fetchone()
cursor.close()
conn.close()
# 打印下查询结果
print(ret)

查询多条数据

# 导入pymysql模块
import pymysql
# 连接database
conn = pymysql.connect(host=“你的数据库地址”, user=“用户名”,password=“密码”,database=“数据库名”,charset=“utf8”)
# 得到一个可以执行SQL语句的光标对象
cursor = conn.cursor()
# 查询数据的SQL语句
sql = "SELECT id,name,age from USER1;"
# 执行SQL语句
cursor.execute(sql)
# 获取多条查询数据
ret = cursor.fetchall()
cursor.close()
conn.close()
# 打印下查询结果
print(ret)

进阶

import pymysql

# 连接
conn = pymysql.connect(host="localhost", user="root", password="123456", database="s8", charset="utf8")
# 获取光标
cursor = conn.cursor()
# 写SQL语句
sql = "select * from userinfo;"
ret = cursor.execute(sql)
print("-->",ret)  # 受影响的行数
# 一次取一条
# print(cursor.fetchone())  # (1, 'root', '123456')
# print(cursor.fetchone())  # (4, 'Egon2', '1234')
# 一次取所有
# print(cursor.fetchall())  # ((1, 'root', '123456'), (4, 'Egon2', '1234'))
# 一次取2条
# print(cursor.fetchmany(2))

# 进阶用法
print(cursor.fetchone())  # 取1条
print(cursor.fetchall())  # 接着上一条取所有
print(cursor.fetchall())  # 取不到了，为()
# 移动取数据的光标
cursor.scroll(-2)  # 往上移两个，默认是相对移动
cursor.scroll(0, mode="absolute")  # 按照绝对位置移动，移动到开始位置

cursor.close()
conn.close()