remote/sqlite-utils
1
0
Fork 0
mirror of https://github.com/simonw/sqlite-utils.git synced 2026-07-26 10:54:32 +02:00
Code Issues Projects Releases Packages Wiki Activity

Improved foreign_keys= argument, closes #17

Browse source
This commit is contained in:
Simon Willison 2019-02-24 14:12:45 -08:00
commit 557dc3f9a7
3 changed files with 57 additions and 6 deletions
Download patch file Download diff file Expand all files Collapse all files

12
sqlite_utils/db.py
View file

@ -109,6 +109,14 @@ class Database:
if hash_id:
column_items.insert(0, (hash_id, str))
pk = hash_id
# Sanity check foreign_keys point to existing tables
for _, fk_other_table, fk_other_column in foreign_keys:
if not any(
c for c in self[fk_other_table].columns if c.name == fk_other_column
):
raise AlterError(
"No such column: {}.{}".format(fk_other_table, fk_other_column)
)
extra = ""
columns_sql = ",\n".join(
" [{col_name}] {col_type}{primary_key}{references}".format(
@ -117,8 +125,8 @@ class Database:
primary_key=" PRIMARY KEY" if (pk == col_name) else "",
references=(
" REFERENCES [{other_table}]([{other_column}])".format(
other_table=foreign_keys_by_name[col_name][2],
other_column=foreign_keys_by_name[col_name][3],
other_table=foreign_keys_by_name[col_name][1],
other_column=foreign_keys_by_name[col_name][2],
)
if col_name in foreign_keys_by_name
else ""