Revision b84ed662 snf-cyclades-app/synnefo/api/management/commands/server-list.py
b/snf-cyclades-app/synnefo/api/management/commands/server-list.py | ||
---|---|---|
34 | 34 |
from optparse import make_option |
35 | 35 |
|
36 | 36 |
from django.core.management.base import BaseCommand, CommandError |
37 |
from synnefo.management.common import format_vm_state, get_backend |
|
37 | 38 |
|
38 | 39 |
from synnefo.api.util import get_image |
39 |
from synnefo.db.models import VirtualMachine, Backend |
|
40 |
from ._common import format_vm_state |
|
40 |
from synnefo.db.models import VirtualMachine |
|
41 | 41 |
|
42 | 42 |
|
43 | 43 |
class Command(BaseCommand): |
... | ... | |
71 | 71 |
raise CommandError("Command doesn't accept any arguments") |
72 | 72 |
|
73 | 73 |
if options['backend_id']: |
74 |
try: |
|
75 |
servers = Backend.objects.get(id=options['backend_id'])\ |
|
76 |
.virtual_machines |
|
77 |
except Backend.DoesNotExist: |
|
78 |
raise CommandError("Backend not found in DB") |
|
74 |
backend = get_backend(options['backend_id']) |
|
75 |
servers = backend.virtual_machines |
|
79 | 76 |
else: |
80 | 77 |
servers = VirtualMachine.objects |
81 | 78 |
|
Also available in: Unified diff