remote/sqlite-utils
1
0
Fork 0
mirror of https://github.com/simonw/sqlite-utils.git synced 2026-07-23 17:34:32 +02:00
Code Issues Projects Releases Packages Wiki Activity

--raw option, refs #123

Browse source
This commit is contained in:
Simon Willison 2020-07-26 09:43:45 -07:00
commit bc8409941f
3 changed files with 35 additions and 2 deletions
Download patch file Download diff file Expand all files Collapse all files

11
sqlite_utils/cli.py
View file

@ -675,7 +675,8 @@ def drop_view(path, view):
)
@click.argument("sql")
@output_options
def query(path, sql, nl, arrays, csv, no_headers, table, fmt, json_cols):
@click.option("-r", "--raw", is_flag=True, help="Raw output, first column of first row")
def query(path, sql, nl, arrays, csv, no_headers, table, fmt, json_cols, raw):
"Execute SQL query and return the results as JSON"
db = sqlite_utils.Database(path)
with db.conn:
@ -686,7 +687,13 @@ def query(path, sql, nl, arrays, csv, no_headers, table, fmt, json_cols):
cursor = [[cursor.rowcount]]
else:
headers = [c[0] for c in cursor.description]
if table:
if raw:
data = cursor.fetchone()[0]
if isinstance(data, bytes):
sys.stdout.buffer.write(data)
else:
sys.stdout.write(str(data))
elif table:
print(tabulate.tabulate(list(cursor), headers=headers, tablefmt=fmt))
elif csv:
writer = csv_std.writer(sys.stdout)