bash 或 python 脚本从 STDIN 读取二进制数据并保存到文件

Question

在任何人告诉我之前，我已经完全意识到我正在尝试做的事情是荒谬的，但我有我的理由。

我有一个服务器，我可以在上面编写脚本（bash 或 python）。该脚本需要接受来自 STDIN 的二进制数据并将其保存到文件中。准确地说，二进制数据是一个WAV文件。

我正在使用一个名为 Workflow 的 iOS 应用程序。该应用程序将让我录制音频文件，然后将其作为 STDIN 输入发送到它将通过 SSH 运行的脚本。所以我的脚本需要接受该 STDIN 输入并将其保存到文件中。

有什么想法/想法吗？我完全不知道从哪里开始。

Answer 1

将标准输入保存到 POSIXy 系统上的文件中：

1. 确保标准输入以二进制模式打开
2. 检查fadvise 是否在您的情况下提高了时间性能，例如，请参阅copyfileobj() in contrib
3. 从标准输入读取块并将它们写入文件

一般来说，您也可以try os.sendfile() if stdin might be redirected from a file (mmap-able file descriptor)（不适用于您的情况）。

#!/usr/bin/env python
import os
import shutil
import sys

with os.fdopen(sys.stdin.fileno(), 'rb') as input_file,\
     open(sys.argv[1], 'wb') as output_file:
    shutil.copyfileobj(input_file, output_file)

例子：

$ python -msave_stdin output_file

应该相当于：

$ cat >| output_file

Answer 2

在 python 中从标准输入读取：

import sys
input = sys.stdin.read()

将二进制写入文件：

with open('name_of_file_here','wb') as output:
   output.write(input)

'wb' 以二进制模式发送写入的位置。