Google 开源的 Python 命令行库：fire 实现 git 命令

做者：HelloGitHub- Prodesire

涉及的示例代码和历史文章，已同步更新到 HelloGitHub-Team 仓库

1、前言

在前面三篇介绍 fire 的文章中，咱们全面了解了 fire 强大而不失简洁的能力。按照惯例，咱们要像使用 argparse、docopt 和 click 同样使用 fire 来实现 git 命令。

本文的关注点并不在 git 的各类命令是如何实现的，而是怎么使用 fire 去打造一个实用命令行程序，代码结构是怎样的。所以，和 git 相关的操做，将会使用 gitpython 库来简单实现。

为了让没读过 使用 xxx 实现 git 命令（xxx 指 argparse、docopt 和 click） 的小伙伴也能读明白本文，咱们仍会对 git 经常使用命令和 gitpython 作一个简单介绍。

本系列文章默认使用 Python 3 做为解释器进行讲解。
若你仍在使用 Python 2，请注意二者之间语法和库的使用差别哦~
复制代码

2、git 经常使用命令

当你写好一段代码或增删一些文件后，会用以下命令查看文件状态：

git status
复制代码

确认文件状态后，会用以下命令将的一个或多个文件（夹）添加到暂存区：

git add [pathspec [pathspec ...]]
复制代码

而后使用以下命令提交信息：

git commit -m "your commit message"
复制代码

最后使用以下命令将提交推送到远程仓库：

git push
复制代码

咱们将使用 fire 和 gitpython 库来实现这 4 个子命令。

3、关于 gitpython

gitpython 是一个和 git 仓库交互的 Python 第三方库。 咱们将借用它的能力来实现真正的 git 逻辑。

安装：

pip install gitpython
复制代码

4、思考

在实现前，咱们不妨先思考下会用到 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 程序。

5、实现

假定咱们在 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 。

6、小结

本文简单介绍了平常工做中经常使用的 git 命令，而后提出实现它的思路，最终一步步地使用 fire 和 gitpython 实现了 git 程序。

对比 argparse、docopt 和 click 的实现版本，你会发现使用 fire 来实现是最简单的：

• 相较于 argparse，子解析器、参数类型什么的通通不须要关心
• 相较于 docopt，参数解析和命令调用处理也不须要关心
• 相较于 click，装饰器所定义的命令行参数信息也必需要关心

无疑，fire 把能简化的都简化了，简直就是懒人福音。

关于 fire 的讲解将告一段落，回顾下 fire 的至简之道，你会深爱上它。这也体现出了 Python 之美。

如今，你已学会了四个特色各异的主流命令行解析库的使用了，不再须要为命令行程序的实现而烦恼了。

什么，你为要使用哪个库而发愁？在下一篇也是最后一篇文章中，咱们将对这些库作一个横向对比，以对什么场景下使用什么样的命令行库了然于胸~

---

『讲解开源项目系列』——让对开源项目感兴趣的人再也不畏惧、让开源项目的发起者再也不孤单。跟着咱们的文章，你会发现编程的乐趣、使用和发现参与开源项目如此简单。欢迎留言联系咱们、加入咱们，让更多人爱上开源、贡献开源～