快速使用
https://github.com/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, ('[email protected]', 'very-secret'))
connection.commit()
with connection.cursor() as cursor:
# 查询数据
sql = "SELECT `id`, `password` FROM `users` WHERE `email`=%s"
cursor.execute(sql, ('[email protected]',))
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()函数,然而数据库并不支持事务或者事务已关闭。 |