Python 作为一种脚本语言,可以非常方便地用于系统(尤其是*nix系统)命令行工具的开发。Python 自身也集成了一些标准库,专门用于处理命令行相关的问题。
命令行工具的一般结构
1. 标准输入输出
*nix 系统中,一切皆为文件,因此标准输入、输出可以完全可以看做是对文件的操作。标准化输入可以通过管道(pipe)或重定向(redirect)的方式传递:
1 2 3 4 5 |
# script reverse.py #!/usr/bin/env python import sys for l in sys.stdin.readlines(): sys.stdout.write(l[::-1]) |
保存为 reverse.py
,通过管道 |
传递:
1 2 3 4 5 6 7 |
chmod +x reverse.py cat reverse.py | ./reverse.py nohtyp vne/nib/rsu/!# sys tropmi :)(senildaer.nidts.sys ni l rof )]1-::[l(etirw.tuodts.sys |
通过重定向 传递:
1 2 |
./reverse.py reverse.py # 输出结果同上 |
2. 命令行参数
一般在命令行后追加的参数可以通过 sys.argv
获取, sys.argv
是一个列表,其中第一个元素为当前脚本的文件名:
1 2 3 4 |
# script argv.py #!/usr/bin/env python import sys print(sys.argv) # 下面返回的是 Jupyter 运行的结果 |
1 |
['/Users/rainy/Projects/GitHub/pytips/venv3/lib/python3.5/site-packages/ipykernel/__main__.py', '-f', '/Users/rainy/Library/Jupyter/runtime/kernel-0533e681-bd7c-4c4d-9094-a78fde7fc2ed.json'] |
运行上面的脚本:
1 2 3 4 5 6 |
chmod +x argv.py ./argv.py hello world python argv.py hello world # 返回的结果是相同的 # ['./test.py', 'hello', 'world'] |
对于比较复杂的命令行参数,例如通过 --option
传递的选项参数,如果是对 sys.argv
逐项进行解析会很麻烦,Python 提供标准库 argparse
(旧的库为 optparse
,已经停止维护)专门解析命令行参数:
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 |
# script convert.py #!/usr/bin/env python import argparse as apa def loadConfig(config): print("Load config from: {}".format(config)) def setTheme(theme): print("Set theme: {}".format(theme)) def main(): parser = apa.ArgumentParser(prog="convert") # 设定命令信息,用于输出帮助信息 parser.add_argument("-c", "--config", required=False, default="config.ini") parser.add_argumentt">False, default="config.ini") parser.add_argument
命令行工具的一般结构1. 标准输入输出 *nix 系统中,一切皆为文件,因此标准输入、输出可以完全可以看做是对文件的操作。标准化输入可以通过管道(pipe)或重定向(redirect)的方式传递:
保存为
通过重定向
2. 命令行参数 一般在命令行后追加的参数可以通过
运行上面的脚本:
对于比较复杂的命令行参数,例如通过
|