def create_app(config_obj, no_sql=False):
"""Flask application factory. Initializes and returns the Flask application.
Blueprints are registered in here.
Modeled after: http://flask.pocoo.org/docs/patterns/appfactories/
Positional arguments:
config_obj -- configuration object to load into app.config.
Keyword arguments:
no_sql -- does not run init_app() for the SQLAlchemy instance. For Celery compatibility.
Returns:
The initialized Flask application.
"""
# Initialize app. Flatten config_obj to dictionary (resolve properties).
app = Flask(__name__,
template_folder=TEMPLATE_FOLDER,
static_folder=STATIC_FOLDER)
config_dict = dict([(k, getattr(config_obj, k)) for k in dir(config_obj)
if not k.startswith('_')])
app.config.update(config_dict)
# Import DB models. Flask-SQLAlchemy doesn't do this automatically like Celery does.
with app.app_context():
for module in app.config.get('DB_MODELS_IMPORTS', list()):
import_module(module)
# Setup redirects and register blueprints.
app.add_url_rule('/favicon.ico', 'favicon',
lambda: app.send_static_file('favicon.ico'))
for bp in all_blueprints:
import_module(bp.import_name)
app.register_blueprint(bp)
# Initialize extensions/add-ons/plugins.
if not no_sql:
db.init_app(app)
Statics(app) # Enable Flask-Statics-Helper features.
redis.init_app(app)
celery.init_app(app)
mail.init_app(app)
# Activate middleware.
locale.setlocale(locale.LC_ALL,
'en_US.UTF-8') # For filters inside the middleware file.
with app.app_context():
import_module('pypi_portal.middleware')
# Return the application instance.
return app
def celery_worker_init_db(**_):
"""Initialize SQLAlchemy right after the Celery worker process forks.
This ensures each Celery worker has its own dedicated connection to the MySQL database. Otherwise
one worker may close the connection while another worker is using it, raising exceptions.
Without this, the existing session to the MySQL server is cloned to all Celery workers, so they
all share a single session. A SQLAlchemy session is not thread/concurrency-safe, causing weird
exceptions to be raised by workers.
Based on http://stackoverflow.com/a/14146403/1198943
"""
LOG.debug('Initializing SQLAlchemy for PID {}.'.format(os.getpid()))
db.init_app(current_app)
def celery_worker_init_db(**_):
"""Initialize SQLAlchemy right after the Celery worker process forks.
This ensures each Celery worker has its own dedicated connection to the MySQL database. Otherwise
one worker may close the connection while another worker is using it, raising exceptions.
Without this, the existing session to the MySQL server is cloned to all Celery workers, so they
all share a single session. A SQLAlchemy session is not thread/concurrency-safe, causing weird
exceptions to be raised by workers.
Based on http://stackoverflow.com/a/14146403/1198943
"""
LOG.debug('Initializing SQLAlchemy for PID {}.'.format(os.getpid()))
db.init_app(current_app)
def create_app(config_obj, no_sql=False):
"""Flask application factory. Initializes and returns the Flask application.
Blueprints are registered in here.
Modeled after: http://flask.pocoo.org/docs/patterns/appfactories/
Positional arguments:
config_obj -- configuration object to load into app.config.
Keyword arguments:
no_sql -- does not run init_app() for the SQLAlchemy instance. For Celery compatibility.
Returns:
The initialized Flask application.
"""
# Initialize app. Flatten config_obj to dictionary (resolve properties).
app = Flask(__name__, template_folder=TEMPLATE_FOLDER, static_folder=STATIC_FOLDER)
config_dict = dict([(k, getattr(config_obj, k)) for k in dir(config_obj) if not k.startswith('_')])
app.config.update(config_dict)
# Import DB models. Flask-SQLAlchemy doesn't do this automatically like Celery does.
with app.app_context():
for module in app.config.get('DB_MODELS_IMPORTS', list()):
import_module(module)
# Setup redirects and register blueprints.
app.add_url_rule('/favicon.ico', 'favicon', lambda: app.send_static_file('favicon.ico'))
for bp in all_blueprints:
import_module(bp.import_name)
app.register_blueprint(bp)
# Initialize extensions/add-ons/plugins.
if not no_sql:
db.init_app(app)
Statics(app) # Enable Flask-Statics-Helper features.
redis.init_app(app)
celery.init_app(app)
mail.init_app(app)
# Activate middleware.
locale.setlocale(locale.LC_ALL, 'en_US.UTF-8') # For filters inside the middleware file.
with app.app_context():
import_module('pypi_portal.middleware')
# Return the application instance.
return app
