remote/sqlite-utils
1
0
Fork 0
mirror of https://github.com/simonw/sqlite-utils.git synced 2026-09-17 22:14:09 +02:00
Code Issues Projects Releases Packages Wiki Activity

Merge 8d641ab08a into dce9bb05b6

Browse source
This commit is contained in:
Simon Willison 2021-06-19 22:33:55 -04:00 • committed by GitHub
commit 0bb6c7a389
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23
2 changed files with 42 additions and 1 deletions
Download patch file Download diff file Expand all files Collapse all files

19
sqlite_utils/db.py
View file

@ -1195,6 +1195,25 @@ class Table(Queryable):
self.add_foreign_key(fk_column, table, "id")
return self
def extract_expand(
self, column, expand, table=None, fk_column=None, fk_column_type=int, pk=None
):
"Use expand function to transform values in column and extract them into a new table"
table = table or column
fk_column = fk_column or "{}_id".format(table)
self.add_column(fk_column, fk_column_type)
for row_pk, row in self.pks_and_rows_where():
value = row[column]
expanded = expand(value)
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})
# Can drop the original column now
self.transform(drop=[column])
# And add that foreign key
self.add_foreign_key(fk_column, table, "id")
return self
def create_index(self, columns, index_name=None, unique=False, if_not_exists=False):
if index_name is None:
index_name = "idx_{}_{}".format(