使用pymysql模块操作MySQL

安装

$ pip3 install pymysql

使用

输出 MySQL 版本

import pymysql

# 打开数据库连接
db = pymysql.connect("localhost", "root", "root", "mysql")
# 使用 cursor() 方法创建一个游标对象 cursor
cursor = db.cursor()
# 使用 execute()  方法执行 SQL 查询
cursor.execute("SELECT VERSION()")
# 使用 fetchone() 方法获取单条数据.
data = cursor.fetchone()
print("Database version : %s " % data)
# 关闭数据库连接
db.close()

'''
Database version : 5.6.45
'''

创建表

import pymysql

# 打开数据库连接
db = pymysql.connect("127.0.0.1", "root", "root", "testdb")
# 使用 cursor() 方法创建一个游标对象 cursor
cursor = db.cursor()
# 使用 execute() 方法执行 SQL，如果表存在则删除
cursor.execute("DROP TABLE IF EXISTS UserInfo;")
# 使用预处理语句创建表
sql = """CREATE TABLE UserInfo (
         Name  CHAR(20) NOT NULL,
         AGE INT,  
         SEX CHAR(1) );"""
cursor.execute(sql)
# 关闭数据库连接
db.close()

新增操作

import pymysql

# 打开数据库连接
db = pymysql.connect("localhost", "root", "root", "testdb")
# 使用cursor()方法获取操作游标
cursor = db.cursor()
# SQL 插入语句
sql = """INSERT INTO UserInfo(Name,AGE, SEX)
         VALUES ('张三', 18, '男')"""
try:
    # 执行sql语句
    cursor.execute(sql)
    # 提交到数据库执行
    db.commit()
except:
    # 如果发生错误则回滚
    db.rollback()
# 关闭数据库连接
db.close()

查询

一次获取所有行

import pymysql

# 打开数据库连接
db = pymysql.connect("localhost", "root", "root", "testdb")
# 使用cursor()方法获取操作游标
cursor = db.cursor()
# SQL 查询语句
sql = "SELECT * FROM UserInfo"
try:
    # 执行SQL语句
    cursor.execute(sql)
    # 获取所有记录列表
    results = cursor.fetchall()
    for row in results:
        name = row[0]
        age = row[1]
        sex = row[2]
        # 打印结果
        print("name=%s,age=%d,sex=%s" %
              (name, age, sex))
except:
    print("Error: unable to fetch data")
# 关闭数据库连接
db.close()

'''
name=张三,age=18,sex=男
'''

单行获取

import pymysql

# 打开数据库连接
db = pymysql.connect("localhost", "root", "root", "testdb")
# 使用cursor()方法获取操作游标
cursor = db.cursor()
# SQL 查询语句
sql = "SELECT * FROM UserInfo"
try:
    # 执行SQL语句
    cursor.execute(sql)
    # 获取所有记录列表
    while 1:
        one = cursor.fetchone()
        if one:
            print(one)
        else:
            break
except:
    print("Error: unable to fetch data")
# 关闭数据库连接
db.close()

'''
('张三', 18, '男')
'''

可通过 cursor = db.cursor(cursor=pymysql.cursors.DictCursor) 的方式让游标以字典方式返回数据。

事务

在使用 pymysql 操作支持事务的数据库时，当游标建立之时，就自动开始了一个隐形的数据库事务。commit() 方法提交所有更新操作，rollback() 方法回滚当前游标的所有操作。

import pymysql

# 打开数据库连接
db = pymysql.connect("localhost", "root", "root", "testdb")
# 使用cursor()方法获取操作游标
cursor = db.cursor()
# SQL 删除语句
sql = "DELETE FROM UserInfo WHERE AGE >= %d" % (18)
try:
    # 执行SQL语句
    cursor.execute(sql)
    # 提交修改
    db.commit()
except:
    # 发生错误时回滚
    db.rollback()
# 关闭连接
db.close()

可能出现的错误

异常	描述
Warning	当有严重警告时触发，例如插入数据是被截断等等。必须是 StandardError 的子类。
Error	警告以外所有其他错误类。必须是 StandardError 的子类。
InterfaceError	当有数据库接口模块本身的错误（而不是数据库的错误）发生时触发。 必须是 Error 的子类。
DatabaseError	和数据库有关的错误发生时触发。 必须是 Error 的子类。
DataError	当有数据处理时的错误发生时触发，例如：除零错误，数据超范围等等。 必须是 DatabaseError 的子类。
OperationalError	指非用户控制的，而是操作数据库时发生的错误。例如：连接意外断开、 数据库名未找到、事务处理失败、内存分配错误等等操作数据库是发生的错误。 必须是 DatabaseError 的子类。
IntegrityError	完整性相关的错误，例如外键检查失败等。必须是 DatabaseError 子类。
InternalError	数据库的内部错误，例如游标（cursor）失效了、事务同步失败等等。 必须是DatabaseError子类。
ProgrammingError	程序错误，例如数据表（table）没找到或已存在、SQL 语句语法错误、 参数数量错误等等。必须是 DatabaseError 的子类。
NotSupportedError	不支持错误，指使用了数据库不支持的函数或 API 等。例如在连接对象上 使用 rollback() 函数，然而数据库并不支持事务或者事务已关闭。 必须是 DatabaseError 的子类。