pagure/runserver.py

#!/usr/bin/env python

###############################################################################
# Please note that this script is only used for development purposes.         #
# Do not start the Flask app with this script in a production environment,    #
# use files/pagure.wsgi instead!                                              #
###############################################################################

from __future__ import unicode_literals, absolute_import

import argparse
import sys
import os

parser = argparse.ArgumentParser(description="Run the Pagure app")
parser.add_argument(
    "--config",
    "-c",
    dest="config",
    help="Configuration file to use for pagure.",
)
parser.add_argument(
    "--plugins",
    dest="plugins",
    help="Configuration file for pagure plugins. This argument takes "
    "precedence over the environment variable PAGURE_PLUGINS_CONFIG, which in "
    "turn takes precedence over the configuration variable "
    "PAGURE_PLUGINS_CONFIG.",
)
parser.add_argument(
    "--debug",
    dest="debug",
    action="store_true",
    default=False,
    help="Expand the level of data returned.",
)
parser.add_argument(
    "--profile",
    dest="profile",
    action="store_true",
    default=False,
    help="Profile Pagure.",
)
parser.add_argument(
    "--perf-verbose",
    dest="perfverbose",
    action="store_true",
    default=False,
    help="Enable per-request printing of performance statistics.",
)
parser.add_argument(
    "--port", "-p", default=5000, help="Port for the Pagure to run on."
)
parser.add_argument(
    "--no-debug", action="store_true", help="Disable debugging"
)
parser.add_argument(
    "--host",
    default="127.0.0.1",
    help="Hostname to listen on. When set to 0.0.0.0 the server is available "
    "externally. Defaults to 127.0.0.1 making the it only visible on localhost",
)

parser.add_argument(
    "--cert", "-s", default=None, help="Filename of SSL cert for the flask application."
)
parser.add_argument(
    "--key",
    "-k",
    default=None,
    help="Filename of the SSL key for the flask application.",
)

args = parser.parse_args()

if args.config:
    config = args.config
    if not config.startswith("/"):
        here = os.path.join(os.path.dirname(os.path.abspath(__file__)))
        config = os.path.join(here, config)
    os.environ["PAGURE_CONFIG"] = config

if args.plugins:
    config = args.plugins
    if not config.startswith("/"):
        here = os.path.join(os.path.dirname(os.path.abspath(__file__)))
        config = os.path.join(here, config)
    os.environ["PAGURE_PLUGINS_CONFIG"] = config
    
    # If this script is ran with the deprecated env. variable PAGURE_PLUGIN
    # and --plugins, we need to override it too.
    if "PAGURE_PLUGIN" in os.environ:
        os.environ["PAGURE_PLUGIN"] = config

if args.perfverbose:
    os.environ["PAGURE_PERFREPO"] = "true"
    os.environ["PAGURE_PERFREPO_VERBOSE"] = "true"

from pagure.flask_app import create_app

APP = create_app()

if args.profile:
    from werkzeug.contrib.profiler import ProfilerMiddleware

    APP.config["PROFILE"] = True
    APP.wsgi_app = ProfilerMiddleware(APP.wsgi_app, restrictions=[30])

APP.debug = not args.no_debug

if args.cert and args.key:
    APP.run(host=args.host, port=int(args.port), ssl_context=(args.cert, args.key))
else:
    APP.run(host=args.host, port=int(args.port))