Python数据类批量插入SQLite：SQLite3适配器如何优雅地解决
2024-05-29 03:37:05
用 SQLite 适配器从 Python 数据类中优雅地插入数据
问题
在计算机科学中，数据持久化是一项至关重要的技术，它使应用程序能够将数据存储在持久性存储中，以便即使在进程终止或系统重启后也能保留数据。在 Python 中，使用 SQLite3 等关系数据库管理系统（RDBMS）来存储数据非常普遍。
假设你有一个名为 Person 的数据类，其中包含 name 和 age 属性。你希望将存储在 Person 数据类实例列表中的数据批量插入 SQLite 数据库的 people 表中。
内置机制或惯用方法
遗憾的是，Python 没有内置机制或惯用方法可以自动将数据类实例转换为 SQLite 数据库中表的行。你必须手动指定要插入的属性，如下所示：
cursor.executemany(
f"INSERT INTO {table} (name, age) VALUES (?, ?)",
[(person.name, person.age) for person in people]
)
这种方法虽然有效，但对于具有大量属性的数据类来说会变得冗长且容易出错。
适配器解决方案
为了解决这个问题，你可以使用 SQLite3 适配器。适配器是一种机制，允许你将自定义 Python 类型转换为 SQLite 值，反之亦然。
要创建适用于 Person 数据类的适配器，你可以执行以下步骤：
- 导入
sqlite3模块。 - 定义一个名为
dataclass_adapter的函数，该函数接受一个Person数据类实例并返回一个元组，其中包含数据类属性的值。 - 使用
sqlite3.register_adapter函数注册适配器。
代码如下：
import sqlite3
def dataclass_adapter(obj):
return tuple(obj.__dict__.values())
sqlite3.register_adapter(Person, dataclass_adapter)
使用适配器
注册适配器后，你就可以使用它来将数据类实例批量插入 SQLite 数据库：
cursor.executemany(
f"INSERT INTO {table} ({headers}) VALUES ({placeholders})",
people
)
在这种情况下，dataclass_adapter 适配器将自动应用于 people 列表中的每个 Person 实例，并将其转换为 SQLite 值。
结论
使用 SQLite3 适配器，你可以优雅地将存储在 Python 数据类实例中的数据批量插入 SQLite 数据库。这消除了手动指定要插入的属性的需要，并简化了数据持久化过程。
常见问题解答
-
为什么使用数据类而不是常规的 Python 类？
数据类是一种更简单的类定义语法，它提供了不可变性和类型提示，这对于数据持久化很有用。
-
适配器是如何工作的？
适配器充当 SQLite3 和 Python 数据类之间的桥梁，允许在两者之间转换值。
-
除了
name和age属性外，适配器是否支持其他属性？是的，只要你为要插入的每个属性定义一个数据类属性，适配器就会自动转换所有属性。
-
是否可以使用适配器将数据从 SQLite 数据库插入到 Python 数据类？
可以，你可以使用
sqlite3.register_converter函数注册一个反向适配器。 -
适配器是否与其他 SQLite3 版本兼容？
是的，适配器与 SQLite3 的所有主要版本兼容。
