使用pymysql模块操作MySQL

使用pymysql模块操作MySQL

微信搜索 zze_coding 或扫描 👉 二维码关注我的微信公众号获取更多资源推送:

安装

$ 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 的子类。

Copyright: 采用 知识共享署名4.0 国际许可协议进行许可

Links: https://www.zze.xyz/archives/python-module-pymysql.html

Buy me a cup of coffee ☕.