Run everything in a transaction

simonw · Jan 24, 2024 · 4c19d5e · 4c19d5e
1 parent 7a34bb1
commit 4c19d5e
Showing 1 changed file with 9 additions and 8 deletions.
diff --git a/datasette_upload_csvs/__init__.py b/datasette_upload_csvs/__init__.py
@@ -159,19 +159,20 @@ def docs_with_progress():
  "completed": str(datetime.datetime.utcnow()),
  },
  )
- # Trasform columns to detected types
+ # Transform columns to detected types
  database[table_name].transform(types=tracker.types)
  return database[table_name].count
 
  def insert_docs_catch_errors(conn):
  database = sqlite_utils.Database(conn)
- try:
- insert_docs(database)
- except Exception as error:
- database["_csv_progress_"].update(
- task_id,
- {"error": str(error)},
- )
+ with conn:
+ try:
+ insert_docs(database)
+ except Exception as error:
+ database["_csv_progress_"].update(
+ task_id,
+ {"error": str(error)},
+ )
 
  await db.execute_write_fn(insert_docs_catch_errors, block=False)