Классы данных Python и адаптеры sqlite3
Каков самый чистый способ фиксации данных, хранящихся в экземплярах класса данных, содержащегося в списке, в SQLite с помощью sqlite3executemany?
Например:
@dataclass
class Person:
name: str
age: int
a = Person("Alice", 21)
b = Person("Bob", 22)
c = Person("Charlie", 23)
people = [a, b, c]
# ...
cursor.executemany(f"INSERT INTO {table} ({headers}) ({placeholders})", people) # <-- fails, obviously
Есть ли встроенный механизм, который выполняет всю работу за меня (перебирает каждый атрибут класса данных), или идиоматический способ, или, если нет, то как мне реализовать адаптер для этого без явного указания атрибутов одним?
Я мог бы преобразовать его в список кортежей, который работает «из коробки», но кажется излишним.
Возможно, вы захотите использовать ORM (например, sqlalchemy), чтобы определить сопоставление между экземпляром Person (класс данных или нет) и вашей базой данных.
@ekhumoro «Адапторы здесь не имеют значения, поскольку вы не конвертируете отдельные типы значений» — это явно та часть, которую я не понял.
@chepner Я знаю об этом, но для данного конкретного варианта использования это излишество.
Ответы 1
Используйте функцию astuple() для преобразования экземпляров класса данных в кортежи.
from dataclasses import dataclass, fields, astuple
headers = ','.join(f.name for f in fields(Person))
people = map(astuple, [a, b, c])
placeholders = ','.join(['?'] * len(headers))
table = 'persons'
cursor.executemany(f"INSERT INTO {table} ({headers}) ({placeholders})", people)
Используйте поля и asdict из модуля классов данных. Например:
headers = ','.join(f.name for f in fields(Person)),placeholders = ','.join(f':{f.name}' for f in fields(Person))иpeople = [asdict(d) for d in (a, b, c)]. Должно быть достаточно просто разработать общую функцию, выполняющую всю операцию. (Аппаторы здесь не имеют значения, поскольку вы не конвертируете отдельные типы значений).