Python 開發(fā)命令行工具 - PyTips 0x06

PyTips

項(xiàng)目地址:https://git.io/pytips

Python 作為一種腳本語言,可以非常方便地用于系統(tǒng)(尤其是*nix系統(tǒng))命令行工具的開發(fā)骤公。Python 自身也集成了一些標(biāo)準(zhǔn)庫(kù)幼东,專門用于處理命令行相關(guān)的問題。

命令行工具的一般結(jié)構(gòu)

CL-in-Python

1. 標(biāo)準(zhǔn)輸入輸出

*nix 系統(tǒng)中,一切皆為文件铭拧,因此標(biāo)準(zhǔn)輸入时肿、輸出可以完全可以看做是對(duì)文件的操作庇茫。標(biāo)準(zhǔn)化輸入可以通過管道(pipe)或重定向(redirect)的方式傳遞:

# script reverse.py
#!/usr/bin/env python
import sys
for l in sys.stdin.readlines():
    sys.stdout.write(l[::-1])

保存為 reverse.py,通過管道 | 傳遞:

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

通過重定向 < 傳遞:

./reverse.py < reverse.py
# 輸出結(jié)果同上

2. 命令行參數(shù)

一般在命令行后追加的參數(shù)可以通過 sys.argv 獲取螃成, sys.argv 是一個(gè)列表旦签,其中第一個(gè)元素為當(dāng)前腳本的文件名:

# script argv.py
#!/usr/bin/env python
import sys
print(sys.argv) # 下面返回的是 Jupyter 運(yùn)行的結(jié)果
['/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']

運(yùn)行上面的腳本:

chmod +x argv.py
./argv.py hello world
python argv.py hello world

# 返回的結(jié)果是相同的
# ['./test.py', 'hello', 'world']

對(duì)于比較復(fù)雜的命令行參數(shù),例如通過 --option 傳遞的選項(xiàng)參數(shù)寸宏,如果是對(duì) sys.argv 逐項(xiàng)進(jìn)行解析會(huì)很麻煩宁炫,Python 提供標(biāo)準(zhǔn)庫(kù) argparse(舊的庫(kù)為 optparse,已經(jīng)停止維護(hù))專門解析命令行參數(shù):

# 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") # 設(shè)定命令信息氮凝,用于輸出幫助信息
    parser.add_argument("-c", "--config", required=False, default="config.ini")
    parser.add_argument("-t", "--theme", required=False, default="default.theme")
    parser.add_argument("-f") # Accept Jupyter runtime option
    args = parser.parse_args()
    loadConfig(args.config)
    setTheme(args.theme)

if __name__ == "__main__":
    main()
Load config from: config.ini
Set theme: default.theme

利用 argparse 可以很方便地解析選項(xiàng)參數(shù)羔巢,同時(shí)可以定義指定參數(shù)的相關(guān)屬性(是否必須、默認(rèn)值等)罩阵,同時(shí)還可以自動(dòng)生成幫助文檔竿秆。執(zhí)行上面的腳本:

./convert.py -h
usage: convert [-h] [-c CONFIG] [-t THEME]

optional arguments:
  -h, --help            show this help message and exit
  -c CONFIG, --config CONFIG
  -t THEME, --theme THEME

3. 執(zhí)行系統(tǒng)命令

當(dāng) Python 能夠準(zhǔn)確地解讀輸入信息或參數(shù)之后,就可以通過 Python 去做任何事情了永脓。這里主要介紹通過 Python 調(diào)用系統(tǒng)命令袍辞,也就是替代 Shell 腳本完成系統(tǒng)管理的功能。我以前的習(xí)慣是將命令行指令通過 os.system(command) 執(zhí)行常摧,但是更好的做法應(yīng)該是用 subprocess 標(biāo)準(zhǔn)庫(kù)搅吁,它的存在就是為了替代舊的 os.system; os.spawn*

subprocess 模塊提供簡(jiǎn)便的直接調(diào)用系統(tǒng)指令的call()方法落午,以及較為復(fù)雜可以讓用戶更加深入地與系統(tǒng)命令進(jìn)行交互的Popen對(duì)象谎懦。

# script list_files.py
#!/usr/bin/env python
import subprocess as sb
res = sb.check_output("ls -lh ./*.ipynb", shell=True) # 為了安全起見,默認(rèn)不通過系統(tǒng) Shell 執(zhí)行溃斋,因此需要設(shè)定 shell=True
print(res.decode()) # 默認(rèn)返回值為 bytes 類型界拦,需要進(jìn)行解碼操作
-rw-r--r--  1 rainy  staff   3.4K  3  8 17:36 ./2016-03-06-The-Zen-of-Python.ipynb
-rw-r--r--  1 rainy  staff   6.7K  3  8 17:45 ./2016-03-07-iterator-and-generator.ipynb
-rw-r--r--  1 rainy  staff   6.0K  3 10 12:35 ./2016-03-08-Functional-Programming-in-Python.ipynb
-rw-r--r--  1 rainy  staff   5.9K  3  9 16:28 ./2016-03-09-List-Comprehension.ipynb
-rw-r--r--  1 rainy  staff    10K  3 10 14:14 ./2016-03-10-Scope-and-Closure.ipynb
-rw-r--r--  1 rainy  staff   8.0K  3 11 16:30 ./2016-03-11-Arguments-and-Unpacking.ipynb
-rw-r--r--  1 rainy  staff   8.5K  3 14 19:31 ./2016-03-14-Command-Line-tools-in-Python.ipynb

如果只是簡(jiǎn)單地執(zhí)行系統(tǒng)命令還不能滿足你的需求,可以使用 subprocess.Popen 與生成的子進(jìn)程進(jìn)行更多交互:

import subprocess as sb

p = sb.Popen(['grep', 'communicate'], stdin=sb.PIPE, stdout=sb.PIPE)
res, err = p.communicate(sb.check_output('cat ./*', shell=True))
if not err:
    print(res.decode())
      "    \"p = sb.Popen(['grep', 'communicate'], stdout=sb.PIPE)\\n\",\n",
      "    \"# res = p.communicate(sb.check_output('cat ./*'))\"\n",
    "p = sb.Popen(['grep', 'communicate'], stdin=sb.PIPE, stdout=sb.PIPE)\n",
    "res, err = p.communicate(sb.check_output('cat ./*', shell=True))\n",
歡迎關(guān)注公眾號(hào) PyHub梗劫!
最后編輯于
?著作權(quán)歸作者所有,轉(zhuǎn)載或內(nèi)容合作請(qǐng)聯(lián)系作者
  • 序言:七十年代末享甸,一起剝皮案震驚了整個(gè)濱河市,隨后出現(xiàn)的幾起案子梳侨,更是在濱河造成了極大的恐慌蛉威,老刑警劉巖,帶你破解...
    沈念sama閱讀 212,383評(píng)論 6 493
  • 序言:濱河連續(xù)發(fā)生了三起死亡事件走哺,死亡現(xiàn)場(chǎng)離奇詭異蚯嫌,居然都是意外死亡,警方通過查閱死者的電腦和手機(jī),發(fā)現(xiàn)死者居然都...
    沈念sama閱讀 90,522評(píng)論 3 385
  • 文/潘曉璐 我一進(jìn)店門择示,熙熙樓的掌柜王于貴愁眉苦臉地迎上來束凑,“玉大人,你說我怎么就攤上這事栅盲⊥羲撸” “怎么了?”我有些...
    開封第一講書人閱讀 157,852評(píng)論 0 348
  • 文/不壞的土叔 我叫張陵剪菱,是天一觀的道長(zhǎng)摩瞎。 經(jīng)常有香客問我,道長(zhǎng)孝常,這世上最難降的妖魔是什么? 我笑而不...
    開封第一講書人閱讀 56,621評(píng)論 1 284
  • 正文 為了忘掉前任蚓哩,我火速辦了婚禮构灸,結(jié)果婚禮上,老公的妹妹穿的比我還像新娘岸梨。我一直安慰自己喜颁,他們只是感情好,可當(dāng)我...
    茶點(diǎn)故事閱讀 65,741評(píng)論 6 386
  • 文/花漫 我一把揭開白布曹阔。 她就那樣靜靜地躺著半开,像睡著了一般。 火紅的嫁衣襯著肌膚如雪赃份。 梳的紋絲不亂的頭發(fā)上寂拆,一...
    開封第一講書人閱讀 49,929評(píng)論 1 290
  • 那天,我揣著相機(jī)與錄音抓韩,去河邊找鬼纠永。 笑死,一個(gè)胖子當(dāng)著我的面吹牛谒拴,可吹牛的內(nèi)容都是我干的尝江。 我是一名探鬼主播,決...
    沈念sama閱讀 39,076評(píng)論 3 410
  • 文/蒼蘭香墨 我猛地睜開眼英上,長(zhǎng)吁一口氣:“原來是場(chǎng)噩夢(mèng)啊……” “哼炭序!你這毒婦竟也來了?” 一聲冷哼從身側(cè)響起苍日,我...
    開封第一講書人閱讀 37,803評(píng)論 0 268
  • 序言:老撾萬榮一對(duì)情侶失蹤惭聂,失蹤者是張志新(化名)和其女友劉穎,沒想到半個(gè)月后易遣,有當(dāng)?shù)厝嗽跇淞掷锇l(fā)現(xiàn)了一具尸體彼妻,經(jīng)...
    沈念sama閱讀 44,265評(píng)論 1 303
  • 正文 獨(dú)居荒郊野嶺守林人離奇死亡,尸身上長(zhǎng)有42處帶血的膿包…… 初始之章·張勛 以下內(nèi)容為張勛視角 年9月15日...
    茶點(diǎn)故事閱讀 36,582評(píng)論 2 327
  • 正文 我和宋清朗相戀三年,在試婚紗的時(shí)候發(fā)現(xiàn)自己被綠了侨歉。 大學(xué)時(shí)的朋友給我發(fā)了我未婚夫和他白月光在一起吃飯的照片。...
    茶點(diǎn)故事閱讀 38,716評(píng)論 1 341
  • 序言:一個(gè)原本活蹦亂跳的男人離奇死亡幽邓,死狀恐怖,靈堂內(nèi)的尸體忽然破棺而出牵舵,到底是詐尸還是另有隱情,我是刑警寧澤畸颅,帶...
    沈念sama閱讀 34,395評(píng)論 4 333
  • 正文 年R本政府宣布担巩,位于F島的核電站,受9級(jí)特大地震影響没炒,放射性物質(zhì)發(fā)生泄漏。R本人自食惡果不足惜送火,卻給世界環(huán)境...
    茶點(diǎn)故事閱讀 40,039評(píng)論 3 316
  • 文/蒙蒙 一、第九天 我趴在偏房一處隱蔽的房頂上張望种吸。 院中可真熱鬧弃衍,春花似錦、人聲如沸坚俗。這莊子的主人今日做“春日...
    開封第一講書人閱讀 30,798評(píng)論 0 21
  • 文/蒼蘭香墨 我抬頭看了看天上的太陽。三九已至辙浑,卻和暖如春,著一層夾襖步出監(jiān)牢的瞬間倦踢,已是汗流浹背。 一陣腳步聲響...
    開封第一講書人閱讀 32,027評(píng)論 1 266
  • 我被黑心中介騙來泰國(guó)打工辱挥, 沒想到剛下飛機(jī)就差點(diǎn)兒被人妖公主榨干…… 1. 我叫王不留边涕,地道東北人褂微。 一個(gè)月前我還...
    沈念sama閱讀 46,488評(píng)論 2 361
  • 正文 我出身青樓园爷,卻偏偏與公主長(zhǎng)得像,于是被迫代替她去往敵國(guó)和親童社。 傳聞我的和親對(duì)象是個(gè)殘疾皇子,可洞房花燭夜當(dāng)晚...
    茶點(diǎn)故事閱讀 43,612評(píng)論 2 350

推薦閱讀更多精彩內(nèi)容