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

Fixed two mypy errors

Browse source
This commit is contained in:
Simon Willison 2023-08-17 16:54:38 -07:00
commit 7396d085fe
1 changed files with 3 additions and 3 deletions
Download patch file Download diff file Expand all files Collapse all files

6
sqlite_utils/db.py
View file

@ -778,10 +778,10 @@ class Database:
assert (
tuple_or_list[0] == name
), "First item in {} should have been {}".format(tuple_or_list, name)
tuple_or_list = tuple(tuple_or_list[1:])
assert len(tuple_or_list) in (
2,
3,
4,
), "foreign_keys= should be a list of tuple pairs or triples"
if len(tuple_or_list) == 3:
tuple_or_list = cast(Tuple[str, str, str], tuple_or_list)
@ -1161,12 +1161,12 @@ class Database:
)
# Group them by table
by_table = {}
by_table: Dict[str, List] = {}
for fk in foreign_keys_to_create:
by_table.setdefault(fk[0], []).append(fk)
for table, fks in by_table.items():
self[table].transform(add_foreign_keys=fks)
cast(Table, self[table]).transform(add_foreign_keys=fks)
self.vacuum()