Changeset f00fb7d for fedd

Timestamp:

May 17, 2010 5:39:01 AM (15 years ago)

Author:

Ted Faber <faber@…>

Branches:

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

Children:

2e0a952

Parents:

eab6ae1

Message:

Add a call to a ns2 -> topdl converter

File:

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

fedd/fedd_client.py

@@ -220 +220 @@
         self.add_option("--file", dest="file", 
                 help="experiment description file")
+
+class fedd_ns_topdl_opts(fedd_client_opts):
+    def __init__(self):
+        fedd_client_opts.__init__(self)
+        self.add_option("--file", dest="file", 
+                help="experiment description file")
+        self.add_option("--output", dest="outfile", type="string",
+                help="output topdl file")
 
 def exit_with_fault(dict, out=sys.stderr):
@@ -1844 +1852 @@
             print resp_dict
 
+class ns_topdl(fedd_rpc):
+    def __init__(self): 
+        fedd_rpc.__init__(self)
+    def __call__(self):
+        # Process the options using the customized option parser defined above
+        parser = fedd_ns_topdl_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.file:
+            try:
+                f = open(opts.file,"r")
+                contents = "".join(f)
+                f.close()
+            except EnvironmentError, e:
+                sys.exit("Can't read %s: %s" % (opts.file, e))
+        else:
+            sys.exit("Must specify an experiment description (--file)")
+
+        msg = { 'description': { 'ns2description': contents }, }
+
+        if opts.debug > 1: print >>sys.stderr, msg
+
+        try:
+            resp_dict = self.do_rpc(msg, 
+                    opts.url, opts.transport, cert, opts.trusted, 
+                    serialize_only=opts.serialize_only,
+                    tracefile=opts.tracefile,
+                    caller=self.caller('Ns2Topdl'))
+        except self.RPCException, e:
+            exit_with_fault(\
+                    {'desc': e.desc, 'errstr': e.errstr, 'code': e.code})
+        except RuntimeError, e:
+            sys.exit("Error processing RPC: %s" % e)
+
+        if 'experimentdescription' in resp_dict:
+            if 'topdldescription' in resp_dict['experimentdescription']:
+                exp = resp_dict['experimentdescription']['topdldescription']
+                top = topdl.Topology(**exp)
+            else:
+                sys.exit("Bad response: could not translate")
+        elif opts.serialze_only:
+            sys.exit(0)
+        else:
+            sys.exit("Bad response. %s" % e.message)
+
+        if opts.outfile:
+            try:
+                f = open(opts.outfile, "w")
+                print >>f, topdl.topology_to_xml(top, top="experiment")
+                f.close()
+            except EnvironmentError, e:
+                sys.exit("Can't write to %s: %s" % (opts.outfile, e))
+        else:
+            print topdl.topology_to_xml(top, top="experiment")
+
 cmds = {\
         'new': new(),\
@@ -1863 +1945 @@
         'start_segment': start_segment(),\
         'terminate_segment': terminate_segment(),\
+        'ns_topdl': ns_topdl(),\
     }
 if len(sys.argv) > 1: