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

Now complies with flake8, refs #291

Browse source
This commit is contained in:
Simon Willison 2021-06-22 18:22:08 -07:00
commit 90e211e3e2
20 changed files with 252 additions and 67 deletions
Download patch file Download diff file Expand all files Collapse all files

6
sqlite_utils/db.py
View file

@ -1,5 +1,5 @@
from .utils import sqlite3, OperationalError, suggest_column_types, column_affinity
from collections import namedtuple, OrderedDict
from collections import namedtuple
from collections.abc import Mapping
import contextlib
import datetime
@ -1116,8 +1116,6 @@ class Table(Queryable):
)
)
table = table or "_".join(columns)
first_column = columns[0]
pks = self.pks
lookup_table = self.db[table]
fk_column = fk_column or "{}_id".format(table)
magic_lookup_column = "{}_{}".format(fk_column, os.urandom(6).hex())
@ -1236,7 +1234,7 @@ class Table(Queryable):
fk_col_type = None
if fk is not None:
# fk must be a valid table
if not fk in self.db.table_names():
if fk not in self.db.table_names():
raise AlterError("table '{}' does not exist".format(fk))
# if fk_col specified, must be a valid column
if fk_col is not None: