From d1d780379ca5338fda046f363fcf2836f60dc96a Mon Sep 17 00:00:00 2001 From: phaer Date: Sun, 19 Sep 2021 09:46:56 +0200 Subject: [PATCH] Support lists in .extract_expand() --- sqlite_utils/db.py | 4 ++++ 1 file changed, 4 insertions(+) diff --git a/sqlite_utils/db.py b/sqlite_utils/db.py index fb1ae4f..649ec3a 100644 --- a/sqlite_utils/db.py +++ b/sqlite_utils/db.py @@ -1164,6 +1164,10 @@ class Table(Queryable): if isinstance(expanded, dict): new_pk = self.db[table].insert(expanded, pk="id", replace=True).last_pk self.update(row_pk, {fk_column: new_pk}) + elif isinstance(expanded, list): + for new_row in expanded: + new_pk = self.db[table].insert(new_row, pk="id", replace=True).last_pk + self.update(row_pk, {fk_column: new_pk}) # Can drop the original column now self.transform(drop=[column]) # And add that foreign key