Flask context decorator for Celery tasks

gistfile1.py

def push_context(fn):
    @wraps(fn)
    def decorator(*args, **kwargs):
        app_host = app.config.get('SERVER_NAME')
        app_root = app.config.get('APPLICATION_ROOT')
        base_url = 'https://{}/'.format(app_host if app_host else 'localhost')
        if app_root:
            base_url += app.root.lstrip('/')
        app.test_request_context(base_url=base_url).push()
        return fn(*args, **kwargs)
    return decorator