如何在列表或数组中的 python 脚本中保存 shell-script-printout？

Question

我想在 python 脚本中运行 shell 命令。并将打印输出保存在列表或数组中。

例如shell命令的打印输出如下

root@cic-1:~# for i in $(neutron net-list | grep -i cscf2 | awk '{print $2}'); do neutron net-show $i | grep -i segmentation|awk '{print $4}' ; done
2006
2022
2023
2081
root@cic-1:~#

我希望将此打印输出保存在这样的列表中

[2006,2022,2023,2081]

我已经完成了如下的python代码

#!/usr/pkg/bin/python
import subprocess

vlans= subprocess.Popen ("for i in $(neutron net-list | grep -i cscf2 | awk '{print $2}'); do neutron net-show $i | grep -i segmentation|awk '{print $4}' ; done", shell=True, stdout=subprocess.PIPE).communicate()[0]
print vlans[0]

输出是 2 ，但是我希望 vlans[0] 是 2006

Answer 1

如果您希望vlans[0] 成为输出的第一行，则需要将输出拆分为行。为此，我在vlans = ... 命令的末尾添加了.split('\n')：

>>> vlans= subprocess.Popen ('printf "%s\n" 2006 2022 2023 2081', shell=True, stdout=subprocess.PIPE).communicate()[0].split('\n')
>>> vlans
['2006', '2022', '2023', '2081', '']
>>> vlans[0]
'2006'

当然，因为我没有 neutron 或 net-list，所以我替换了另一个 shell 命令，它提供了与您显示的相同的输出。

改进

Shell 输出通常包含最后一个换行符。但是，当 python 的 split 函数看到最后的换行符时，它会创建一个最终的空字符串。为了消除这种情况，我们可以将strip 应用于输出，然后将其传递给split：

>>> vlans= subprocess.Popen ('printf "%s\n" 2006 2022 2023 2081', shell=True, stdout=subprocess.PIPE).communicate()[0].strip().split('\n')
>>> vlans
['2006', '2022', '2023', '2081']
>>> vlans[0]

根据您的目标，您可能希望使用rstrip('\n') 代替更饥饿的strip()。

Answer 2

您可以使用 subprocess.check_output 来（更）轻松地捕获命令的输出。这在功能上等同于您所拥有的。

vlans = subprocess.check_output(
    "for i in $(neutron net-list | grep -i cscf2 | awk '{print $2}'); "
    "do neutron net-show $i | grep -i segmentation|awk '{print $4}' ; done")

如果您希望将 vlan 作为字符串列表，您可以这样做：

vlans = vlans.splitlines()

如果您想要一个整数列表，请改为：

vlans = [int(v) for v in vlans.splitlines()]