python

关注公众号 jb51net

关闭
首页 > 脚本专栏 > python > python subprocess

python subprocess.run()、subprocess.Popen()、subprocess.check_output()

作者：Dontla

Python的subprocess模块是用于创建和管理子进程的模块,本文主要介绍了python subprocess.run()、subprocess.Popen()、subprocess.check_output(),具有一定的参考价值,感兴趣的可以了解一下

Python的subprocess模块是用于创建和管理子进程的模块。它提供了一种在Python中调用外部命令的方式，可以执行系统命令、启动新的进程、连接到子进程的输入/输出管道等。

基本函数

subprocess.run

subprocess.run(args, *, stdin=None, input=None, stdout=None,
               stderr=None, capture_output=False, text=None,
               check=False, timeout=None, encoding=None, errors=None)

示例（只可以捕获标准输出，标准错误不知道咋获取，异常也没抛。。。以后再看，现在有事）

import subprocess


def execute_update_ip_sh(ip):
    """
    执行修改配置文件 ip 脚本，将会修改ky_ai_solution.json、nginx.conf、vsftpd.conf 等文件中的ip
    """
    try:
        # 构建命令
        command = ['/ky/update_ip.sh', ip]

        # 执行命令并获取输出（这个出错直接就抛异常了，不能捕获标准输出和标准错误）
        # result = subprocess.check_output(command, universal_newlines=True)

        result = subprocess.run(command, stdout=subprocess.PIPE,
                                stderr=subprocess.PIPE, universal_newlines=True)

        # 打印脚本的输出
        print(f"[{command}] 标准输出:", result.stdout)

        # 打印脚本的错误输出（打印不出来）
        print(f"[{command}] 标准错误输出:", result.stderr)

        # 执行成功，返回True
        return result.returncode == 0
    # except subprocess.CalledProcessError as e:
    except Exception as e:
        # 执行失败，返回False
        print(e)
        print("ddddddddddddddddddddddddddddddddddddddd")
        return False


if __name__ == "__main__":
    execute_update_ip_sh("192.168.1.140")

在这里插入图片描述

subprocess.Popen

subprocess.Popen(args, bufsize=-1, executable=None, 
                 stdin=None, stdout=None, stderr=None, 
                 preexec_fn=None, close_fds=True, 
                 shell=False, cwd=None, env=None, 
                 universal_newlines=False, startupinfo=None, 
                 creationflags=0, restore_signals=True, 
                 start_new_session=False, pass_fds=(), *, 
                 encoding=None, errors=None)

subprocess.Popen参数解释

示例：使用preexec_fn来将Popen对象绑定进程组

当使用shell=True时，subprocess.Popen()函数创建的子进程实际上是一个shell进程，而不是直接执行的命令。因此，向Popen对象发送信号并不能直接影响到子进程。

如果需要向通过shell执行的命令发送信号，可以使用os.killpg()函数来发送信号给进程组。具体步骤如下：

下面是一个示例，演示了如何向通过shell执行的命令发送信号：

import subprocess
import os
import signal

# 通过shell执行命令
process = subprocess.Popen('sleep 10', shell=True, preexec_fn=os.setsid)

# 发送信号给进程组
os.killpg(os.getpgid(process.pid), signal.SIGTERM)

在这个示例中，首先通过subprocess.Popen()函数创建了一个通过shell执行的命令。在创建Popen对象时，通过preexec_fn=os.setsid设置了preexec_fn参数，以创建一个新的进程组。

然后，使用os.killpg()函数发送信号给进程组。在这个示例中，发送了SIGTERM信号，即终止信号。

需要注意的是，向进程组发送信号会影响到进程组中的所有进程。因此，如果有多个通过shell执行的命令，它们都属于同一个进程组，发送信号时需要注意。

当shell参数设置为True时，表示通过shell来执行命令。这意味着可以使用shell的语法和功能，比如管道、重定向、通配符等。在这种情况下，args参数可以是一个字符串，表示要执行的完整命令。

当shell参数设置为False时，表示不通过shell来执行命令。这是默认的行为。在这种情况下，args参数应该是一个字符串列表，其中第一个元素是要执行的命令，后续元素是命令的参数。

需要注意的是，使用shell=True时，应该谨慎处理输入参数，避免命令注入等安全问题。应该始终对用户输入进行验证和过滤，避免直接将用户输入拼接到命令中。

下面是一个示例，演示了shell参数的用法：

import subprocess

# 通过shell执行命令
subprocess.Popen('ls -l', shell=True)

# 不通过shell执行命令
subprocess.Popen(['ls', '-l'])

在这个示例中，第一个Popen()函数调用通过shell执行了ls -l命令，而第二个Popen()函数调用不通过shell执行了ls -l命令。

使用shell执行和不使用shell执行的差异（以管道为例）

使用shell执行：

下面是一个示例，演示了如何通过shell执行命令并使用管道：

import subprocess

# 通过shell执行命令并使用管道
process = subprocess.Popen('ls -l | grep .txt', shell=True)

# 等待子进程结束
process.wait()

在这个示例中，通过subprocess.Popen()函数的shell参数设置为True，表示通过shell执行命令。然后，可以使用shell的管道语法|将两个命令连接起来，实现输出过滤。

不使用shell执行：

import subprocess

# 不通过shell执行命令并使用管道
command1 = subprocess.Popen(['ls', '-l'], stdout=subprocess.PIPE)
command2 = subprocess.Popen(['grep', '.txt'], stdin=command1.stdout, stdout=subprocess.PIPE)

# 获取命令2的输出
output = command2.communicate()[0]

# 打印输出结果
print(output.decode())

在这个示例中，首先创建了两个Popen()对象，分别对应两个命令。command1执行了ls -l命令，并将输出通过管道传递给command2。command2执行了grep .txt命令，并将结果输出到标准输出。

通过command2.communicate()方法获取命令2的输出，并将其打印出来。

不通过shell执行命令时，需要手动设置命令之间的输入输出关系，以及处理命令的输出。这需要更多的代码来实现，但也提供了更多的灵活性和控制能力。

总结

subprocess.Popen()函数会返回一个Popen对象，可以用于与子进程进行交互。Popen对象具有一些常用的方法和属性，比如communicate()、wait()、poll()、terminate()、kill()、stdout、stderr和returncode等。

Popen对象的常用方法和属性

下面是一个使用subprocess模块的示例，演示了如何执行一个系统命令并获取其输出：

import subprocess

# 执行系统命令
result = subprocess.run(['ls', '-l'], capture_output=True, text=True)

# 获取命令的输出
output = result.stdout
print(output)

以上代码会执行ls -l命令，并将其输出打印出来。

需要注意的是，subprocess模块在执行外部命令时，存在一定的安全风险，特别是在使用shell=True参数时。因此，在使用subprocess模块时，应该谨慎处理输入参数，避免命令注入等安全问题。

subprocess.check_output

subprocess.check_output函数是Python标准库subprocess模块中的一个函数，用于执行系统命令并返回命令的输出结果。

它的用法是：

subprocess.check_output(args, *, stdin=None, stderr=None, 
                        shell=False, universal_newlines=False)

其中，args是一个字符串或者字符串列表，表示要执行的命令及其参数。stdin、stderr、shell和universal_newlines是可选参数，用于控制输入、错误输出、是否使用shell以及输出结果的编码方式。

check_output函数会执行指定的命令，并等待命令执行完成。如果命令执行成功，它会返回命令的输出结果（以字节字符串的形式），如果命令执行失败，它会抛出一个CalledProcessError异常。

在上述代码中，subprocess.check_output函数被用于执行一些系统命令，如ip addr show和ip route show，以获取当前以太网接口的IP配置信息。

到此这篇关于python subprocess.run()、subprocess.Popen()、subprocess.check_output()的文章就介绍到这了,更多相关python subprocess 内容请搜索脚本之家以前的文章或继续浏览下面的相关文章希望大家以后多多支持脚本之家！

您可能感兴趣的文章:
阅读全文