Revision 91f7a08c
| b/snf-cyclades-app/synnefo/api/management/commands/server-modify.py | ||
|---|---|---|
| 33 | 33 |
|
| 34 | 34 |
from optparse import make_option |
| 35 | 35 |
|
| 36 |
from django.db import transaction |
|
| 36 | 37 |
from django.core.management.base import BaseCommand, CommandError |
| 37 | 38 |
from synnefo.management.common import get_vm |
| 38 |
|
|
| 39 |
from synnefo.db.models import VirtualMachine
|
|
| 39 |
from synnefo.webproject.management.utils import parse_bool |
|
| 40 |
from synnefo.logic import servers
|
|
| 40 | 41 |
|
| 41 | 42 |
|
| 42 | 43 |
class Command(BaseCommand): |
| 43 | 44 |
args = "<server ID>" |
| 44 |
help = "Modify a server" |
|
| 45 |
help = "Modify a server."
|
|
| 45 | 46 |
|
| 46 | 47 |
option_list = BaseCommand.option_list + ( |
| 47 | 48 |
make_option( |
| 48 | 49 |
'--name', |
| 49 | 50 |
dest='name', |
| 50 | 51 |
metavar='NAME', |
| 51 |
help="Set server's name"),
|
|
| 52 |
help="Rename server"),
|
|
| 52 | 53 |
make_option( |
| 53 | 54 |
'--owner', |
| 54 | 55 |
dest='owner', |
| 55 |
metavar='USER_ID', |
|
| 56 |
help="Set server's owner"), |
|
| 57 |
make_option( |
|
| 58 |
'--state', |
|
| 59 |
dest='state', |
|
| 60 |
metavar='STATE', |
|
| 61 |
help="Set server's state"), |
|
| 62 |
make_option( |
|
| 63 |
'--set-deleted', |
|
| 64 |
action='store_true', |
|
| 65 |
dest='deleted', |
|
| 66 |
help="Mark a server as deleted"), |
|
| 67 |
make_option( |
|
| 68 |
'--set-undeleted', |
|
| 69 |
action='store_true', |
|
| 70 |
dest='undeleted', |
|
| 71 |
help="Mark a server as not deleted"), |
|
| 56 |
metavar='USER_UUID', |
|
| 57 |
help="Change ownership of server. Value must be a user UUID"), |
|
| 72 | 58 |
make_option( |
| 73 |
'--set-suspended', |
|
| 74 |
action='store_true', |
|
| 75 |
dest='suspended', |
|
| 76 |
help="Mark a server as suspended"), |
|
| 77 |
make_option( |
|
| 78 |
'--set-unsuspended', |
|
| 79 |
action='store_true', |
|
| 80 |
dest='unsuspended', |
|
| 81 |
help="Mark a server as not suspended") |
|
| 59 |
"--suspended", |
|
| 60 |
dest="suspended", |
|
| 61 |
default=None, |
|
| 62 |
choices=["True", "False"], |
|
| 63 |
metavar="True|False", |
|
| 64 |
help="Mark a server as suspended/non-suspended."), |
|
| 82 | 65 |
) |
| 83 | 66 |
|
| 67 |
@transaction.commit_on_success |
|
| 84 | 68 |
def handle(self, *args, **options): |
| 85 | 69 |
if len(args) != 1: |
| 86 | 70 |
raise CommandError("Please provide a server ID")
|
| 87 | 71 |
|
| 88 | 72 |
server = get_vm(args[0]) |
| 89 | 73 |
|
| 90 |
name = options.get('name')
|
|
| 91 |
if name is not None: |
|
| 92 |
server.name = name |
|
| 93 |
|
|
| 94 |
owner = options.get('owner')
|
|
| 95 |
if owner is not None: |
|
| 96 |
server.userid = owner |
|
| 97 |
|
|
| 98 |
state = options.get('state')
|
|
| 99 |
if state is not None: |
|
| 100 |
allowed = [x[0] for x in VirtualMachine.OPER_STATES] |
|
| 101 |
if state not in allowed: |
|
| 102 |
msg = "Invalid state, must be one of %s" % ', '.join(allowed) |
|
| 103 |
raise CommandError(msg) |
|
| 104 |
server.operstate = state |
|
| 105 |
|
|
| 106 |
if options.get('deleted'):
|
|
| 107 |
server.deleted = True |
|
| 108 |
elif options.get('undeleted'):
|
|
| 109 |
server.deleted = False |
|
| 74 |
new_name = options.get("name", None)
|
|
| 75 |
if new_name is not None: |
|
| 76 |
old_name = server.name |
|
| 77 |
server = servers.rename(server, new_name) |
|
| 78 |
self.stdout.write("Renamed server '%s' from '%s' to '%s'\n" %
|
|
| 79 |
(server, old_name, new_name)) |
|
| 110 | 80 |
|
| 111 |
if options.get('suspended'):
|
|
| 112 |
server.suspended = True |
|
| 113 |
elif options.get('unsuspended'):
|
|
| 114 |
server.suspended = False |
|
| 81 |
suspended = options.get("suspended", None)
|
|
| 82 |
if suspended is not None: |
|
| 83 |
suspended = parse_bool(suspended) |
|
| 84 |
server.suspended = suspended |
|
| 85 |
server.save() |
|
| 86 |
self.stdout.write("Set server '%s' as suspended=%s\n" %
|
|
| 87 |
(server, suspended)) |
|
| 115 | 88 |
|
| 116 |
server.save() |
|
| 89 |
new_owner = options.get('owner')
|
|
| 90 |
if new_owner is not None: |
|
| 91 |
if "@" in new_owner: |
|
| 92 |
raise CommandError("Invalid owner UUID.")
|
|
| 93 |
old_owner = server.userid |
|
| 94 |
server.userid = new_owner |
|
| 95 |
server.save() |
|
| 96 |
msg = "Changed the owner of server '%s' from '%s' to '%s'.\n" |
|
| 97 |
self.stdout.write(msg % (server, old_owner, new_owner)) |
|
Also available in: Unified diff