python | tinydb，一个非常厉害的 关于数据库的 Python 库！

本文来源公众号“python”，仅用于学术分享，侵权删，干货满满。

原文链接：tinydb，一个非常厉害的 Python 库！

大家好，今天为大家分享一个非常厉害的 Python 库 - tinydb。

Github地址：https://github.com/msiemens/tinydb

在现代应用开发中，使用数据库来存储和管理数据是非常常见的需求。对于简单的数据存储需求，关系型数据库可能显得过于复杂。TinyDB 是一个纯 Python 实现的轻量级 NoSQL 数据库，专为嵌入式场景设计，适用于小型项目、原型开发和教学等场景。本文将详细介绍 TinyDB 库，包括其安装方法、主要特性、基本和高级功能，以及实际应用场景，帮助全面了解并掌握该库的使用。

1 安装

1.1 使用 pip 安装

可以通过 pip 直接安装 TinyDB：

pip install tinydb

2 特性

1. 轻量级：纯 Python 实现，单个文件即可满足需求。

2. 易于使用：提供简单直观的 API，快速上手。

3. 嵌入式：适用于嵌入式场景，无需独立的数据库服务器。

4. 文档存储：以 JSON 格式存储数据，支持复杂的数据结构。

5. 可扩展：支持自定义存储引擎和查询操作。

3 基本功能

3.1 创建数据库和表

可以使用 TinyDB 创建数据库和表：

from tinydb import TinyDB, Query

# 创建数据库
db = TinyDB('db.json')

# 创建表
table = db.table('users')

3.2 插入数据

可以向表中插入数据：

# 插入单条数据
table.insert({'name': 'Alice', 'age': 25})

# 插入多条数据
table.insert_multiple([
    {'name': 'Bob', 'age': 30},
    {'name': 'Charlie', 'age': 35}
])

3.3 查询数据

可以使用查询对象从表中查询数据：

# 创建查询对象
User = Query()

# 查询单条数据
result = table.get(User.name == 'Alice')
print(result)

# 查询多条数据
results = table.search(User.age > 30)
print(results)

3.4 更新数据

可以根据查询条件更新数据：

# 更新单条数据
table.update({'age': 26}, User.name == 'Alice')

# 更新多条数据
table.update({'age': 40}, User.age > 30)

3.5 删除数据

可以根据查询条件删除数据：

# 删除单条数据
table.remove(User.name == 'Alice')

# 删除多条数据
table.remove(User.age > 30)

4 高级功能

4.1 自定义存储引擎

可以自定义存储引擎，实现数据的自定义存储：

from tinydb.storages import Storage

class CustomStorage(Storage):
    def __init__(self, filename):
        self.filename = filename

    def read(self):
        # 自定义读取数据的逻辑
        pass

    def write(self, data):
        # 自定义写入数据的逻辑
        pass

# 使用自定义存储引擎
db = TinyDB('custom_db.json', storage=CustomStorage)

4.2 查询扩展

可以扩展查询操作，支持自定义查询：

from tinydb.queries import QueryInstance

class CustomQuery(QueryInstance):
    def startswith(self, value):
        return lambda doc: doc[self.name].startswith(value)

# 使用自定义查询
User = CustomQuery()
results = table.search(User.name.startswith('A'))
print(results)

4.3 数据迁移

可以使用 TinyDB 进行数据迁移，确保数据结构的一致性：

def migrate_data(db):
    for user in db.table('users'):
        if 'email' not in user:
            db.table('users').update({'email': ''}, doc_ids=[user.doc_id])

# 进行数据迁移
migrate_data(db)

5 实际应用场景

5.1 小型项目

在小型项目中使用 TinyDB 存储和管理数据，简化开发过程。

from tinydb import TinyDB, Query

# 创建数据库和表
db = TinyDB('small_project.json')
table = db.table('tasks')

# 插入数据
table.insert({'task': 'Write blog post', 'status': 'in progress'})

# 查询数据
Task = Query()
result = table.get(Task.task == 'Write blog post')
print(result)

5.2 原型开发

在原型开发中使用 TinyDB 快速存储和检索数据，加快开发速度。

from tinydb import TinyDB

# 创建数据库和表
db = TinyDB('prototype.json')
table = db.table('users')

# 插入数据
table.insert_multiple([
    {'name': 'Alice', 'age': 25},
    {'name': 'Bob', 'age': 30}
])

# 查询数据
results = table.all()
print(results)

5.3 配置管理

使用 TinyDB 存储和管理应用程序配置，方便配置的读取和更新。

from tinydb import TinyDB

# 创建数据库和表
db = TinyDB('config.json')
table = db.table('settings')

# 插入配置
table.insert({'setting': 'theme', 'value': 'dark'})

# 更新配置
table.update({'value': 'light'}, Query().setting == 'theme')

# 读取配置
config = table.get(Query().setting == 'theme')
print(config)

5.4 教学示例

在教学中使用 TinyDB 演示数据库操作的基本概念，帮助学生理解数据库的使用。

from tinydb import TinyDB, Query

# 创建数据库和表
db = TinyDB('students.json')
table = db.table('students')

# 插入数据
table.insert({'name': 'Alice', 'grade': 'A'})
table.insert({'name': 'Bob', 'grade': 'B'})

# 查询数据
Student = Query()
results = table.search(Student.grade == 'A')
print(results)

6 总结

TinyDB 是一个功能强大且易于使用的轻量级 NoSQL 数据库，能够帮助开发者在各种应用场景中高效地存储和管理数据。通过支持简单易用的 API、灵活的查询和更新操作、强大的扩展功能和自定义存储引擎，TinyDB 提供了强大的功能和灵活的扩展能力。本文详细介绍了 TinyDB 库的安装方法、主要特性、基本和高级功能，以及实际应用场景。希望本文能帮助大家全面掌握 TinyDB 的使用，并在实际项目中发挥其优势。

THE END !

文章结束，感谢阅读。您的点赞，收藏，评论是我继续更新的动力。大家有推荐的公众号可以评论区留言，共同学习，一起进步。