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

Sanity check add_foreign_key() column exists, closes #29

Browse source
This commit is contained in:
Simon Willison 2019-06-20 16:58:09 -07:00 • committed by Simon Willison
commit 976263effc
2 changed files with 11 additions and 0 deletions
Download patch file Download diff file Expand all files Collapse all files

3
sqlite_utils/db.py
View file

@ -466,6 +466,9 @@ class Table:
return pks[0].name
def add_foreign_key(self, column, other_table=None, other_column=None):
# Ensure column exists
if column not in self.columns_dict:
raise AlterError("No such column: {}".format(column))
# If other_table is not specified, attempt to guess it from the column
if other_table is None:
other_table = self.guess_foreign_table(column)