def create_app(config_name):
connexion_options = {"swagger_ui": True}
# This feels like a hack but it is needed. The logging configuration
# needs to be setup before the flask app is initialized.
configure_logging(config_name)
app_config = Config()
app_config.log_configuration(config_name)
connexion_app = connexion.App(
"inventory", specification_dir="./swagger/", options=connexion_options
)
# Read the swagger.yml file to configure the endpoints
with open("swagger/api.spec.yaml", "rb") as fp:
spec = yaml.safe_load(fp)
for api_url in app_config.api_urls:
if api_url:
connexion_app.add_api(
spec,
arguments={"title": "RestyResolver Example"},
resolver=RestyResolver("api"),
validate_responses=True,
strict_validation=True,
base_path=api_url,
)
app_config.logger.info("Listening on API: %s" % api_url)
# Add an error handler that will convert our top level exceptions
# into error responses
connexion_app.add_error_handler(InventoryException, render_exception)
flask_app = connexion_app.app
flask_app.config["SQLALCHEMY_ECHO"] = False
flask_app.config["SQLALCHEMY_TRACK_MODIFICATIONS"] = False
flask_app.config["SQLALCHEMY_DATABASE_URI"] = app_config.db_uri
flask_app.config["SQLALCHEMY_POOL_SIZE"] = app_config.db_pool_size
flask_app.config["SQLALCHEMY_POOL_TIMEOUT"] = app_config.db_pool_timeout
db.init_app(flask_app)
flask_app.register_blueprint(monitoring_blueprint,
url_prefix=app_config.mgmt_url_path_prefix)
@flask_app.before_request
def set_request_id():
threadctx.request_id = request.headers.get(
REQUEST_ID_HEADER,
UNKNOWN_REQUEST_ID_VALUE)
init_tasks(app_config, flask_app)
return flask_app
def main(logger):
config = _init_config()
init_tasks(config)
registry = CollectorRegistry()
for metric in COLLECTED_METRICS:
registry.register(metric)
Session = _init_db(config)
session = Session()
try:
with session_guard(session):
run(config, logger, session)
finally:
flush()
job = _prometheus_job(config.kubernetes_namespace)
push_to_gateway(config.prometheus_pushgateway, job, registry)
def main(config_name):
config = _init_config(config_name)
init_tasks(config)
registry = CollectorRegistry()
for metric in COLLECTED_METRICS:
registry.register(metric)
Session = _init_db(config)
session = Session()
try:
with session_guard(session):
run(config, session)
except Exception as exception:
logger = get_logger(LOGGER_NAME)
logger.exception(exception)
finally:
flush()
job = _prometheus_job(config.kubernetes_namespace)
push_to_gateway(config.prometheus_pushgateway, job, registry)
def create_app(config_name, start_tasks=False, start_payload_tracker=False):
connexion_options = {"swagger_ui": True}
# This feels like a hack but it is needed. The logging configuration
# needs to be setup before the flask app is initialized.
configure_logging(config_name)
app_config = Config(RuntimeEnvironment.server)
app_config.log_configuration(config_name)
connexion_app = connexion.App("inventory", specification_dir="./swagger/", options=connexion_options)
# Read the swagger.yml file to configure the endpoints
with open("swagger/api.spec.yaml", "rb") as fp:
spec = yaml.safe_load(fp)
for api_url in app_config.api_urls:
if api_url:
connexion_app.add_api(
spec,
arguments={"title": "RestyResolver Example"},
resolver=RestyResolver("api"),
validate_responses=True,
strict_validation=True,
base_path=api_url,
)
logger.info("Listening on API: %s", api_url)
# Add an error handler that will convert our top level exceptions
# into error responses
connexion_app.add_error_handler(InventoryException, render_exception)
flask_app = connexion_app.app
flask_app.config["SQLALCHEMY_ECHO"] = False
flask_app.config["SQLALCHEMY_TRACK_MODIFICATIONS"] = False
flask_app.config["SQLALCHEMY_DATABASE_URI"] = app_config.db_uri
flask_app.config["SQLALCHEMY_POOL_SIZE"] = app_config.db_pool_size
flask_app.config["SQLALCHEMY_POOL_TIMEOUT"] = app_config.db_pool_timeout
flask_app.config["INVENTORY_CONFIG"] = app_config
db.init_app(flask_app)
flask_app.register_blueprint(monitoring_blueprint, url_prefix=app_config.mgmt_url_path_prefix)
@flask_app.before_request
def set_request_id():
threadctx.request_id = request.headers.get(REQUEST_ID_HEADER, UNKNOWN_REQUEST_ID_VALUE)
if start_tasks:
init_tasks(app_config, flask_app)
else:
logger.warning(
'WARNING: The "tasks" subsystem has been disabled. '
"The message queue based system_profile consumer "
"and message queue based event notifications have been disabled."
)
payload_tracker_producer = None
if start_payload_tracker is False:
# If we are running in "testing" mode, then inject the NullProducer.
payload_tracker_producer = payload_tracker.NullProducer()
logger.warning(
"WARNING: Using the NullProducer for the payload tracker producer. "
"No payload tracker events will be sent to to payload tracker."
)
payload_tracker.init_payload_tracker(app_config, producer=payload_tracker_producer)
# HTTP request metrics
if config_name != "testing":
PrometheusMetrics(
flask_app,
defaults_prefix="inventory",
group_by="url_rule",
path=None,
excluded_paths=["^/metrics$", "^/health$", "^/version$", r"^/favicon\.ico$"],
)
return flask_app
from utils import resolve_task_type
from tasks import init_tasks
app = Flask(__name__)
app.config.from_object('goodpackagenamehere.settings')
try:
app.config.from_object('local_settings')
except ImportError:
pass
assets_init(app)
db = MongoEngine(app)
init_social_login(app, db)
TASKS_TYPES = init_tasks(app)
@app.route('/', methods=["GET"], defaults={'task_type': None})
@app.route('/type/<string:task_type>/', methods=["GET"])
@resolve_task_type
def index(task_type):
return render_template("index.html", task_type=task_type)
@app.route('/logout', methods=['POST'])
def logout():
login.logout_user()
return redirect(url_for('index'))