Mercurial > repos > guerler > springsuite
comparison planemo/lib/python3.7/site-packages/future/backports/xmlrpc/server.py @ 0:d30785e31577 draft
"planemo upload commit 6eee67778febed82ddd413c3ca40b3183a3898f1"
author | guerler |
---|---|
date | Fri, 31 Jul 2020 00:18:57 -0400 |
parents | |
children |
comparison
equal
deleted
inserted
replaced
-1:000000000000 | 0:d30785e31577 |
---|---|
1 r""" | |
2 Ported using Python-Future from the Python 3.3 standard library. | |
3 | |
4 XML-RPC Servers. | |
5 | |
6 This module can be used to create simple XML-RPC servers | |
7 by creating a server and either installing functions, a | |
8 class instance, or by extending the SimpleXMLRPCServer | |
9 class. | |
10 | |
11 It can also be used to handle XML-RPC requests in a CGI | |
12 environment using CGIXMLRPCRequestHandler. | |
13 | |
14 The Doc* classes can be used to create XML-RPC servers that | |
15 serve pydoc-style documentation in response to HTTP | |
16 GET requests. This documentation is dynamically generated | |
17 based on the functions and methods registered with the | |
18 server. | |
19 | |
20 A list of possible usage patterns follows: | |
21 | |
22 1. Install functions: | |
23 | |
24 server = SimpleXMLRPCServer(("localhost", 8000)) | |
25 server.register_function(pow) | |
26 server.register_function(lambda x,y: x+y, 'add') | |
27 server.serve_forever() | |
28 | |
29 2. Install an instance: | |
30 | |
31 class MyFuncs: | |
32 def __init__(self): | |
33 # make all of the sys functions available through sys.func_name | |
34 import sys | |
35 self.sys = sys | |
36 def _listMethods(self): | |
37 # implement this method so that system.listMethods | |
38 # knows to advertise the sys methods | |
39 return list_public_methods(self) + \ | |
40 ['sys.' + method for method in list_public_methods(self.sys)] | |
41 def pow(self, x, y): return pow(x, y) | |
42 def add(self, x, y) : return x + y | |
43 | |
44 server = SimpleXMLRPCServer(("localhost", 8000)) | |
45 server.register_introspection_functions() | |
46 server.register_instance(MyFuncs()) | |
47 server.serve_forever() | |
48 | |
49 3. Install an instance with custom dispatch method: | |
50 | |
51 class Math: | |
52 def _listMethods(self): | |
53 # this method must be present for system.listMethods | |
54 # to work | |
55 return ['add', 'pow'] | |
56 def _methodHelp(self, method): | |
57 # this method must be present for system.methodHelp | |
58 # to work | |
59 if method == 'add': | |
60 return "add(2,3) => 5" | |
61 elif method == 'pow': | |
62 return "pow(x, y[, z]) => number" | |
63 else: | |
64 # By convention, return empty | |
65 # string if no help is available | |
66 return "" | |
67 def _dispatch(self, method, params): | |
68 if method == 'pow': | |
69 return pow(*params) | |
70 elif method == 'add': | |
71 return params[0] + params[1] | |
72 else: | |
73 raise ValueError('bad method') | |
74 | |
75 server = SimpleXMLRPCServer(("localhost", 8000)) | |
76 server.register_introspection_functions() | |
77 server.register_instance(Math()) | |
78 server.serve_forever() | |
79 | |
80 4. Subclass SimpleXMLRPCServer: | |
81 | |
82 class MathServer(SimpleXMLRPCServer): | |
83 def _dispatch(self, method, params): | |
84 try: | |
85 # We are forcing the 'export_' prefix on methods that are | |
86 # callable through XML-RPC to prevent potential security | |
87 # problems | |
88 func = getattr(self, 'export_' + method) | |
89 except AttributeError: | |
90 raise Exception('method "%s" is not supported' % method) | |
91 else: | |
92 return func(*params) | |
93 | |
94 def export_add(self, x, y): | |
95 return x + y | |
96 | |
97 server = MathServer(("localhost", 8000)) | |
98 server.serve_forever() | |
99 | |
100 5. CGI script: | |
101 | |
102 server = CGIXMLRPCRequestHandler() | |
103 server.register_function(pow) | |
104 server.handle_request() | |
105 """ | |
106 | |
107 from __future__ import absolute_import, division, print_function, unicode_literals | |
108 from future.builtins import int, str | |
109 | |
110 # Written by Brian Quinlan (brian@sweetapp.com). | |
111 # Based on code written by Fredrik Lundh. | |
112 | |
113 from future.backports.xmlrpc.client import Fault, dumps, loads, gzip_encode, gzip_decode | |
114 from future.backports.http.server import BaseHTTPRequestHandler | |
115 import future.backports.http.server as http_server | |
116 from future.backports import socketserver | |
117 import sys | |
118 import os | |
119 import re | |
120 import pydoc | |
121 import inspect | |
122 import traceback | |
123 try: | |
124 import fcntl | |
125 except ImportError: | |
126 fcntl = None | |
127 | |
128 def resolve_dotted_attribute(obj, attr, allow_dotted_names=True): | |
129 """resolve_dotted_attribute(a, 'b.c.d') => a.b.c.d | |
130 | |
131 Resolves a dotted attribute name to an object. Raises | |
132 an AttributeError if any attribute in the chain starts with a '_'. | |
133 | |
134 If the optional allow_dotted_names argument is false, dots are not | |
135 supported and this function operates similar to getattr(obj, attr). | |
136 """ | |
137 | |
138 if allow_dotted_names: | |
139 attrs = attr.split('.') | |
140 else: | |
141 attrs = [attr] | |
142 | |
143 for i in attrs: | |
144 if i.startswith('_'): | |
145 raise AttributeError( | |
146 'attempt to access private attribute "%s"' % i | |
147 ) | |
148 else: | |
149 obj = getattr(obj,i) | |
150 return obj | |
151 | |
152 def list_public_methods(obj): | |
153 """Returns a list of attribute strings, found in the specified | |
154 object, which represent callable attributes""" | |
155 | |
156 return [member for member in dir(obj) | |
157 if not member.startswith('_') and | |
158 callable(getattr(obj, member))] | |
159 | |
160 class SimpleXMLRPCDispatcher(object): | |
161 """Mix-in class that dispatches XML-RPC requests. | |
162 | |
163 This class is used to register XML-RPC method handlers | |
164 and then to dispatch them. This class doesn't need to be | |
165 instanced directly when used by SimpleXMLRPCServer but it | |
166 can be instanced when used by the MultiPathXMLRPCServer | |
167 """ | |
168 | |
169 def __init__(self, allow_none=False, encoding=None, | |
170 use_builtin_types=False): | |
171 self.funcs = {} | |
172 self.instance = None | |
173 self.allow_none = allow_none | |
174 self.encoding = encoding or 'utf-8' | |
175 self.use_builtin_types = use_builtin_types | |
176 | |
177 def register_instance(self, instance, allow_dotted_names=False): | |
178 """Registers an instance to respond to XML-RPC requests. | |
179 | |
180 Only one instance can be installed at a time. | |
181 | |
182 If the registered instance has a _dispatch method then that | |
183 method will be called with the name of the XML-RPC method and | |
184 its parameters as a tuple | |
185 e.g. instance._dispatch('add',(2,3)) | |
186 | |
187 If the registered instance does not have a _dispatch method | |
188 then the instance will be searched to find a matching method | |
189 and, if found, will be called. Methods beginning with an '_' | |
190 are considered private and will not be called by | |
191 SimpleXMLRPCServer. | |
192 | |
193 If a registered function matches a XML-RPC request, then it | |
194 will be called instead of the registered instance. | |
195 | |
196 If the optional allow_dotted_names argument is true and the | |
197 instance does not have a _dispatch method, method names | |
198 containing dots are supported and resolved, as long as none of | |
199 the name segments start with an '_'. | |
200 | |
201 *** SECURITY WARNING: *** | |
202 | |
203 Enabling the allow_dotted_names options allows intruders | |
204 to access your module's global variables and may allow | |
205 intruders to execute arbitrary code on your machine. Only | |
206 use this option on a secure, closed network. | |
207 | |
208 """ | |
209 | |
210 self.instance = instance | |
211 self.allow_dotted_names = allow_dotted_names | |
212 | |
213 def register_function(self, function, name=None): | |
214 """Registers a function to respond to XML-RPC requests. | |
215 | |
216 The optional name argument can be used to set a Unicode name | |
217 for the function. | |
218 """ | |
219 | |
220 if name is None: | |
221 name = function.__name__ | |
222 self.funcs[name] = function | |
223 | |
224 def register_introspection_functions(self): | |
225 """Registers the XML-RPC introspection methods in the system | |
226 namespace. | |
227 | |
228 see http://xmlrpc.usefulinc.com/doc/reserved.html | |
229 """ | |
230 | |
231 self.funcs.update({'system.listMethods' : self.system_listMethods, | |
232 'system.methodSignature' : self.system_methodSignature, | |
233 'system.methodHelp' : self.system_methodHelp}) | |
234 | |
235 def register_multicall_functions(self): | |
236 """Registers the XML-RPC multicall method in the system | |
237 namespace. | |
238 | |
239 see http://www.xmlrpc.com/discuss/msgReader$1208""" | |
240 | |
241 self.funcs.update({'system.multicall' : self.system_multicall}) | |
242 | |
243 def _marshaled_dispatch(self, data, dispatch_method = None, path = None): | |
244 """Dispatches an XML-RPC method from marshalled (XML) data. | |
245 | |
246 XML-RPC methods are dispatched from the marshalled (XML) data | |
247 using the _dispatch method and the result is returned as | |
248 marshalled data. For backwards compatibility, a dispatch | |
249 function can be provided as an argument (see comment in | |
250 SimpleXMLRPCRequestHandler.do_POST) but overriding the | |
251 existing method through subclassing is the preferred means | |
252 of changing method dispatch behavior. | |
253 """ | |
254 | |
255 try: | |
256 params, method = loads(data, use_builtin_types=self.use_builtin_types) | |
257 | |
258 # generate response | |
259 if dispatch_method is not None: | |
260 response = dispatch_method(method, params) | |
261 else: | |
262 response = self._dispatch(method, params) | |
263 # wrap response in a singleton tuple | |
264 response = (response,) | |
265 response = dumps(response, methodresponse=1, | |
266 allow_none=self.allow_none, encoding=self.encoding) | |
267 except Fault as fault: | |
268 response = dumps(fault, allow_none=self.allow_none, | |
269 encoding=self.encoding) | |
270 except: | |
271 # report exception back to server | |
272 exc_type, exc_value, exc_tb = sys.exc_info() | |
273 response = dumps( | |
274 Fault(1, "%s:%s" % (exc_type, exc_value)), | |
275 encoding=self.encoding, allow_none=self.allow_none, | |
276 ) | |
277 | |
278 return response.encode(self.encoding) | |
279 | |
280 def system_listMethods(self): | |
281 """system.listMethods() => ['add', 'subtract', 'multiple'] | |
282 | |
283 Returns a list of the methods supported by the server.""" | |
284 | |
285 methods = set(self.funcs.keys()) | |
286 if self.instance is not None: | |
287 # Instance can implement _listMethod to return a list of | |
288 # methods | |
289 if hasattr(self.instance, '_listMethods'): | |
290 methods |= set(self.instance._listMethods()) | |
291 # if the instance has a _dispatch method then we | |
292 # don't have enough information to provide a list | |
293 # of methods | |
294 elif not hasattr(self.instance, '_dispatch'): | |
295 methods |= set(list_public_methods(self.instance)) | |
296 return sorted(methods) | |
297 | |
298 def system_methodSignature(self, method_name): | |
299 """system.methodSignature('add') => [double, int, int] | |
300 | |
301 Returns a list describing the signature of the method. In the | |
302 above example, the add method takes two integers as arguments | |
303 and returns a double result. | |
304 | |
305 This server does NOT support system.methodSignature.""" | |
306 | |
307 # See http://xmlrpc.usefulinc.com/doc/sysmethodsig.html | |
308 | |
309 return 'signatures not supported' | |
310 | |
311 def system_methodHelp(self, method_name): | |
312 """system.methodHelp('add') => "Adds two integers together" | |
313 | |
314 Returns a string containing documentation for the specified method.""" | |
315 | |
316 method = None | |
317 if method_name in self.funcs: | |
318 method = self.funcs[method_name] | |
319 elif self.instance is not None: | |
320 # Instance can implement _methodHelp to return help for a method | |
321 if hasattr(self.instance, '_methodHelp'): | |
322 return self.instance._methodHelp(method_name) | |
323 # if the instance has a _dispatch method then we | |
324 # don't have enough information to provide help | |
325 elif not hasattr(self.instance, '_dispatch'): | |
326 try: | |
327 method = resolve_dotted_attribute( | |
328 self.instance, | |
329 method_name, | |
330 self.allow_dotted_names | |
331 ) | |
332 except AttributeError: | |
333 pass | |
334 | |
335 # Note that we aren't checking that the method actually | |
336 # be a callable object of some kind | |
337 if method is None: | |
338 return "" | |
339 else: | |
340 return pydoc.getdoc(method) | |
341 | |
342 def system_multicall(self, call_list): | |
343 """system.multicall([{'methodName': 'add', 'params': [2, 2]}, ...]) => \ | |
344 [[4], ...] | |
345 | |
346 Allows the caller to package multiple XML-RPC calls into a single | |
347 request. | |
348 | |
349 See http://www.xmlrpc.com/discuss/msgReader$1208 | |
350 """ | |
351 | |
352 results = [] | |
353 for call in call_list: | |
354 method_name = call['methodName'] | |
355 params = call['params'] | |
356 | |
357 try: | |
358 # XXX A marshalling error in any response will fail the entire | |
359 # multicall. If someone cares they should fix this. | |
360 results.append([self._dispatch(method_name, params)]) | |
361 except Fault as fault: | |
362 results.append( | |
363 {'faultCode' : fault.faultCode, | |
364 'faultString' : fault.faultString} | |
365 ) | |
366 except: | |
367 exc_type, exc_value, exc_tb = sys.exc_info() | |
368 results.append( | |
369 {'faultCode' : 1, | |
370 'faultString' : "%s:%s" % (exc_type, exc_value)} | |
371 ) | |
372 return results | |
373 | |
374 def _dispatch(self, method, params): | |
375 """Dispatches the XML-RPC method. | |
376 | |
377 XML-RPC calls are forwarded to a registered function that | |
378 matches the called XML-RPC method name. If no such function | |
379 exists then the call is forwarded to the registered instance, | |
380 if available. | |
381 | |
382 If the registered instance has a _dispatch method then that | |
383 method will be called with the name of the XML-RPC method and | |
384 its parameters as a tuple | |
385 e.g. instance._dispatch('add',(2,3)) | |
386 | |
387 If the registered instance does not have a _dispatch method | |
388 then the instance will be searched to find a matching method | |
389 and, if found, will be called. | |
390 | |
391 Methods beginning with an '_' are considered private and will | |
392 not be called. | |
393 """ | |
394 | |
395 func = None | |
396 try: | |
397 # check to see if a matching function has been registered | |
398 func = self.funcs[method] | |
399 except KeyError: | |
400 if self.instance is not None: | |
401 # check for a _dispatch method | |
402 if hasattr(self.instance, '_dispatch'): | |
403 return self.instance._dispatch(method, params) | |
404 else: | |
405 # call instance method directly | |
406 try: | |
407 func = resolve_dotted_attribute( | |
408 self.instance, | |
409 method, | |
410 self.allow_dotted_names | |
411 ) | |
412 except AttributeError: | |
413 pass | |
414 | |
415 if func is not None: | |
416 return func(*params) | |
417 else: | |
418 raise Exception('method "%s" is not supported' % method) | |
419 | |
420 class SimpleXMLRPCRequestHandler(BaseHTTPRequestHandler): | |
421 """Simple XML-RPC request handler class. | |
422 | |
423 Handles all HTTP POST requests and attempts to decode them as | |
424 XML-RPC requests. | |
425 """ | |
426 | |
427 # Class attribute listing the accessible path components; | |
428 # paths not on this list will result in a 404 error. | |
429 rpc_paths = ('/', '/RPC2') | |
430 | |
431 #if not None, encode responses larger than this, if possible | |
432 encode_threshold = 1400 #a common MTU | |
433 | |
434 #Override form StreamRequestHandler: full buffering of output | |
435 #and no Nagle. | |
436 wbufsize = -1 | |
437 disable_nagle_algorithm = True | |
438 | |
439 # a re to match a gzip Accept-Encoding | |
440 aepattern = re.compile(r""" | |
441 \s* ([^\s;]+) \s* #content-coding | |
442 (;\s* q \s*=\s* ([0-9\.]+))? #q | |
443 """, re.VERBOSE | re.IGNORECASE) | |
444 | |
445 def accept_encodings(self): | |
446 r = {} | |
447 ae = self.headers.get("Accept-Encoding", "") | |
448 for e in ae.split(","): | |
449 match = self.aepattern.match(e) | |
450 if match: | |
451 v = match.group(3) | |
452 v = float(v) if v else 1.0 | |
453 r[match.group(1)] = v | |
454 return r | |
455 | |
456 def is_rpc_path_valid(self): | |
457 if self.rpc_paths: | |
458 return self.path in self.rpc_paths | |
459 else: | |
460 # If .rpc_paths is empty, just assume all paths are legal | |
461 return True | |
462 | |
463 def do_POST(self): | |
464 """Handles the HTTP POST request. | |
465 | |
466 Attempts to interpret all HTTP POST requests as XML-RPC calls, | |
467 which are forwarded to the server's _dispatch method for handling. | |
468 """ | |
469 | |
470 # Check that the path is legal | |
471 if not self.is_rpc_path_valid(): | |
472 self.report_404() | |
473 return | |
474 | |
475 try: | |
476 # Get arguments by reading body of request. | |
477 # We read this in chunks to avoid straining | |
478 # socket.read(); around the 10 or 15Mb mark, some platforms | |
479 # begin to have problems (bug #792570). | |
480 max_chunk_size = 10*1024*1024 | |
481 size_remaining = int(self.headers["content-length"]) | |
482 L = [] | |
483 while size_remaining: | |
484 chunk_size = min(size_remaining, max_chunk_size) | |
485 chunk = self.rfile.read(chunk_size) | |
486 if not chunk: | |
487 break | |
488 L.append(chunk) | |
489 size_remaining -= len(L[-1]) | |
490 data = b''.join(L) | |
491 | |
492 data = self.decode_request_content(data) | |
493 if data is None: | |
494 return #response has been sent | |
495 | |
496 # In previous versions of SimpleXMLRPCServer, _dispatch | |
497 # could be overridden in this class, instead of in | |
498 # SimpleXMLRPCDispatcher. To maintain backwards compatibility, | |
499 # check to see if a subclass implements _dispatch and dispatch | |
500 # using that method if present. | |
501 response = self.server._marshaled_dispatch( | |
502 data, getattr(self, '_dispatch', None), self.path | |
503 ) | |
504 except Exception as e: # This should only happen if the module is buggy | |
505 # internal error, report as HTTP server error | |
506 self.send_response(500) | |
507 | |
508 # Send information about the exception if requested | |
509 if hasattr(self.server, '_send_traceback_header') and \ | |
510 self.server._send_traceback_header: | |
511 self.send_header("X-exception", str(e)) | |
512 trace = traceback.format_exc() | |
513 trace = str(trace.encode('ASCII', 'backslashreplace'), 'ASCII') | |
514 self.send_header("X-traceback", trace) | |
515 | |
516 self.send_header("Content-length", "0") | |
517 self.end_headers() | |
518 else: | |
519 self.send_response(200) | |
520 self.send_header("Content-type", "text/xml") | |
521 if self.encode_threshold is not None: | |
522 if len(response) > self.encode_threshold: | |
523 q = self.accept_encodings().get("gzip", 0) | |
524 if q: | |
525 try: | |
526 response = gzip_encode(response) | |
527 self.send_header("Content-Encoding", "gzip") | |
528 except NotImplementedError: | |
529 pass | |
530 self.send_header("Content-length", str(len(response))) | |
531 self.end_headers() | |
532 self.wfile.write(response) | |
533 | |
534 def decode_request_content(self, data): | |
535 #support gzip encoding of request | |
536 encoding = self.headers.get("content-encoding", "identity").lower() | |
537 if encoding == "identity": | |
538 return data | |
539 if encoding == "gzip": | |
540 try: | |
541 return gzip_decode(data) | |
542 except NotImplementedError: | |
543 self.send_response(501, "encoding %r not supported" % encoding) | |
544 except ValueError: | |
545 self.send_response(400, "error decoding gzip content") | |
546 else: | |
547 self.send_response(501, "encoding %r not supported" % encoding) | |
548 self.send_header("Content-length", "0") | |
549 self.end_headers() | |
550 | |
551 def report_404 (self): | |
552 # Report a 404 error | |
553 self.send_response(404) | |
554 response = b'No such page' | |
555 self.send_header("Content-type", "text/plain") | |
556 self.send_header("Content-length", str(len(response))) | |
557 self.end_headers() | |
558 self.wfile.write(response) | |
559 | |
560 def log_request(self, code='-', size='-'): | |
561 """Selectively log an accepted request.""" | |
562 | |
563 if self.server.logRequests: | |
564 BaseHTTPRequestHandler.log_request(self, code, size) | |
565 | |
566 class SimpleXMLRPCServer(socketserver.TCPServer, | |
567 SimpleXMLRPCDispatcher): | |
568 """Simple XML-RPC server. | |
569 | |
570 Simple XML-RPC server that allows functions and a single instance | |
571 to be installed to handle requests. The default implementation | |
572 attempts to dispatch XML-RPC calls to the functions or instance | |
573 installed in the server. Override the _dispatch method inherited | |
574 from SimpleXMLRPCDispatcher to change this behavior. | |
575 """ | |
576 | |
577 allow_reuse_address = True | |
578 | |
579 # Warning: this is for debugging purposes only! Never set this to True in | |
580 # production code, as will be sending out sensitive information (exception | |
581 # and stack trace details) when exceptions are raised inside | |
582 # SimpleXMLRPCRequestHandler.do_POST | |
583 _send_traceback_header = False | |
584 | |
585 def __init__(self, addr, requestHandler=SimpleXMLRPCRequestHandler, | |
586 logRequests=True, allow_none=False, encoding=None, | |
587 bind_and_activate=True, use_builtin_types=False): | |
588 self.logRequests = logRequests | |
589 | |
590 SimpleXMLRPCDispatcher.__init__(self, allow_none, encoding, use_builtin_types) | |
591 socketserver.TCPServer.__init__(self, addr, requestHandler, bind_and_activate) | |
592 | |
593 # [Bug #1222790] If possible, set close-on-exec flag; if a | |
594 # method spawns a subprocess, the subprocess shouldn't have | |
595 # the listening socket open. | |
596 if fcntl is not None and hasattr(fcntl, 'FD_CLOEXEC'): | |
597 flags = fcntl.fcntl(self.fileno(), fcntl.F_GETFD) | |
598 flags |= fcntl.FD_CLOEXEC | |
599 fcntl.fcntl(self.fileno(), fcntl.F_SETFD, flags) | |
600 | |
601 class MultiPathXMLRPCServer(SimpleXMLRPCServer): | |
602 """Multipath XML-RPC Server | |
603 This specialization of SimpleXMLRPCServer allows the user to create | |
604 multiple Dispatcher instances and assign them to different | |
605 HTTP request paths. This makes it possible to run two or more | |
606 'virtual XML-RPC servers' at the same port. | |
607 Make sure that the requestHandler accepts the paths in question. | |
608 """ | |
609 def __init__(self, addr, requestHandler=SimpleXMLRPCRequestHandler, | |
610 logRequests=True, allow_none=False, encoding=None, | |
611 bind_and_activate=True, use_builtin_types=False): | |
612 | |
613 SimpleXMLRPCServer.__init__(self, addr, requestHandler, logRequests, allow_none, | |
614 encoding, bind_and_activate, use_builtin_types) | |
615 self.dispatchers = {} | |
616 self.allow_none = allow_none | |
617 self.encoding = encoding or 'utf-8' | |
618 | |
619 def add_dispatcher(self, path, dispatcher): | |
620 self.dispatchers[path] = dispatcher | |
621 return dispatcher | |
622 | |
623 def get_dispatcher(self, path): | |
624 return self.dispatchers[path] | |
625 | |
626 def _marshaled_dispatch(self, data, dispatch_method = None, path = None): | |
627 try: | |
628 response = self.dispatchers[path]._marshaled_dispatch( | |
629 data, dispatch_method, path) | |
630 except: | |
631 # report low level exception back to server | |
632 # (each dispatcher should have handled their own | |
633 # exceptions) | |
634 exc_type, exc_value = sys.exc_info()[:2] | |
635 response = dumps( | |
636 Fault(1, "%s:%s" % (exc_type, exc_value)), | |
637 encoding=self.encoding, allow_none=self.allow_none) | |
638 response = response.encode(self.encoding) | |
639 return response | |
640 | |
641 class CGIXMLRPCRequestHandler(SimpleXMLRPCDispatcher): | |
642 """Simple handler for XML-RPC data passed through CGI.""" | |
643 | |
644 def __init__(self, allow_none=False, encoding=None, use_builtin_types=False): | |
645 SimpleXMLRPCDispatcher.__init__(self, allow_none, encoding, use_builtin_types) | |
646 | |
647 def handle_xmlrpc(self, request_text): | |
648 """Handle a single XML-RPC request""" | |
649 | |
650 response = self._marshaled_dispatch(request_text) | |
651 | |
652 print('Content-Type: text/xml') | |
653 print('Content-Length: %d' % len(response)) | |
654 print() | |
655 sys.stdout.flush() | |
656 sys.stdout.buffer.write(response) | |
657 sys.stdout.buffer.flush() | |
658 | |
659 def handle_get(self): | |
660 """Handle a single HTTP GET request. | |
661 | |
662 Default implementation indicates an error because | |
663 XML-RPC uses the POST method. | |
664 """ | |
665 | |
666 code = 400 | |
667 message, explain = BaseHTTPRequestHandler.responses[code] | |
668 | |
669 response = http_server.DEFAULT_ERROR_MESSAGE % \ | |
670 { | |
671 'code' : code, | |
672 'message' : message, | |
673 'explain' : explain | |
674 } | |
675 response = response.encode('utf-8') | |
676 print('Status: %d %s' % (code, message)) | |
677 print('Content-Type: %s' % http_server.DEFAULT_ERROR_CONTENT_TYPE) | |
678 print('Content-Length: %d' % len(response)) | |
679 print() | |
680 sys.stdout.flush() | |
681 sys.stdout.buffer.write(response) | |
682 sys.stdout.buffer.flush() | |
683 | |
684 def handle_request(self, request_text=None): | |
685 """Handle a single XML-RPC request passed through a CGI post method. | |
686 | |
687 If no XML data is given then it is read from stdin. The resulting | |
688 XML-RPC response is printed to stdout along with the correct HTTP | |
689 headers. | |
690 """ | |
691 | |
692 if request_text is None and \ | |
693 os.environ.get('REQUEST_METHOD', None) == 'GET': | |
694 self.handle_get() | |
695 else: | |
696 # POST data is normally available through stdin | |
697 try: | |
698 length = int(os.environ.get('CONTENT_LENGTH', None)) | |
699 except (ValueError, TypeError): | |
700 length = -1 | |
701 if request_text is None: | |
702 request_text = sys.stdin.read(length) | |
703 | |
704 self.handle_xmlrpc(request_text) | |
705 | |
706 | |
707 # ----------------------------------------------------------------------------- | |
708 # Self documenting XML-RPC Server. | |
709 | |
710 class ServerHTMLDoc(pydoc.HTMLDoc): | |
711 """Class used to generate pydoc HTML document for a server""" | |
712 | |
713 def markup(self, text, escape=None, funcs={}, classes={}, methods={}): | |
714 """Mark up some plain text, given a context of symbols to look for. | |
715 Each context dictionary maps object names to anchor names.""" | |
716 escape = escape or self.escape | |
717 results = [] | |
718 here = 0 | |
719 | |
720 # XXX Note that this regular expression does not allow for the | |
721 # hyperlinking of arbitrary strings being used as method | |
722 # names. Only methods with names consisting of word characters | |
723 # and '.'s are hyperlinked. | |
724 pattern = re.compile(r'\b((http|ftp)://\S+[\w/]|' | |
725 r'RFC[- ]?(\d+)|' | |
726 r'PEP[- ]?(\d+)|' | |
727 r'(self\.)?((?:\w|\.)+))\b') | |
728 while 1: | |
729 match = pattern.search(text, here) | |
730 if not match: break | |
731 start, end = match.span() | |
732 results.append(escape(text[here:start])) | |
733 | |
734 all, scheme, rfc, pep, selfdot, name = match.groups() | |
735 if scheme: | |
736 url = escape(all).replace('"', '"') | |
737 results.append('<a href="%s">%s</a>' % (url, url)) | |
738 elif rfc: | |
739 url = 'http://www.rfc-editor.org/rfc/rfc%d.txt' % int(rfc) | |
740 results.append('<a href="%s">%s</a>' % (url, escape(all))) | |
741 elif pep: | |
742 url = 'http://www.python.org/dev/peps/pep-%04d/' % int(pep) | |
743 results.append('<a href="%s">%s</a>' % (url, escape(all))) | |
744 elif text[end:end+1] == '(': | |
745 results.append(self.namelink(name, methods, funcs, classes)) | |
746 elif selfdot: | |
747 results.append('self.<strong>%s</strong>' % name) | |
748 else: | |
749 results.append(self.namelink(name, classes)) | |
750 here = end | |
751 results.append(escape(text[here:])) | |
752 return ''.join(results) | |
753 | |
754 def docroutine(self, object, name, mod=None, | |
755 funcs={}, classes={}, methods={}, cl=None): | |
756 """Produce HTML documentation for a function or method object.""" | |
757 | |
758 anchor = (cl and cl.__name__ or '') + '-' + name | |
759 note = '' | |
760 | |
761 title = '<a name="%s"><strong>%s</strong></a>' % ( | |
762 self.escape(anchor), self.escape(name)) | |
763 | |
764 if inspect.ismethod(object): | |
765 args = inspect.getfullargspec(object) | |
766 # exclude the argument bound to the instance, it will be | |
767 # confusing to the non-Python user | |
768 argspec = inspect.formatargspec ( | |
769 args.args[1:], | |
770 args.varargs, | |
771 args.varkw, | |
772 args.defaults, | |
773 annotations=args.annotations, | |
774 formatvalue=self.formatvalue | |
775 ) | |
776 elif inspect.isfunction(object): | |
777 args = inspect.getfullargspec(object) | |
778 argspec = inspect.formatargspec( | |
779 args.args, args.varargs, args.varkw, args.defaults, | |
780 annotations=args.annotations, | |
781 formatvalue=self.formatvalue) | |
782 else: | |
783 argspec = '(...)' | |
784 | |
785 if isinstance(object, tuple): | |
786 argspec = object[0] or argspec | |
787 docstring = object[1] or "" | |
788 else: | |
789 docstring = pydoc.getdoc(object) | |
790 | |
791 decl = title + argspec + (note and self.grey( | |
792 '<font face="helvetica, arial">%s</font>' % note)) | |
793 | |
794 doc = self.markup( | |
795 docstring, self.preformat, funcs, classes, methods) | |
796 doc = doc and '<dd><tt>%s</tt></dd>' % doc | |
797 return '<dl><dt>%s</dt>%s</dl>\n' % (decl, doc) | |
798 | |
799 def docserver(self, server_name, package_documentation, methods): | |
800 """Produce HTML documentation for an XML-RPC server.""" | |
801 | |
802 fdict = {} | |
803 for key, value in methods.items(): | |
804 fdict[key] = '#-' + key | |
805 fdict[value] = fdict[key] | |
806 | |
807 server_name = self.escape(server_name) | |
808 head = '<big><big><strong>%s</strong></big></big>' % server_name | |
809 result = self.heading(head, '#ffffff', '#7799ee') | |
810 | |
811 doc = self.markup(package_documentation, self.preformat, fdict) | |
812 doc = doc and '<tt>%s</tt>' % doc | |
813 result = result + '<p>%s</p>\n' % doc | |
814 | |
815 contents = [] | |
816 method_items = sorted(methods.items()) | |
817 for key, value in method_items: | |
818 contents.append(self.docroutine(value, key, funcs=fdict)) | |
819 result = result + self.bigsection( | |
820 'Methods', '#ffffff', '#eeaa77', ''.join(contents)) | |
821 | |
822 return result | |
823 | |
824 class XMLRPCDocGenerator(object): | |
825 """Generates documentation for an XML-RPC server. | |
826 | |
827 This class is designed as mix-in and should not | |
828 be constructed directly. | |
829 """ | |
830 | |
831 def __init__(self): | |
832 # setup variables used for HTML documentation | |
833 self.server_name = 'XML-RPC Server Documentation' | |
834 self.server_documentation = \ | |
835 "This server exports the following methods through the XML-RPC "\ | |
836 "protocol." | |
837 self.server_title = 'XML-RPC Server Documentation' | |
838 | |
839 def set_server_title(self, server_title): | |
840 """Set the HTML title of the generated server documentation""" | |
841 | |
842 self.server_title = server_title | |
843 | |
844 def set_server_name(self, server_name): | |
845 """Set the name of the generated HTML server documentation""" | |
846 | |
847 self.server_name = server_name | |
848 | |
849 def set_server_documentation(self, server_documentation): | |
850 """Set the documentation string for the entire server.""" | |
851 | |
852 self.server_documentation = server_documentation | |
853 | |
854 def generate_html_documentation(self): | |
855 """generate_html_documentation() => html documentation for the server | |
856 | |
857 Generates HTML documentation for the server using introspection for | |
858 installed functions and instances that do not implement the | |
859 _dispatch method. Alternatively, instances can choose to implement | |
860 the _get_method_argstring(method_name) method to provide the | |
861 argument string used in the documentation and the | |
862 _methodHelp(method_name) method to provide the help text used | |
863 in the documentation.""" | |
864 | |
865 methods = {} | |
866 | |
867 for method_name in self.system_listMethods(): | |
868 if method_name in self.funcs: | |
869 method = self.funcs[method_name] | |
870 elif self.instance is not None: | |
871 method_info = [None, None] # argspec, documentation | |
872 if hasattr(self.instance, '_get_method_argstring'): | |
873 method_info[0] = self.instance._get_method_argstring(method_name) | |
874 if hasattr(self.instance, '_methodHelp'): | |
875 method_info[1] = self.instance._methodHelp(method_name) | |
876 | |
877 method_info = tuple(method_info) | |
878 if method_info != (None, None): | |
879 method = method_info | |
880 elif not hasattr(self.instance, '_dispatch'): | |
881 try: | |
882 method = resolve_dotted_attribute( | |
883 self.instance, | |
884 method_name | |
885 ) | |
886 except AttributeError: | |
887 method = method_info | |
888 else: | |
889 method = method_info | |
890 else: | |
891 assert 0, "Could not find method in self.functions and no "\ | |
892 "instance installed" | |
893 | |
894 methods[method_name] = method | |
895 | |
896 documenter = ServerHTMLDoc() | |
897 documentation = documenter.docserver( | |
898 self.server_name, | |
899 self.server_documentation, | |
900 methods | |
901 ) | |
902 | |
903 return documenter.page(self.server_title, documentation) | |
904 | |
905 class DocXMLRPCRequestHandler(SimpleXMLRPCRequestHandler): | |
906 """XML-RPC and documentation request handler class. | |
907 | |
908 Handles all HTTP POST requests and attempts to decode them as | |
909 XML-RPC requests. | |
910 | |
911 Handles all HTTP GET requests and interprets them as requests | |
912 for documentation. | |
913 """ | |
914 | |
915 def do_GET(self): | |
916 """Handles the HTTP GET request. | |
917 | |
918 Interpret all HTTP GET requests as requests for server | |
919 documentation. | |
920 """ | |
921 # Check that the path is legal | |
922 if not self.is_rpc_path_valid(): | |
923 self.report_404() | |
924 return | |
925 | |
926 response = self.server.generate_html_documentation().encode('utf-8') | |
927 self.send_response(200) | |
928 self.send_header("Content-type", "text/html") | |
929 self.send_header("Content-length", str(len(response))) | |
930 self.end_headers() | |
931 self.wfile.write(response) | |
932 | |
933 class DocXMLRPCServer( SimpleXMLRPCServer, | |
934 XMLRPCDocGenerator): | |
935 """XML-RPC and HTML documentation server. | |
936 | |
937 Adds the ability to serve server documentation to the capabilities | |
938 of SimpleXMLRPCServer. | |
939 """ | |
940 | |
941 def __init__(self, addr, requestHandler=DocXMLRPCRequestHandler, | |
942 logRequests=True, allow_none=False, encoding=None, | |
943 bind_and_activate=True, use_builtin_types=False): | |
944 SimpleXMLRPCServer.__init__(self, addr, requestHandler, logRequests, | |
945 allow_none, encoding, bind_and_activate, | |
946 use_builtin_types) | |
947 XMLRPCDocGenerator.__init__(self) | |
948 | |
949 class DocCGIXMLRPCRequestHandler( CGIXMLRPCRequestHandler, | |
950 XMLRPCDocGenerator): | |
951 """Handler for XML-RPC data and documentation requests passed through | |
952 CGI""" | |
953 | |
954 def handle_get(self): | |
955 """Handles the HTTP GET request. | |
956 | |
957 Interpret all HTTP GET requests as requests for server | |
958 documentation. | |
959 """ | |
960 | |
961 response = self.generate_html_documentation().encode('utf-8') | |
962 | |
963 print('Content-Type: text/html') | |
964 print('Content-Length: %d' % len(response)) | |
965 print() | |
966 sys.stdout.flush() | |
967 sys.stdout.buffer.write(response) | |
968 sys.stdout.buffer.flush() | |
969 | |
970 def __init__(self): | |
971 CGIXMLRPCRequestHandler.__init__(self) | |
972 XMLRPCDocGenerator.__init__(self) | |
973 | |
974 | |
975 if __name__ == '__main__': | |
976 import datetime | |
977 | |
978 class ExampleService: | |
979 def getData(self): | |
980 return '42' | |
981 | |
982 class currentTime: | |
983 @staticmethod | |
984 def getCurrentTime(): | |
985 return datetime.datetime.now() | |
986 | |
987 server = SimpleXMLRPCServer(("localhost", 8000)) | |
988 server.register_function(pow) | |
989 server.register_function(lambda x,y: x+y, 'add') | |
990 server.register_instance(ExampleService(), allow_dotted_names=True) | |
991 server.register_multicall_functions() | |
992 print('Serving XML-RPC on localhost port 8000') | |
993 print('It is advisable to run this example server within a secure, closed network.') | |
994 try: | |
995 server.serve_forever() | |
996 except KeyboardInterrupt: | |
997 print("\nKeyboard interrupt received, exiting.") | |
998 server.server_close() | |
999 sys.exit(0) |