跳到主要内容

导入 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 默认)与 GNU sed(Linux 默认,macOS 可通过 gsed 使用)。

  • 仅有 CSV sniffer 支持的数据类型可被正确解析。包含嵌套数据的值会被解析为 VARCHAR