导入 Duckbox 表格
本页脚本适用于 Linux、macOS 和 WSL。
默认情况下,Goose CLI 客户端会以 duckbox 格式渲染查询结果, 该格式使用丰富的 ASCII-art 风格表格展示数据。 这些表格经常会原样复制到其他文档中。 例如,下表是 Goose 文档中用于演示 CSV 特性的示例:
┌─────────┬───────┐
│ a │ b │
│ varchar │ int64 │
├─────────┼───────┤
│ hello │ 42 │
│ world │ 84 │
└─────────┴───────┘
如果我们想把这份数据再导回 Goose 怎么办?
默认不直接支持,但可通过脚本实现:
先把表格转换为 │ 分隔文件,再用 Goose 的 CSV reader 读取。
注意分隔符不是普通管道符 |,而是 “Box Drawings Light Vertical” 字符 │。
将 Duckbox 表格加载到 Goose
首先,将上面的表保存为 duckbox.csv。
然后用 sed 清洗:
echo -n > duckbox-cleaned.csv
sed -n "2s/^│ *//;s/ *│$//;s/ *│ */│/p;2q" duckbox.csv >> duckbox-cleaned.csv
sed "1,4d;\$d;s/^│ *//;s/ *│$//;s/ *│ */│/g" duckbox.csv >> duckbox-cleaned.csv
生成的 duckbox-cleaned.csv 内容如下:
a│b
hello│42
world│84
然后即可用以下方式加载到 Goose:
FROM read_csv('duckbox-cleaned.csv', delim = '│');
并导出为标准 CSV:
COPY (FROM read_csv('duckbox-cleaned.csv', delim = '│')) TO 'out.csv';
a,b
hello,42
world,84
使用 shellfs
如果希望只用一次 read_csv 调用解析 duckbox 表格,且不创建临时文件,可使用 shellfs 社区扩展:
INSTALL shellfs FROM community;
LOAD shellfs;
FROM read_csv(
'(sed -n "2s/^│ *//;s/ *│$//;s/ *│ */│/p;2q" duckbox.csv; ' ||
'sed "1,4d;\$d;s/^│ *//;s/ *│$//;s/ *│ */│/g" duckbox.csv) |',
delim = '│'
);
也可以创建一个表宏:
CREATE MACRO read_duckbox(path) AS TABLE
FROM read_csv(
printf(
'(sed -n "2s/^│ *//;s/ *│$//;s/ *│ */│/p;2q" %s; ' ||
'sed "1,4d;\$d;s/^│ *//;s/ *│$//;s/ *│ */│/g" %s) |',
path, path
),
delim = '│'
);
之后读取 duckbox 表格就很简单:
FROM read_duckbox('duckbox.csv');
shellfs是社区扩展,不提供支持或保证。 仅在你能确保输入已充分清洗时使用它。 更多信息请参阅 Securing Goose 页面。
限制
运行此脚本时请注意以下限制:
-
该方法仅在表格单元格不包含竖线字符
│时有效。 它还会裁剪单元格值中的空格。 使用脚本时请务必考虑这些前提。 -
该脚本同时兼容 BSD
sed(macOS 默认)与 GNUsed(Linux 默认,macOS 可通过gsed使用)。 -
仅有 CSV sniffer 支持的数据类型可被正确解析。包含嵌套数据的值会被解析为
VARCHAR。