如何从 Python 运行 shell 命令

简介

本教程将探讨直接从 Python 运行 shell 命令的基本技术，为开发者提供与系统级操作进行交互的强大方法。Python 提供了多种执行 shell 命令的方法，能够实现 Python 脚本与系统环境的无缝集成。

Shell 命令基础

Python 中的 Shell 命令简介

Shell 命令是系统交互的基础，它允许开发者直接从 Python 脚本中执行操作系统命令。在系统管理、自动化和 DevOps 的场景中，理解如何运行 Shell 命令至关重要。

为什么要从 Python 运行 Shell 命令？

从 Python 运行 Shell 命令有几个关键优势：

运行 Shell 命令的基本方法

1. os.system() 方法

运行 Shell 命令的最简单方法，但功能有限：

import os

## 执行一个基本的 Shell 命令
os.system('ls -l')

2. subprocess 模块（推荐）

运行 Shell 命令最强大、最灵活的方法：

import subprocess

## 运行命令并捕获输出
result = subprocess.run(['ls', '-l'], capture_output=True, text=True)
print(result.stdout)

关键注意事项

• 错误处理
• 输出捕获
• 安全影响
• 性能优化

LabEx Pro 提示

在学习 Shell 命令执行时，在像 LabEx 这样的受控环境中进行实践，以了解 Python 和系统命令之间细微的交互。

subprocess 执行

理解 subprocess 模块

subprocess 模块是在 Python 中执行 shell 命令的推荐方式，它提供了强大且灵活的命令执行能力。

subprocess 方法概述

1. subprocess.run() - 现代方法

import subprocess

## 基本命令执行
result = subprocess.run(['ls', '-l'], capture_output=True, text=True)
print(result.stdout)

## 使用 shell 的复杂命令
result = subprocess.run('grep -R "error" /var/log', shell=True, capture_output=True, text=True)

2. subprocess.Popen() - 高级执行

import subprocess

## 详细的进程控制
process = subprocess.Popen(['ping', '-c', '4', 'google.com'],
                            stdout=subprocess.PIPE,
                            stderr=subprocess.PIPE,
                            text=True)

stdout, stderr = process.communicate()
print(stdout)

执行参数

高级执行技术

超时处理

import subprocess

try:
    result = subprocess.run(['sleep', '10'],
                             timeout=5,
                             capture_output=True)
except subprocess.TimeoutExpired:
    print("命令超时")

环境定制

import subprocess
import os

custom_env = os.environ.copy()
custom_env['CUSTOM_VAR'] = '值'

subprocess.run(['env'], env=custom_env)

LabEx Pro 提示

在实践 subprocess 技术时，始终要考虑安全影响和潜在的命令注入风险。

错误处理

Shell 命令执行中的错误处理策略

在运行 shell 命令时，正确的错误处理对于确保 Python 脚本的健壮性和可靠性至关重要。

subprocess 中的异常类型

1. 基本错误处理

import subprocess

try:
    ## 当 check=True 时，非零退出码会引发异常
    result = subprocess.run(['ls', '/nonexistent'],
                             check=True,
                             capture_output=True,
                             text=True)
except subprocess.CalledProcessError as e:
    print(f"命令以退出码 {e.returncode} 失败")
    print(f"错误输出: {e.stderr}")

常见错误场景

2. 高级错误处理

import subprocess
import sys

def run_safe_command(command):
    try:
        result = subprocess.run(command,
                                capture_output=True,
                                text=True,
                                check=True)
        return result.stdout
    except subprocess.CalledProcessError as e:
        print(f"执行 {command} 时出错")
        print(f"错误输出: {e.stderr}")
        sys.exit(1)
    except FileNotFoundError:
        print(f"命令 {command} 未找到")
        sys.exit(1)

## 使用方法
output = run_safe_command(['grep', 'error', '/var/log/syslog'])

3. 超时处理

import subprocess

try:
    result = subprocess.run(['ping', '-c', '10', 'google.com'],
                             timeout=5,
                             capture_output=True,
                             text=True)
except subprocess.TimeoutExpired:
    print("命令超时")

最佳实践

• 始终使用 check=True 来捕获命令失败
• 同时捕获标准输出和标准错误
• 实现全面的错误处理
• 对可能长时间运行的命令使用超时设置

LabEx Pro 提示

在学习错误处理时，通过各种场景进行实践，以培养健壮的 shell 命令执行技能。

总结

通过掌握在 Python 中执行 shell 命令，开发者可以创建出更通用、强大的脚本，与操作系统功能进行高效交互。理解 subprocess 方法、错误处理和命令执行策略，能使 Python 程序员构建出健壮的跨平台自动化和系统管理工具。