在Python阵列中捕获BASH输出

Question

我有一个bash脚本，需要转换为python程序。

我运行命令并在bash数组中捕获其输出，并通过它迭代。

diskArr=(`lsblk | grep 'disk' | awk -v col1=1 '{print $col1}'`)

该命令在系统中为我提供了所有HDD的列表，并将其存储在数组" diskarr"中。

我尝试使用OS.System和subprocess.popen，但尚未成功。

>>> import shlex, subprocess
>>> command_line = raw_input()
lsblk | grep 'disk' | awk -v col1=1 '{print $col1}'
>>> args = shlex.split(command_line)
>>> print args
['lsblk', '|', 'grep', 'disk', '|', 'awk', '-v', 'col1=1', '{print $col1}']
>>>
>>>
>>> subprocess.Popen(args)
<subprocess.Popen object at 0x7f8e083ce590>
>>> lsblk: invalid option -- 'v'
Usage:
 lsblk [options] [<device> ...]

Answer 1

到目前为止，您实际上并没有将程序转换为Python，您只是想将Python用作外壳的包装器。但是您也可以在python中进行格雷普和尴尬：

import subprocess
import re
lsblk = subprocess.Popen(['lsblk'], stdout=subprocess.PIPE, stderr=subprocess.STDOUT)
for line in lsblk.stdout:
    if 'disk' in line:
        parts = re.split(r's+', line.strip())
        name, majmin, rm, size, ro, devtype = parts[:6]
        if len(parts) > 6:
            mountpoint = parts[6]
        else:
            mountpoint = None
        print(majmin)
returncode = lsblk.wait()
if returncode:
    print("things got bad. real bad.")

这只是一个例子。如果您想要一个参考磁盘的行列表，则可以构建一个包含"磁盘"线的列表：

lsblk = subprocess.Popen(['lsblk'], stdout=subprocess.PIPE, stderr=subprocess.STDOUT)
blockdevs = [line.strip() for line in lsblk.stdout if 'disk' in line]
returncode = lsblk.wait()
if returncode:
    print("things got bad. real bad.")
print(blockdevs)

Answer 2

您可能会在官方文档中替换外壳管道，它有一个很好的例子。