From 597cadcc35abc2aa3b748355408159ad60778106 Mon Sep 17 00:00:00 2001
From: Seblu <sebastien.luttringer@smartjog.com>
Date: Wed, 2 Mar 2011 19:44:13 +0100
Subject: [PATCH] Add migrate command
---
cccli/command/__init__.py | 1 +
cccli/command/migrate.py | 83 +++++++++++++++++++++++++++++++++++++++
2 files changed, 84 insertions(+)
create mode 100644 cccli/command/migrate.py
diff --git a/cccli/command/__init__.py b/cccli/command/__init__.py
index 4a9e2fa..a04b164 100644
--- a/cccli/command/__init__.py
+++ b/cccli/command/__init__.py
@@ -20,6 +20,7 @@ from cccli.command.expert import Command_expert
from cccli.command.jobs import Command_jobs
from cccli.command.kill import Command_kill
from cccli.command.list import Command_list
+from cccli.command.migrate import Command_migrate
from cccli.command.server import Command_server
from cccli.command.shutdown import Command_shutdown
from cccli.command.tagdisplay import Command_tagdisplay
diff --git a/cccli/command/migrate.py b/cccli/command/migrate.py
new file mode 100644
index 0000000..8de9c4c
--- /dev/null
+++ b/cccli/command/migrate.py
@@ -0,0 +1,83 @@
+#!/usr/bin/env python
+#coding=utf8
+
+'''
+CloudControl jobs command
+'''
+
+from cccli.exception import *
+from sjrpc.core.exceptions import *
+from cccli.printer import Printer, color
+from cccli.command.command import TqlCommand
+
+class Command_migrate(TqlCommand):
+ '''Migrate vm'''
+
+ def __init__(self, cli, argv0):
+ TqlCommand.__init__(self, cli, argv0)
+ self.set_usage("%prog [options] [source tql] [dest tql]")
+ self.remove_option("--direct")
+ self.remove_option("--raw")
+ self.remove_option("--print-tql")
+ self.add_option("-l", "--list", action="store_true", dest="list", default=False,
+ help="List migration types and algo")
+ self.add_option("-t", "--type", action="store", dest="type", default="",
+ help="Selection migration type")
+ self.add_option("-a", "--algo", action="store", dest="algo", default="",
+ help="Select migration algorithm")
+
+ def __call__(self, argv):
+ # Parse argline
+ self.parse_args(argv)
+ # Retrieve election types
+ self.etypes = self.get_electiontypes()
+ # Check args and do listings
+ if self.options.list:
+ self.list()
+ return
+ elif self.options.type not in self.etypes:
+ raise cmdBadArgument("No such type: %s"%self.options.type)
+ elif self.options.algo not in self.etypes[self.options.type]:
+ raise cmdBadArgument("No such algo: %s for type: %s"%(self.options.algo, self.options.type))
+ elif len(self.args) != 2:
+ raise cmdBadArgument()
+ stql = self.args[0]
+ dtql = self.args[1]
+ # election(query_vm, query_dest, mtype='cold', algo='fair', **kwargs)
+ scrutin = self.rpccall("election", stql, dtql,
+ mtype=self.options.type,
+ algo=self.options.algo,
+ _direct=True, _status=False)
+ # check election result
+ if len(scrutin) == 0:
+ raise cmdError("No migration plan found")
+ # print election
+ for (i,o) in enumerate(scrutin):
+ if self.options.index:
+ self.printer.out("[%d] "%i, nl="")
+ self.printer.out("%s%s %s-> %s%s%s (%s)"%(
+ self.tdc("id"),
+ o["sid"],
+ color["reset"],
+ self.tdc("id"),
+ o["did"],
+ color["reset"],
+ o["type"]))
+ # ask confirmation
+ if self.printer.ask("Do you confirm election? (Yes baby) ") != "Yes baby":
+ raise cmdWarning("User resign")
+ # run migration
+ self.rpccall("migrate", scrutin, _direct=True)
+
+ def get_electiontypes(self):
+ '''Return a list of migration type'''
+ try:
+ return self.cli.rpc.call("electiontypes")
+ except RpcError as e:
+ raise cmdError(e)
+
+ def list(self):
+ '''Print a list of migration type'''
+ self.printer.out("Migration types: Election Algo")
+ for t in self.etypes.keys():
+ self.printer.out("%s: %s"%(t, ",".join(self.etypes[t])))
--
GitLab