summaryrefslogtreecommitdiffstats
path: root/cloudify/scripts/onap/create_namespace.py
blob: c0f1f19680ff4ed8198715f86298953a6195020d (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
import pip

from cloudify import ctx
from cloudify.exceptions import NonRecoverableError


def _import_or_install():
    try:
        import yaml
    except ImportError:
        pip.main(["install", "pyaml"])

    try:
        import cloudify_kubernetes.tasks as kubernetes_plugin
    except ImportError:
        pip.main([
            "install",
            "https://github.com/cloudify-incubator/cloudify-kubernetes-plugin/archive/1.2.1rc1.zip"
        ])

    import yaml
    import cloudify_kubernetes.tasks as kubernetes_plugin

    return yaml, kubernetes_plugin


def _retrieve_namespace():
    namespace = ctx.node.properties.get(
        'namespace',
        ctx.node.properties
            .get('options', {})
            .get('namespace', None)
    )

    if not namespace:
        raise NonRecoverableError(
            'Namespace is not defined (node={})'.format(ctx.node.name)
        )

    return namespace


def _prepare_namespace_resource_template(name):
    return {
        'definition': {
            'apiVersion': 'v1',
            'kind': 'Namespace',
            'metadata': {
                'name': name,
                'labels': {
                    'name': name
                },
            },
        },
        'api_mapping': {
            'create': {
                'api': 'CoreV1Api',
                'method': 'create_namespace',
                'payload': 'V1Namespace'
            },
            'read': {
                'api': 'CoreV1Api',
                'method': 'read_namespace',
            },
            'delete': {
                'api': 'CoreV1Api',
                'method': 'delete_namespace',
                'payload': 'V1DeleteOptions'
            }
        }
    }


def _save_deployment_result(key):
    result = ctx.instance.runtime_properties['kubernetes']
    ctx.instance.runtime_properties[key] = result
    ctx.instance.runtime_properties['kubernetes'] = {}


def _do_create_namespace(kubernetes_plugin):
    namespace = _retrieve_namespace()
    ctx.logger.info('Creating namespace: {0}'.format(namespace))

    namespace_resource_template = _prepare_namespace_resource_template(
        namespace
    )

    ctx.logger.debug(
        'Kubernetes object which will be deployed: {0}'
            .format(namespace_resource_template)
    )

    kubernetes_plugin.custom_resource_create(**namespace_resource_template)
    _save_deployment_result('namespace')
    ctx.logger.info('Namespace created successfully')


if __name__ == '__main__':
    _, kubernetes_plugin = _import_or_install()

    _do_create_namespace(kubernetes_plugin)