LUSetInstanceParams: save cluster

[ganeti-local] / lib / cli.py

diff --git a/lib/cli.py b/lib/cli.py
index 124f602..281fb5c 100644 (file)
--- a/lib/cli.py
+++ b/lib/cli.py
@@ -51,7 +51,7 @@ __all__ = ["DEBUG_OPT", "NOHDR_OPT", "SEP_OPT", "GenericMain",
            "FormatError", "SplitNodeOption", "SubmitOrSend",
            "JobSubmittedException", "FormatTimestamp", "ParseTimespec",
            "ToStderr", "ToStdout", "UsesRPC",
            "FormatError", "SplitNodeOption", "SubmitOrSend",
            "JobSubmittedException", "FormatTimestamp", "ParseTimespec",
            "ToStderr", "ToStdout", "UsesRPC",

-           "GetOnlineNodes", "JobExecutor", "SYNC_OPT",
+           "GetOnlineNodes", "JobExecutor", "SYNC_OPT", "CONFIRM_OPT",

            ]
 
 
            ]
 
 

@@ -182,6 +182,9 @@ FIELDS_OPT = make_option("-o", "--output", dest="output", action="store",
 FORCE_OPT = make_option("-f", "--force", dest="force", action="store_true",
                         default=False, help="Force the operation")
 
 FORCE_OPT = make_option("-f", "--force", dest="force", action="store_true",
                         default=False, help="Force the operation")
 

+CONFIRM_OPT = make_option("--yes", dest="confirm", action="store_true",
+                          default=False, help="Do not require confirmation")
+

 TAG_SRC_OPT = make_option("--from", dest="tags_source",
                           default=None, help="File with tag names")
 
 TAG_SRC_OPT = make_option("--from", dest="tags_source",
                           default=None, help="File with tag names")
 

@@ -815,6 +818,8 @@ def GenerateTable(headers, fields, separator, data,
     format = separator.replace("%", "%%").join(format_fields)
 
   for row in data:
     format = separator.replace("%", "%%").join(format_fields)
 
   for row in data:

+    if row is None:
+      continue

     for idx, val in enumerate(row):
       if unitfields.Matches(fields[idx]):
         try:
     for idx, val in enumerate(row):
       if unitfields.Matches(fields[idx]):
         try:

@@ -840,6 +845,8 @@ def GenerateTable(headers, fields, separator, data,
 
   for line in data:
     args = []
 
   for line in data:
     args = []

+    if line is None:
+      line = ['-' for _ in fields]

     for idx in xrange(len(fields)):
       if separator is None:
         args.append(mlens[idx])
     for idx in xrange(len(fields)):
       if separator is None:
         args.append(mlens[idx])

@@ -856,7 +863,7 @@ def FormatTimestamp(ts):
   @param ts: a timeval-type timestamp, a tuple of seconds and microseconds
 
   @rtype: string
   @param ts: a timeval-type timestamp, a tuple of seconds and microseconds
 
   @rtype: string

-  @returns: a string with the formatted timestamp
+  @return: a string with the formatted timestamp

 
   """
   if not isinstance (ts, (tuple, list)) or len(ts) != 2:
 
   """
   if not isinstance (ts, (tuple, list)) or len(ts) != 2:

@@ -988,15 +995,24 @@ class JobExecutor(object):
       cl = GetClient()
     self.cl = cl
     self.verbose = verbose
       cl = GetClient()
     self.cl = cl
     self.verbose = verbose

+    self.jobs = []

 
   def QueueJob(self, name, *ops):
 
   def QueueJob(self, name, *ops):

-    """Submit a job for execution.
+    """Record a job for later submit.

 
     @type name: string
     @param name: a description of the job, will be used in WaitJobSet
     """
 
     @type name: string
     @param name: a description of the job, will be used in WaitJobSet
     """

-    job_id = SendJob(ops, cl=self.cl)
-    self.queue.append((job_id, name))
+    self.queue.append((name, ops))
+
+
+  def SubmitPending(self):
+    """Submit all pending jobs.
+
+    """
+    results = self.cl.SubmitManyJobs([row[1] for row in self.queue])
+    for ((status, data), (name, _)) in zip(results, self.queue):
+      self.jobs.append((status, data, name))

 
   def GetResults(self):
     """Wait for and return the results of all jobs.
 
   def GetResults(self):
     """Wait for and return the results of all jobs.

@@ -1007,10 +1023,18 @@ class JobExecutor(object):
         there will be the error message
 
     """
         there will be the error message
 
     """

+    if not self.jobs:
+      self.SubmitPending()

     results = []
     if self.verbose:
     results = []
     if self.verbose:

-      ToStdout("Submitted jobs %s", ", ".join(row[0] for row in self.queue))
-    for jid, name in self.queue:
+      ok_jobs = [row[1] for row in self.jobs if row[0]]
+      if ok_jobs:
+        ToStdout("Submitted jobs %s", ", ".join(ok_jobs))
+    for submit_status, jid, name in self.jobs:
+      if not submit_status:
+        ToStderr("Failed to submit job for %s: %s", name, jid)
+        results.append((False, jid))
+        continue

       if self.verbose:
         ToStdout("Waiting for job %s for %s...", jid, name)
       try:
       if self.verbose:
         ToStdout("Waiting for job %s for %s...", jid, name)
       try:

@@ -1035,5 +1059,10 @@ class JobExecutor(object):
     if wait:
       return self.GetResults()
     else:
     if wait:
       return self.GetResults()
     else:

-      for jid, name in self.queue:
-        ToStdout("%s: %s", jid, name)
+      if not self.jobs:
+        self.SubmitPending()
+      for status, result, name in self.jobs:
+        if status:
+          ToStdout("%s: %s", result, name)
+        else:
+          ToStderr("Failure for %s: %s", name, result)