Changeset f76d3d7 for fedd/fedd_client.py

Timestamp:

Jun 30, 2009 6:09:32 PM (15 years ago)

Author:

Ted Faber <faber@…>

Branches:

axis_example, compt_changes, info-ops, master, version-1.30, version-2.00, version-3.01, version-3.02

Children:

7a1918e

Parents:

b57df44

Message:

add status command

File:

• fedd/fedd_client.py (modified) (2 diffs)

fedd/fedd_client.py

@@ -425 +425 @@
 
 
+# Querying experiment data follows the same control flow regardless of the
+# specific data retrieved.  This class encapsulates that control flow.
+class simple_info(fedd_rpc):
+    def __init__(self): 
+        """
+        Specialize the class for the type of data requested (op)
+        """
+
+        fedd_rpc.__init__(self, 'Info')
+
+    def print_xml(self, d, out=sys.stdout):
+        """
+        Print the retrieved data is a simple xml representation of the dict.
+        """
+        str = "<%s>\n" % self.xml
+        for t in ('node', 'lan'):
+            if d.has_key(t): 
+                for x in d[t]:
+                    str += "<%s>" % t
+                    for k in x.keys():
+                        str += "<%s>%s</%s>" % (k, x[k],k)
+                    str += "</%s>\n" % t
+        str+= "</%s>" % self.xml
+        print >>out, str
+
+    def __call__(self):
+        """
+        The control flow.  Compose the request and print the response.
+        """
+        # Process the options using the customized option parser defined above
+        parser = fedd_exp_data_opts()
+
+        (opts, args) = parser.parse_args()
+
+        if opts.trusted:
+            if ( not os.access(opts.trusted, os.R_OK) ) :
+                sys.exit("Cannot read trusted certificates (%s)" % opts.trusted)
+
+        if opts.debug > 0: opts.tracefile=sys.stderr
+
+        (user, cert) = self.get_user_info([])
+
+        if opts.cert != None: cert = opts.cert
+
+        if cert == None:
+            sys.exit("No certificate given (--cert) or found")
+
+        if os.access(cert, os.R_OK):
+            fid = fedid(file=cert)
+        else:
+            sys.exit("Cannot read certificate (%s)" % cert)
+
+        if opts.exp_name and opts.exp_certfile:
+            sys.exit("Only one of --experiment_cert and " +\
+                    "--experiment_name permitted");
+
+        if opts.exp_certfile:
+            exp_id = { 'fedid': fedid(file=opts.exp_certfile) }
+
+        if opts.exp_name:
+            exp_id = { 'localname' : opts.exp_name }
+
+        req = { 'experiment': exp_id }
+
+        status = "active"
+        try:
+            resp_dict = self.do_rpc(req,
+                    opts.url, opts.transport, cert, opts.trusted, 
+                    serialize_only=opts.serialize_only,
+                    tracefile=opts.tracefile)
+        except self.RPCException, e:
+            if e.code == 5:
+                status = "swapping"
+            elif e.code == 1:
+                status = "inactive"
+            else:
+                status = "unknown"
+        except RuntimeError, e:
+            print e
+            sys.exit("Error processing RPC: %s" % e)
+
+        print status
+
+
 class image(fedd_rpc):
     def __init__(self, op): 
@@ -1125 +1209 @@
         'ns_image': ns_image('Ns2Split'),\
         'info': exp_data('Info'),\
+        'status': simple_info(),\
         'terminate': terminate(),\
     }