Files
fm-orchestrator/module_build_service/__init__.py
Chenxiong Qi 1c0715662a Cleanup module_build_service/__init__.py
Most of the code are moved to dedicated subpackages, but some others
can't due to the cycle dependencies.

Signed-off-by: Chenxiong Qi <cqi@redhat.com>
2020-03-03 14:48:48 -05:00

93 lines
2.7 KiB
Python

# -*- coding: utf-8 -*-
# SPDX-License-Identifier: MIT
"""The module build orchestrator for Modularity.
The orchestrator coordinates module builds and is responsible
for a number of tasks:
- Providing an interface for module client-side tooling via
which module build submission and build state queries are
possible.
- Verifying the input data (modulemd, RPM SPEC files and others)
is available and correct.
- Preparing the build environment in the supported build systems,
such as koji.
- Scheduling and building of the module components and tracking
the build state.
- Emitting bus messages about all state changes so that other
infrastructure services can pick up the work.
"""
from __future__ import absolute_import
import pkg_resources
from flask import Flask, has_app_context, url_for
from flask_sqlalchemy import SQLAlchemy
from sqlalchemy.pool import StaticPool
from module_build_service.common.config import config_section
from module_build_service.web.proxy import ReverseProxy
try:
version = pkg_resources.get_distribution("module-build-service").version
except pkg_resources.DistributionNotFound:
version = "unknown"
api_version = 2
app = Flask(__name__)
app.wsgi_app = ReverseProxy(app.wsgi_app)
app.config.from_object(config_section)
class MBSSQLAlchemy(SQLAlchemy):
"""
Inherits from SQLAlchemy and if SQLite in-memory database is used,
sets the driver options so multiple threads can share the same database.
This is used *only* during tests to make them faster.
"""
def apply_driver_hacks(self, app, info, options):
if info.drivername == "sqlite" and info.database in (None, "", ":memory:"):
options["poolclass"] = StaticPool
options["connect_args"] = {"check_same_thread": False}
try:
del options["pool_size"]
except KeyError:
pass
super(MBSSQLAlchemy, self).apply_driver_hacks(app, info, options)
db = MBSSQLAlchemy(app)
def get_url_for(*args, **kwargs):
"""
flask.url_for wrapper which creates the app_context on-the-fly.
"""
if has_app_context():
return url_for(*args, **kwargs)
# Localhost is right URL only when the scheduler runs on the same
# system as the web views.
app.config["SERVER_NAME"] = "localhost"
with app.app_context():
from module_build_service.common import log
log.debug(
"WARNING: get_url_for() has been called without the Flask "
"app_context. That can lead to SQLAlchemy errors caused by "
"multiple session being used in the same time."
)
return url_for(*args, **kwargs)
def load_views():
from module_build_service.web import views
assert views
load_views()