resource-import.py 3.32 KB
Newer Older
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
# Copyright 2013 GRNET S.A. All rights reserved.
#
# Redistribution and use in source and binary forms, with or
# without modification, are permitted provided that the following
# conditions are met:
#
#   1. Redistributions of source code must retain the above
#      copyright notice, this list of conditions and the following
#      disclaimer.
#
#   2. Redistributions in binary form must reproduce the above
#      copyright notice, this list of conditions and the following
#      disclaimer in the documentation and/or other materials
#      provided with the distribution.
#
# THIS SOFTWARE IS PROVIDED BY GRNET S.A. ``AS IS'' AND ANY EXPRESS
# OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED
# WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR
# PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL GRNET S.A OR
# CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL,
# SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT
# LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF
# USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED
# AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT
# LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN
# ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE
# POSSIBILITY OF SUCH DAMAGE.
#
# The views and conclusions contained in the software and
# documentation are those of the authors and should not be
# interpreted as representing official policies, either expressed
# or implied, of GRNET S.A.

from optparse import make_option

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

40
from astakos.im.register import add_resource, RegisterException
41
from ._common import read_from_file
42
43
44


class Command(BaseCommand):
45
    help = "Register resources"
46
47
48
49
50

    option_list = BaseCommand.option_list + (
        make_option('--json',
                    dest='json',
                    metavar='<json.file>',
51
                    help="Load resource definitions from a json file"),
52
53
54
55
56
    )

    def handle(self, *args, **options):

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

61
        else:
62
63
64
65
66
67
68
69
            data = read_from_file(json_file)
            m = 'Input should be a JSON list.'
            try:
                data = json.loads(data)
            except json.JSONDecodeError:
                raise CommandError(m)
            if not isinstance(data, list):
                raise CommandError(m)
70
        self.add_resources(data)
71

72
    @transaction.commit_on_success
73
74
    def add_resources(self, resources):
        output = []
75
        for resource in resources:
76
77
            if not isinstance(resource, dict):
                raise CommandError("Malformed resource dict.")
78
79
            try:
                r, exists = add_resource(resource)
80
            except RegisterException as e:
81
                raise CommandError(e.message)
82
            name = r.name
83
84
85
86
87
            if exists:
                m = "Resource '%s' updated in database.\n" % (name)
            else:
                m = ("Resource '%s' created in database with default "
                     "quota limit 0.\n" % (name))
88
89
90
91
            output.append(m)

        for line in output:
            self.stdout.write(line)