sqlite-utils query can now run DML (#120)

* Failing test showing that DML in `sqlite-utils query` doesn't work
* Run `sqlite-utils query` in a transaction so that DML is committed

Thanks, @tsibley!

sqlite_utils/cli.py

@@ -669,24 +669,25 @@ def drop_view(path, view):
 def query(path, sql, nl, arrays, csv, no_headers, table, fmt, json_cols):
     "Execute SQL query and return the results as JSON"
     db = sqlite_utils.Database(path)
-    cursor = db.conn.execute(sql)
-    if cursor.description is None:
-        # This was an update/insert
-        headers = ["rows_affected"]
-        cursor = [[cursor.rowcount]]
-    else:
-        headers = [c[0] for c in cursor.description]
-    if table:
-        print(tabulate.tabulate(list(cursor), headers=headers, tablefmt=fmt))
-    elif csv:
-        writer = csv_std.writer(sys.stdout)
-        if not no_headers:
-            writer.writerow(headers)
-        for row in cursor:
-            writer.writerow(row)
-    else:
-        for line in output_rows(cursor, headers, nl, arrays, json_cols):
-            click.echo(line)
+    with db.conn:
+        cursor = db.conn.execute(sql)
+        if cursor.description is None:
+            # This was an update/insert
+            headers = ["rows_affected"]
+            cursor = [[cursor.rowcount]]
+        else:
+            headers = [c[0] for c in cursor.description]
+        if table:
+            print(tabulate.tabulate(list(cursor), headers=headers, tablefmt=fmt))
+        elif csv:
+            writer = csv_std.writer(sys.stdout)
+            if not no_headers:
+                writer.writerow(headers)
+            for row in cursor:
+                writer.writerow(row)
+        else:
+            for line in output_rows(cursor, headers, nl, arrays, json_cols):
+                click.echo(line)
 
 
 @cli.command()

tests/test_cli.py

@@ -1126,3 +1126,6 @@ def test_query_update(db_path, args, expected):
         cli.cli, [db_path, "update dogs set age = 5 where name = 'Cleo'"] + args
     )
     assert expected == result.output.strip()
+    assert db.execute_returning_dicts("select * from dogs") == [
+        {"id": 1, "age": 5, "name": "Cleo"},
+    ]