Revision a17ce1de snf-cyclades-app/synnefo/management/common.py

b/snf-cyclades-app/synnefo/management/common.py
32 32
# or implied, of GRNET S.A.
33 33

  
34 34
from django.core.management import CommandError
35
from synnefo.db.models import Backend, VirtualMachine, Network, Flavor
35
from synnefo.db.models import Backend, VirtualMachine, Network, Flavor, Subnet
36 36
from functools import wraps
37 37

  
38 38
from snf_django.lib.api import faults
......
127 127
                           " available network IDs." % network_id)
128 128

  
129 129

  
130
def get_subnet(subnet_id, for_update=True):
131
    """Get a Subnet object by its ID."""
132
    try:
133
        return Subnet.objects.get(id=subnet_id)
134
    except Subnet.DoesNotExist:
135
        raise CommandError("Subnet with ID %s not found in DB."
136
                           " Use snf-manage subnet-list to find out"
137
                           " available subnet IDs" % subnet_id)
138

  
139

  
130 140
def get_flavor(flavor_id):
131 141
    try:
132 142
        flavor_id = int(flavor_id)

Also available in: Unified diff