今天要介绍的这个python第三方库非常厉害,完美操作各种数据库。
名字叫records,在网上很少有这个库的相关资料,但是在开源社区可是很火热的哦。
如果这还不能打消你的顾虑,再告诉你一件事:如果你用python编程,你一定听过requests,这两个库是同一个作者写的。
![d7b4780bd3b11ea3534121cd503539c7.png](https://i-blog.csdnimg.cn/blog_migrate/d909d3fa359b22a5a311e883f2485a8c.jpeg)
![5b647507f6319df17f6b2236b7bd042e.png](https://i-blog.csdnimg.cn/blog_migrate/61e674621c3813679f637294fb02fcd3.png)
使用步骤:
1、连接数据库,返回 db 数据库对象
2、db 对象执行 sql 语句
# pip install records 安装
import records
# 获取数据库
db = records.Database('mysql+pymysql://root:@localhost:3306/dev01_git')
# 查询
rows = db.query('select * from lemon_user')
获取数据库的格式是标准的 URL 格式,如果使用的不是 mysql 数据库,只需要换掉数据库类型就可以了:
![6d789d47478df6f675089ac1f393b396.png](https://i-blog.csdnimg.cn/blog_migrate/3ebc9bcb256cb0f1e8eff470a62cb484.jpeg)
![e5c022b00883f7c26d93c964f6c09965.png](https://i-blog.csdnimg.cn/blog_migrate/b078834995f8fdb346294ebd87c45929.png)
# 连接数据库
db = records.Database('mysql+pymysql://root:@localhost:3306/dev01_git')
# 创建表
sql_create_table = """CREATE TABLE IF NOT EXISTS lemon_user (
name varchar(20),
age int
) DEFAULT CHARSET=utf8 ;"""
db.query(sql_create_table)
![dfc755b97e36fe94ae96f5cd473777fd.png](https://i-blog.csdnimg.cn/blog_migrate/bb694a1735562acbe1bf90c0c50a1470.png)
records 支持使用 :variable
定义变量,通过参数传入完成动态传值,在需要动态加载数据的时候非常有用:
user = {"name": "yuze5", "age": 20}
db.query('INSERT INTO lemon_user(name,age) values (:name, :age)', **user)
![4d94526f09c443b1f1ddd4eb67f8b913.png](https://i-blog.csdnimg.cn/blog_migrate/d2cdaeab53c4660e91ab897ee22e705b.png)
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)
![058c14876b8d703a58e725fd6b324a68.png](https://i-blog.csdnimg.cn/blog_migrate/d12fe5194ba9922c5fce4c5ac1934c3d.png)
查询到数据以后,可以通过 all() 方法获取所有的记录,支持 3 种类型。
- 默认是自己封装的 RecordCollection 对象,
- 也可以通过 as_dict=True 参数转成字典形式,
- 可以通过 as_ordereddict=True 转成排序字典形式
rows = db.query('SELECT * FROM lemon_user;')
# 得到所有数据
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())
![e02e49fb88191360e3a0dc92ad20f74f.png](https://i-blog.csdnimg.cn/blog_migrate/1e08772d69ca1df36ea57995f088e0ee.png)
数据库事务是经常需要使用到的数据库操作,他通常是为了保持数据原子性和一致性。
比如一个转账的数据库操作:
1、从 yuze 账号中读取余额 , 2、对 yuze 账号余额减去转账 - 400 3、从 chaoge 账号中把余额读出来 4、对 chaoge 账号做加法操作(+400)。
我们必须保证这 4 步同时执行成功,要么同时都不成功。如果前 2 步已经执行成功,但是到第 3 步发生了错误导致后面都不能执行,就会出现问题:yuze 的账号被扣了钱,但是 chaoge 的账号却没有加钱。
通过数据库事务就能避免这种情况。
with db.transaction() as tx:
user = {"name": "yuze9", "age": 20}
tx.query('INSERT INTO lemon_user(name,age) values (:name, :age)', **user)
# 下面是错误的 sql 语句,有错误,则上面的 sql 语句不会成功执行。
tx.query('sof')
![6a59964d7101d9487c58987d3c81e0d3.png](https://i-blog.csdnimg.cn/blog_migrate/faa51050535ccde76fc351089de55386.png)
数据库数据得到以后还需要进一步的操作,比如保存起来。
rows = db.query('SELECT * FROM lemon_user;')
json_rows = rows.export('yaml')
print(json_rows)
![73321584c12d399b792a99edbf27b438.png](https://i-blog.csdnimg.cn/blog_migrate/11b3e5791c574c7b64bd70bc8cf40f4c.png)
最常用的是如果需要各种各样的报表,需要放到 excel 文件中保存起来。
rows = db.query('SELECT * FROM lemon_user;')
with open('users.xlsx', 'wb') as f:
f.write(rows.export('xlsx'))
导出的数据效果:
![2f154d5916d92b5673b6abde0b634e24.png](https://i-blog.csdnimg.cn/blog_migrate/bfa4b31c491440cb2476e3f172f5bb1f.jpeg)
![567493a0deeebf85d116497f53b8cc27.png](https://i-blog.csdnimg.cn/blog_migrate/9bad52d9ddb10213423cc373063aa1c1.png)
- records 支持多种不同的数据库;
- records 的方法非常简单和统一,不像 pymysql 还需要记住如游标对象这样的底层信息;
- 自动实现上下文管理器,不需要关注链接状态;
- 支持
:variable
方言,以安全的形式进行动态参数传递,防止 sql 语句不规范导致的安全问题; - 支持数据库事务,保持数据的原子性和一致性;
- 轻松导出为 json, yaml, xls, xlsx, pandas, html 等多种数据格式。
- 源码非常简单,你一定看得懂。
作者:雨泽
来源:柠檬班社区