使用pyinstaller将flask应用打包

Pyinstaller

用户将python程序打包成各个平台可直接运行的程序,也可以算作是对代码加密的一种方式。pyinstaller的安装及使用方式请参考官网

注:该文章的系统环境是ubuntu

将flask应用打包

项目结构

这是我开发的一个项目,并且已经成功打包并上线运行


项目结构
  • api 所有的代码都在里面
  • app.py只有一行代码,from api import create_app
开始打包

下面我们来将该项目打包pyinstaller -F app.py -name app, 通过这个命令,我们就能将整个项目打包成一个名为app的bin文件。直接运行./app,你会发现程序没有运行,因为app.py里面只是单纯的引入了app模块,如果你想通过flask run来执行的话,抱歉,app是个bin文件,不是python模块,会提示找不到app的,简单的解决办法就是在app.py文件中添加以下代码.

from api import create_app

if __name__ == "__main__":
    app = create_app()
    app.run()

然后在执行打包命令pyinstaller -F app.py -name app,这个时候我们的app就可以直接运行了./app,想要在启动的时候指定端口,主机名等等的参数,使用click.

使用gunicorn

总所周知,flask使用的是Werkzeug来作为它的WSGI server,但是性能很一般,生产环境一般会使用其他的WSGI server, 网上查到有以下WSGI server:

  • Gunicorn 独角兽,从RubyUnicorn移植过来的。
  • uWSGI 比较全能的一个WSGI server
  • mod_wsgi 这个包提供了一个Apache模块,并实现了与wsgi兼容的接口,可以让python程序运行在Apache web server之上。
  • CherryPy CherryPyPython的一个HTTP Framework,然后它也有WSGI server

可能还有其他的一些WSGI server,对于这几种,哪个好,我也不知道,我只对于gunicorn熟悉,那么要使用gunicornapp.py需添加以下代码:

import gunicorn.app.base


class StandaloneApplication(gunicorn.app.base.BaseApplication):
    """
    Custom application
    """

    def init(self, parser, opts, args):
        pass

    def __init__(self, app, options=None):
        self.options = options or {}
        self.application = app
        super(StandaloneApplication, self).__init__()

    def load_config(self):
        config = dict([(key, value) for key, value in iteritems(self.options)
                       if key in self.cfg.settings and value is not None])
        for key, value in iteritems(config):
            self.cfg.set(key.lower(), value)

    def load(self):
        return self.application


if __name__ == "__main__":
    options = {
        "bind": "127.0.0.1:8000"
    }
    StandaloneApplication(app, options=options).run()

接下来再执行pyinstaller -F app.py -name app./app就会使用gunicorn来运行服务了,对于gunicorn的参数,依然使用click来搞定。然而,当你开开心心运行程序的时候,突然报错了:

gunicorn.glogging是啥?它为什么找不到?我要去哪里找它?这是gunicorn的日志包,但是pyinstaller在打包的时候没有将它一起打入进去,所以运行是找不到,这里我们需要在打包的时候加个参数:

pyinstaller -F app.py --name app --hidden-import=gunicorn.glogging

这是啥意思呢,因为gunicorn自身的代码,并没有直接引入这个包,所以需要手动添加,--hidden-import参数含义请翻阅官方文档。接下来运行./app,还是报错,为什么路途就这么不顺呢?

这个包是gunicorn默认的工作类,pyinstaller在打包的时候也没有将它一起打入进去

pyinstaller -F app.py --name app \ 
    --hidden-import=gunicorn.glogging \
    --hidden-import=gunicorn.workers.sync

再次执行./app,程序就完美使用gunicorn来运行了。如果你想使用其他的worker_class,请在打包的时候传入对应的包名,如:

pyinstaller -F app.py --name app \ 
    --hidden-import=gunicorn.glogging \
    --hidden-import=gunicorn.workers.sync \
    --hidden-import=gunicorn.workers.ggevent

添加命令行工具

像flask一样

flask 0.11版本开始,就内建了一个命令行工具flask,而我们在开发项目的时候,也会添加一些自定义命令,然后通过flask来执行。为了让我们的打包后的可执行文件能够实现这一功能,修改app.py代码:

...

if __name__ == "__main__":
    ...
    app.cli()

但是酱紫之后,服务如何来启动呢,我的解决办法是添加一个run命令到app.cli里面,大家如果有更好的方法,还望不吝赐教。

...

def run():
    """运行服务"""
    options = {
        ...
    }
    StandaloneApplication(app, options=options).run()

if __name__ == "__main__":
    app.cli.add_command(run)
    app.cli()

打包之后,运行./app./app run,运行十分顺利。

支持db命令

flask_migrate数据库迁移库是个相当棒的工具,flask命令会自动去添加db命令,我们也可以把它添加到我们的命令中去:

...
from flask_migrate.cli import db

if __name__ == "__main__":
    ...
    app.cli.add_command(db)
    ...

之后当你兴高采烈的运行db命令的时候,又一个拦路虎出现了

找不到flask应用,我不是app = create_app()已经创建了么,为啥还要去找FLASK_APP这个环境变量呢,其实不单单是db命令会报这个错,就连我们自己写的命令也可能会报这个错,我们先来查看源代码flask_migrate/cli.py,大概在85行的位置:

...
@with_appcontext
def migrate(directory, message, sql, head, splice, branch_label, version_path,
            rev_id, x_arg):
    """Autogenerate a new revision file (Alias for 'revision --autogenerate')"""
    _migrate(directory, message, sql, head, splice, branch_label, version_path,
             rev_id, x_arg)

这里使用了with_appcontext这个装饰器,它来自于flask/cli.py文件:

def with_appcontext(f):
    @click.pass_context
    def decorator(__ctx, *args, **kwargs):
        with __ctx.ensure_object(ScriptInfo).load_app().app_context():
            return __ctx.invoke(f, *args, **kwargs)
    return update_wrapper(decorator, f)

这个装饰器的作用就是让被装饰的函数在app的上下文去执行,__ctx.ensure_object(ScriptInfo).load_app()这个函数就是flask根据FLASK_APP环境变量,或者默认的文件名app.py, wsgi.py,去找到app,所以db使用的app都是它自己去找到位置然后定义。如果使用的是flask.cli.AppGroup来定义自己的命令,那么也是一样的逻辑。所以现在要解决的问题是如何把我们手动创建的app传入进去。很直接的我想到的是current_app,只要把我们的app压入栈就可以了。

# overide.py
import click
from functools import update_wrapper
from flask import current_app
from flask.cli import with_appcontext as origin_with_appcontext


def override_with_appcontext(f):
    @click.pass_context
    def decorator(__ctx, *args, **kwargs):
        with current_app.app_context():
            return __ctx.invoke(f, *args, **kwargs)

    return update_wrapper(decorator, f)


# If the app starts up in Pyinstaller binary mode, the bootloader will set sys.frozen attribute.
if hasattr(sys, "frozen"):
    with_appcontext = override_with_appcontext
    print("Use override with_appcontext")
else:
    with_appcontext = origin_with_appcontext
    print("Use original with_appcontext")

# app.py
...
ctx = app.app_context()
ctx.push()
app.cli()
ctx.pop() # 这里的pop运行不到这里来

这里我兼容pyinstaller打包的运行的和常规运行两种,然后在需要上下文的命令函数加上重写后的with_appcontext就可以了,而对于flask_migrate.cli.db,我采用暴力的方式,直接拷贝了它的源代码,然后使用重写的with_appcontext,然后再打包就可以了。

其他注意点

非python文件的使用

如果你代码里面读取了其他文件的内容,那么在打包的时候,需要把这些文件加上,通过--add-data来添加, 代码中通过下面代码来判断
base_path = getattr(sys, "_MEIPASS",os.path.realpath(os.path.dirname(__file__)))

gunicorn使用gevent

gunicorn使用gevent的时候,需要在代码最前面加上

from gevent import monkey
monkey.patch_all(subprocess=True)

本文作者: Lim
版权声明: 著作权归作者所有。商业转载请联系作者获得授权,非商业转载请注明出处。

最后编辑于
©著作权归作者所有,转载或内容合作请联系作者
  • 序言:七十年代末,一起剥皮案震惊了整个滨河市,随后出现的几起案子,更是在滨河造成了极大的恐慌,老刑警刘岩,带你破解...
    沈念sama阅读 199,519评论 5 468
  • 序言:滨河连续发生了三起死亡事件,死亡现场离奇诡异,居然都是意外死亡,警方通过查阅死者的电脑和手机,发现死者居然都...
    沈念sama阅读 83,842评论 2 376
  • 文/潘晓璐 我一进店门,熙熙楼的掌柜王于贵愁眉苦脸地迎上来,“玉大人,你说我怎么就摊上这事。” “怎么了?”我有些...
    开封第一讲书人阅读 146,544评论 0 330
  • 文/不坏的土叔 我叫张陵,是天一观的道长。 经常有香客问我,道长,这世上最难降的妖魔是什么? 我笑而不...
    开封第一讲书人阅读 53,742评论 1 271
  • 正文 为了忘掉前任,我火速办了婚礼,结果婚礼上,老公的妹妹穿的比我还像新娘。我一直安慰自己,他们只是感情好,可当我...
    茶点故事阅读 62,646评论 5 359
  • 文/花漫 我一把揭开白布。 她就那样静静地躺着,像睡着了一般。 火红的嫁衣衬着肌肤如雪。 梳的纹丝不乱的头发上,一...
    开封第一讲书人阅读 48,027评论 1 275
  • 那天,我揣着相机与录音,去河边找鬼。 笑死,一个胖子当着我的面吹牛,可吹牛的内容都是我干的。 我是一名探鬼主播,决...
    沈念sama阅读 37,513评论 3 390
  • 文/苍兰香墨 我猛地睁开眼,长吁一口气:“原来是场噩梦啊……” “哼!你这毒妇竟也来了?” 一声冷哼从身侧响起,我...
    开封第一讲书人阅读 36,169评论 0 254
  • 序言:老挝万荣一对情侣失踪,失踪者是张志新(化名)和其女友刘颖,没想到半个月后,有当地人在树林里发现了一具尸体,经...
    沈念sama阅读 40,324评论 1 294
  • 正文 独居荒郊野岭守林人离奇死亡,尸身上长有42处带血的脓包…… 初始之章·张勋 以下内容为张勋视角 年9月15日...
    茶点故事阅读 35,268评论 2 317
  • 正文 我和宋清朗相恋三年,在试婚纱的时候发现自己被绿了。 大学时的朋友给我发了我未婚夫和他白月光在一起吃饭的照片。...
    茶点故事阅读 37,299评论 1 329
  • 序言:一个原本活蹦乱跳的男人离奇死亡,死状恐怖,灵堂内的尸体忽然破棺而出,到底是诈尸还是另有隐情,我是刑警宁泽,带...
    沈念sama阅读 32,996评论 3 315
  • 正文 年R本政府宣布,位于F岛的核电站,受9级特大地震影响,放射性物质发生泄漏。R本人自食恶果不足惜,却给世界环境...
    茶点故事阅读 38,591评论 3 303
  • 文/蒙蒙 一、第九天 我趴在偏房一处隐蔽的房顶上张望。 院中可真热闹,春花似锦、人声如沸。这庄子的主人今日做“春日...
    开封第一讲书人阅读 29,667评论 0 19
  • 文/苍兰香墨 我抬头看了看天上的太阳。三九已至,却和暖如春,着一层夹袄步出监牢的瞬间,已是汗流浃背。 一阵脚步声响...
    开封第一讲书人阅读 30,911评论 1 255
  • 我被黑心中介骗来泰国打工, 没想到刚下飞机就差点儿被人妖公主榨干…… 1. 我叫王不留,地道东北人。 一个月前我还...
    沈念sama阅读 42,288评论 2 345
  • 正文 我出身青楼,却偏偏与公主长得像,于是被迫代替她去往敌国和亲。 传闻我的和亲对象是个残疾皇子,可洞房花烛夜当晚...
    茶点故事阅读 41,871评论 2 341

推荐阅读更多精彩内容