Statistics
| Branch: | Tag: | Revision:

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

History | View | Annotate | Download (3.8 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

    
38

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

    
44
    option_list = BaseCommand.option_list + (
45
        make_option('--clustername',
46
            dest='clustername',
47
            help="Set backend's clustername"),
48
        make_option('--port',
49
            dest='port',
50
            help="Set backend's port"),
51
        make_option('--username',
52
            dest='username',
53
            help="Set backend'username"),
54
        make_option('--password',
55
            dest='password',
56
            help="Set backend's password"),
57
        make_option('--drained',
58
            dest='drained',
59
            action='store_true',
60
            default=False,
61
            help="Set the backend as drained to exclude from allocation "\
62
                 "operations"),
63
        make_option('--no-drained',
64
            dest='drained',
65
            action='store_false'),
66
        make_option('--offline',
67
            dest='offline',
68
            action='store_true',
69
            default=False,
70
            help="Set the backend as offline to not communicate in order "\
71
                 "to avoid delays"),
72
        make_option('--no-offline',
73
            dest='offline',
74
            action='store_false')
75
        )
76

    
77
    def handle(self, *args, **options):
78
        if len(args) != 1:
79
            raise CommandError("Please provide a backend ID")
80

    
81
        backend = get_backend(args[0])
82

    
83
        # Ensure fields correspondence with options and Backend model
84
        fields = ('clustername', 'port', 'username', 'password', 'drained',
85
                  'offline')
86
        for field in fields:
87
            value = options.get(field)
88
            if value is not None:
89
                backend.__setattr__(field, value)
90

    
91
        credentials = ('clustername', 'port', 'username', 'password')
92
        for field in credentials:
93
            if options.get(field):
94
                # check credentials, if any of them changed!
95
                check_backend_credentials(backend.clustername, backend.port,
96
                                          backend.username, backend.password)
97
                return
98

    
99
        backend.save()