Python 运行 shell 命令的方法汇总

from pathlib import Path
# 创建一个Path对象表示当前工作目录
current_directory = Path.cwd()
print("当前工作目录:", current_directory)
# 创建一个新的目录
new_directory = current_directory / "my_folder"
new_directory.mkdir()
print("创建新目录:", new_directory)
# 创建一个新的文件
new_file = new_directory / "my_file.txt"
new_file.touch()
print("创建新文件:", new_file)
# 写入文件
with new_file.open(mode='w') as f:
    f.write("Hello, World!")
# 读取文件内容
with new_file.open() as f:
    content = f.read()
print("文件内容:", content)
# 遍历目录中的文件
for file in new_directory.iterdir():
    print("文件:", file)
# 删除文件和目录
new_file.unlink()
new_directory.rmdir()
print("删除文件和目录:", new_file, new_directory)

import tempfile
# 创建临时文件
temp_file = tempfile.NamedTemporaryFile(delete=False)
temp_file.write(b'This is a temporary file.')
temp_file.close()
# 打印临时文件路径
print("临时文件路径:", temp_file.name)
# 打开临时文件并读取内容
with open(temp_file.name, 'r') as f:
    content = f.read()
print("临时文件内容:", content)
# 创建临时目录
temp_dir = tempfile.TemporaryDirectory()
# 打印临时目录路径
print("临时目录路径:", temp_dir.name)
# 自动清理临时目录
temp_dir.cleanup()

import shutil
# 复制文件
shutil.copy('source_file.txt', 'destination_folder/')
# 移动文件
shutil.move('source_file.txt', 'destination_folder/')
# 删除文件
shutil.remove('file_to_be_deleted.txt')
# 删除目录
shutil.rmtree('directory_to_be_deleted/')
# 创建压缩文件
shutil.make_archive('archive', 'zip', 'source_folder/')
# 解压缩文件
shutil.unpack_archive('archive.zip', 'destination_folder/')

import os
print(os.getenv('PATH'))
# 获取环境变量PATH的值，并打印
# 示例输出：/home/martin/.local/bin:/usr/local/sbin:/usr/local/bin:...
print(os.uname())
# 获取操作系统的信息，并打印
# 示例输出：posix.uname_result(sysname='Linux', nodename='...', release='...', version='...', machine='x86_64')
print(os.times())
# 获取进程的CPU时间信息，并打印
# 示例输出：posix.times_result(user=0.01, system=0.0, children_user=0.0, children_system=0.0, elapsed=1740.63)
print(os.cpu_count())
# 获取可用的CPU核心数量，并打印
# 示例输出：16
print(os.getloadavg())
# 获取系统的平均负载，并打印
# 示例输出：(2.021484375, 2.35595703125, 2.04052734375)
old_umask = os.umask(0o022)
# 设置文件创建时的权限掩码，并将旧的掩码保存起来
# 在此处可以执行与文件相关的操作...
os.umask(old_umask)
# 恢复旧的文件权限掩码

pip install sh

import sh
# 在 $PATH 中运行任何命令...
print(sh.ls('-la'))
# 执行ls命令并打印输出
# 示例输出：
# total 36
# drwxrwxr-x  2 martin martin  4096 apr  8 14:18 .
# drwxrwxr-x 41 martin martin 20480 apr  7 15:23 ..
# -rw-rw-r--  1 martin martin    30 apr  8 14:18 examples.py
ls_cmd = sh.Command('ls')
print(ls_cmd('-la'))  # 显式调用
# 使用Command对象执行ls命令并打印输出
# 示例输出与上述相同
# 如果命令不在PATH中：
custom_cmd = sh.Command('/path/to/my/cmd')
custom_cmd('some', 'args') # 执行自定义命令并传递参数
with sh.contrib.sudo:
    # 使用'sudo'执行一些操作...
    ...
# 使用'sudo'执行一些操作的上下文环境

custom_cmd = sh.Command('/path/to/my/cmd')
custom_cmd('some', 'args')  # 执行自定义命令并传递参数

#相当于 ip address > /tmp/ipaddr
sh.ip.address(_out='/tmp/ipaddr')

print(sh.awk('{print $9}', _in=sh.ls('-la')))
# 等同于 "ls -la | awk '{print $9}'"
print(sh.wc('-l', _in=sh.ls('.', '-1')))
# 等同于  "ls -1 | wc -l"

try:
    sh.cat('/tmp/doesnt/exist')
except sh.ErrorReturnCode as e:
    print(f'Command {e.full_cmd} exited with {e.exit_code}')
    # '/usr/bin/cat /tmp/doesnt/exist' 命令结果返回 1
curl = sh.curl('https://httpbin.org/delay/5', _bg=True)
try:
    curl.wait(timeout=3)
except sh.TimeoutException:
    print("Command timed out...")
    curl.kill()