自动安装Python第三方库

0. 背景

近期策划提出一个关于策划表预处理的需求：将两个excel表通过某个规则进行合并。合并的规则并不是本文讨论的范围，故这里不作详述。

但是不管如何，打开excel，获取某个sheet的单元格内容，Python无疑是最方便的语言工具，很快的，我就知道了相关的第三方库：openpyxl，官网：http://openpyxl.readthedocs.org。

安装库有好几种方式：a) 使用pip或是easy_install，b) 使用库中的python setup.py也是非常方便的。首选使用a)方式。但是问题来了，策划并不希望这么麻烦的安装，对于他们来说，其实也不需要知道安装这个东西。最好是在代码中直接实现了。

1. 解决思路

既然策划不关注安装库的细节，我们可以使用方式b) python setup.py，嵌入到代码中。简单步骤如下：

1）. 从pypi社区下载相关安装包(*.tar.gz)，解压后，放在某个目录下；
2）. 使用 python setup.py安装。

2. 参考代码

实现自动安装openpyxl的python代码如下：

# -*- coding: utf-8 -*-

def import_openpyxl():
    '''
    To handle .xls or .xlsx file, we need import package "openpyxl", steps as follows:
    1. Add "pip" or "easy_install" cmd-tools as environment variable. Generally in path: "C:\Python27\Scripts"
    2. In CMD screen, key into "$ pip install openpyxl" or "$ easy_install openpyxl"
    Reference-Linking:  http://openpyxl.readthedocs.org/en/latest/
    '''
    try:
        from openpyxl import load_workbook
    except:
        log.info('\nnow install "openpyxl"')
        setup = os.path.join(JsonLoader.ROOT_DIR, 'dependence', 'openpyxl-2.3.0', 'setup.py') # set "setup.py" file path with yours
        cwd = os.getcwd()
        os.chdir(os.path.dirname(setup))
        os.system(' '.join(('python', setup, 'install')))
        os.chdir(cwd)
        from openpyxl import load_workbook

    return load_workbook

if __name__ == '__main__':
    load_workbook = import_openpyxl()

当第一次使用时，try中会出现exception，catch后，会调用setup.py进行安装。当第一次安装后，代码就不会出现exception了。注意return的模块名和被赋值的模块名要相同。这样就不用管策划的PC上是否安装了openpyxl，都可以正确使用该库了，并完成两个表的合并。