Statistics
| Branch: | Tag: | Revision:

root / snf-cyclades-app / synnefo / logic / management / commands / backend-modify.py @ a6e6fe48

History | View | Annotate | Download (4.5 kB)

1
# Copyright 2012-2014 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
from django.core.management.base import BaseCommand, CommandError
36
from synnefo.db.models import Backend
37
from snf_django.management.utils import parse_bool
38
from synnefo.management import common
39

    
40
HYPERVISORS = [h[0] for h in Backend.HYPERVISORS]
41

    
42

    
43
class Command(BaseCommand):
44
    output_transaction = True
45
    args = "<backend ID>"
46
    help = "Modify a backend"
47

    
48
    option_list = BaseCommand.option_list + (
49
        make_option('--clustername',
50
                    dest='clustername',
51
                    help="Set backend's clustername"),
52
        make_option('--port',
53
                    dest='port',
54
                    help="Set backend's port"),
55
        make_option('--username',
56
                    dest='username',
57
                    help="Set backend'username"),
58
        make_option('--password',
59
                    dest='password',
60
                    help="Set backend's password"),
61
        make_option('--drained',
62
                    dest='drained',
63
                    choices=["True", "False"],
64
                    metavar="True|False",
65
                    help="Set the backend as drained to exclude from"
66
                         " allocation operations"),
67
        make_option('--hypervisor',
68
                    dest='hypervisor',
69
                    default=None,
70
                    choices=HYPERVISORS,
71
                    metavar="|".join(HYPERVISORS),
72
                    help="The hypervisor that the Ganeti backend uses"),
73
        make_option('--offline',
74
                    dest='offline',
75
                    choices=["True", "False"],
76
                    metavar="True|False",
77
                    help="Set the backend as offline to not communicate in"
78
                         " order to avoid delays"),
79
    )
80

    
81
    def handle(self, *args, **options):
82
        if len(args) != 1:
83
            raise CommandError("Please provide a backend ID")
84

    
85
        backend = common.get_resource("backend", args[0], for_update=True)
86

    
87
        # Ensure fields correspondence with options and Backend model
88
        credentials_changed = False
89
        fields = ('clustername', 'port', 'username', 'password')
90
        for field in fields:
91
            value = options.get(field)
92
            if value is not None:
93
                backend.__setattr__(field, value)
94
                credentials_changed = True
95

    
96
        if credentials_changed:
97
                # check credentials, if any of them changed!
98
                common.check_backend_credentials(backend.clustername,
99
                                                 backend.port,
100
                                                 backend.username,
101
                                                 backend.password)
102
        if options['drained']:
103
            backend.drained = parse_bool(options['drained'], strict=True)
104
        if options['offline']:
105
            backend.offline = parse_bool(options['offline'], strict=True)
106
        hypervisor = options["hypervisor"]
107
        if hypervisor:
108
            backend.hypervisor = hypervisor
109

    
110
        backend.save()