mirror of
https://github.com/simonw/sqlite-utils.git
synced 2026-07-23 09:24:31 +02:00
8d74ffc932* Add comprehensive type annotations - mypy.ini: expanded configuration with module-specific settings - hookspecs.py: type annotations for hook functions - plugins.py: typed get_plugins() return value - recipes.py: full type annotations for parsedate, parsedatetime, jsonsplit - utils.py: extensive type annotations including Row type alias, TypeTracker, ValueTracker, and all utility functions - db.py: type annotations for Database methods (__exit__, ensure_autocommit_off, tracer, register_function, etc.) and Queryable class methods - tests/test_docs.py: updated to match new signature display format * Fix type errors caught by ty check - Add type: ignore comments for external library type stub limitations (csv.reader, click.progressbar, IOBase.name, Callable.__name__) - Change Iterable to Sequence for SQL where_args parameters - Use db.table() instead of db[name] for proper Table return type - Fix rebuild_fts return type from None to Table - Update test_tracer to expect fewer queries (optimization side effect) * Fix mypy type errors - Add type: ignore comments for runtime-valid patterns mypy can't verify - Fix new_column_types annotation to Dict[str, Set[type]] - Add type: ignore for Default sentinel values passed to create_table * mypy skip tests directory * Fix CI: exclude typing imports from recipe docs, skip mypy on tests - Add Callable and Optional to exclusion list in _generate_convert_help() - Regenerate docs/cli-reference.rst with cog - Add [mypy-tests.*] ignore_errors = True to skip test type errors --------- Co-authored-by: Claude Opus 4.5 <noreply@anthropic.com>
96 lines
3 KiB
Python
96 lines
3 KiB
Python
from sqlite_utils import Database
|
|
|
|
|
|
def test_tracer():
|
|
collected = []
|
|
db = Database(
|
|
memory=True, tracer=lambda sql, params: collected.append((sql, params))
|
|
)
|
|
dogs = db.table("dogs")
|
|
dogs.insert({"name": "Cleopaws"})
|
|
dogs.enable_fts(["name"])
|
|
dogs.search("Cleopaws")
|
|
assert collected == [
|
|
("PRAGMA recursive_triggers=on;", None),
|
|
("select name from sqlite_master where type = 'view'", None),
|
|
("select name from sqlite_master where type = 'table'", None),
|
|
("select name from sqlite_master where type = 'view'", None),
|
|
("select name from sqlite_master where type = 'view'", None),
|
|
("select name from sqlite_master where type = 'table'", None),
|
|
("select name from sqlite_master where type = 'view'", None),
|
|
('CREATE TABLE "dogs" (\n "name" TEXT\n);\n ', None),
|
|
("select name from sqlite_master where type = 'view'", None),
|
|
('INSERT INTO "dogs" ("name") VALUES (?)', ["Cleopaws"]),
|
|
(
|
|
'CREATE VIRTUAL TABLE "dogs_fts" USING FTS5 (\n "name",\n content="dogs"\n)',
|
|
None,
|
|
),
|
|
(
|
|
'INSERT INTO "dogs_fts" (rowid, "name")\n SELECT rowid, "name" FROM "dogs";',
|
|
None,
|
|
),
|
|
]
|
|
|
|
|
|
def test_with_tracer():
|
|
collected = []
|
|
|
|
def tracer(sql, params):
|
|
return collected.append((sql, params))
|
|
|
|
db = Database(memory=True)
|
|
|
|
dogs = db.table("dogs")
|
|
|
|
dogs.insert({"name": "Cleopaws"})
|
|
dogs.enable_fts(["name"])
|
|
|
|
assert len(collected) == 0
|
|
|
|
with db.tracer(tracer):
|
|
list(dogs.search("Cleopaws"))
|
|
|
|
assert len(collected) == 4
|
|
assert collected == [
|
|
(
|
|
"SELECT name FROM sqlite_master\n"
|
|
" WHERE rootpage = 0\n"
|
|
" AND (\n"
|
|
" sql LIKE :like\n"
|
|
" OR sql LIKE :like2\n"
|
|
" OR (\n"
|
|
" tbl_name = :table\n"
|
|
" AND sql LIKE '%VIRTUAL TABLE%USING FTS%'\n"
|
|
" )\n"
|
|
" )",
|
|
{
|
|
"like": '%VIRTUAL TABLE%USING FTS%content="dogs"%',
|
|
"like2": '%VIRTUAL TABLE%USING FTS%content="dogs"%',
|
|
"table": "dogs",
|
|
},
|
|
),
|
|
("select name from sqlite_master where type = 'view'", None),
|
|
("select sql from sqlite_master where name = ?", ("dogs_fts",)),
|
|
(
|
|
'with "original" as (\n'
|
|
" select\n"
|
|
" rowid,\n"
|
|
" *\n"
|
|
' from "dogs"\n'
|
|
")\n"
|
|
"select\n"
|
|
' "original".*\n'
|
|
"from\n"
|
|
' "original"\n'
|
|
' join "dogs_fts" on "original".rowid = "dogs_fts".rowid\n'
|
|
"where\n"
|
|
' "dogs_fts" match :query\n'
|
|
"order by\n"
|
|
' "dogs_fts".rank',
|
|
{"query": "Cleopaws"},
|
|
),
|
|
]
|
|
|
|
# Outside the with block collected should not be appended to
|
|
dogs.insert({"name": "Cleopaws"})
|
|
assert len(collected) == 4
|