if APP.config.get('ENABLE_TICKETS', True): import pagure.ui.issues # noqa: E402 import pagure.ui.plugins # noqa: E402 import pagure.ui.repo # noqa: E402 from pagure.api import API # noqa: E402 APP.register_blueprint(API) import pagure.internal # noqa: E402 APP.register_blueprint(pagure.internal.PV) # Only import the login controller if the app is set up for local login if APP.config.get('PAGURE_AUTH', None) == 'local': import pagure.ui.login as login APP.before_request_funcs[None].insert(0, login._check_session_cookie) APP.after_request(login._send_session_cookie) # pylint: disable=unused-argument @APP.teardown_request def shutdown_session(exception=None): """ Remove the DB session at the end of each request. """ SESSION.remove() # pylint: disable=unused-argument @APP.teardown_request def gcollect(exception=None): """ Runs a garbage collection to get rid of any open pygit2 handles. Details: https://pagure.io/pagure/issue/2302"""
# Import the application import pagure.ui.app import pagure.ui.admin import pagure.ui.fork import pagure.ui.groups if APP.config.get('ENABLE_TICKETS', True): import pagure.ui.issues import pagure.ui.plugins import pagure.ui.repo from pagure.api import API APP.register_blueprint(API) import pagure.internal APP.register_blueprint(pagure.internal.PV) # Only import the login controller if the app is set up for local login if APP.config.get('PAGURE_AUTH', None) == 'local': import pagure.ui.login as login APP.before_request(login._check_session_cookie) APP.after_request(login._send_session_cookie) # pylint: disable=W0613 @APP.teardown_request def shutdown_session(exception=None): """ Remove the DB session at the end of each request. """ SESSION.remove()
def create_app(config=None): """ Create the flask application. """ app = MultiStaticFlask(__name__) app.config = pagure_config if config: app.config.update(config) if app.config.get('SESSION_TYPE', None) is not None: import flask_session flask_session.Session(app) logging.basicConfig() logging.config.dictConfig(app.config.get('LOGGING') or {'version': 1}) app.jinja_env.trim_blocks = True app.jinja_env.lstrip_blocks = True if perfrepo: # Do this as early as possible. # We want the perfrepo before_request to be the very first thing # to be run, so that we can properly setup the stats before the # request. app.before_request(perfrepo.reset_stats) if pagure_config.get('THEME_TEMPLATE_FOLDER', False): # Jinja can be told to look for templates in different folders # That's what we do here template_folder = pagure_config['THEME_TEMPLATE_FOLDER'] if template_folder[0] != '/': template_folder = os.path.join(app.root_path, app.template_folder, template_folder) import jinja2 # Jinja looks for the template in the order of the folders specified templ_loaders = [ jinja2.FileSystemLoader(template_folder), app.jinja_loader, ] app.jinja_loader = jinja2.ChoiceLoader(templ_loaders) if pagure_config.get('THEME_STATIC_FOLDER', False): static_folder = pagure_config['THEME_STATIC_FOLDER'] if static_folder[0] != '/': static_folder = os.path.join(app.root_path, 'static', static_folder) # Unlike templates, to serve static files from multiples folders we # need flask-multistatic app.static_folder = [ static_folder, os.path.join(app.root_path, 'static'), ] auth = pagure_config.get('PAGURE_AUTH', None) if auth in ['fas', 'openid']: # Only import and set flask_fas_openid if it is needed from pagure.ui.fas_login import FAS FAS.init_app(app) elif auth == 'oidc': # Only import and set flask_fas_openid if it is needed from pagure.ui.oidc_login import oidc, fas_user_from_oidc oidc.init_app(app) app.before_request(fas_user_from_oidc) # Report error by email if not app.debug and not pagure_config.get('DEBUG', False): app.logger.addHandler( pagure.mail_logging.get_mail_handler( smtp_server=pagure_config.get('SMTP_SERVER', '127.0.0.1'), mail_admin=pagure_config.get('MAIL_ADMIN', pagure_config['EMAIL_ERROR']), from_email=pagure_config.get('FROM_EMAIL', '*****@*****.**'))) # Support proxy app.wsgi_app = pagure.proxy.ReverseProxied(app.wsgi_app) # Back port 'equalto' to older version of jinja2 app.jinja_env.tests.setdefault('equalto', lambda value, other: value == other) # Import the application from pagure.api import API # noqa: E402 app.register_blueprint(API) from pagure.ui import UI_NS # noqa: E402 app.register_blueprint(UI_NS) from pagure.internal import PV # noqa: E402 app.register_blueprint(PV) app.before_request(set_request) app.teardown_request(end_request) # Only import the login controller if the app is set up for local login if pagure_config.get('PAGURE_AUTH', None) == 'local': import pagure.ui.login as login app.before_request(login._check_session_cookie) app.after_request(login._send_session_cookie) if perfrepo: # Do this at the very end, so that this after_request comes last. app.after_request(perfrepo.print_stats) app.add_url_rule('/login/', view_func=auth_login, methods=['GET', 'POST']) app.add_url_rule('/logout/', view_func=auth_logout) return app