Merge lp:~stylesen/lava-scheduler-tool/add-job-output-cmd into lp:lava-scheduler-tool

[Senthil Kumaran S]

Add new command to lava-scheduler-tool in order to download job output file.

[Antonio Terceiro (terceiro)]

Looks good to me

 review approve

review: Approve

=== modified file 'lava_scheduler_tool/commands.py'
--- lava_scheduler_tool/commands.py	2012-05-11 04:12:23 +0000
+++ lava_scheduler_tool/commands.py	2013-04-17 09:47:30 +0000
@@ -16,6 +16,9 @@
 # You should have received a copy of the GNU Lesser General Public License
 # along with lava-scheduler-tool.  If not, see <http://www.gnu.org/licenses/>.
 
+import os
+import sys
+import argparse
 import xmlrpclib
 
 from lava_tool.authtoken import AuthenticatingServerProxy, KeyringAuthBackend
@@ -88,3 +91,44 @@
         server = AuthenticatingServerProxy(
             self.args.SERVER, auth_backend=KeyringAuthBackend())
         server.scheduler.cancel_job(self.args.JOB_ID)
+
+
+class job_output(Command):
+    """
+    Get job output from the scheduler.
+    """
+
+    @classmethod
+    def register_arguments(cls, parser):
+        super(job_output, cls).register_arguments(parser)
+        parser.add_argument("SERVER")
+        parser.add_argument("JOB_ID",
+                            type=int,
+                            help="Job ID to download output file")
+        parser.add_argument("--overwrite",
+                            action="store_true",
+                            help="Overwrite files on the local disk")
+        parser.add_argument("--output", "-o",
+                            type=argparse.FileType("wb"),
+                            default=None,
+                            help="Alternate name of the output file")
+
+    def invoke(self):
+        if self.args.output is None:
+            filename = str(self.args.JOB_ID) + '_output.txt'
+            if os.path.exists(filename) and not self.args.overwrite:
+                print >> sys.stderr, "File {filename!r} already exists".format(
+                    filename=filename)
+                print >> sys.stderr, "You may pass --overwrite to write over it"
+                return -1
+            stream = open(filename, "wb")
+        else:
+            stream = self.args.output
+            filename = self.args.output.name
+
+        server = AuthenticatingServerProxy(
+            self.args.SERVER, auth_backend=KeyringAuthBackend())
+        stream.write(server.scheduler.job_output(self.args.JOB_ID).data)
+
+        print "Downloaded job output of {0} to file {1!r}".format(
+            self.args.JOB_ID, filename)
 
=== modified file 'setup.py'
--- setup.py	2012-04-04 05:22:32 +0000
+++ setup.py	2013-04-17 09:47:30 +0000
@@ -38,6 +38,7 @@
     submit-job = lava_scheduler_tool.commands:submit_job
     resubmit-job = lava_scheduler_tool.commands:resubmit_job
     cancel-job = lava_scheduler_tool.commands:cancel_job
+    job-output = lava_scheduler_tool.commands:job_output
     [lava_tool.commands]
     submit-job = lava_scheduler_tool.commands:submit_job
     resubmit-job = lava_scheduler_tool.commands:resubmit_job