Statistics
| Branch: | Tag: | Revision:

root / snf-cyclades-app / synnefo / logic / management / commands / pool-modify.py @ 3e7c63f8

History | View | Annotate | Download (4 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 django.core.management.base import BaseCommand, CommandError
35
from optparse import make_option
36

    
37
from util import pool_table_from_type
38

    
39
POOL_CHOICES = ['bridge', 'mac-prefix']
40

    
41

    
42
class Command(BaseCommand):
43
    args = "<pool ID>"
44
    help = "Modify a pool"
45
    output_transaction = True
46
    option_list = BaseCommand.option_list + (
47
        make_option('--type', dest='type',
48
                    choices=POOL_CHOICES,
49
                    help="Type of pool"
50
                    ),
51
        make_option('--offset', dest='offset'),
52
        make_option('--size', dest='size'),
53
        make_option('--base', dest='base'),
54
        make_option('--add-reserved', dest='add-reserved',
55
                    help="Comma-seperated list of values to reserve"),
56
        make_option('--remove-reserved', dest="remove-reserved",
57
                    help="Comma-seperated list of values to release"),
58
    )
59

    
60
    def handle(self, *args, **options):
61
        type_ = options['type']
62
        offset = options['offset']
63
        base = options['base']
64
        add_reserved = options['add-reserved']
65
        remove_reserved = options['remove-reserved']
66

    
67
        if not type_:
68
            raise CommandError("Type of pool is mandatory")
69

    
70
        pool_table = pool_table_from_type(type_)
71

    
72
        try:
73
            pool_id = int(args[0])
74
            pool_row = pool_table.objects.get(id=pool_id)
75
        except (ValueError, pool_table.DoesNotExist):
76
            raise CommandError("Invalid pool ID")
77

    
78
        pool = pool_row.pool
79
        if add_reserved:
80
            reserved = add_reserved.split(',')
81
            for value in reserved:
82
                pool.reserve(value, external=True)
83
        if remove_reserved:
84
            released = remove_reserved.split(',')
85
            for value in released:
86
                pool.put(value, external=True)
87
        pool.save(db=False)
88

    
89
        if offset:
90
            pool_row.offset = offset
91
        if base:
92
            pool_row.base = base
93

    
94
        # Save now, to avoid conflict with resizing pool.save()
95
        pool_row.save()
96

    
97
        if size:
98
            try:
99
                size = int(size)
100
            except ValueError:
101
                raise CommandError("Invalid size")
102
            pool = pool_row.get_pool()
103
            self.resize_pool(pool, pool_row.size, size)
104
            pool.save()
105

    
106
    def resize_pool(self, pool, old_size, new_size):
107
        if new_size == old_size:
108
            return
109
        elif new_size > old_size:
110
            pool.extend(new_size - old_size)
111
        else:
112
            pool.shrink(old_size - new_size)
113
        pool.save()