需求

有时候需要把 shell 命令在 python 中执行，如果使用 os.system(cmd) 的方式的话，标准输出只会打印在屏幕上，没办法赋给变量，这时候就可以用 subprocess 来实现

实现

1. os.system(cmd)

In [60]: path = "/hdfs_path/.../"

In [61]: cmd = ' /opt/tiger/.../bin/hadoop fs -du -s -h {} '.format(path)
    ...: res = os.system(cmd)
    ...:
# 如下，输出只是打印在屏幕上，没办法用变量存起来
0  0  /hdfs_path/.../

In [62]: res  # 为 0 仅表示命令运行成功，与命令的标准输出无关
Out[62]: 0

2. subprocess

def du_hdfs_file(path):

    from subprocess import PIPE, Popen

    def cmdline(command):
        """获取标准输出"""
        process = Popen(
            args=command,
            stdout=PIPE,
            shell=True
        )
        return process.communicate()[0]

    cmd = ' /opt/tiger/.../bin/hadoop fs -du -s -h {} '.format(path)
    # res = os.system(cmd)
    res = cmdline(cmd)
    return res

######################################
In [65]: res = du_hdfs_file(path)  # 标准输出赋值给 res

In [66]: res
Out[66]: '0  0  /hdfs_path/.../\n'

In [67]: res[0]
Out[67]: '0'

参考

1. Assign output of os.system to a variable and prevent it from being displayed on the screen [duplicate]