X-Git-Url: http://git.onelab.eu/?a=blobdiff_plain;f=wsdl%2Fsfa2wsdl.py;h=2ec130722d2ebde753f8c858e7bdcc53f932a08e;hb=ecc85e0b923922cf7117d29b380f5284edb88f21;hp=4ee12570cb250552bc548f5f5fd700e80b6d1d18;hpb=fc4bc6f03e7737532d8cc5f96fbdf549609e9a26;p=sfa.git diff --git a/wsdl/sfa2wsdl.py b/wsdl/sfa2wsdl.py index 4ee12570..2ec13072 100755 --- a/wsdl/sfa2wsdl.py +++ b/wsdl/sfa2wsdl.py @@ -2,29 +2,22 @@ # # Sapan Bhatia # -# Generates a WSDL for geniwrapper +# Generates a WSDL for sfa import os, sys import time import pdb import xml.dom.minidom -import xml.dom.ext import apistub import inspect from types import * from optparse import OptionParser -from sfa.trust.auth import Auth -from sfa.util.parameter import Parameter,Mixed +from sfa.storage.parameter import Parameter, Mixed -import globals - -complex_types = {} -services = {} - -num_types = 0 +plc_ns="http://www.planet-lab.org/sfa" class SoapError(Exception): def __init__(self, value): @@ -37,272 +30,307 @@ except NameError: from sets import Set set = Set -def filter_argname(argname): - global interface_options - if (not interface_options.lite or (argname!="cred")): - if (argname.find('(') != -1): - # The name has documentation in it :-/ - brackright = argname.split('(')[1] - if (brackright.find(')') == -1): - raise Exception("Please fix the argument %s to be well-formed.\n"%argname) - inbrack = brackright.split(')')[0] - argname = inbrack - return argname - -def fold_complex_type_names(acc, arg): - name = arg.doc - if (type(acc)==list): - acc.append(name) - else: - p_i_b = acc.doc - acc = [p_i_b,name] - return acc - -def fold_complex_type(acc, arg): - global complex_types - name = name_complex_type(arg) - complex_types[arg]=name - if (type(acc)==list): - acc.append(name) - else: - p_i_b = name_complex_type(acc) - acc = [p_i_b,name] - return acc - -def name_complex_type(arg): - global num_types - global types - - types_section = types.getElementsByTagName("xsd:schema")[0] - - #pdb.set_trace() - if (isinstance(arg, Mixed)): - inner_types = reduce(fold_complex_type, arg) - inner_names = reduce(fold_complex_type_names, arg) - if (inner_types[-1]=="none"): - inner_types=inner_types[:-1] - min_args = 0 - else: - min_args = 1 - - num_types=num_types+1 - type_name = "Type%d"%num_types - complex_type = types_section.appendChild(types.createElement("xsd:complexType")) - complex_type.setAttribute("name", type_name) - - choice = complex_type.appendChild(types.createElement("xsd:choice")) - for n,t in zip(inner_names,inner_types): - element = choice.appendChild(types.createElement("element")) - n = filter_argname(n) - element.setAttribute("name", n) - element.setAttribute("type", "%s"%t) - element.setAttribute("minOccurs","%d"%min_args) - return "xsdl:%s"%type_name - elif (isinstance(arg, Parameter)): - return (name_simple_type(arg.type)) - elif type(arg) == ListType or type(arg) == TupleType: - inner_type = name_complex_type(arg[0]) - num_types=num_types+1 - type_name = "Type%d"%num_types - complex_type = types_section.appendChild(types.createElement("xsd:complexType")) - type_name = filter_argname(type_name) - complex_type.setAttribute("name", type_name) - complex_content = complex_type.appendChild(types.createElement("xsd:complexContent")) - restriction = complex_content.appendChild(types.createElement("xsd:restriction")) - restriction.setAttribute("base","soapenc:Array") - attribute = restriction.appendChild(types.createElement("xsd:attribute")) - attribute.setAttribute("ref","soapenc:arrayType") - attribute.setAttribute("wsdl:arrayType","%s[]"%inner_type) - - return "xsdl:%s"%type_name - - elif type(arg) == DictType or arg == DictType or (inspect.isclass(arg) and issubclass(arg, dict)): - num_types=num_types+1 - type_name = "Type%d"%num_types - complex_type = types_section.appendChild(types.createElement("xsd:complexType")) - type_name = filter_argname(type_name) - complex_type.setAttribute("name", type_name) - complex_content = complex_type.appendChild(types.createElement("xsd:sequence")) - - for k in arg.fields: - inner_type = name_complex_type(arg.fields[k]) - element=complex_content.appendChild(types.createElement("xsd:element")) - element.setAttribute("name",k) - element.setAttribute("type",inner_type) - - return "xsdl:%s"%type_name - else: - return (name_simple_type(arg)) - -def name_simple_type(arg_type): - # A Parameter is reported as an instance, even though it is serialized as a type <> - if type(arg_type) == InstanceType: - return (name_simple_type(arg_type.type)) - if arg_type == None: - return "none" - if arg_type == DictType: - return "xsd:anyType" - elif arg_type == IntType or arg_type == LongType: - return "xsd:int" - elif arg_type == bool: - return "xsd:boolean" - elif arg_type == FloatType: - return "xsd:double" - elif arg_type in StringTypes: - return "xsd:string" - else: - pdb.set_trace() - raise SoapError, "Cannot handle %s objects" % arg_type - -def param_type(arg): - return (name_complex_type(arg)) - -def add_wsdl_ports_and_bindings (wsdl): - for method in apistub.methods: - - # Skip system. methods - if "system." in method: - continue - - function = apistub.callable(method) # Commented documentation - #lines = ["// " + line.strip() for line in function.__doc__.strip().split("\n")] - #print "\n".join(lines) - #print - - - in_el = wsdl.firstChild.appendChild(wsdl.createElement("message")) - in_el.setAttribute("name", method + "_in") - - for service_name in function.interfaces: - if (services.has_key(service_name)): - if (not method in services[service_name]): - services[service_name].append(method) +class WSDLGen: + complex_types = {} + services = {} + num_types = 0 + wsdl = None + types = None + + def __init__(self, interface_options): + self.interface_options = interface_options + + def interface_name (self): + if self.interface_options.aggregate and \ + self.interface_options.slicemgr and \ + self.interface_options.registry: + return "complete" + if self.interface_options.aggregate: return "aggregate" + elif self.interface_options.slicemgr: return "slicemgr" + elif self.interface_options.registry: return "registry" + elif self.interface_options.component: return "component" + else: return "unknown" + + def filter_argname(self,argname): + if (not self.interface_options.lite or (argname!="cred")): + if (argname.find('(') != -1): + # The name has documentation in it :-/ + brackright = argname.split('(')[1] + if (brackright.find(')') == -1): + raise Exception("Please fix the argument %s to be well-formed.\n"%argname) + inbrack = brackright.split(')')[0] + argname = inbrack + return argname + +# def fold_complex_type_names(self,acc, arg): +# name = arg.doc +# if (type(acc)==list): +# acc.append(name) +# else: +# p_i_b = acc.doc +# acc = [p_i_b,name] +# return acc +# +# def fold_complex_type(self,acc, arg): +# name = self.name_complex_type(arg) +# self.complex_types[arg]=name +# if (type(acc)==list): +# acc.append(name) +# else: +# p_i_b = self.name_complex_type(acc) +# acc = [p_i_b,name] +# return acc + + def name_complex_type(self,arg): + + types_section = self.types.getElementsByTagName("xsd:schema")[0] + + #pdb.set_trace() + if (isinstance(arg, Mixed)): +# inner_types = reduce(self.fold_complex_type, arg) +# inner_names = reduce(self.fold_complex_type_names, arg) + inner_types = [ self.name_complex_type(x) for x in arg ] + inner_names = [ x.doc for x in arg ] + if (inner_types[-1]=="none"): + inner_types=inner_types[:-1] + min_args = 0 else: - services[service_name]=[method] - - # Arguments - - if (function.accepts): - (min_args, max_args, defaults) = function.args() - for (argname,argtype) in zip(max_args,function.accepts): - argname = filter_argname(argname) - arg_part = in_el.appendChild(wsdl.createElement("part")) - arg_part.setAttribute("name", argname) - arg_part.setAttribute("type", param_type(argtype)) - - # Return type - return_type = function.returns - out_el = wsdl.firstChild.appendChild(wsdl.createElement("message")) - out_el.setAttribute("name", method + "_out") - ret_part = out_el.appendChild(wsdl.createElement("part")) - ret_part.setAttribute("name", "Result") - ret_part.setAttribute("type", param_type(return_type)) - - # Port connecting arguments with return type - - port_el = wsdl.firstChild.appendChild(wsdl.createElement("portType")) - port_el.setAttribute("name", method + "_port") - - op_el = port_el.appendChild(wsdl.createElement("operation")) - op_el.setAttribute("name", method) - inp_el=wsdl.createElement("input") - inp_el.setAttribute("message","tns:" + method + "_in") - inp_el.setAttribute("name",method+"_request") - op_el.appendChild(inp_el) - out_el = wsdl.createElement("output") - out_el.setAttribute("message","tns:" + method + "_out") - out_el.setAttribute("name",method+"_response") - op_el.appendChild(out_el) - - # Bindings - - bind_el = wsdl.firstChild.appendChild(wsdl.createElement("binding")) - bind_el.setAttribute("name", method + "_binding") - bind_el.setAttribute("type", "tns:" + method + "_port") - - soap_bind = bind_el.appendChild(wsdl.createElement("soap:binding")) - soap_bind.setAttribute("style", "rpc") - soap_bind.setAttribute("transport","http://schemas.xmlsoap.org/soap/http") - + min_args = 1 - wsdl_op = bind_el.appendChild(wsdl.createElement("operation")) - wsdl_op.setAttribute("name", method) - wsdl_op.appendChild(wsdl.createElement("soap:operation")).setAttribute("soapAction", - "urn:" + method) - + self.num_types += 1 + type_name = "Type%d"%self.num_types + complex_type = types_section.appendChild(self.types.createElement("xsd:complexType")) + complex_type.setAttribute("name", type_name) + + choice = complex_type.appendChild(self.types.createElement("xsd:choice")) + for (n,t) in zip(inner_names,inner_types): + element = choice.appendChild(self.types.createElement("element")) + n = self.filter_argname(n) + element.setAttribute("name", n) + element.setAttribute("type", "%s"%t) + element.setAttribute("minOccurs","%d"%min_args) + return "xsdl:%s"%type_name + elif (isinstance(arg, Parameter)): + return (self.name_simple_type(arg.type)) + elif type(arg) in ( ListType , TupleType ): + inner_type = self.name_complex_type(arg[0]) + self.num_types=self.num_types+1 + type_name = "Type%d"%self.num_types + complex_type = types_section.appendChild(self.types.createElement("xsd:complexType")) + type_name = self.filter_argname(type_name) + complex_type.setAttribute("name", type_name) + complex_content = complex_type.appendChild(self.types.createElement("xsd:complexContent")) + restriction = complex_content.appendChild(self.types.createElement("xsd:restriction")) + restriction.setAttribute("base","soapenc:Array") + attribute = restriction.appendChild(self.types.createElement("xsd:attribute")) + attribute.setAttribute("ref","soapenc:arrayType") + attribute.setAttribute("wsdl:arrayType","%s[]"%inner_type) + + return "xsdl:%s"%type_name + + elif type(arg) == DictType or arg == DictType or (inspect.isclass(arg) and issubclass(arg, dict)): + self.num_types=self.num_types+1 + type_name = "Type%d"%self.num_types + complex_type = types_section.appendChild(self.types.createElement("xsd:complexType")) + type_name = self.filter_argname(type_name) + complex_type.setAttribute("name", type_name) + complex_content = complex_type.appendChild(self.types.createElement("xsd:sequence")) + + for k in arg.fields: + inner_type = self.name_complex_type(arg.fields[k]) + element=complex_content.appendChild(self.types.createElement("xsd:element")) + element.setAttribute("name",k) + element.setAttribute("type",inner_type) + + return "xsdl:%s"%type_name + else: + return (self.name_simple_type(arg)) + + def name_simple_type(self,arg_type): + # A Parameter is reported as an instance, even though it is serialized as a type <> + if type(arg_type) == InstanceType: + return (self.name_simple_type(arg_type.type)) + if arg_type == None: + return "none" + if arg_type == DictType: + return "xsd:anyType" + if arg_type in (ListType, TupleType): + return "xsd:arrayType" + elif arg_type == IntType or arg_type == LongType: + return "xsd:int" + elif arg_type == bool: + return "xsd:boolean" + elif arg_type == FloatType: + return "xsd:double" + elif arg_type in StringTypes: + return "xsd:string" + else: + pdb.set_trace() + raise SoapError("Cannot handle %s objects" % arg_type) + + def param_type(self, arg): + return (self.name_complex_type(arg)) + + def add_wsdl_ports_and_bindings (self): + for method in apistub.methods: + + # Skip system. methods + if "system." in method: + continue + + function = apistub.callable(method) # Commented documentation + #lines = ["// " + line.strip() for line in function.__doc__.strip().split("\n")] + #print "\n".join(lines) + #print + + + in_el = self.wsdl.lastChild.appendChild(self.wsdl.createElement("message")) + in_el.setAttribute("name", method + "_in") + + for service_name in function.interfaces: + if (self.services.has_key(service_name)): + if (not method in self.services[service_name]): + self.services[service_name].append(method) + else: + self.services[service_name]=[method] + + # Arguments + + if (function.accepts): + (min_args, max_args, defaults) = function.args() + for (argname,argtype) in zip(max_args,function.accepts): + argname = self.filter_argname(argname) + arg_part = in_el.appendChild(self.wsdl.createElement("part")) + arg_part.setAttribute("name", argname) + arg_part.setAttribute("type", self.param_type(argtype)) + + # Return type + return_type = function.returns + out_el = self.wsdl.lastChild.appendChild(self.wsdl.createElement("message")) + out_el.setAttribute("name", method + "_out") + ret_part = out_el.appendChild(self.wsdl.createElement("part")) + ret_part.setAttribute("name", "Result") + ret_part.setAttribute("type", self.param_type(return_type)) + + # Port connecting arguments with return type + + port_el = self.wsdl.lastChild.appendChild(self.wsdl.createElement("portType")) + port_el.setAttribute("name", method + "_port") + + op_el = port_el.appendChild(self.wsdl.createElement("operation")) + op_el.setAttribute("name", method) + inp_el=self.wsdl.createElement("input") + inp_el.setAttribute("message","tns:" + method + "_in") + inp_el.setAttribute("name",method+"_request") + op_el.appendChild(inp_el) + out_el = self.wsdl.createElement("output") + out_el.setAttribute("message","tns:" + method + "_out") + out_el.setAttribute("name",method+"_response") + op_el.appendChild(out_el) + + # Bindings + + bind_el = self.wsdl.lastChild.appendChild(self.wsdl.createElement("binding")) + bind_el.setAttribute("name", method + "_binding") + bind_el.setAttribute("type", "tns:" + method + "_port") + + soap_bind = bind_el.appendChild(self.wsdl.createElement("soap:binding")) + soap_bind.setAttribute("style", "rpc") + soap_bind.setAttribute("transport","http://schemas.xmlsoap.org/soap/http") + + + wsdl_op = bind_el.appendChild(self.wsdl.createElement("operation")) + wsdl_op.setAttribute("name", method) + wsdl_op.appendChild(self.wsdl.createElement("soap:operation")).setAttribute("soapAction", + "urn:" + method) + + + wsdl_input = wsdl_op.appendChild(self.wsdl.createElement("input")) + input_soap_body = wsdl_input.appendChild(self.wsdl.createElement("soap:body")) + input_soap_body.setAttribute("use", "encoded") + input_soap_body.setAttribute("namespace", "urn:" + method) + input_soap_body.setAttribute("encodingStyle","http://schemas.xmlsoap.org/soap/encoding/") + + + wsdl_output = wsdl_op.appendChild(self.wsdl.createElement("output")) + output_soap_body = wsdl_output.appendChild(self.wsdl.createElement("soap:body")) + output_soap_body.setAttribute("use", "encoded") + output_soap_body.setAttribute("namespace", "urn:" + method) + output_soap_body.setAttribute("encodingStyle","http://schemas.xmlsoap.org/soap/encoding/") + + + def add_wsdl_services(self): + for service in self.services.keys(): + if (getattr(self.interface_options,service)): + service_el = self.wsdl.lastChild.appendChild(self.wsdl.createElement("service")) + service_el.setAttribute("name", service) + + for method in self.services[service]: + name=method + servport_el = service_el.appendChild(self.wsdl.createElement("port")) + servport_el.setAttribute("name", name + "_port") + servport_el.setAttribute("binding", "tns:" + name + "_binding") + + soapaddress = servport_el.appendChild(self.wsdl.createElement("soap:address")) + soapaddress.setAttribute("location", "%s/%s" % (plc_ns,service)) + + + def compute_wsdl_definitions(self): + wsdl_text_header = """ + + + """ % (self.interface_name(),plc_ns,plc_ns,plc_ns) + + self.wsdl = xml.dom.minidom.parseString(wsdl_text_header) - wsdl_input = wsdl_op.appendChild(wsdl.createElement("input")) - input_soap_body = wsdl_input.appendChild(wsdl.createElement("soap:body")) - input_soap_body.setAttribute("use", "encoded") - input_soap_body.setAttribute("namespace", "urn:" + method) - input_soap_body.setAttribute("encodingStyle","http://schemas.xmlsoap.org/soap/encoding/") - - wsdl_output = wsdl_op.appendChild(wsdl.createElement("output")) - output_soap_body = wsdl_output.appendChild(wsdl.createElement("soap:body")) - output_soap_body.setAttribute("use", "encoded") - output_soap_body.setAttribute("namespace", "urn:" + method) - output_soap_body.setAttribute("encodingStyle","http://schemas.xmlsoap.org/soap/encoding/") + def compute_wsdl_definitions_and_types(self): + wsdl_text_header = """ + + + + + + """ % (self.interface_name(),plc_ns, plc_ns, plc_ns, plc_ns) + self.types = xml.dom.minidom.parseString(wsdl_text_header) -def add_wsdl_service(wsdl): - for service in services.keys(): - global interface_options - if (getattr(interface_options,service)): - service_el = wsdl.firstChild.appendChild(wsdl.createElement("service")) - service_el.setAttribute("name", service) + def add_wsdl_types(self): + wsdl_types = self.wsdl.importNode(self.types.getElementsByTagName("types")[0], True) + self.wsdl.lastChild.appendChild(wsdl_types) - for method in services[service]: - name=method - servport_el = service_el.appendChild(wsdl.createElement("port")) - servport_el.setAttribute("name", name + "_port") - servport_el.setAttribute("binding", "tns:" + name + "_binding") + def generate_wsdl(self): + self.compute_wsdl_definitions_and_types() + self.compute_wsdl_definitions() + self.add_wsdl_ports_and_bindings() + self.add_wsdl_types() + self.add_wsdl_services() - soapaddress = servport_el.appendChild(wsdl.createElement("soap:address")) - soapaddress.setAttribute("location", "%s/%s" % (globals.plc_ns,service)) - - -def get_wsdl_definitions(): - wsdl_text_header = """ - - """ % (globals.plc_ns,globals.plc_ns,globals.plc_ns) - - wsdl = xml.dom.minidom.parseString(wsdl_text_header) - - return wsdl - -def get_wsdl_definitions_and_types(): - wsdl_text_header = """ - - - - - """ % (globals.plc_ns, globals.plc_ns, globals.plc_ns, globals.plc_ns) - wsdl = xml.dom.minidom.parseString(wsdl_text_header) - return wsdl + def pretty_print(self): + if (self.wsdl): + print xml.dom.minidom.Document.toprettyxml(self.wsdl) + else: + raise Exception("Empty WSDL") def main(): - global types - global interface_options - parser = OptionParser() parser.add_option("-r", "--registry", dest="registry", action="store_true", help="Generate registry.wsdl", metavar="FILE") @@ -311,18 +339,18 @@ def main(): help="Generate sm.wsdl") parser.add_option("-a", "--aggregate", action="store_true", dest="aggregate", help="Generate am.wsdl") + parser.add_option("-c", "--component", action="store_true", dest="component", + help="Generate cm.wsdl") + parser.add_option("-g", "--geni-aggregate", action="store_true", dest="geni_am", + help="Generate gm.wsdl") parser.add_option("-l", "--lite", action="store_true", dest="lite", help="Generate LITE version of the interface, in which calls exclude credentials") (interface_options, args) = parser.parse_args() - types = get_wsdl_definitions_and_types() - wsdl = get_wsdl_definitions() - add_wsdl_ports_and_bindings(wsdl) - wsdl_types = wsdl.importNode(types.getElementsByTagName("types")[0], True) - wsdl.firstChild.appendChild(wsdl_types) - add_wsdl_service(wsdl) - - xml.dom.ext.PrettyPrint(wsdl) + gen = WSDLGen(interface_options) + gen.generate_wsdl() + gen.pretty_print() + if __name__ == "__main__": main()