Пример #1
0
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
Пример #2
0
def create_app(test_config=None):
    # create and configure the app
    app = MultiStaticFlask(__name__, instance_relative_config=True)
    babel = Babel(app)
    CORS(app, supports_credentials=True)

    # Add custom static location
    if custom_id:
        app.static_folder = [
            os.path.join(app.root_path.replace('webApp', ''),
                         'custom/' + custom_id[0] + '/webApp/static/'),
            os.path.join(app.root_path, 'static')
        ]
    else:
        app.static_folder = [
            os.path.join(app.root_path, 'static'),
            os.path.join(app.root_path, 'static')
        ]

    app.config.from_mapping(
        SECRET_KEY='§§SECRET§§',
        CONFIG_FILE=os.path.join(app.instance_path, 'config.ini'),
        CONFIG_FOLDER=os.path.join(app.instance_path, 'config/'),
        UPLOAD_FOLDER=os.path.join(app.instance_path, 'upload/'),
        PER_PAGE=16,
        BABEL_TRANSLATION_DIRECTORIES=os.path.join(app.static_folder[0],
                                                   'babel/translations/')
        if os.path.isdir(app.static_folder[0] + 'babel/translations') else
        os.path.join(app.static_folder[1], 'babel/translations/'),
        # array key is the babel language code
        # index 0 is the label
        # index 1 is the locale language code in the config.ini
        # index 2 is the localeocr language code in the config.ini (tesseract)
        LANGUAGES={
            'fr': ['Français', 'fr_FR', 'fra'],
            'en': ['English', 'en_EN', 'eng']
        },
    )

    app.register_blueprint(ws.bp)
    app.register_blueprint(pdf.bp)
    app.register_blueprint(auth.bp)
    app.register_blueprint(user.bp)
    app.register_blueprint(supplier.bp)
    app.register_blueprint(dashboard.bp)
    app.register_blueprint(splitter.bp)
    app.add_url_rule('/', endpoint='index')

    # Add custom templates location
    if custom_id:
        array_location = [custom_id[1] + '/webApp', 'webApp']
    else:
        array_location = ['webApp']

    templates_locations = jinja2.ChoiceLoader([
        app.jinja_loader,
        jinja2.FileSystemLoader(array_location),
    ])
    app.jinja_loader = templates_locations

    db.init_app(app)

    if test_config is None:
        # load the instance config, if it exists, when not testing
        app.config.from_pyfile('config.py', silent=True)
    else:
        # load the test config if passed in
        app.config.from_mapping(test_config)

    # ensure the instance folder exists
    try:
        os.makedirs(app.instance_path)
    except OSError:
        pass

    @app.route('/')
    def index():
        return redirect(url_for('pdf.index', time='TODAY', status='NEW'))

    @babel.localeselector
    def get_locale():
        if 'lang' not in session:
            session['lang'] = request.accept_languages.best_match(
                app.config['LANGUAGES'].keys())

        return session['lang']

    return app