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

Applied Black

Browse source
This commit is contained in:
Simon Willison 2020-09-22 17:32:40 -07:00
commit 5534c320e4
2 changed files with 8 additions and 8 deletions
Download patch file Download diff file Expand all files Collapse all files

6
sqlite_utils/db.py
View file

@ -741,9 +741,9 @@ class Table(Queryable):
defaults=defaults, defaults=defaults,
drop_foreign_keys=drop_foreign_keys, drop_foreign_keys=drop_foreign_keys,
) )
pragma_foreign_keys_was_on = self.db.execute( pragma_foreign_keys_was_on = self.db.execute("PRAGMA foreign_keys").fetchone()[
"PRAGMA foreign_keys" 0
).fetchone()[0] ]
try: try:
if pragma_foreign_keys_was_on: if pragma_foreign_keys_was_on:
self.db.execute("PRAGMA foreign_keys=0;") self.db.execute("PRAGMA foreign_keys=0;")

10
tests/test_transform.py
View file

@ -103,12 +103,12 @@ def test_transform_sql(fresh_db, params, expected_sql, use_pragma_foreign_keys):
dogs.transform(**params) dogs.transform(**params)
# If use_pragma_foreign_keys, check that we did the right thing # If use_pragma_foreign_keys, check that we did the right thing
if use_pragma_foreign_keys: if use_pragma_foreign_keys:
assert ('PRAGMA foreign_keys=0;', None) in captured assert ("PRAGMA foreign_keys=0;", None) in captured
assert captured[-2] == ('PRAGMA foreign_key_check;', None) assert captured[-2] == ("PRAGMA foreign_key_check;", None)
assert captured[-1] == ('PRAGMA foreign_keys=1;', None) assert captured[-1] == ("PRAGMA foreign_keys=1;", None)
else: else:
assert ('PRAGMA foreign_keys=0;', None) not in captured assert ("PRAGMA foreign_keys=0;", None) not in captured
assert ('PRAGMA foreign_keys=1;', None) not in captured assert ("PRAGMA foreign_keys=1;", None) not in captured
def test_transform_sql_rowid_to_id(fresh_db): def test_transform_sql_rowid_to_id(fresh_db):