datasette argument

Refs datasette/datasette-enrichments#16
datasette · Nov 17, 2023 · 3c2808e · 3c2808e
1 parent 43cd59d
commit 3c2808e
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 2 deletions.
diff --git a/datasette_enrichments_jinja.py b/datasette_enrichments_jinja.py
@@ -1,3 +1,4 @@
+from __future__ import annotations
 from datasette_enrichments import Enrichment
 from datasette import hookimpl
 
@@ -42,7 +43,7 @@ class ConfigForm(Form):
 
         return ConfigForm
 
-    async def initialize(self, db, table, config):
+    async def initialize(self, datasette, db, table, config):
         # Ensure column exists
         output_column = config["output_column"][0]
 
@@ -55,6 +56,7 @@ def add_column_if_not_exists(conn):
 
     async def enrich_batch(
         self,
+        datasette: Datasette,
         db: Database,
         table: str,
         rows: List[dict],

diff --git a/tests/test_enrichments_jinja.py b/tests/test_enrichments_jinja.py
@@ -21,11 +21,12 @@ async def test_enrichment(tmpdir):
     db["items"].insert_all(rows)
     ds_db = datasette.get_database("data")
     enrichment = JinjaSandbox()
-    await enrichment.initialize(ds_db, "items", config)
+    await enrichment.initialize(datasette, ds_db, "items", config)
     # It should now have a template_output column
     assert "template_output" in db["items"].columns_dict
     # Now enrich the batch
     await enrichment.enrich_batch(
+        datasette=datasette,
         db=ds_db,
         table="items",
         rows=rows,