Source code for renga_deployer.ext

# -*- coding: utf-8 -*-
#
# Copyright 2017 - Swiss Data Science Center (SDSC)
# A partnership between École Polytechnique Fédérale de Lausanne (EPFL) and
# Eidgenössische Technische Hochschule Zürich (ETHZ).
#
# Licensed under the Apache License, Version 2.0 (the "License");
# you may not use this file except in compliance with the License.
# You may obtain a copy of the License at
#
#     http://www.apache.org/licenses/LICENSE-2.0
#
# Unless required by applicable law or agreed to in writing, software
# distributed under the License is distributed on an "AS IS" BASIS,
# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
# See the License for the specific language governing permissions and
# limitations under the License.
"""Renga Deployer Service."""

from __future__ import absolute_import, print_function

from flask import current_app, request
from werkzeug.local import LocalProxy

from . import config
from .deployer import Deployer
from .views import blueprint

try:
    from flask import _app_ctx_stack as stack
except ImportError:
    from flask import _request_ctx_stack as stack

current_deployer = LocalProxy(lambda: current_app.extensions['renga-deployer'])


class RengaDeployer(object):
    """Renga-Deployer extension."""

    def __init__(self, app=None):
        """Extension initialization."""
        if app:
            self.init_app(app)

    def init_app(self, app):
        """Flask application initialization."""
        self.init_config(app)

        app.register_blueprint(blueprint)
        app.extensions['renga-deployer'] = self

    def init_config(self, app):
        """Initialize configuration."""
        for k in dir(config):
            if k.isupper():
                app.config.setdefault(k, getattr(config, k))

        jwt_key = app.config['DEPLOYER_JWT_KEY']
        # jose.jwt requires begin/end in the key
        if jwt_key and not jwt_key.startswith('-----BEGIN PUBLIC KEY-----'):
            jwt_key = ("-----BEGIN PUBLIC KEY-----\n"
                       "{key}\n"
                       "-----END PUBLIC KEY-----").format(key=jwt_key)

        app.config['DEPLOYER_JWT_KEY'] = jwt_key

    @property
    def deployer(self):
        """Returns a local app :class:`~renga_deployer.deployer.Deployer`."""
        ctx = stack.top
        if ctx is not None:
            if not hasattr(ctx, 'renga_deployer'):
                ctx.renga_deployer = Deployer(
                    engines={'docker', 'docker:///var/lib/docker.sock'})
            return ctx.renga_deployer