[GITLAB] - UPGRADE TO v12 on Wednesday the 18th of December at 11.30AM

gen_json.py 2.79 KB
Newer Older
Daniele Venzano's avatar
Daniele Venzano committed
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15
# Copyright (c) 2016, Daniele Venzano
#
# 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.

16
"""Jupyter Zoe application description generator."""
Daniele Venzano's avatar
Daniele Venzano committed
17 18 19 20 21

import json
import sys
import os

22
UPSTREAM_NOTEBOOKS = [
23 24 25 26
    ("r-notebook", "jupyter/r-notebook:latest"),
    ("datasci-notebook", "jupyter/datascience-notebook:latest"),
    ("tf-notebook", "jupyter/tensorflow-notebook:latest"),
    ("scipy-notebook", "jupyter/scipy-notebook:latest")
27 28
]

Daniele Venzano's avatar
Daniele Venzano committed
29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45
ZOE_APPLICATION_DESCRIPTION_VERSION = 3

options = {
    'core_limit': {
        'value': 4,
        'description': 'Notebook core limit'
    },
    'memory_limit': {
        'value': 4 * (1024**3),
        'description': 'Notebook memory limit (bytes)'
    }
}

REGISTRY = os.getenv("DOCKER_REGISTRY", default="docker-engine:5000")
REPOSITORY = os.getenv("REPOSITORY", default="zapps")
VERSION = os.getenv("VERSION", default="latest")

46
def jupyter_service(memory_limit, core_limit, image):
Daniele Venzano's avatar
Daniele Venzano committed
47 48 49 50 51
    """
    :rtype: dict
    """
    service = {
        'name': "jupyter",
52
        'image': image,
Daniele Venzano's avatar
Daniele Venzano committed
53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71
        'monitor': True,
        'resources': {
            "memory": {
                "min": memory_limit,
                "max": memory_limit
            },
            "cores": {
                "min": core_limit,
                "max": core_limit
            }
        },
        'ports': [
            {
                'name': 'Jupyter Notebook interface',
                'protocol': 'tcp',
                'port_number': 8888,
                'url_template': 'http://{ip_port}/'
            }
        ],
72 73 74 75
        'environment': [
            ['NB_UID', '1000'],
            ['HOME', '/mnt/workspace']
        ],
Daniele Venzano's avatar
Daniele Venzano committed
76 77
        'volumes': [],
        'command': None,
78
        'work_dir': '/mnt/workspace',
Daniele Venzano's avatar
Daniele Venzano committed
79 80
        'total_count': 1,
        'essential_count': 1,
81
        'replicas': 1,
Daniele Venzano's avatar
Daniele Venzano committed
82 83 84 85 86 87 88
        'startup_order': 0
    }

    return service


if __name__ == '__main__':
89 90 91 92 93 94 95 96 97 98
    for app_name, image in UPSTREAM_NOTEBOOKS:
        app = {
            'name': app_name,
            'version': ZOE_APPLICATION_DESCRIPTION_VERSION,
            'will_end': False,
            'size': 512,
            'services': [
                jupyter_service(options["memory_limit"]["value"], options["core_limit"]["value"], image)
            ]
        }
Daniele Venzano's avatar
Daniele Venzano committed
99

100
        json.dump(app, open(app_name + ".json", "w"), sort_keys=True, indent=4)
Daniele Venzano's avatar
Daniele Venzano committed
101

102
    print("ZApps written")
Daniele Venzano's avatar
Daniele Venzano committed
103