Changeset 65f3f29 for fedd/fedd_client.py

Timestamp:

Jul 25, 2009 10:52:45 AM (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:

ca489e8

Parents:

d27fd76

Message:

Added multi info functions and some helpers to fedd_client

File:

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

fedd/fedd_client.py

@@ -157 +157 @@
                 help="data to extract")
 
+class fedd_multi_exp_data_opts(fedd_client_opts):
+    def __init__(self):
+        fedd_client_opts.__init__(self)
+        self.add_option("--data", dest="data", default=[],
+                action="append", type="choice",
+                choices=("id", "federant", "vtopo", "vis", "log", "status"),
+                help="data to extract")
+
 class fedd_spew_opts(fedd_client_opts):
     def __init__(self):
@@ -353 +361 @@
             raise RuntimeError("No body in response??")
 
-# Querying experiment data follows the same control flow regardless of the
-# specific data retrieved.  This class encapsulates that control flow.
-class exp_data(fedd_rpc):
-    def __init__(self): 
+class exp_data_base(fedd_rpc):
+    def __init__(self, op='Info'):
         """
         Init the various conversions
         """
 
-        fedd_rpc.__init__(self, 'Info')
+        fedd_rpc.__init__(self, op)
         # List of things one could ask for and what formatting routine is
         # called.
@@ -374 +380 @@
             }
 
-
+    # Utility functions
     def print_string(self, d, out=sys.stdout):
         print >>out, d
@@ -403 +409 @@
             print >>out, str
 
+        
+
+# Querying experiment data follows the same control flow regardless of the
+# specific data retrieved.  This class encapsulates that control flow.
+class exp_data(exp_data_base):
+    def __init__(self): 
+        exp_data_base.__init__(self, 'Info')
+
     def __call__(self):
         """
@@ -495 +509 @@
         sys.argv.append('--data=status')
         exp_data.__call__(self)
+
+class multi_exp_data(exp_data_base):
+    def __init__(self): 
+        exp_data_base.__init__(self, 'MultiInfo')
+
+
+    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_multi_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)
+
+        req = { }
+
+        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:
+            exit_with_fault(\
+                    {'desc': e.desc, 'errstr': e.errstr, 'code': e.code})
+        except RuntimeError, e:
+            print e
+            sys.exit("Error processing RPC: %s" % e)
+
+        exps = resp_dict.get('info', [])
+        if exps:
+            print '---'
+            for exp in exps:
+                for d in opts.data:
+                    key, output = self.params[d]
+                    try:
+                        if exp.has_key(key):
+                            output(exp[key])
+                    except RuntimeError, e:
+                        sys.exit("Bad response. %s" % e.message)
+                print '---'
+
+
+class multi_status(exp_data_base):
+    def __init__(self): 
+        exp_data_base.__init__(self, 'MultiInfo')
+
+
+    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_client_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)
+
+        req = { }
+
+        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:
+            exit_with_fault(\
+                    {'desc': e.desc, 'errstr': e.errstr, 'code': e.code})
+        except RuntimeError, e:
+            print e
+            sys.exit("Error processing RPC: %s" % e)
+
+        for exp in resp_dict.get('info', []):
+            out = []
+            for eid in exp.get('experimentID', []):
+                if eid.has_key('localname'):
+                    out.append(eid['localname'])
+                    break
+            else:
+                out.append("")
+            for eid in exp.get('experimentID', []):
+                if eid.has_key('fedid'):
+                    out.append("%s" % eid['fedid'])
+                    break
+            else:
+                out.append("")
+
+            out.append(exp.get('experimentStatus', ""))
+
+            for f in exp.get('federant', []):
+                if f.get('master', False):
+                    em = f.get('emulab', None)
+                    if em:
+                        project = em.get('project', None)
+                        if project:
+                            tb = project.get('testbed', None)
+                            if tb and tb.has_key('localname'):
+                                out.append(tb['localname'])
+                            else:
+                                out.append("")
+                            pn = project.get('name', None)
+                            if pn and pn.has_key('localname'):
+                                out.append(pn['localname'])
+                            else:
+                                out.append("")
+                        else:
+                            out.extend(("", ""))
+                    else:
+                        out.extend(("", ""))
+                    break
+            else:
+                out.extend(("",""))
+
+            print ":".join(out)
 
 class image(fedd_rpc):
@@ -1302 +1465 @@
         'vis': vis(),\
         'info': exp_data(),\
+        'multiinfo': multi_exp_data(),\
+        'multistatus': multi_status(),\
         'image': image(),\
         'ns_image': ns_image(),\