Revision b84ed662 snf-cyclades-app/synnefo/logic/management/commands/backend-modify.py
b/snf-cyclades-app/synnefo/logic/management/commands/backend-modify.py | ||
---|---|---|
33 | 33 |
|
34 | 34 |
from optparse import make_option |
35 | 35 |
from django.core.management.base import BaseCommand, CommandError |
36 |
|
|
37 |
from synnefo.db.models import Backend |
|
36 |
from synnefo.management.common import get_backend |
|
38 | 37 |
|
39 | 38 |
|
40 | 39 |
class Command(BaseCommand): |
... | ... | |
78 | 77 |
if len(args) != 1: |
79 | 78 |
raise CommandError("Please provide a backend ID") |
80 | 79 |
|
81 |
try: |
|
82 |
backend_id = int(args[0]) |
|
83 |
backend = Backend.objects.get(id=backend_id) |
|
84 |
except ValueError: |
|
85 |
raise CommandError("Invalid backend ID") |
|
86 |
except Backend.DoesNotExist: |
|
87 |
raise CommandError("Backend not found in DB") |
|
80 |
backend = get_backend(args[0]) |
|
88 | 81 |
|
89 | 82 |
# Ensure fields correspondence with options and Backend model |
90 | 83 |
fields = ('clustername', 'port', 'username', 'password', 'drained', |
Also available in: Unified diff