python subprocess.run()、subprocess.Popen()、subprocess.check_output()(执行系统命令、启动新的进程、连接到子进程的输入/输出管道)

目录

文章目录

    • python subprocess
    • 基本函数
      • subprocess.run
        • 示例(只可以捕获标准输出,标准错误不知道咋获取,异常也没抛。。。以后再看,现在有事)
      • subprocess.Popen
        • subprocess.Popen参数解释
          • – `args`:要执行的命令及其参数。可以是一个字符串或一个字符串列表。
          • – `bufsize`:指定缓冲区的大小。默认值为-1,表示使用系统默认的缓冲区大小。
          • – `executable`:指定要执行的可执行文件的路径。默认值为None,表示使用系统默认的可执行文件。
          • – `stdin`:指定子进程的标准输入。可以是一个文件对象或一个文件描述符。默认值为None,表示使用父进程的标准输入。
          • – `stdout`:指定子进程的标准输出。可以是一个文件对象或一个文件描述符。默认值为None,表示使用父进程的标准输出。
          • – `stderr`:指定子进程的标准错误输出。可以是一个文件对象或一个文件描述符。默认值为None,表示使用父进程的标准错误输出。
          • – `preexec_fn`:在子进程执行前调用的可调用对象。默认值为None,表示不调用任何函数。
            • 示例:使用preexec_fn来将Popen对象绑定进程组
          • – `close_fds`:指定是否在子进程中关闭父进程中打开的文件描述符。默认值为True,表示关闭文件描述符。
          • – `shell`:指定是否通过shell来执行命令。默认值为False,表示不通过shell执行命令。
            • 使用shell执行和不使用shell执行的差异(以管道为例)
          • – `cwd`:指定子进程的当前工作目录。默认值为None,表示使用父进程的当前工作目录。
          • – `env`:指定子进程的环境变量。可以是一个字典,表示环境变量的键值对。默认值为None,表示使用父进程的环境变量。
          • – `universal_newlines`:指定是否将输入/输出流以文本模式打开。默认值为False,表示以二进制模式打开。
          • – `startupinfo`:指定子进程的启动信息。可以是一个subprocess.STARTUPINFO对象,用于设置子进程的一些属性。默认值为None。
          • – `creationflags`:指定子进程的创建标志。默认值为0,表示使用默认的创建标志。
          • – `restore_signals`:指定是否在子进程中恢复信号处理程序。默认值为True,表示恢复信号处理程序。
          • – `start_new_session`:指定是否在新的会话中启动子进程。默认值为False,表示在当前会话中启动子进程。
          • – `pass_fds`:指定要传递给子进程的文件描述符列表。默认值为(),表示不传递文件描述符。
          • – `encoding`:指定输入/输出的编码方式。默认值为None,表示使用系统默认的编码方式。
          • – `errors`:指定输入/输出的错误处理方式。默认值为None,表示使用系统默认的错误处理方式。
          • 总结
        • Popen对象的常用方法和属性
      • subprocess.check_output

python subprocess

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)
  • 运行指定的命令,并等待其完成。args参数是一个字符串或列表,表示要执行的命令和参数。
  • stdin、stdout和stderr参数分别用于指定子进程的标准输入、标准输出和标准错误输出的处理方式。
  • capture_output参数用于指定是否捕获子进程的输出。
  • check参数用于指定是否检查子进程的返回值,如果返回值不为0,则会抛出CalledProcessError异常。
  • timeout参数用于指定子进程的超时时间。
  • encoding和errors参数用于指定输入/输出的编码方式和错误处理方式。
示例(只可以捕获标准输出,标准错误不知道咋获取,异常也没抛。。。以后再看,现在有事)
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)
  • 启动一个新的进程,并返回一个Popen对象,可以用于与子进程进行交互。
  • args参数和其他参数的含义与subprocess.run()函数相同。
subprocess.Popen参数解释
args:要执行的命令及其参数。可以是一个字符串或一个字符串列表。
bufsize:指定缓冲区的大小。默认值为-1,表示使用系统默认的缓冲区大小。
executable:指定要执行的可执行文件的路径。默认值为None,表示使用系统默认的可执行文件。
stdin:指定子进程的标准输入。可以是一个文件对象或一个文件描述符。默认值为None,表示使用父进程的标准输入。
stdout:指定子进程的标准输出。可以是一个文件对象或一个文件描述符。默认值为None,表示使用父进程的标准输出。
stderr:指定子进程的标准错误输出。可以是一个文件对象或一个文件描述符。默认值为None,表示使用父进程的标准错误输出。
preexec_fn:在子进程执行前调用的可调用对象。默认值为None,表示不调用任何函数。
示例:使用preexec_fn来将Popen对象绑定进程组

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

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

  1. 创建Popen对象时,设置preexec_fn=os.setsid,以创建一个新的进程组。
  2. 使用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执行的命令,它们都属于同一个进程组,发送信号时需要注意。

close_fds:指定是否在子进程中关闭父进程中打开的文件描述符。默认值为True,表示关闭文件描述符。
shell:指定是否通过shell来执行命令。默认值为False,表示不通过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命令,并将输出通过管道传递给command2command2执行了grep .txt命令,并将结果输出到标准输出。

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

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

cwd:指定子进程的当前工作目录。默认值为None,表示使用父进程的当前工作目录。
env:指定子进程的环境变量。可以是一个字典,表示环境变量的键值对。默认值为None,表示使用父进程的环境变量。
universal_newlines:指定是否将输入/输出流以文本模式打开。默认值为False,表示以二进制模式打开。
startupinfo:指定子进程的启动信息。可以是一个subprocess.STARTUPINFO对象,用于设置子进程的一些属性。默认值为None。
creationflags:指定子进程的创建标志。默认值为0,表示使用默认的创建标志。
restore_signals:指定是否在子进程中恢复信号处理程序。默认值为True,表示恢复信号处理程序。
start_new_session:指定是否在新的会话中启动子进程。默认值为False,表示在当前会话中启动子进程。
pass_fds:指定要传递给子进程的文件描述符列表。默认值为(),表示不传递文件描述符。
encoding:指定输入/输出的编码方式。默认值为None,表示使用系统默认的编码方式。
errors:指定输入/输出的错误处理方式。默认值为None,表示使用系统默认的错误处理方式。
总结

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

Popen对象的常用方法和属性
  • communicate(input=None, timeout=None):与子进程进行交互,发送输入并获取输出。
  • wait():等待子进程结束,并返回其返回值。
  • poll():检查子进程是否结束,如果结束则返回其返回值,否则返回None。
  • terminate():终止子进程。
  • kill():杀死子进程。
  • stdout:子进程的标准输出。
  • stderr:子进程的标准错误输出。
  • returncode:子进程的返回值。

下面是一个使用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是一个字符串或者字符串列表,表示要执行的命令及其参数。stdinstderrshelluniversal_newlines是可选参数,用于控制输入、错误输出、是否使用shell以及输出结果的编码方式。

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

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

文章出处登录后可见!

已经登录?立即刷新

共计人评分,平均

到目前为止还没有投票!成为第一位评论此文章。

(0)
青葱年少的头像青葱年少普通用户
上一篇 2023年11月3日
下一篇 2023年11月3日

相关推荐