Python获取命令实时输出-原样彩色输出并返回输出结果

Posted 2022-12-06 亦游

经试验显示效果不错。

#!/usr/bin/python3
# -*- coding: utf-8 -*-
import os
import subprocess


# 与在命令窗口执行显示效果相同，如有彩色输出可保留，但不能返回结果
def run(command):
    subprocess.call(command, shell=True)


# 实时输出但不可显示彩色，可以返回结果
def sh(command, print_msg=True):
    p = subprocess.Popen(command, shell=True, stdout=subprocess.PIPE, stderr=subprocess.STDOUT)
    lines = []
    for line in iter(p.stdout.readline, b''):
        line = line.rstrip().decode('utf8')
        if print_msg:
            print(">>>", line)
        lines.append(line)
    return lines


print('run():')
run("ping www.baidu.com")
print('\\n\\nsh():')
run("ping www.baidu.com")