3 # Sapan Bhatia <sapanb@cs.princeton.edu>
5 # Generates a WSDL for geniwrapper
11 import xml.dom.minidom
17 from optparse import OptionParser
19 from sfa.trust.auth import Auth
20 from sfa.util.parameter import Parameter,Mixed
29 class SoapError(Exception):
30 def __init__(self, value):
33 return repr(self.value)
40 def fold_complex_type_names(acc, arg):
45 python_is_braindead = acc.doc
46 acc = [python_is_braindead,name]
49 def fold_complex_type(acc, arg):
51 name = name_complex_type(arg)
52 complex_types[arg]=name
56 python_is_braindead = name_complex_type(acc)
57 acc = [python_is_braindead,name]
60 def name_complex_type(arg):
64 types_section = types.getElementsByTagName("xsd:schema")[0]
67 if (isinstance(arg, Mixed)):
68 inner_types = reduce(fold_complex_type, arg)
69 inner_names = reduce(fold_complex_type_names, arg)
70 if (inner_types[-1]=="none"):
71 inner_types=inner_types[:-1]
77 type_name = "Type%d"%num_types
78 complex_type = types_section.appendChild(types.createElement("xsd:complexType"))
79 complex_type.setAttribute("name", type_name)
81 choice = complex_type.appendChild(types.createElement("xsd:choice"))
82 for n,t in zip(inner_names,inner_types):
83 element = choice.appendChild(types.createElement("element"))
84 element.setAttribute("name", n)
85 element.setAttribute("type", "%s"%t)
86 element.setAttribute("minOccurs","%d"%min_args)
87 return "xsdl:%s"%type_name
88 elif (isinstance(arg, Parameter)):
89 return (name_simple_type(arg.type))
90 elif type(arg) == ListType or type(arg) == TupleType:
91 inner_type = name_complex_type(arg[0])
93 type_name = "Type%d"%num_types
94 complex_type = types_section.appendChild(types.createElement("xsd:simpleType"))
95 complex_type.setAttribute("name", type_name)
96 complex_content = complex_type.appendChild(types.createElement("xsd:list"))
97 complex_content.setAttribute("itemType",inner_type)
98 return "xsdl:%s"%type_name
99 elif type(arg) == DictType or arg == DictType or (inspect.isclass(arg) and issubclass(arg, dict)):
100 num_types=num_types+1
101 type_name = "Type%d"%num_types
102 complex_type = types_section.appendChild(types.createElement("xsd:complexType"))
103 complex_type.setAttribute("name", type_name)
104 complex_content = complex_type.appendChild(types.createElement("xsd:sequence"))
107 inner_type = name_complex_type(arg.fields[k])
108 element=complex_content.appendChild(types.createElement("xsd:element"))
109 element.setAttribute("name",k)
110 element.setAttribute("type",inner_type)
112 return "xsdl:%s"%type_name
114 return (name_simple_type(arg))
116 def name_simple_type(arg_type):
117 # A Parameter is reported as an instance, even though it is serialized as a type <>
118 if type(arg_type) == InstanceType:
119 return (name_simple_type(arg_type.type))
122 if arg_type == DictType:
124 elif arg_type == IntType or arg_type == LongType:
126 elif arg_type == bool:
128 elif arg_type == FloatType:
130 elif arg_type in StringTypes:
134 raise SoapError, "Cannot handle %s objects" % arg_type
137 return (name_complex_type(arg))
139 def add_wsdl_ports_and_bindings (wsdl):
140 for method in apistub.methods:
142 # Skip system. methods
143 if "system." in method:
146 function = apistub.callable(method) # Commented documentation
147 #lines = ["// " + line.strip() for line in function.__doc__.strip().split("\n")]
148 #print "\n".join(lines)
152 in_el = wsdl.firstChild.appendChild(wsdl.createElement("message"))
153 in_el.setAttribute("name", method + "_in")
155 for service_name in function.interfaces:
156 if (services.has_key(service_name)):
157 if (not method in services[service_name]):
158 services[service_name].append(method)
160 services[service_name]=[method]
164 if (function.accepts):
165 (min_args, max_args, defaults) = function.args()
166 for (argname,argtype) in zip(max_args,function.accepts):
167 arg_part = in_el.appendChild(wsdl.createElement("part"))
168 arg_part.setAttribute("name", argname)
169 arg_part.setAttribute("type", param_type(argtype))
172 return_type = function.returns
173 out_el = wsdl.firstChild.appendChild(wsdl.createElement("message"))
174 out_el.setAttribute("name", method + "_out")
175 ret_part = out_el.appendChild(wsdl.createElement("part"))
176 ret_part.setAttribute("name", "returnvalue")
177 ret_part.setAttribute("type", param_type(return_type))
179 # Port connecting arguments with return type
181 port_el = wsdl.firstChild.appendChild(wsdl.createElement("portType"))
182 port_el.setAttribute("name", method + "_port")
184 op_el = port_el.appendChild(wsdl.createElement("operation"))
185 op_el.setAttribute("name", method)
186 inp_el=wsdl.createElement("input")
187 inp_el.setAttribute("message","tns:" + method + "_in")
188 inp_el.setAttribute("name",method+"_request")
189 op_el.appendChild(inp_el)
190 out_el = wsdl.createElement("output")
191 out_el.setAttribute("message","tns:" + method + "_out")
192 out_el.setAttribute("name",method+"_response")
193 op_el.appendChild(out_el)
197 bind_el = wsdl.firstChild.appendChild(wsdl.createElement("binding"))
198 bind_el.setAttribute("name", method + "_binding")
199 bind_el.setAttribute("type", "tns:" + method + "_port")
201 soap_bind = bind_el.appendChild(wsdl.createElement("soap:binding"))
202 soap_bind.setAttribute("style", "rpc")
203 soap_bind.setAttribute("transport","http://schemas.xmlsoap.org/soap/http")
206 wsdl_op = bind_el.appendChild(wsdl.createElement("operation"))
207 wsdl_op.setAttribute("name", method)
208 wsdl_op.appendChild(wsdl.createElement("soap:operation")).setAttribute("soapAction",
212 wsdl_input = wsdl_op.appendChild(wsdl.createElement("input"))
213 input_soap_body = wsdl_input.appendChild(wsdl.createElement("soap:body"))
214 input_soap_body.setAttribute("use", "encoded")
215 input_soap_body.setAttribute("namespace", "urn:" + method)
216 input_soap_body.setAttribute("encodingStyle","http://schemas.xmlsoap.org/soap/encoding/")
219 wsdl_output = wsdl_op.appendChild(wsdl.createElement("output"))
220 output_soap_body = wsdl_output.appendChild(wsdl.createElement("soap:body"))
221 output_soap_body.setAttribute("use", "encoded")
222 output_soap_body.setAttribute("namespace", "urn:" + method)
223 output_soap_body.setAttribute("encodingStyle","http://schemas.xmlsoap.org/soap/encoding/")
226 def add_wsdl_service(wsdl):
227 for service in services.keys():
228 global interface_options
229 if (getattr(interface_options,service)):
230 service_el = wsdl.firstChild.appendChild(wsdl.createElement("service"))
231 service_el.setAttribute("name", service)
233 for method in services[service]:
235 servport_el = service_el.appendChild(wsdl.createElement("port"))
236 servport_el.setAttribute("name", name + "_port")
237 servport_el.setAttribute("binding", "tns:" + name + "_binding")
239 soapaddress = servport_el.appendChild(wsdl.createElement("soap:address"))
240 soapaddress.setAttribute("location", "%s/%s" % (globals.plc_ns,service))
243 def get_wsdl_definitions():
244 wsdl_text_header = """
246 name="geniwrapper_autogenerated"
247 targetNamespace="%s/2009/07/sfa.wsdl"
248 xmlns="http://schemas.xmlsoap.org/wsdl/"
249 xmlns:xsd="http://www.w3.org/2001/XMLSchema"
250 xmlns:xsdl="%s/2009/07/schema"
251 xmlns:tns="%s/2009/07/sfa.wsdl"
252 xmlns:soap="http://schemas.xmlsoap.org/wsdl/soap/"
253 xmlns:soapenc="http://schemas.xmlsoap.org/wsdl/soap/encoding"
254 xmlns:wsdl="http://schemas.xmlsoap.org/wsdl/"/>
255 """ % (globals.plc_ns,globals.plc_ns,globals.plc_ns)
257 wsdl = xml.dom.minidom.parseString(wsdl_text_header)
261 def get_wsdl_definitions_and_types():
262 wsdl_text_header = """
264 name="geniwrapper_autogenerated"
265 targetNamespace="%s/2009/07/sfa.wsdl"
266 xmlns="http://schemas.xmlsoap.org/wsdl/"
267 xmlns:xsd="http://www.w3.org/2001/XMLSchema"
268 xmlns:xsdl="%s/2009/07/schema"
269 xmlns:tns="%s/2009/07/sfa.wsdl"
270 xmlns:soap="http://schemas.xmlsoap.org/wsdl/soap/"
271 xmlns:soapenc="http://schemas.xmlsoap.org/wsdl/soap/encoding"
272 xmlns:wsdl="http://schemas.xmlsoap.org/wsdl/">
274 <xsd:schema xmlns="http://www.w3.org/2001/XMLSchema" targetNamespace="%s/2009/07/schema"/>
276 </wsdl:definitions> """ % (globals.plc_ns, globals.plc_ns, globals.plc_ns, globals.plc_ns)
277 wsdl = xml.dom.minidom.parseString(wsdl_text_header)
282 global interface_options
284 parser = OptionParser()
285 parser.add_option("-r", "--registry", dest="registry", action="store_true",
286 help="Generate registry.wsdl", metavar="FILE")
287 parser.add_option("-s", "--slice-manager",
288 action="store_true", dest="slicemgr",
289 help="Generate sm.wsdl")
290 parser.add_option("-a", "--aggregate", action="store_true", dest="aggregate",
291 help="Generate am.wsdl")
293 (interface_options, args) = parser.parse_args()
294 types = get_wsdl_definitions_and_types()
295 wsdl = get_wsdl_definitions()
296 add_wsdl_ports_and_bindings(wsdl)
297 wsdl_types = wsdl.importNode(types.getElementsByTagName("types")[0], True)
298 wsdl.firstChild.appendChild(wsdl_types)
299 add_wsdl_service(wsdl)
301 xml.dom.ext.PrettyPrint(wsdl)
303 if __name__ == "__main__":