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

.transform() can now change the primary key

Browse source
This commit is contained in:
Simon Willison 2020-09-21 19:15:43 -07:00
commit 9a6e7dab4d
2 changed files with 27 additions and 6 deletions
Download patch file Download diff file Expand all files Collapse all files

13
sqlite_utils/db.py
View file

@ -734,7 +734,7 @@ class Table(Queryable):
columns=columns,
rename=rename,
drop=None,
# pk=pk,
pk=pk,
# foreign_keys=foreign_keys,
# column_order=column_order,
# not_null=not_null,
@ -752,6 +752,7 @@ class Table(Queryable):
columns=None,
rename=None,
drop=None,
pk=None,
tmp_suffix=None,
):
columns = columns or {}
@ -773,11 +774,11 @@ class Table(Queryable):
copy_from_to[name] = new_name
sqls = []
pk = None
if len(self.pks) == 1:
pk = self.pks[0]
else:
pk = self.pks
if pk is None:
if len(self.pks) == 1:
pk = self.pks[0]
else:
pk = self.pks
sqls.append(
self.db.create_table_sql(
new_table_name,

20
tests/test_transform.py
View file

@ -54,6 +54,26 @@ import pytest
"ALTER TABLE [dogs_new_suffix] RENAME TO [dogs]",
],
),
# Change primary key
(
{"pk": "age"},
[
"CREATE TABLE [dogs_new_suffix] (\n [id] INTEGER,\n [name] TEXT,\n [age] TEXT PRIMARY KEY\n);",
"INSERT INTO [dogs_new_suffix] ([id], [name], [age]) SELECT [id], [name], [age] FROM [dogs]",
"DROP TABLE [dogs]",
"ALTER TABLE [dogs_new_suffix] RENAME TO [dogs]",
],
),
# Change primary key to a compound pk
(
{"pk": ("age", "name")},
[
"CREATE TABLE [dogs_new_suffix] (\n [id] INTEGER,\n [name] TEXT,\n [age] TEXT,\n PRIMARY KEY ([age], [name])\n);",
"INSERT INTO [dogs_new_suffix] ([id], [name], [age]) SELECT [id], [name], [age] FROM [dogs]",
"DROP TABLE [dogs]",
"ALTER TABLE [dogs_new_suffix] RENAME TO [dogs]",
],
),
],
)
def test_transform_sql(fresh_db, params, expected_sql):