Google 開源的 Python 命令列庫:fire 實現 git 命令
作者:HelloGitHub-Prodesire
HelloGitHub 的《講解開源專案》系列,專案地址:https://github.com/HelloGitHub-Team/Article
一、前言
在前面三篇介紹 fire
的文章中,我們全面瞭解了 fire
強大而不失簡潔的能力。按照慣例,我們要像使用 argparse
、docopt
和 click
一樣使用 fire
來實現 git 命令。
本文的關注點並不在 git
的各種命令是如何實現的,而是怎麼使用 fire
去打造一個實用命令列程式,程式碼結構是怎樣的。因此,和 git
相關的操作,將會使用 gitpython
庫來簡單實現。
為了讓沒讀過 使用 xxx 實現 git 命令
(xxx
指 argparse
、docopt
和 click
) 的小夥伴也能讀明白本文,我們仍會對 git
常用命令和 gitpython
做一個簡單介紹。
本系列文章預設使用 Python 3 作為直譯器進行講解。
若你仍在使用 Python 2,請注意兩者之間語法和庫的使用差異哦~
二、git 常用命令
當你寫好一段程式碼或增刪一些檔案後,會用如下命令檢視檔案狀態:
git status
確認檔案狀態後,會用如下命令將的一個或多個檔案(夾)新增到暫存區:
git add [pathspec [pathspec ...]]
然後使用如下命令提交資訊:
git commit -m "your commit message"
最後使用如下命令將提交推送到遠端倉庫:
git push
我們將使用 fire
和 gitpython
庫來實現這 4 個子命令。
三、關於 gitpython
gitpython 是一個和 git
倉庫互動的 Python 第三方庫。
我們將借用它的能力來實現真正的 git
邏輯。
安裝:
pip install gitpython
四、思考
在實現前,我們不妨先思考下會用到 fire
的哪些功能?整個程式的結構是怎樣的?
fire
git
的 4 個子命令的實現其實對應於四個函式,我們可以都放到一個類中,實現四個例項方法。
而對於 git add
*pathspecs
引數來表達。對於
git commit
命令,需要接受 -m
選項,在例項方法中用 m
引數來表達。
程式結構
程式結構上:
- 例項化
Git
物件,供全域性使用 - 在
GitCli
類中定義四個命令對應的例項方法status
、add
、commit
、push
則基本結構如下:
import os
import fire
from git.cmd import Git
git = Git(os.getcwd())
class GitCli:
def status(self):
"""
處理 status 命令
"""
pass
def add(self, *pathspecs):
"""
處理 add 命令
"""
pass
def commit(self, m):
"""
處理 -m <msg> 命令
"""
pass
def push(self):
"""
處理 push 命令
"""
pass
if __name__ == '__main__':
fire.Fire(GitCli())
下面我們將一步步地實現我們的 git
程式。
五、實現
假定我們在 fire-git.py 檔案中實現我們的 git
程式。
5.1 status 子命令
status
子命令不接受任何引數和選項,因此 status
方法無需任何入參。
class GitCli:
def status(self):
"""
處理 status 命令
"""
cmd = ['git', 'status']
output = git.execute(cmd)
return output
不難看出,我們最後呼叫了真正的 git status
來實現,並列印了輸出。
5.2 add 子命令
add
子命令相對於 status
子命令,需要接受任意個 pathspec 引數,因此 add
方法需要增加 *pathspecs
入參。
fire 最終傳入的是一個元組,我們需要將其轉換成 list 以便後續處理。
class GitCli:
def add(self, *pathspecs):
"""
處理 add 命令
"""
cmd = ['git', 'add'] + list(pathspecs)
output = git.execute(cmd)
return output
當我們執行 python3 fire-git.py add --help
時,結果如下:
INFO: Showing help with the command 'fire-git.py add -- --help'.
NAME
fire-git.py add - 處理 add 命令
SYNOPSIS
fire-git.py add [PATHSPECS]...
DESCRIPTION
處理 add 命令
POSITIONAL ARGUMENTS
PATHSPECS
5.3 commit 子命令
commit
子命令相對於 status
子命令,需要接受 -m
選項,因此 commit
方法需要增加 m
入參。
class GitCli:
def commit(self, m):
"""
處理 -m <msg> 命令
"""
cmd = ['git', 'commit', '-m', m]
output = git.execute(cmd)
return output
5.4 push 子命令
push
子命令同 status
子命令一樣,不接受任何引數和選項,因此 push
方法無需任何入參。
class GitCli:
def push(self):
"""
處理 push 命令
"""
cmd = ['git', 'push']
output = git.execute(cmd)
return output
至此,我們就實現了一個簡單的 git
命令列,使用 python fire-git.py status
便可查詢專案狀態。
非常方便的是,每個命令函式的 docstring
都將作為這個命令的幫助資訊,因此,當我們執行 python3 fire-git.py --help
會自動生成如下幫助內容:
INFO: Showing help with the command 'fire-git.py -- --help'.
NAME
fire-git.py
SYNOPSIS
fire-git.py COMMAND
COMMANDS
COMMAND is one of the following:
add
處理 add 命令
commit
處理 -m <msg> 命令
push
處理 push 命令
status
處理 status 命令
想看整個原始碼,請戳 fire-git.py 。
六、小結
本文簡單介紹了日常工作中常用的 git
命令,然後提出實現它的思路,最終一步步地使用 fire
和 gitpython
實現了 git
程式。
對比 argparse
、docopt
和 click
的實現版本,你會發現使用 fire
來實現是最簡單的:
- 相較於
argparse
,子解析器、引數型別什麼的統統不需要關心 - 相較於
docopt
,引數解析和命令呼叫處理也不需要關心 - 相較於
click
,裝飾器所定義的命令列引數資訊也必須要關心
無疑,fire
把能簡化的都簡化了,簡直就是懶人福音。
關於 fire
的講解將告一段落,回顧下 fire
的至簡之道,你會深愛上它。這也體現出了 Python 之美。
現在,你已學會了四個特點各異的主流命令列解析庫的使用了,再也不需要為命令列程式的實現而煩惱了。
什麼,你為要使用哪一個庫而發愁?在下一篇也是最後一篇文章中,我們將對這些庫做一個橫向對比,以對什麼場景下使用什麼樣的命令列庫瞭然於胸~
『講解開源專案系列』——讓對開源專案感興趣的人不再畏懼、讓開源專案的發起者不再孤單。跟著我們的文章,你會發現程式設計的樂趣、使用和發現參與開源專案如此簡單。歡迎留言聯絡我們、加入我們,讓更多人愛上開源、貢獻開源~