diff --git a/manage.py b/manage.py index aa51357..ee979dc 100644 --- a/manage.py +++ b/manage.py @@ -6,20 +6,22 @@ # @File Name: manage.py +import os from app import create_app -app = create_app('default') + +config = os.getenv('FLASK_CONFIG') or os.getenv('DYNO') or 'default' +app = create_app(config) if __name__ == '__main__': - from app import logger from gevent import pywsgi - logger.info('服务运行在 http://127.0.0.1:5000') + app.logger.info('服务运行在 http://127.0.0.1:5000') try: server = pywsgi.WSGIServer(('0.0.0.0', 5000), app) server.serve_forever() except KeyboardInterrupt: - logger.info('已退出') + app.logger.info('已退出') except OSError: - logger.critical('5000 端口已被占用') + app.logger.critical('5000 端口已被占用') exit(1)