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

db.close() method, closes #504

Browse source
This commit is contained in:
Simon Willison 2022-10-25 13:57:43 -07:00
commit 05e2bb85fc
2 changed files with 18 additions and 0 deletions
Download patch file Download diff file Expand all files Collapse all files

4
sqlite_utils/db.py
View file

@ -336,6 +336,10 @@ class Database:
self._registered_functions: set = set() self._registered_functions: set = set()
self.use_counts_table = use_counts_table self.use_counts_table = use_counts_table
def close(self):
"Close the SQLite connection, and the underlying database file"
self.conn.close()
@contextlib.contextmanager @contextlib.contextmanager
def tracer(self, tracer: Callable = None): def tracer(self, tracer: Callable = None):
""" """

14
tests/test_constructor.py
View file

@ -1,4 +1,6 @@
from sqlite_utils import Database from sqlite_utils import Database
from sqlite_utils.utils import sqlite3
import pytest
def test_recursive_triggers(): def test_recursive_triggers():
@ -25,3 +27,15 @@ def test_sqlite_version():
as_string = ".".join(map(str, version)) as_string = ".".join(map(str, version))
actual = next(db.query("select sqlite_version() as v"))["v"] actual = next(db.query("select sqlite_version() as v"))["v"]
assert actual == as_string assert actual == as_string
@pytest.mark.parametrize("memory", [True, False])
def test_database_close(tmpdir, memory):
if memory:
db = Database(memory=True)
else:
db = Database(str(tmpdir / "test.db"))
assert db.execute("select 1 + 1").fetchone()[0] == 2
db.close()
with pytest.raises(sqlite3.ProgrammingError):
db.execute("select 1 + 1")