1

我设置了一个 Flask 应用程序,因此它可以运行两个不同的站点(这些站点共享业务逻辑和数据库模型,因此它们使用相同的后端)。一个只是普通网站,另一个是运行任务的站点(这是一个将长时间运行的任务拍摄到 celery 中的 Web 界面)

该应用程序当前使用 Flask-Script(manage.py 命令)来启动常规网站,但我也想使用相同的脚本来启动任务站点。

在 Flask-Script 中,似乎所有命令都是在单个应用程序上运行的进程。是否可以让 manage.py 启动两个不同的应用程序?

我的代码现在如下,其中create_app是为网站创建 Flask 应用程序的工厂函数,taskserver是创建任务服务器网站的工厂函数。

import os
from flask_script import Manager
from app import create_app, database, taskserver

if os.getenv('FLASK_CONFIG'):
    app = create_app(os.getenv('FLASK_CONFIG'))
    manager = Manager(app)
else:
    app = create_app
    manager = Manager(app)
    manager.add_option('-c', '--config_name', dest='config_name', required=False)

@manager.shell
def make_shell_context():
    from app.models.data import base_models
    return dict(app=app, db=database, models=base_models)

if __name__ == "__main__":
    manager.run()

我希望有人知道如何调整 manage.py 脚本以便能够启动其中任何一个。

更新:

感谢@Joro Tenev,问题解决了。我最终使用的代码是:

import os
from flask_script import Manager
from app import create_website, create_taskserver


def make_app(app_type, config_name):
    if os.getenv('FLASK_CONFIG'):
        config_name = os.getenv('FLASK_CONFIG')
    if app_type == 'webserver':
        return create_website(config_name)
    else:
        return create_taskserver(config_name)  # i don't know how your factory works, so adjust accordingly

manager = Manager(make_app)

manager.add_option('-a', '--app_type', dest='app_type', required=True)
manager.add_option('-c', '--config_name', dest='config_name', required=False)

为了启动不同的应用程序,我使用:

python manage.py --app_type=webserver --config_name=development runserver
python manage.py --app_type=taskserver --config_name=development runserver
4

1 回答 1

2

您可以将函数传递给 Manager 的构造函数。该函数应该返回一个 Flask 应用程序。您还可以使用manager.add_option

manager.add_option('-a','--app_type', dest='app_type',required=True)
def make_app(app_type):
  if app_type =='webserver':
    return create_app(os.getenv('FLASK_CONFIG'))
  elif:
    return taskserver(os.getenv('FLASK_CONFIG')) # i don't know how your factory works, so adjust accordingly

manager = Manager(make_app)

你会用它

$ python manage.py --app_type='webserver'

希望这可以帮助 :)

于 2016-08-07T13:21:00.880 回答