本文总结使用 shell
脚本的几种方法。
Updated: 2022 / 7 / 27
Python
作为一门脚本语言,有时候需要与 shell
命令交互式使用,在 Python
中提供了很多的方法可以调用并执行 shell
脚本,本文做一个简单的总结 1。
本文的开发环境是 macOS Big Sur
。
os.system('command')
python
自带的执行 shell
命令的方法,其中最后一个 0
是这个命令的返回值,为 0
表示命令执行成功。system()
无法将执行的结果保存起来,如下所示:print(os.system("touch a.txt"))
# 0
#
# 会返回一个 0,表示执行成功了,然后在当前文件夹之下创建了一个新的 a.txt 文件
print(os.system("ls -lh"))
# total 416
# -rw-r--r-- 1 staff 49B Jul 27 20:02 main.py
# -rw-r--r-- 1 staff 63B Jul 16 17:38 temporary.py
# -rw-r--r-- 1 staff 4.1K Jul 24 20:02 test.py
# -rw-r--r-- 1 staff 94B Jul 24 19:31 usrs_info.pickle
# 0
#
#
os.popen('command')
os.popen()
返回的是一个文件对象,对其进行读取 read()
的操作可以看到执行的输出。f = os.popen("ls -l")
# 返回的是一个文件对象
print(f.read())
# 通过文件的read()读取所返回的内容
#
#
# total 416
# -rw-r--r-- 1 staff 49B Jul 27 20:02 main.py
# -rw-r--r-- 1 staff 63B Jul 16 17:38 temporary.py
# -rw-r--r-- 1 staff 4.1K Jul 24 20:02 test.py
# -rw-r--r-- 1 staff 94B Jul 24 19:31 usrs_info.pickle
f = os.popen("touch b.txt")
# 创建一个文件
print(f.read())
# 无返回值
subprocess
模块是 python
从 2.4
版本开始引入的模块,也是系统自带的,不需要再额外安装了。
主要用来取代一些旧的模块方法,如 os.system
、os.spawn*
、os.popen*
、commands.*
等。
subprocess
通过子进程来执行外部指令,并通过 input / output / error
管道,获取子进程的执行的返回信息。
subprocess.call()
执行命令,并返回执行状态。其中,
shell
参数为 False
时,命令以及命令的参数需要通过列表的方式传入;print(subprocess.call(["ls", "-l"], shell=False))
#
#
# total 416
# -rw-r--r-- 1 staff 49B Jul 27 20:02 main.py
# -rw-r--r-- 1 staff 63B Jul 16 17:38 temporary.py
# -rw-r--r-- 1 staff 4.1K Jul 24 20:02 test.py
# -rw-r--r-- 1 staff 94B Jul 24 19:31 usrs_info.pickle
# 0
print(subprocess.call(["touch", "c.txt"], shell=False))
# 直接返回 0,表示操作成功
# 0
shell
参数为 True
时,可通过一个字符串直接传入命令以及命令所需要的参数;print(subprocess.call(("ls -l"), shell=True)
#
#
# total 416
# -rw-r--r-- 1 staff 49B Jul 27 20:02 main.py
# -rw-r--r-- 1 staff 63B Jul 16 17:38 temporary.py
# -rw-r--r-- 1 staff 4.1K Jul 24 20:02 test.py
# -rw-r--r-- 1 staff 94B Jul 24 19:31 usrs_info.pickle
# 0
print(subprocess.call(["mkdir newdir1"], shell=True))
# 直接返回 0,表示操作成功
# 0
subprocess.check_call()
subprocess.call()
类似,区别是,0
时,直接抛出异常。subprocess.check_output()
0
时,直接返回输出结果;0
,直接抛出异常。python3.x
中才有。a = subprocess.check_output(["ls","-l"], shell=False)
print(a)
#
# 不是直接返回0了,而是直接返回执行结果的内容
# b'total 416\n-rw-r--r-- 1 xueshanzhang staff 4326 Jul 24 17:59 MotivationLetterGenerator.py\ndrwxr
subprocess
的功能还有更多,比如还有 Popen
类。
subprocess
模块中定义了一个 Popen
类,通过它可以来创建进程,并与其进行复杂的交互。
首先安装 sh
库,pip install sh
。
Python
是一种伟大的脚本语言,不过有时使用标准 os
和 subprocess
库会有点棘手。
sh
库提供了一种不错的替代方案。sh
库:http://amoffat.github.io/sh/
库允许用户像使用普通函数一样调用任意程序,这对自动化工作流和任务非常有用。
它的一般工作模式如下:
sh.command_name("参数一", "参数二", "参数三")
如下,
print(sh.pwd())
# /Users/PycharmProjects/pythonProject0312
sh.mkdir('FolderA')
sh.touch('File.txt')
print(sh.whoami())
# your username
#
#
print(sh.echo('This is great!'))
# This is great!
#
#
sh.ls("-l")
# 等价于 ls -l
print(sh.ls("-l"))
# total 416
# -rw-r--r-- 1 staff 49B Jul 27 20:02 main.py
# -rw-r--r-- 1 staff 63B Jul 16 17:38 temporary.py
# -rw-r--r-- 1 staff 4.1K Jul 24 20:02 test.py
# -rw-r--r-- 1 staff 94B Jul 24 19:31 usrs_info.pickle
sh.df("-h")
# 等价于 df -h
print(sh.df("-h"))
# Filesystem Size Used Avail Capacity iused ifree %iused Mounted on
# /dev/disk3s1s1 228Gi 21Gi 31Gi 41% 553757 2393071203 0% /
# devfs 204Ki 204Ki 0Bi 100% 705 0 100% /dev
# ...
#
#
sh.du("-h", "-d 1")
# 等价于 du -h -d 1
print(sh.du("-h", "-d 1"))
# 当有多个参数的情况,且参数可以赋值
# 0B ./FolderA
# 92K ./.git
# 40K ./.idea
# 340K .
#
#
注意:上面在通过 sh.**
编写的时候可能没有代码提示,因为 sh
模块里面并没有直接定义像上面的 pwd()
,touch()
,du()
等这些函数,他是通过其他的方式实现的,所以没有智能敏感提示。