Statistics
| Branch: | Tag: | Revision:

root / snf-astakos-app / astakos / im / management / commands / service-import.py @ eb765213

History | View | Annotate | Download (4.6 kB)

1
# Copyright 2013 GRNET S.A. All rights reserved.
2
#
3
# Redistribution and use in source and binary forms, with or
4
# without modification, are permitted provided that the following
5
# conditions are met:
6
#
7
#   1. Redistributions of source code must retain the above
8
#      copyright notice, this list of conditions and the following
9
#      disclaimer.
10
#
11
#   2. Redistributions in binary form must reproduce the above
12
#      copyright notice, this list of conditions and the following
13
#      disclaimer in the documentation and/or other materials
14
#      provided with the distribution.
15
#
16
# THIS SOFTWARE IS PROVIDED BY GRNET S.A. ``AS IS'' AND ANY EXPRESS
17
# OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED
18
# WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR
19
# PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL GRNET S.A OR
20
# CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL,
21
# SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT
22
# LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF
23
# USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED
24
# AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT
25
# LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN
26
# ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE
27
# POSSIBILITY OF SUCH DAMAGE.
28
#
29
# The views and conclusions contained in the software and
30
# documentation are those of the authors and should not be
31
# interpreted as representing official policies, either expressed
32
# or implied, of GRNET S.A.
33

    
34
from optparse import make_option
35

    
36
from django.core.management.base import BaseCommand, CommandError
37
from django.utils import simplejson as json
38

    
39
from snf_django.lib.db.transaction import commit_on_success_strict
40
from astakos.im.register import add_service, add_resource, RegisterException
41
from astakos.im.models import Component
42
from ._common import read_from_file
43

    
44

    
45
class Command(BaseCommand):
46
    help = "Register services"
47

    
48
    option_list = BaseCommand.option_list + (
49
        make_option('--json',
50
                    dest='json',
51
                    metavar='<json.file>',
52
                    help="Load service definitions from a json file"),
53
    )
54

    
55
    @commit_on_success_strict()
56
    def handle(self, *args, **options):
57

    
58
        json_file = options['json']
59
        if not json_file:
60
            m = "Expecting option --json."
61
            raise CommandError(m)
62

    
63
        else:
64
            data = read_from_file(json_file)
65
            m = ('Input should be a JSON dict mapping service names '
66
                 'to definitions.')
67
            try:
68
                data = json.loads(data)
69
            except json.JSONDecodeError:
70
                raise CommandError(m)
71
            if not isinstance(data, dict):
72
                raise CommandError(m)
73
        self.add_services(data)
74

    
75
    def add_services(self, data):
76
        write = self.stdout.write
77
        output = []
78
        for name, service_dict in data.iteritems():
79
            try:
80
                component_name = service_dict['component']
81
                service_type = service_dict['type']
82
                endpoints = service_dict['endpoints']
83
            except KeyError:
84
                raise CommandError('Malformed service definition.')
85

    
86
            try:
87
                component = Component.objects.get(name=component_name)
88
            except Component.DoesNotExist:
89
                m = "Component '%s' is not registered." % component_name
90
                raise CommandError(m)
91

    
92
            try:
93
                existed = add_service(component, name, service_type, endpoints,
94
                                      out=self.stdout)
95
            except RegisterException as e:
96
                raise CommandError(e.message)
97

    
98
            m = "%s service %s.\n" % ("Updated" if existed else "Added", name)
99
            output.append(m)
100

    
101
            resources = service_dict.get('resources', {}).values()
102
            for resource in resources:
103
                if not isinstance(resource, dict):
104
                    raise CommandError("Malformed resource dict.")
105

    
106
                service_origin = resource.get('service_origin')
107
                if name != service_origin:
108
                    raise CommandError("service_origin mismatch.")
109
                try:
110
                    r, exists = add_resource(resource)
111
                except RegisterException as e:
112
                    raise CommandError(e.message)
113
                if exists:
114
                    m = "Resource '%s' updated in database.\n" % (r.name)
115
                else:
116
                    m = ("Resource '%s' created in database with default "
117
                         "quota limit 0.\n" % (r.name))
118
                output.append(m)
119

    
120
        for line in output:
121
            write(line)