pymysql语法_PyMySQL模块的使用

快速使用

官方教程

import pymysql.cursors

# 连接数据库

connection = pymysql.connect(host='localhost',

user='user',

password='passwd',

db='db',

charset='utf8mb4',

cursorclass=pymysql.cursors.DictCursor)

try:

with connection.cursor() as cursor:

# 增加一条数据

sql = "INSERT INTO `users` (`email`, `password`) VALUES (%s, %s)"

cursor.execute(sql, ('webmaster@python.org', 'very-secret'))

connection.commit()

with connection.cursor() as cursor:

# 查询数据

sql = "SELECT `id`, `password` FROM `users` WHERE `email`=%s"

cursor.execute(sql, ('webmaster@python.org',))

result = cursor.fetchone()

print(result)

finally:

connection.close()

备查

# 获取输入SQL语句的光标对象，指定返回的数据格式为字典格式

cursor = conn.cursor(cursor=pymysql.cursors.DictCursor)

# 执行sql语句

cursor.execute(sql, (username, pwd))

# 提交事务

conn.commit()

# 获取单条查询数据

ret = cursor.fetchone()

# 返回所有数据

ret = cursor.fetchall()

# 回滚

conn.rollback()

基本使用

# 导入模块

import pymysql

username = input("请输入用户名: ")

pwd = input("请输入密码: ")

# 连接数据库

conn = pymysql.connect(host="localhost",

port=3306,

database="userinfo",

user='root',

password="123456",

charset="utf8"

)

# 获取输入SQL语句的光标对象

cursor = conn.cursor()

# 定义要执行的SQL语句

sql = "select * from info WHERE username=%s and password=%s;"

# 执行sql语句，让pymysql帮我们拼接SQL语句

cursor.execute(sql, (username, pwd))

# 关闭光标对象

cursor.close()

# 关闭数据库连接

conn.close()

返回字典格式的数据

# 获取输入SQL语句的光标对象，指定返回的数据格式为字典格式

cursor = conn.cursor(cursor=pymysql.cursors.DictCursor)

增删改查操作

增

import pymysql

# 获取用户输入

username = input("请输入用户名: ")

pwd = input("请输入密码: ")

# 连接数据库

conn = pymysql.connect(host="localhost",

port=3306,

database="userinfo",

user='root',

password="123456",

charset="utf8"

)

# 获取输入SQL语句的光标对象

cursor = conn.cursor()

# sql语句

sql = "insert into info (username, password) VALUES (%s, %s);"

# 执行sql语句

cursor.execute(sql, [username, pwd])

# 提交

conn.commit()

# 关闭光标对象和连接

cursor.close()

conn.close()

插入数据失败回滚

在执行增删改操作时，如果不想提交前面的操作，可以使用 rollback() 回滚取消操作。

import pymysql

# 获取用户输入

username = input("请输入用户名: ")

pwd = input("请输入密码: ")

# 连接数据库

conn = pymysql.connect(host="localhost",

port=3306,

database="userinfo",

user='root',

password="123456",

charset="utf8"

)

# 获取输入SQL语句的光标对象

cursor = conn.cursor()

# 拼接sql语句

sql = "insert into info (username, password) VALUES (%s, %s);"

# 执行sql语句

# 让pymysql帮我们拼接SQL语句(传入的参数可以是列表也可以是元祖，一定要是一个可迭代对象)

try:

cursor.execute(sql, [username, pwd])

# 提交

conn.commit()

except Exception as e:

print("报错啦：", str(e))

conn.rollback() # 回滚

# 关闭光标对象和连接

cursor.close()

conn.close()

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

import pymysql

# 获取用户输入

username = input("请输入用户名: ")

pwd = input("请输入密码: ")

# 连接数据库

conn = pymysql.connect(host="localhost",

port=3306,

database="userinfo",

user='root',

password="123456",

charset="utf8"

)

# 获取输入SQL语句的光标对象

cursor = conn.cursor()

# 拼接sql语句

sql = "insert into info (username, password) VALUES (%s,%s);"

# 执行sql语句

cursor.execute(sql, [username, pwd])

# 提交

conn.commit()

# 提交之后，获取刚插入的数据的ID

last_id = cursor.lastrowid

# 关闭光标对象和连接

cursor.close()

conn.close()

批量执行

import pymysql

# 连接数据库

conn = pymysql.connect(host="localhost",

port=3306,

database="userinfo",

user='root',

password="123456",

charset="utf8"

)

# 获取光标对象

cursor = conn.cursor()

data = [("奔驰哥", 123456), ("小宝宝", 123456), ("教主", 123456)]

# sql语句

sql = "insert into info(username, password) VALUES (%s, %s)"

# 批量插入

cursor.executemany(sql, data)

# 提交

conn.commit()

# 关闭光标对象和连接

cursor.close()

conn.close()

删

import pymysql

# 连接数据库

conn = pymysql.connect(host="localhost",

port=3306,

database="userinfo",

user='root',

password="123456",

charset="utf8"

)

# 获取光标对象

cursor = conn.cursor()

# sql语句

sql = "delete from info where id=%s"

# 执行SQL语句

cursor.execute(sql, [1])

# 提交事务

conn.commit()

# 关闭光标对象和连接

cursor.close()

conn.close()

改

import pymysql

# 连接数据库

conn = pymysql.connect(host="localhost",

port=3306,

database="userinfo",

user='root',

password="123456",

charset="utf8"

)

# 获取输入SQL语句的光标对象

cursor = conn.cursor()

# sql语句

sql = "update info set password=%s where id=%s"

password = "654321"

id = 2

cursor.execute(sql, [password, id])

conn.commit()

cursor.close()

conn.close()

查

查询单条数据

import pymysql

# 连接数据库

conn = pymysql.connect(host="localhost",

port=3306,

database="userinfo",

user='root',

password="123456",

charset="utf8"

)

# 获取输入SQL语句的光标对象，指定返回的数据格式为字典格式

cursor = conn.cursor(cursor=pymysql.cursors.DictCursor)

sql = "select * from info"

# 提交事务

cursor.execute(sql)

# 获取单条查询数据

ret = cursor.fetchone()

# 返回所有数据

ret = cursor.fetchall()

# 关闭光标和连接

cursor.close()

conn.close()

进阶用法

查询后光标就会移动，下次查询就会再上次查询的数据之后开始新的查询

# 可以获取指定数量的数据

cursor.fetchmany(3)

# 光标按绝对位置移动1

cursor.scroll(1, mode="absolute")

# 光标按照相对位置(当前位置)移动1

cursor.scroll(1, mode="relative")

错误处理

异常

描述

Warning

当有严重警告时触发，例如插入数据是被截断等等。

Error

警告以外所有其他错误类。

InterfaceError

当有数据库接口模块本身的错误(而不是数据库的错误)发生时触发。

DatabaseError

和数据库有关的错误发生时触发。

DataError

当有数据处理时的错误发生时触发，例如：除零错误，数据超范围等等。

OperationalError

指非用户控制的，而是操作数据库时发生的错误。例如：连接意外断开、 数据库名未找到、事务处理失败、内存分配错误等等操作数据库是发生的错误。

IntegrityError

完整性相关的错误，例如外键检查失败等。

InternalError

数据库的内部错误，例如游标(cursor)失效了、事务同步失败等等。

ProgrammingError

程序错误，例如数据表(table)没找到或已存在、SQL语句语法错误、 参数数量错误等等。

NotSupportedError

不支持错误，指使用了数据库不支持的函数或API等。例如在连接对象上 使用.rollback()函数，然而数据库并不支持事务或者事务已关闭。