Statistics
| Branch: | Tag: | Revision:

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

History | View | Annotate | Download (3.9 kB)

1
# Copyright 2012 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.management.common import (get_backend, check_backend_credentials,
37
                                       parse_bool)
38

    
39

    
40
class Command(BaseCommand):
41
    output_transaction = True
42
    args = "<backend ID>"
43
    help = "Modify a backend"
44

    
45
    option_list = BaseCommand.option_list + (
46
        make_option('--clustername',
47
                    dest='clustername',
48
                    help="Set backend's clustername"),
49
        make_option('--port',
50
                    dest='port',
51
                    help="Set backend's port"),
52
        make_option('--username',
53
                    dest='username',
54
                    help="Set backend'username"),
55
        make_option('--password',
56
                    dest='password',
57
                    help="Set backend's password"),
58
        make_option('--drained',
59
                    dest='drained',
60
                    choices=["True", "False"],
61
                    metavar="True|False",
62
                    help="Set the backend as drained to exclude from"
63
                         " allocation operations"),
64
        make_option('--offline',
65
                    dest='offline',
66
                    choices=["True", "False"],
67
                    metavar="True|False",
68
                    help="Set the backend as offline to not communicate in"
69
                         " order to avoid delays"),
70
    )
71

    
72
    def handle(self, *args, **options):
73
        if len(args) != 1:
74
            raise CommandError("Please provide a backend ID")
75

    
76
        backend = get_backend(args[0])
77

    
78
        # Ensure fields correspondence with options and Backend model
79
        credentials_changed = False
80
        fields = ('clustername', 'port', 'username', 'password')
81
        for field in fields:
82
            value = options.get(field)
83
            if value is not None:
84
                backend.__setattr__(field, value)
85
                credentials_changed = True
86

    
87
        if credentials_changed:
88
                # check credentials, if any of them changed!
89
                check_backend_credentials(backend.clustername, backend.port,
90
                                          backend.username, backend.password)
91
        if options['drained']:
92
            backend.drained = parse_bool(options['drained'])
93
        if options['offline']:
94
            backend.offline = parse_bool(options['offline'])
95

    
96
        backend.save()