Python操作数据库之records

安装

pip install records

快速使用

使用步骤：

1. 连接数据库，返回 db 数据库对象

2. db 对象执行 sql 语句

import records

# 获取数据库
db = records.Database('mysql+pymysql://root:@localhost:3306/dev01_git')
# 查询
rows = db.query('select * from student')

获取数据库的格式是标准的 URL 格式，如果使用的不是 mysql 数据库，只需要换掉数据库类型就可以了

image-20200801160643981

创建表

# 连接数据库
db = records.Database('mysql+pymysql://root:@localhost:3306/dev01_git')

# 创建表
sql_create_table = """CREATE TABLE IF NOT EXISTS student (
    name varchar(20),
    age int
) DEFAULT CHARSET=utf8 ;"""

db.query(sql_create_table)

INSTER

插入单条数据

records 支持使用 :variable 定义变量，通过参数传入完成动态传值，在需要动态加载数据的时候非常有用：

user = {"name": "tom", "age": 20}
db.query('INSERT INTO student(name,age) values (:name, :age)', **user)

插入多条数据

sql 语句在执行多条数据操作的时候非常不方便，尤其是当值还是变化的时候。 records 提供的bulk_query 方法能快捷的插入和更新多条数据：

users = [
    {"name":"yuze", "age": 13},
    {"name":"yuze2", "age": 15},
    {"name":"yuze3", "age": 16}
]
db.bulk_query('INSERT INTO lemon_user(name,age) values (:name, :age)', users)

SELECT

查询到数据以后，可以通过 all()方法获取所有的记录，支持 3 种类型。

• 默认是自己封装的 RecordCollection 对象，
• 也可以通过 as_dict=True 参数转成字典形式，
• 可以通过 as_ordereddict=True 转成有序字典形式

rows = db.query('SELECT * FROM student;')
# 得到所有数据
print(rows.all())
# 字典形式展示
print(rows.all(as_dict=True))
# 获取第一个
print(rows.first())
# 以字典形式获取第一个
print(rows.first(as_dict=True))
# 排序字典
print(rows.first(as_ordereddict=True))
# 查询唯一的一个
print(rows.one())

简洁的数据库事务支持

数据库事务是经常需要使用到的数据库操作，他通常是为了保持数据原子性和一致性。

比如一个转账的数据库操作：

1、从 A 账号中读取余额 ，
2、对 A 账号余额减去转账 - 400
3、从 B 账号中把余额读出来
4、对 B 账号做加法操作（+400）。

我们必须保证这 4 步同时执行成功，要么同时都不成功。如果前 2 步已经执行成功，但是到第 3 步发生了错误导致后面都不能执行，就会出现问题：yuze 的账号被扣了钱，但是 chaoge 的账号却没有加钱。

通过数据库事务就能避免这种情况。

with db.transaction() as tx:
    user = {"name": "tom", "age": 20}
    tx.query('INSERT INTO student(name,age) values (:name, :age)', **user)
    # 下面是错误的 sql 语句，有错误，则上面的 sql 语句不会成功执行。
    tx.query('test')

数据导出

数据导出为json

rows = db.query('SELECT * FROM student;')
json_rows = rows.export('json')
print(json_rows)

数据导出为Excel

rows = db.query('SELECT * FROM student;')
with open('users.xlsx', 'wb') as f:
    f.write(rows.export('xlsx'))

总结

• records 支持多种不同的数据库；
• records 的方法非常简单和统一，不像 pymysql 还需要记住如游标对象这样的底层信息；
• 自动实现上下文管理器，不需要关注链接状态；
• 支持 :variable 方言，以安全的形式进行动态参数传递，防止 sql 语句不规范导致的安全问题；
• 支持数据库事务，保持数据的原子性和一致性；
• 轻松导出为 json, yaml, xls, xlsx, pandas, html 等多种数据格式