blob: 2b5ed6e51f41f879a51d2178d1fe72d0402dac9c [file] [log] [blame]
Fredrik Lundhb9056332001-07-11 17:42:21 +00001#
2# XML-RPC CLIENT LIBRARY
3# $Id$
4#
Fredrik Lundhc4c062f2001-09-10 19:45:02 +00005# an XML-RPC client interface for Python.
6#
7# the marshalling and response parser code can also be used to
8# implement XML-RPC servers.
9#
10# Notes:
Martin v. Löwis37af9862004-08-20 07:31:37 +000011# this version is designed to work with Python 2.1 or newer.
Fredrik Lundhc4c062f2001-09-10 19:45:02 +000012#
Fredrik Lundhb9056332001-07-11 17:42:21 +000013# History:
14# 1999-01-14 fl Created
15# 1999-01-15 fl Changed dateTime to use localtime
16# 1999-01-16 fl Added Binary/base64 element, default to RPC2 service
17# 1999-01-19 fl Fixed array data element (from Skip Montanaro)
18# 1999-01-21 fl Fixed dateTime constructor, etc.
19# 1999-02-02 fl Added fault handling, handle empty sequences, etc.
20# 1999-02-10 fl Fixed problem with empty responses (from Skip Montanaro)
21# 1999-06-20 fl Speed improvements, pluggable parsers/transports (0.9.8)
22# 2000-11-28 fl Changed boolean to check the truth value of its argument
23# 2001-02-24 fl Added encoding/Unicode/SafeTransport patches
24# 2001-02-26 fl Added compare support to wrappers (0.9.9/1.0b1)
25# 2001-03-28 fl Make sure response tuple is a singleton
26# 2001-03-29 fl Don't require empty params element (from Nicholas Riley)
Fredrik Lundh78eedce2001-08-23 20:04:33 +000027# 2001-06-10 fl Folded in _xmlrpclib accelerator support (1.0b2)
28# 2001-08-20 fl Base xmlrpclib.Error on built-in Exception (from Paul Prescod)
Fredrik Lundhc4c062f2001-09-10 19:45:02 +000029# 2001-09-03 fl Allow Transport subclass to override getparser
30# 2001-09-10 fl Lazy import of urllib, cgi, xmllib (20x import speedup)
Fredrik Lundh1538c232001-10-01 19:42:03 +000031# 2001-10-01 fl Remove containers from memo cache when done with them
32# 2001-10-01 fl Use faster escape method (80% dumps speedup)
Fredrik Lundh3d9addd2002-06-27 21:36:21 +000033# 2001-10-02 fl More dumps microtuning
34# 2001-10-04 fl Make sure import expat gets a parser (from Guido van Rossum)
Skip Montanaro5e9c71b2001-10-10 15:56:34 +000035# 2001-10-10 sm Allow long ints to be passed as ints if they don't overflow
Fredrik Lundh3d9addd2002-06-27 21:36:21 +000036# 2001-10-17 sm Test for int and long overflow (allows use on 64-bit systems)
Fredrik Lundhb6ab93f2001-12-19 21:40:04 +000037# 2001-11-12 fl Use repr() to marshal doubles (from Paul Felix)
Fredrik Lundh3d9addd2002-06-27 21:36:21 +000038# 2002-03-17 fl Avoid buffered read when possible (from James Rucker)
39# 2002-04-07 fl Added pythondoc comments
40# 2002-04-16 fl Added __str__ methods to datetime/binary wrappers
41# 2002-05-15 fl Added error constants (from Andrew Kuchling)
42# 2002-06-27 fl Merged with Python CVS version
Fredrik Lundh1303c7c2002-10-22 18:23:00 +000043# 2002-10-22 fl Added basic authentication (based on code from Phillip Eby)
Martin v. Löwis541342f2003-07-12 07:53:04 +000044# 2003-01-22 sm Add support for the bool type
45# 2003-02-27 gvr Remove apply calls
46# 2003-04-24 sm Use cStringIO if available
47# 2003-04-25 ak Add support for nil
48# 2003-06-15 gn Add support for time.struct_time
49# 2003-07-12 gp Correct marshalling of Faults
Martin v. Löwis45394c22003-10-31 13:49:36 +000050# 2003-10-31 mvl Add multicall support
Martin v. Löwis37af9862004-08-20 07:31:37 +000051# 2004-08-20 mvl Bump minimum supported Python version to 2.1
Fredrik Lundhb9056332001-07-11 17:42:21 +000052#
Fredrik Lundh3d9addd2002-06-27 21:36:21 +000053# Copyright (c) 1999-2002 by Secret Labs AB.
54# Copyright (c) 1999-2002 by Fredrik Lundh.
Fredrik Lundhb9056332001-07-11 17:42:21 +000055#
56# info@pythonware.com
57# http://www.pythonware.com
58#
59# --------------------------------------------------------------------
60# The XML-RPC client interface is
61#
Fredrik Lundh3d9addd2002-06-27 21:36:21 +000062# Copyright (c) 1999-2002 by Secret Labs AB
63# Copyright (c) 1999-2002 by Fredrik Lundh
Fredrik Lundhb9056332001-07-11 17:42:21 +000064#
65# By obtaining, using, and/or copying this software and/or its
66# associated documentation, you agree that you have read, understood,
67# and will comply with the following terms and conditions:
68#
69# Permission to use, copy, modify, and distribute this software and
70# its associated documentation for any purpose and without fee is
71# hereby granted, provided that the above copyright notice appears in
72# all copies, and that both that copyright notice and this permission
73# notice appear in supporting documentation, and that the name of
74# Secret Labs AB or the author not be used in advertising or publicity
75# pertaining to distribution of the software without specific, written
76# prior permission.
77#
78# SECRET LABS AB AND THE AUTHOR DISCLAIMS ALL WARRANTIES WITH REGARD
79# TO THIS SOFTWARE, INCLUDING ALL IMPLIED WARRANTIES OF MERCHANT-
80# ABILITY AND FITNESS. IN NO EVENT SHALL SECRET LABS AB OR THE AUTHOR
81# BE LIABLE FOR ANY SPECIAL, INDIRECT OR CONSEQUENTIAL DAMAGES OR ANY
82# DAMAGES WHATSOEVER RESULTING FROM LOSS OF USE, DATA OR PROFITS,
83# WHETHER IN AN ACTION OF CONTRACT, NEGLIGENCE OR OTHER TORTIOUS
84# ACTION, ARISING OUT OF OR IN CONNECTION WITH THE USE OR PERFORMANCE
85# OF THIS SOFTWARE.
86# --------------------------------------------------------------------
87
88#
Fredrik Lundh3d9addd2002-06-27 21:36:21 +000089# things to look into some day:
Fredrik Lundhb9056332001-07-11 17:42:21 +000090
Fredrik Lundh3d9addd2002-06-27 21:36:21 +000091# TODO: sort out True/False/boolean issues for Python 2.3
Fredrik Lundhb9056332001-07-11 17:42:21 +000092
Fred Drake1b410792001-09-04 18:55:03 +000093"""
94An XML-RPC client interface for Python.
95
96The marshalling and response parser code can also be used to
97implement XML-RPC servers.
98
Fred Drake1b410792001-09-04 18:55:03 +000099Exported exceptions:
100
Fredrik Lundhc4c062f2001-09-10 19:45:02 +0000101 Error Base class for client errors
102 ProtocolError Indicates an HTTP protocol error
103 ResponseError Indicates a broken response package
104 Fault Indicates an XML-RPC fault package
Fred Drake1b410792001-09-04 18:55:03 +0000105
106Exported classes:
107
Fredrik Lundhc4c062f2001-09-10 19:45:02 +0000108 ServerProxy Represents a logical connection to an XML-RPC server
Fred Drake1b410792001-09-04 18:55:03 +0000109
Raymond Hettingercc523fc2003-11-02 09:47:05 +0000110 MultiCall Executor of boxcared xmlrpc requests
Fredrik Lundhc4c062f2001-09-10 19:45:02 +0000111 Boolean boolean wrapper to generate a "boolean" XML-RPC value
112 DateTime dateTime wrapper for an ISO 8601 string or time tuple or
113 localtime integer value to generate a "dateTime.iso8601"
114 XML-RPC value
115 Binary binary data wrapper
Fred Drake1b410792001-09-04 18:55:03 +0000116
Fredrik Lundhc4c062f2001-09-10 19:45:02 +0000117 SlowParser Slow but safe standard parser (based on xmllib)
118 Marshaller Generate an XML-RPC params chunk from a Python data structure
119 Unmarshaller Unmarshal an XML-RPC response from incoming XML event message
120 Transport Handles an HTTP transaction to an XML-RPC server
121 SafeTransport Handles an HTTPS transaction to an XML-RPC server
Fred Drake1b410792001-09-04 18:55:03 +0000122
123Exported constants:
124
Fredrik Lundhc4c062f2001-09-10 19:45:02 +0000125 True
126 False
Fred Drake1b410792001-09-04 18:55:03 +0000127
128Exported functions:
129
Fredrik Lundhc4c062f2001-09-10 19:45:02 +0000130 boolean Convert any Python value to an XML-RPC boolean
131 getparser Create instance of the fastest available parser & attach
132 to an unmarshalling object
133 dumps Convert an argument tuple or a Fault instance to an XML-RPC
134 request (or response, if the methodresponse option is used).
135 loads Convert an XML-RPC packet to unmarshalled data plus a method
136 name (None if not present).
Fred Drake1b410792001-09-04 18:55:03 +0000137"""
138
Neal Norwitzff113342007-04-17 08:42:15 +0000139import re, time, operator
Fredrik Lundh1538c232001-10-01 19:42:03 +0000140
Raymond Hettingerfe59dc12005-02-07 15:28:45 +0000141from types import *
Fredrik Lundhb9056332001-07-11 17:42:21 +0000142
Fredrik Lundh3d9addd2002-06-27 21:36:21 +0000143# --------------------------------------------------------------------
144# Internal stuff
145
Fredrik Lundhb9056332001-07-11 17:42:21 +0000146try:
Fred Drakeba613c32005-02-10 18:33:30 +0000147 import datetime
148except ImportError:
149 datetime = None
150
151try:
Skip Montanaro9a7c96a2003-01-22 18:17:25 +0000152 _bool_is_builtin = False.__class__.__name__ == "bool"
153except NameError:
154 _bool_is_builtin = 0
155
Fredrik Lundhb9056332001-07-11 17:42:21 +0000156def _decode(data, encoding, is8bit=re.compile("[\x80-\xff]").search):
157 # decode non-ascii string (if possible)
Guido van Rossum54ad5232007-05-27 09:17:48 +0000158 if encoding and is8bit(data):
Guido van Rossumef87d6e2007-05-02 19:09:54 +0000159 data = str(data, encoding)
Fredrik Lundhb9056332001-07-11 17:42:21 +0000160 return data
161
Neal Norwitzff113342007-04-17 08:42:15 +0000162def escape(s):
163 s = s.replace("&", "&")
164 s = s.replace("<", "&lt;")
165 return s.replace(">", "&gt;",)
Fredrik Lundh1538c232001-10-01 19:42:03 +0000166
Guido van Rossum54ad5232007-05-27 09:17:48 +0000167def _stringify(string):
168 # convert to 7-bit ascii if possible
169 try:
170 return string.encode("ascii")
171 except UnicodeError:
Fredrik Lundhb9056332001-07-11 17:42:21 +0000172 return string
173
Fredrik Lundh3d9addd2002-06-27 21:36:21 +0000174__version__ = "1.0.1"
175
176# xmlrpc integer limits
Guido van Rossume2a383d2007-01-15 16:59:06 +0000177MAXINT = 2**31-1
178MININT = -2**31
Fredrik Lundh3d9addd2002-06-27 21:36:21 +0000179
180# --------------------------------------------------------------------
181# Error constants (from Dan Libby's specification at
182# http://xmlrpc-epi.sourceforge.net/specs/rfc.fault_codes.php)
183
184# Ranges of errors
185PARSE_ERROR = -32700
186SERVER_ERROR = -32600
187APPLICATION_ERROR = -32500
188SYSTEM_ERROR = -32400
189TRANSPORT_ERROR = -32300
190
191# Specific errors
192NOT_WELLFORMED_ERROR = -32700
193UNSUPPORTED_ENCODING = -32701
194INVALID_ENCODING_CHAR = -32702
195INVALID_XMLRPC = -32600
196METHOD_NOT_FOUND = -32601
197INVALID_METHOD_PARAMS = -32602
198INTERNAL_ERROR = -32603
Fredrik Lundhb9056332001-07-11 17:42:21 +0000199
200# --------------------------------------------------------------------
201# Exceptions
202
Fredrik Lundh3d9addd2002-06-27 21:36:21 +0000203##
204# Base class for all kinds of client-side errors.
205
Fredrik Lundh78eedce2001-08-23 20:04:33 +0000206class Error(Exception):
Fred Drake1b410792001-09-04 18:55:03 +0000207 """Base class for client errors."""
Fredrik Lundhb0e8e9b2001-09-10 21:45:42 +0000208 def __str__(self):
209 return repr(self)
Fredrik Lundhb9056332001-07-11 17:42:21 +0000210
Fredrik Lundh3d9addd2002-06-27 21:36:21 +0000211##
212# Indicates an HTTP-level protocol error. This is raised by the HTTP
213# transport layer, if the server returns an error code other than 200
214# (OK).
215#
216# @param url The target URL.
217# @param errcode The HTTP error code.
218# @param errmsg The HTTP error message.
219# @param headers The HTTP header dictionary.
220
Fredrik Lundhb9056332001-07-11 17:42:21 +0000221class ProtocolError(Error):
Fred Drake1b410792001-09-04 18:55:03 +0000222 """Indicates an HTTP protocol error."""
Fredrik Lundhb9056332001-07-11 17:42:21 +0000223 def __init__(self, url, errcode, errmsg, headers):
Fredrik Lundhb0e8e9b2001-09-10 21:45:42 +0000224 Error.__init__(self)
Fredrik Lundhb9056332001-07-11 17:42:21 +0000225 self.url = url
226 self.errcode = errcode
227 self.errmsg = errmsg
228 self.headers = headers
229 def __repr__(self):
230 return (
231 "<ProtocolError for %s: %s %s>" %
232 (self.url, self.errcode, self.errmsg)
233 )
234
Fredrik Lundh3d9addd2002-06-27 21:36:21 +0000235##
236# Indicates a broken XML-RPC response package. This exception is
237# raised by the unmarshalling layer, if the XML-RPC response is
238# malformed.
239
Fredrik Lundhb9056332001-07-11 17:42:21 +0000240class ResponseError(Error):
Fredrik Lundhc4c062f2001-09-10 19:45:02 +0000241 """Indicates a broken response package."""
Fredrik Lundhb9056332001-07-11 17:42:21 +0000242 pass
243
Fredrik Lundh3d9addd2002-06-27 21:36:21 +0000244##
245# Indicates an XML-RPC fault response package. This exception is
246# raised by the unmarshalling layer, if the XML-RPC response contains
247# a fault string. This exception can also used as a class, to
248# generate a fault XML-RPC message.
249#
250# @param faultCode The XML-RPC fault code.
251# @param faultString The XML-RPC fault string.
252
Fredrik Lundhb9056332001-07-11 17:42:21 +0000253class Fault(Error):
Fredrik Lundhc4c062f2001-09-10 19:45:02 +0000254 """Indicates an XML-RPC fault package."""
Fredrik Lundhb9056332001-07-11 17:42:21 +0000255 def __init__(self, faultCode, faultString, **extra):
Fredrik Lundhb0e8e9b2001-09-10 21:45:42 +0000256 Error.__init__(self)
Fredrik Lundhb9056332001-07-11 17:42:21 +0000257 self.faultCode = faultCode
258 self.faultString = faultString
259 def __repr__(self):
260 return (
261 "<Fault %s: %s>" %
Fredrik Lundhb0e8e9b2001-09-10 21:45:42 +0000262 (self.faultCode, repr(self.faultString))
Fredrik Lundhb9056332001-07-11 17:42:21 +0000263 )
264
265# --------------------------------------------------------------------
266# Special values
267
Fredrik Lundh3d9addd2002-06-27 21:36:21 +0000268##
269# Wrapper for XML-RPC boolean values. Use the xmlrpclib.True and
270# xmlrpclib.False constants, or the xmlrpclib.boolean() function, to
271# generate boolean XML-RPC values.
272#
273# @param value A boolean value. Any true value is interpreted as True,
274# all other values are interpreted as False.
275
Guido van Rossum47b9ff62006-08-24 00:41:19 +0000276boolean = Boolean = bool
Fredrik Lundh3d9addd2002-06-27 21:36:21 +0000277
278##
279# Wrapper for XML-RPC DateTime values. This converts a time value to
280# the format used by XML-RPC.
281# <p>
282# The value can be given as a string in the format
283# "yyyymmddThh:mm:ss", as a 9-item time tuple (as returned by
284# time.localtime()), or an integer value (as returned by time.time()).
285# The wrapper uses time.localtime() to convert an integer to a time
286# tuple.
287#
288# @param value The time, given as an ISO 8601 string, a time
289# tuple, or a integer time value.
Fredrik Lundhb9056332001-07-11 17:42:21 +0000290
Fredrik Lundhb9056332001-07-11 17:42:21 +0000291class DateTime:
Fredrik Lundhc4c062f2001-09-10 19:45:02 +0000292 """DateTime wrapper for an ISO 8601 string or time tuple or
293 localtime integer value to generate 'dateTime.iso8601' XML-RPC
294 value.
295 """
Fredrik Lundhb9056332001-07-11 17:42:21 +0000296
297 def __init__(self, value=0):
Guido van Rossum54ad5232007-05-27 09:17:48 +0000298 if not isinstance(value, basestring):
Fred Drakeba613c32005-02-10 18:33:30 +0000299 if datetime and isinstance(value, datetime.datetime):
300 self.value = value.strftime("%Y%m%dT%H:%M:%S")
301 return
Skip Montanaro174dd222005-05-14 20:54:16 +0000302 if datetime and isinstance(value, datetime.date):
303 self.value = value.strftime("%Y%m%dT%H:%M:%S")
304 return
305 if datetime and isinstance(value, datetime.time):
306 today = datetime.datetime.now().strftime("%Y%m%d")
307 self.value = value.strftime(today+"T%H:%M:%S")
308 return
309 if not isinstance(value, (TupleType, time.struct_time)):
Fredrik Lundhb9056332001-07-11 17:42:21 +0000310 if value == 0:
311 value = time.time()
312 value = time.localtime(value)
313 value = time.strftime("%Y%m%dT%H:%M:%S", value)
314 self.value = value
315
Guido van Rossum47b9ff62006-08-24 00:41:19 +0000316 def __eq__(self, other):
Fredrik Lundhb9056332001-07-11 17:42:21 +0000317 if isinstance(other, DateTime):
318 other = other.value
Guido van Rossum47b9ff62006-08-24 00:41:19 +0000319 return self.value == other
320
321 def __ne__(self, other):
322 if isinstance(other, DateTime):
323 other = other.value
324 return self.value != other
Fredrik Lundhb9056332001-07-11 17:42:21 +0000325
Fredrik Lundh3d9addd2002-06-27 21:36:21 +0000326 ##
327 # Get date/time value.
328 #
329 # @return Date/time value, as an ISO 8601 string.
330
331 def __str__(self):
332 return self.value
333
Fredrik Lundhb9056332001-07-11 17:42:21 +0000334 def __repr__(self):
Fredrik Lundh3d9addd2002-06-27 21:36:21 +0000335 return "<DateTime %s at %x>" % (repr(self.value), id(self))
Fredrik Lundhb9056332001-07-11 17:42:21 +0000336
337 def decode(self, data):
Neal Norwitzff113342007-04-17 08:42:15 +0000338 self.value = str(data).strip()
Fredrik Lundhb9056332001-07-11 17:42:21 +0000339
340 def encode(self, out):
341 out.write("<value><dateTime.iso8601>")
342 out.write(self.value)
343 out.write("</dateTime.iso8601></value>\n")
344
Fredrik Lundh3d9addd2002-06-27 21:36:21 +0000345def _datetime(data):
346 # decode xml element contents into a DateTime structure.
Fredrik Lundhb9056332001-07-11 17:42:21 +0000347 value = DateTime()
348 value.decode(data)
349 return value
350
Skip Montanaro174dd222005-05-14 20:54:16 +0000351def _datetime_type(data):
352 t = time.strptime(data, "%Y%m%dT%H:%M:%S")
353 return datetime.datetime(*tuple(t)[:6])
354
Fredrik Lundh3d9addd2002-06-27 21:36:21 +0000355##
356# Wrapper for binary data. This can be used to transport any kind
357# of binary data over XML-RPC, using BASE64 encoding.
358#
359# @param data An 8-bit string containing arbitrary data.
360
Skip Montanarobfcbfa72003-04-24 19:51:31 +0000361import base64
Guido van Rossum68937b42007-05-18 00:51:22 +0000362import io
Skip Montanarobfcbfa72003-04-24 19:51:31 +0000363
Fredrik Lundhb9056332001-07-11 17:42:21 +0000364class Binary:
Fred Drake1b410792001-09-04 18:55:03 +0000365 """Wrapper for binary data."""
Fredrik Lundhb9056332001-07-11 17:42:21 +0000366
367 def __init__(self, data=None):
368 self.data = data
369
Fredrik Lundh3d9addd2002-06-27 21:36:21 +0000370 ##
371 # Get buffer contents.
372 #
373 # @return Buffer contents, as an 8-bit string.
374
375 def __str__(self):
376 return self.data or ""
377
Guido van Rossum47b9ff62006-08-24 00:41:19 +0000378 def __eq__(self, other):
Fredrik Lundhb9056332001-07-11 17:42:21 +0000379 if isinstance(other, Binary):
380 other = other.data
Guido van Rossum47b9ff62006-08-24 00:41:19 +0000381 return self.data == other
382
383 def __ne__(self, other):
384 if isinstance(other, Binary):
385 other = other.data
386 return self.data != other
Fredrik Lundhb9056332001-07-11 17:42:21 +0000387
388 def decode(self, data):
Fredrik Lundhb9056332001-07-11 17:42:21 +0000389 self.data = base64.decodestring(data)
390
391 def encode(self, out):
Fredrik Lundhb9056332001-07-11 17:42:21 +0000392 out.write("<value><base64>\n")
Guido van Rossum68937b42007-05-18 00:51:22 +0000393 base64.encode(io.StringIO(self.data), out)
Fredrik Lundhb9056332001-07-11 17:42:21 +0000394 out.write("</base64></value>\n")
395
Fredrik Lundh3d9addd2002-06-27 21:36:21 +0000396def _binary(data):
397 # decode xml element contents into a Binary structure
Fredrik Lundhb9056332001-07-11 17:42:21 +0000398 value = Binary()
399 value.decode(data)
400 return value
401
Skip Montanaro9a7c96a2003-01-22 18:17:25 +0000402WRAPPERS = (DateTime, Binary)
403if not _bool_is_builtin:
404 WRAPPERS = WRAPPERS + (Boolean,)
Fredrik Lundhb9056332001-07-11 17:42:21 +0000405
406# --------------------------------------------------------------------
407# XML parsers
408
409try:
Fredrik Lundh7069c312004-10-13 06:48:37 +0000410 # optional xmlrpclib accelerator
Fredrik Lundhb9056332001-07-11 17:42:21 +0000411 import _xmlrpclib
412 FastParser = _xmlrpclib.Parser
413 FastUnmarshaller = _xmlrpclib.Unmarshaller
414except (AttributeError, ImportError):
415 FastParser = FastUnmarshaller = None
416
Fredrik Lundh3d9addd2002-06-27 21:36:21 +0000417try:
418 import _xmlrpclib
419 FastMarshaller = _xmlrpclib.Marshaller
420except (AttributeError, ImportError):
421 FastMarshaller = None
422
Fredrik Lundhb9056332001-07-11 17:42:21 +0000423#
424# the SGMLOP parser is about 15x faster than Python's builtin
425# XML parser. SGMLOP sources can be downloaded from:
426#
427# http://www.pythonware.com/products/xml/sgmlop.htm
428#
429
430try:
431 import sgmlop
432 if not hasattr(sgmlop, "XMLParser"):
433 raise ImportError
434except ImportError:
435 SgmlopParser = None # sgmlop accelerator not available
436else:
437 class SgmlopParser:
438 def __init__(self, target):
439
440 # setup callbacks
441 self.finish_starttag = target.start
442 self.finish_endtag = target.end
443 self.handle_data = target.data
444 self.handle_xml = target.xml
445
446 # activate parser
447 self.parser = sgmlop.XMLParser()
448 self.parser.register(self)
449 self.feed = self.parser.feed
450 self.entity = {
451 "amp": "&", "gt": ">", "lt": "<",
452 "apos": "'", "quot": '"'
453 }
454
455 def close(self):
456 try:
457 self.parser.close()
458 finally:
459 self.parser = self.feed = None # nuke circular reference
460
461 def handle_proc(self, tag, attr):
462 m = re.search("encoding\s*=\s*['\"]([^\"']+)[\"']", attr)
463 if m:
464 self.handle_xml(m.group(1), 1)
465
466 def handle_entityref(self, entity):
467 # <string> entity
468 try:
469 self.handle_data(self.entity[entity])
470 except KeyError:
471 self.handle_data("&%s;" % entity)
472
473try:
474 from xml.parsers import expat
Guido van Rossumb8551342001-10-02 18:33:11 +0000475 if not hasattr(expat, "ParserCreate"):
Fredrik Lundh3d9addd2002-06-27 21:36:21 +0000476 raise ImportError
Fredrik Lundhb9056332001-07-11 17:42:21 +0000477except ImportError:
Fredrik Lundh3d9addd2002-06-27 21:36:21 +0000478 ExpatParser = None # expat not available
Fredrik Lundhb9056332001-07-11 17:42:21 +0000479else:
480 class ExpatParser:
Fredrik Lundh3d9addd2002-06-27 21:36:21 +0000481 # fast expat parser for Python 2.0 and later. this is about
482 # 50% slower than sgmlop, on roundtrip testing
Fredrik Lundhb9056332001-07-11 17:42:21 +0000483 def __init__(self, target):
484 self._parser = parser = expat.ParserCreate(None, None)
485 self._target = target
486 parser.StartElementHandler = target.start
487 parser.EndElementHandler = target.end
488 parser.CharacterDataHandler = target.data
489 encoding = None
490 if not parser.returns_unicode:
491 encoding = "utf-8"
492 target.xml(encoding, None)
493
494 def feed(self, data):
495 self._parser.Parse(data, 0)
496
497 def close(self):
498 self._parser.Parse("", 1) # end of data
499 del self._target, self._parser # get rid of circular references
500
Fredrik Lundhb9056332001-07-11 17:42:21 +0000501# --------------------------------------------------------------------
502# XML-RPC marshalling and unmarshalling code
503
Fredrik Lundh3d9addd2002-06-27 21:36:21 +0000504##
505# XML-RPC marshaller.
506#
507# @param encoding Default encoding for 8-bit strings. The default
508# value is None (interpreted as UTF-8).
509# @see dumps
510
Fredrik Lundhb9056332001-07-11 17:42:21 +0000511class Marshaller:
Fred Drake1b410792001-09-04 18:55:03 +0000512 """Generate an XML-RPC params chunk from a Python data structure.
Fredrik Lundhb9056332001-07-11 17:42:21 +0000513
Fredrik Lundhc4c062f2001-09-10 19:45:02 +0000514 Create a Marshaller instance for each set of parameters, and use
515 the "dumps" method to convert your data (represented as a tuple)
516 to an XML-RPC params chunk. To write a fault response, pass a
517 Fault instance instead. You may prefer to use the "dumps" module
518 function for this purpose.
Fred Drake1b410792001-09-04 18:55:03 +0000519 """
Fredrik Lundhb9056332001-07-11 17:42:21 +0000520
521 # by the way, if you don't understand what's going on in here,
522 # that's perfectly ok.
523
Andrew M. Kuchlinga4c2b742003-04-25 00:26:51 +0000524 def __init__(self, encoding=None, allow_none=0):
Fredrik Lundhb9056332001-07-11 17:42:21 +0000525 self.memo = {}
526 self.data = None
527 self.encoding = encoding
Andrew M. Kuchlinga4c2b742003-04-25 00:26:51 +0000528 self.allow_none = allow_none
Tim Petersc2659cf2003-05-12 20:19:37 +0000529
Fredrik Lundhb9056332001-07-11 17:42:21 +0000530 dispatch = {}
531
532 def dumps(self, values):
Fredrik Lundh3d9addd2002-06-27 21:36:21 +0000533 out = []
534 write = out.append
535 dump = self.__dump
Fredrik Lundhb9056332001-07-11 17:42:21 +0000536 if isinstance(values, Fault):
537 # fault instance
538 write("<fault>\n")
Martin v. Löwis541342f2003-07-12 07:53:04 +0000539 dump({'faultCode': values.faultCode,
540 'faultString': values.faultString},
541 write)
Fredrik Lundhb9056332001-07-11 17:42:21 +0000542 write("</fault>\n")
543 else:
544 # parameter block
Fredrik Lundhc266bb02001-08-23 20:13:08 +0000545 # FIXME: the xml-rpc specification allows us to leave out
546 # the entire <params> block if there are no parameters.
547 # however, changing this may break older code (including
548 # old versions of xmlrpclib.py), so this is better left as
549 # is for now. See @XMLRPC3 for more information. /F
Fredrik Lundhb9056332001-07-11 17:42:21 +0000550 write("<params>\n")
551 for v in values:
552 write("<param>\n")
Fredrik Lundh3d9addd2002-06-27 21:36:21 +0000553 dump(v, write)
Fredrik Lundhb9056332001-07-11 17:42:21 +0000554 write("</param>\n")
555 write("</params>\n")
Neal Norwitzff113342007-04-17 08:42:15 +0000556 result = "".join(out)
Fredrik Lundhb9056332001-07-11 17:42:21 +0000557 return result
558
Fredrik Lundh3d9addd2002-06-27 21:36:21 +0000559 def __dump(self, value, write):
Fredrik Lundhb9056332001-07-11 17:42:21 +0000560 try:
561 f = self.dispatch[type(value)]
562 except KeyError:
Thomas Wouters89f507f2006-12-13 04:49:30 +0000563 # check if this object can be marshalled as a structure
564 try:
565 value.__dict__
566 except:
567 raise TypeError, "cannot marshal %s objects" % type(value)
568 # check if this class is a sub-class of a basic type,
569 # because we don't know how to marshal these types
570 # (e.g. a string sub-class)
571 for type_ in type(value).__mro__:
572 if type_ in self.dispatch.keys():
573 raise TypeError, "cannot marshal %s objects" % type(value)
574 # XXX(twouters): using "_arbitrary_instance" as key as a quick-fix
575 # for the p3yk merge, this should probably be fixed more neatly.
576 f = self.dispatch["_arbitrary_instance"]
577 f(self, value, write)
Fredrik Lundhb9056332001-07-11 17:42:21 +0000578
Andrew M. Kuchlinga4c2b742003-04-25 00:26:51 +0000579 def dump_nil (self, value, write):
580 if not self.allow_none:
581 raise TypeError, "cannot marshal None unless allow_none is enabled"
582 write("<value><nil/></value>")
Raymond Hettingerfe59dc12005-02-07 15:28:45 +0000583 dispatch[NoneType] = dump_nil
Tim Petersc2659cf2003-05-12 20:19:37 +0000584
Fredrik Lundh3d9addd2002-06-27 21:36:21 +0000585 def dump_int(self, value, write):
Skip Montanaro5449e082001-10-17 22:53:33 +0000586 # in case ints are > 32 bits
587 if value > MAXINT or value < MININT:
588 raise OverflowError, "int exceeds XML-RPC limits"
Fredrik Lundh3d9addd2002-06-27 21:36:21 +0000589 write("<value><int>")
590 write(str(value))
591 write("</int></value>\n")
Raymond Hettingerfe59dc12005-02-07 15:28:45 +0000592 dispatch[IntType] = dump_int
Fredrik Lundhb9056332001-07-11 17:42:21 +0000593
Skip Montanaro9a7c96a2003-01-22 18:17:25 +0000594 if _bool_is_builtin:
595 def dump_bool(self, value, write):
596 write("<value><boolean>")
597 write(value and "1" or "0")
598 write("</boolean></value>\n")
599 dispatch[bool] = dump_bool
600
Fredrik Lundh3d9addd2002-06-27 21:36:21 +0000601 def dump_long(self, value, write):
Skip Montanaro5449e082001-10-17 22:53:33 +0000602 if value > MAXINT or value < MININT:
603 raise OverflowError, "long int exceeds XML-RPC limits"
Fredrik Lundh3d9addd2002-06-27 21:36:21 +0000604 write("<value><int>")
605 write(str(int(value)))
606 write("</int></value>\n")
Raymond Hettingerfe59dc12005-02-07 15:28:45 +0000607 dispatch[LongType] = dump_long
Skip Montanaro5e9c71b2001-10-10 15:56:34 +0000608
Fredrik Lundh3d9addd2002-06-27 21:36:21 +0000609 def dump_double(self, value, write):
610 write("<value><double>")
611 write(repr(value))
612 write("</double></value>\n")
Raymond Hettingerfe59dc12005-02-07 15:28:45 +0000613 dispatch[FloatType] = dump_double
Fredrik Lundhb9056332001-07-11 17:42:21 +0000614
Fredrik Lundh3d9addd2002-06-27 21:36:21 +0000615 def dump_string(self, value, write, escape=escape):
616 write("<value><string>")
617 write(escape(value))
618 write("</string></value>\n")
Guido van Rossum54ad5232007-05-27 09:17:48 +0000619 dispatch[str8] = dump_string
Fredrik Lundhb9056332001-07-11 17:42:21 +0000620
Guido van Rossum54ad5232007-05-27 09:17:48 +0000621 def dump_unicode(self, value, write, escape=escape):
622 value = value.encode(self.encoding)
623 write("<value><string>")
624 write(escape(value))
625 write("</string></value>\n")
626 dispatch[str] = dump_unicode
Fredrik Lundhb9056332001-07-11 17:42:21 +0000627
Fredrik Lundh3d9addd2002-06-27 21:36:21 +0000628 def dump_array(self, value, write):
629 i = id(value)
Guido van Rossume2b70bc2006-08-18 22:13:04 +0000630 if i in self.memo:
Fredrik Lundh3d9addd2002-06-27 21:36:21 +0000631 raise TypeError, "cannot marshal recursive sequences"
632 self.memo[i] = None
Fredrik Lundh1538c232001-10-01 19:42:03 +0000633 dump = self.__dump
Fredrik Lundhb9056332001-07-11 17:42:21 +0000634 write("<value><array><data>\n")
635 for v in value:
Fredrik Lundh3d9addd2002-06-27 21:36:21 +0000636 dump(v, write)
Fredrik Lundhb9056332001-07-11 17:42:21 +0000637 write("</data></array></value>\n")
Fredrik Lundh3d9addd2002-06-27 21:36:21 +0000638 del self.memo[i]
Raymond Hettingerfe59dc12005-02-07 15:28:45 +0000639 dispatch[TupleType] = dump_array
640 dispatch[ListType] = dump_array
Fredrik Lundhb9056332001-07-11 17:42:21 +0000641
Fredrik Lundh3d9addd2002-06-27 21:36:21 +0000642 def dump_struct(self, value, write, escape=escape):
643 i = id(value)
Guido van Rossume2b70bc2006-08-18 22:13:04 +0000644 if i in self.memo:
Fredrik Lundh3d9addd2002-06-27 21:36:21 +0000645 raise TypeError, "cannot marshal recursive dictionaries"
646 self.memo[i] = None
Fredrik Lundh1538c232001-10-01 19:42:03 +0000647 dump = self.__dump
Fredrik Lundhb9056332001-07-11 17:42:21 +0000648 write("<value><struct>\n")
Andrew M. Kuchling5962f452004-06-05 12:35:58 +0000649 for k, v in value.items():
Fredrik Lundhb9056332001-07-11 17:42:21 +0000650 write("<member>\n")
Guido van Rossum54ad5232007-05-27 09:17:48 +0000651 if isinstance(k, basestring):
652 k = k.encode(self.encoding)
653 else:
654 raise TypeError, "dictionary key must be string"
Fredrik Lundh1538c232001-10-01 19:42:03 +0000655 write("<name>%s</name>\n" % escape(k))
Andrew M. Kuchling5962f452004-06-05 12:35:58 +0000656 dump(v, write)
Fredrik Lundhb9056332001-07-11 17:42:21 +0000657 write("</member>\n")
658 write("</struct></value>\n")
Fredrik Lundh3d9addd2002-06-27 21:36:21 +0000659 del self.memo[i]
Raymond Hettingerfe59dc12005-02-07 15:28:45 +0000660 dispatch[DictType] = dump_struct
Fredrik Lundhb9056332001-07-11 17:42:21 +0000661
Fred Drakeba613c32005-02-10 18:33:30 +0000662 if datetime:
663 def dump_datetime(self, value, write):
664 write("<value><dateTime.iso8601>")
665 write(value.strftime("%Y%m%dT%H:%M:%S"))
666 write("</dateTime.iso8601></value>\n")
667 dispatch[datetime.datetime] = dump_datetime
668
Skip Montanaro174dd222005-05-14 20:54:16 +0000669 def dump_date(self, value, write):
670 write("<value><dateTime.iso8601>")
671 write(value.strftime("%Y%m%dT00:00:00"))
672 write("</dateTime.iso8601></value>\n")
673 dispatch[datetime.date] = dump_date
674
675 def dump_time(self, value, write):
676 write("<value><dateTime.iso8601>")
677 write(datetime.datetime.now().date().strftime("%Y%m%dT"))
678 write(value.strftime("%H:%M:%S"))
679 write("</dateTime.iso8601></value>\n")
680 dispatch[datetime.time] = dump_time
681
Fredrik Lundh3d9addd2002-06-27 21:36:21 +0000682 def dump_instance(self, value, write):
Fredrik Lundhb9056332001-07-11 17:42:21 +0000683 # check for special wrappers
684 if value.__class__ in WRAPPERS:
Fredrik Lundh3d9addd2002-06-27 21:36:21 +0000685 self.write = write
Fredrik Lundhb9056332001-07-11 17:42:21 +0000686 value.encode(self)
Fredrik Lundh3d9addd2002-06-27 21:36:21 +0000687 del self.write
Fredrik Lundhb9056332001-07-11 17:42:21 +0000688 else:
689 # store instance attributes as a struct (really?)
Fredrik Lundh3d9addd2002-06-27 21:36:21 +0000690 self.dump_struct(value.__dict__, write)
Guido van Rossume4dea982006-04-21 13:45:00 +0000691 dispatch[DateTime] = dump_instance
692 dispatch[Binary] = dump_instance
Thomas Wouters89f507f2006-12-13 04:49:30 +0000693 # XXX(twouters): using "_arbitrary_instance" as key as a quick-fix
694 # for the p3yk merge, this should probably be fixed more neatly.
695 dispatch["_arbitrary_instance"] = dump_instance
Fredrik Lundhb9056332001-07-11 17:42:21 +0000696
Fredrik Lundh3d9addd2002-06-27 21:36:21 +0000697##
698# XML-RPC unmarshaller.
699#
700# @see loads
701
Fredrik Lundhb9056332001-07-11 17:42:21 +0000702class Unmarshaller:
Fred Drake1b410792001-09-04 18:55:03 +0000703 """Unmarshal an XML-RPC response, based on incoming XML event
Fredrik Lundhb0e8e9b2001-09-10 21:45:42 +0000704 messages (start, data, end). Call close() to get the resulting
Fred Drake1b410792001-09-04 18:55:03 +0000705 data structure.
Fredrik Lundhb9056332001-07-11 17:42:21 +0000706
Fredrik Lundhb0e8e9b2001-09-10 21:45:42 +0000707 Note that this reader is fairly tolerant, and gladly accepts bogus
708 XML-RPC data without complaining (but not bogus XML).
Fred Drake1b410792001-09-04 18:55:03 +0000709 """
Fredrik Lundhb9056332001-07-11 17:42:21 +0000710
711 # and again, if you don't understand what's going on in here,
712 # that's perfectly ok.
713
Skip Montanaro174dd222005-05-14 20:54:16 +0000714 def __init__(self, use_datetime=0):
Fredrik Lundhb9056332001-07-11 17:42:21 +0000715 self._type = None
716 self._stack = []
717 self._marks = []
718 self._data = []
719 self._methodname = None
720 self._encoding = "utf-8"
721 self.append = self._stack.append
Skip Montanaro174dd222005-05-14 20:54:16 +0000722 self._use_datetime = use_datetime
723 if use_datetime and not datetime:
724 raise ValueError, "the datetime module is not available"
Fredrik Lundhb9056332001-07-11 17:42:21 +0000725
726 def close(self):
727 # return response tuple and target method
728 if self._type is None or self._marks:
729 raise ResponseError()
730 if self._type == "fault":
Guido van Rossum68468eb2003-02-27 20:14:51 +0000731 raise Fault(**self._stack[0])
Fredrik Lundhb9056332001-07-11 17:42:21 +0000732 return tuple(self._stack)
733
734 def getmethodname(self):
735 return self._methodname
736
737 #
738 # event handlers
739
740 def xml(self, encoding, standalone):
741 self._encoding = encoding
742 # FIXME: assert standalone == 1 ???
743
744 def start(self, tag, attrs):
745 # prepare to handle this element
746 if tag == "array" or tag == "struct":
747 self._marks.append(len(self._stack))
748 self._data = []
749 self._value = (tag == "value")
750
751 def data(self, text):
752 self._data.append(text)
753
Neal Norwitzff113342007-04-17 08:42:15 +0000754 def end(self, tag):
Fredrik Lundhb9056332001-07-11 17:42:21 +0000755 # call the appropriate end tag handler
756 try:
757 f = self.dispatch[tag]
758 except KeyError:
759 pass # unknown tag ?
760 else:
Neal Norwitzff113342007-04-17 08:42:15 +0000761 return f(self, "".join(self._data))
Fredrik Lundhb9056332001-07-11 17:42:21 +0000762
763 #
764 # accelerator support
765
766 def end_dispatch(self, tag, data):
767 # dispatch data
768 try:
769 f = self.dispatch[tag]
770 except KeyError:
771 pass # unknown tag ?
772 else:
773 return f(self, data)
774
775 #
776 # element decoders
777
778 dispatch = {}
779
Andrew M. Kuchlinga4c2b742003-04-25 00:26:51 +0000780 def end_nil (self, data):
781 self.append(None)
782 self._value = 0
783 dispatch["nil"] = end_nil
Tim Petersc2659cf2003-05-12 20:19:37 +0000784
Fredrik Lundh1538c232001-10-01 19:42:03 +0000785 def end_boolean(self, data):
Fredrik Lundhb9056332001-07-11 17:42:21 +0000786 if data == "0":
787 self.append(False)
788 elif data == "1":
789 self.append(True)
790 else:
791 raise TypeError, "bad boolean value"
792 self._value = 0
793 dispatch["boolean"] = end_boolean
794
Fredrik Lundh1538c232001-10-01 19:42:03 +0000795 def end_int(self, data):
796 self.append(int(data))
Fredrik Lundhb9056332001-07-11 17:42:21 +0000797 self._value = 0
798 dispatch["i4"] = end_int
799 dispatch["int"] = end_int
800
Fredrik Lundh1538c232001-10-01 19:42:03 +0000801 def end_double(self, data):
802 self.append(float(data))
Fredrik Lundhb9056332001-07-11 17:42:21 +0000803 self._value = 0
804 dispatch["double"] = end_double
805
Fredrik Lundh1538c232001-10-01 19:42:03 +0000806 def end_string(self, data):
Fredrik Lundhb9056332001-07-11 17:42:21 +0000807 if self._encoding:
808 data = _decode(data, self._encoding)
809 self.append(_stringify(data))
810 self._value = 0
811 dispatch["string"] = end_string
812 dispatch["name"] = end_string # struct keys are always strings
813
814 def end_array(self, data):
Raymond Hettinger46ac8eb2002-06-30 03:39:14 +0000815 mark = self._marks.pop()
Fredrik Lundhb9056332001-07-11 17:42:21 +0000816 # map arrays to Python lists
817 self._stack[mark:] = [self._stack[mark:]]
818 self._value = 0
819 dispatch["array"] = end_array
820
821 def end_struct(self, data):
Raymond Hettinger46ac8eb2002-06-30 03:39:14 +0000822 mark = self._marks.pop()
Fredrik Lundhb9056332001-07-11 17:42:21 +0000823 # map structs to Python dictionaries
824 dict = {}
825 items = self._stack[mark:]
826 for i in range(0, len(items), 2):
827 dict[_stringify(items[i])] = items[i+1]
828 self._stack[mark:] = [dict]
829 self._value = 0
830 dispatch["struct"] = end_struct
831
Fredrik Lundh1538c232001-10-01 19:42:03 +0000832 def end_base64(self, data):
Fredrik Lundhb9056332001-07-11 17:42:21 +0000833 value = Binary()
Fredrik Lundh1538c232001-10-01 19:42:03 +0000834 value.decode(data)
Fredrik Lundhb9056332001-07-11 17:42:21 +0000835 self.append(value)
836 self._value = 0
837 dispatch["base64"] = end_base64
838
Fredrik Lundh1538c232001-10-01 19:42:03 +0000839 def end_dateTime(self, data):
Fredrik Lundhb9056332001-07-11 17:42:21 +0000840 value = DateTime()
Fredrik Lundh1538c232001-10-01 19:42:03 +0000841 value.decode(data)
Skip Montanaro174dd222005-05-14 20:54:16 +0000842 if self._use_datetime:
843 value = _datetime_type(data)
Fredrik Lundhb9056332001-07-11 17:42:21 +0000844 self.append(value)
845 dispatch["dateTime.iso8601"] = end_dateTime
846
847 def end_value(self, data):
Fredrik Lundh3d9addd2002-06-27 21:36:21 +0000848 # if we stumble upon a value element with no internal
Fredrik Lundhb9056332001-07-11 17:42:21 +0000849 # elements, treat it as a string element
850 if self._value:
851 self.end_string(data)
852 dispatch["value"] = end_value
853
854 def end_params(self, data):
855 self._type = "params"
856 dispatch["params"] = end_params
857
858 def end_fault(self, data):
859 self._type = "fault"
860 dispatch["fault"] = end_fault
861
Fredrik Lundh1538c232001-10-01 19:42:03 +0000862 def end_methodName(self, data):
Fredrik Lundhb9056332001-07-11 17:42:21 +0000863 if self._encoding:
864 data = _decode(data, self._encoding)
865 self._methodname = data
866 self._type = "methodName" # no params
867 dispatch["methodName"] = end_methodName
868
Martin v. Löwis45394c22003-10-31 13:49:36 +0000869## Multicall support
870#
Fredrik Lundhb9056332001-07-11 17:42:21 +0000871
Martin v. Löwis45394c22003-10-31 13:49:36 +0000872class _MultiCallMethod:
873 # some lesser magic to store calls made to a MultiCall object
874 # for batch execution
875 def __init__(self, call_list, name):
876 self.__call_list = call_list
877 self.__name = name
878 def __getattr__(self, name):
879 return _MultiCallMethod(self.__call_list, "%s.%s" % (self.__name, name))
880 def __call__(self, *args):
881 self.__call_list.append((self.__name, args))
882
Martin v. Löwis12237b32004-08-22 16:04:50 +0000883class MultiCallIterator:
Martin v. Löwis45394c22003-10-31 13:49:36 +0000884 """Iterates over the results of a multicall. Exceptions are
885 thrown in response to xmlrpc faults."""
Raymond Hettingercc523fc2003-11-02 09:47:05 +0000886
Martin v. Löwis12237b32004-08-22 16:04:50 +0000887 def __init__(self, results):
888 self.results = results
889
890 def __getitem__(self, i):
891 item = self.results[i]
892 if type(item) == type({}):
893 raise Fault(item['faultCode'], item['faultString'])
894 elif type(item) == type([]):
895 return item[0]
Martin v. Löwis45394c22003-10-31 13:49:36 +0000896 else:
897 raise ValueError,\
898 "unexpected type in multicall result"
Raymond Hettingercc523fc2003-11-02 09:47:05 +0000899
Martin v. Löwis45394c22003-10-31 13:49:36 +0000900class MultiCall:
901 """server -> a object used to boxcar method calls
902
903 server should be a ServerProxy object.
904
905 Methods can be added to the MultiCall using normal
906 method call syntax e.g.:
907
908 multicall = MultiCall(server_proxy)
909 multicall.add(2,3)
910 multicall.get_address("Guido")
911
912 To execute the multicall, call the MultiCall object e.g.:
913
914 add_result, address = multicall()
915 """
Raymond Hettingercc523fc2003-11-02 09:47:05 +0000916
Martin v. Löwis45394c22003-10-31 13:49:36 +0000917 def __init__(self, server):
918 self.__server = server
919 self.__call_list = []
Raymond Hettingercc523fc2003-11-02 09:47:05 +0000920
Martin v. Löwis45394c22003-10-31 13:49:36 +0000921 def __repr__(self):
922 return "<MultiCall at %x>" % id(self)
Raymond Hettingercc523fc2003-11-02 09:47:05 +0000923
Martin v. Löwis45394c22003-10-31 13:49:36 +0000924 __str__ = __repr__
925
926 def __getattr__(self, name):
927 return _MultiCallMethod(self.__call_list, name)
928
929 def __call__(self):
930 marshalled_list = []
931 for name, args in self.__call_list:
932 marshalled_list.append({'methodName' : name, 'params' : args})
933
934 return MultiCallIterator(self.__server.system.multicall(marshalled_list))
Raymond Hettingercc523fc2003-11-02 09:47:05 +0000935
Fredrik Lundhb9056332001-07-11 17:42:21 +0000936# --------------------------------------------------------------------
937# convenience functions
938
Fredrik Lundh3d9addd2002-06-27 21:36:21 +0000939##
940# Create a parser object, and connect it to an unmarshalling instance.
941# This function picks the fastest available XML parser.
942#
943# return A (parser, unmarshaller) tuple.
944
Skip Montanaro174dd222005-05-14 20:54:16 +0000945def getparser(use_datetime=0):
Fredrik Lundhb9056332001-07-11 17:42:21 +0000946 """getparser() -> parser, unmarshaller
947
Fredrik Lundhb0e8e9b2001-09-10 21:45:42 +0000948 Create an instance of the fastest available parser, and attach it
949 to an unmarshalling object. Return both objects.
Fredrik Lundhb9056332001-07-11 17:42:21 +0000950 """
Skip Montanaro174dd222005-05-14 20:54:16 +0000951 if use_datetime and not datetime:
952 raise ValueError, "the datetime module is not available"
Fredrik Lundhb9056332001-07-11 17:42:21 +0000953 if FastParser and FastUnmarshaller:
Skip Montanaro174dd222005-05-14 20:54:16 +0000954 if use_datetime:
955 mkdatetime = _datetime_type
956 else:
957 mkdatetime = _datetime
958 target = FastUnmarshaller(True, False, _binary, mkdatetime, Fault)
Fredrik Lundhb9056332001-07-11 17:42:21 +0000959 parser = FastParser(target)
960 else:
Skip Montanaro174dd222005-05-14 20:54:16 +0000961 target = Unmarshaller(use_datetime=use_datetime)
Fredrik Lundhb9056332001-07-11 17:42:21 +0000962 if FastParser:
963 parser = FastParser(target)
964 elif SgmlopParser:
965 parser = SgmlopParser(target)
966 elif ExpatParser:
967 parser = ExpatParser(target)
968 else:
969 parser = SlowParser(target)
970 return parser, target
971
Fredrik Lundh3d9addd2002-06-27 21:36:21 +0000972##
973# Convert a Python tuple or a Fault instance to an XML-RPC packet.
974#
975# @def dumps(params, **options)
976# @param params A tuple or Fault instance.
977# @keyparam methodname If given, create a methodCall request for
978# this method name.
979# @keyparam methodresponse If given, create a methodResponse packet.
980# If used with a tuple, the tuple must be a singleton (that is,
981# it must contain exactly one element).
982# @keyparam encoding The packet encoding.
983# @return A string containing marshalled data.
984
Andrew M. Kuchlinga4c2b742003-04-25 00:26:51 +0000985def dumps(params, methodname=None, methodresponse=None, encoding=None,
986 allow_none=0):
Fredrik Lundhb9056332001-07-11 17:42:21 +0000987 """data [,options] -> marshalled data
988
989 Convert an argument tuple or a Fault instance to an XML-RPC
990 request (or response, if the methodresponse option is used).
991
Fredrik Lundhb0e8e9b2001-09-10 21:45:42 +0000992 In addition to the data object, the following options can be given
993 as keyword arguments:
Fredrik Lundhb9056332001-07-11 17:42:21 +0000994
995 methodname: the method name for a methodCall packet
996
997 methodresponse: true to create a methodResponse packet.
998 If this option is used with a tuple, the tuple must be
999 a singleton (i.e. it can contain only one element).
1000
1001 encoding: the packet encoding (default is UTF-8)
1002
1003 All 8-bit strings in the data structure are assumed to use the
1004 packet encoding. Unicode strings are automatically converted,
Fredrik Lundhb0e8e9b2001-09-10 21:45:42 +00001005 where necessary.
Fredrik Lundhb9056332001-07-11 17:42:21 +00001006 """
1007
Raymond Hettingerfe59dc12005-02-07 15:28:45 +00001008 assert isinstance(params, TupleType) or isinstance(params, Fault),\
Fredrik Lundhb9056332001-07-11 17:42:21 +00001009 "argument must be tuple or Fault instance"
1010
1011 if isinstance(params, Fault):
1012 methodresponse = 1
Raymond Hettingerfe59dc12005-02-07 15:28:45 +00001013 elif methodresponse and isinstance(params, TupleType):
Fredrik Lundhb9056332001-07-11 17:42:21 +00001014 assert len(params) == 1, "response tuple must be a singleton"
1015
Fredrik Lundh3d9addd2002-06-27 21:36:21 +00001016 if not encoding:
Fredrik Lundhb9056332001-07-11 17:42:21 +00001017 encoding = "utf-8"
1018
Fredrik Lundh3d9addd2002-06-27 21:36:21 +00001019 if FastMarshaller:
1020 m = FastMarshaller(encoding)
1021 else:
Andrew M. Kuchlinga4c2b742003-04-25 00:26:51 +00001022 m = Marshaller(encoding, allow_none)
Fredrik Lundh3d9addd2002-06-27 21:36:21 +00001023
Fredrik Lundhb9056332001-07-11 17:42:21 +00001024 data = m.dumps(params)
1025
1026 if encoding != "utf-8":
Fredrik Lundh3d9addd2002-06-27 21:36:21 +00001027 xmlheader = "<?xml version='1.0' encoding='%s'?>\n" % str(encoding)
Fredrik Lundhb9056332001-07-11 17:42:21 +00001028 else:
1029 xmlheader = "<?xml version='1.0'?>\n" # utf-8 is default
1030
1031 # standard XML-RPC wrappings
Fredrik Lundh3d9addd2002-06-27 21:36:21 +00001032 if methodname:
Fredrik Lundhb9056332001-07-11 17:42:21 +00001033 # a method call
Guido van Rossum54ad5232007-05-27 09:17:48 +00001034 if not isinstance(methodname, basestring):
Fredrik Lundhb9056332001-07-11 17:42:21 +00001035 methodname = methodname.encode(encoding)
1036 data = (
1037 xmlheader,
1038 "<methodCall>\n"
1039 "<methodName>", methodname, "</methodName>\n",
1040 data,
1041 "</methodCall>\n"
1042 )
1043 elif methodresponse:
1044 # a method response, or a fault structure
1045 data = (
1046 xmlheader,
1047 "<methodResponse>\n",
1048 data,
1049 "</methodResponse>\n"
1050 )
1051 else:
1052 return data # return as is
Neal Norwitzff113342007-04-17 08:42:15 +00001053 return "".join(data)
Fredrik Lundhb9056332001-07-11 17:42:21 +00001054
Fredrik Lundh3d9addd2002-06-27 21:36:21 +00001055##
1056# Convert an XML-RPC packet to a Python object. If the XML-RPC packet
1057# represents a fault condition, this function raises a Fault exception.
1058#
1059# @param data An XML-RPC packet, given as an 8-bit string.
Walter Dörwaldf0dfc7a2003-10-20 14:01:56 +00001060# @return A tuple containing the unpacked data, and the method name
Fredrik Lundh3d9addd2002-06-27 21:36:21 +00001061# (None if not present).
1062# @see Fault
1063
Skip Montanaro174dd222005-05-14 20:54:16 +00001064def loads(data, use_datetime=0):
Fredrik Lundhb9056332001-07-11 17:42:21 +00001065 """data -> unmarshalled data, method name
1066
1067 Convert an XML-RPC packet to unmarshalled data plus a method
1068 name (None if not present).
1069
1070 If the XML-RPC packet represents a fault condition, this function
1071 raises a Fault exception.
1072 """
Skip Montanaro174dd222005-05-14 20:54:16 +00001073 p, u = getparser(use_datetime=use_datetime)
Fredrik Lundhb9056332001-07-11 17:42:21 +00001074 p.feed(data)
1075 p.close()
1076 return u.close(), u.getmethodname()
1077
1078
1079# --------------------------------------------------------------------
1080# request dispatcher
1081
1082class _Method:
1083 # some magic to bind an XML-RPC method to an RPC server.
1084 # supports "nested" methods (e.g. examples.getStateName)
1085 def __init__(self, send, name):
1086 self.__send = send
1087 self.__name = name
1088 def __getattr__(self, name):
1089 return _Method(self.__send, "%s.%s" % (self.__name, name))
1090 def __call__(self, *args):
1091 return self.__send(self.__name, args)
1092
Fredrik Lundh3d9addd2002-06-27 21:36:21 +00001093##
1094# Standard transport class for XML-RPC over HTTP.
1095# <p>
1096# You can create custom transports by subclassing this method, and
1097# overriding selected methods.
Fredrik Lundhb9056332001-07-11 17:42:21 +00001098
1099class Transport:
Fredrik Lundhc4c062f2001-09-10 19:45:02 +00001100 """Handles an HTTP transaction to an XML-RPC server."""
Fredrik Lundhb9056332001-07-11 17:42:21 +00001101
1102 # client identifier (may be overridden)
1103 user_agent = "xmlrpclib.py/%s (by www.pythonware.com)" % __version__
1104
Skip Montanaro174dd222005-05-14 20:54:16 +00001105 def __init__(self, use_datetime=0):
1106 self._use_datetime = use_datetime
1107
Fredrik Lundh3d9addd2002-06-27 21:36:21 +00001108 ##
1109 # Send a complete request, and parse the response.
1110 #
1111 # @param host Target host.
1112 # @param handler Target PRC handler.
1113 # @param request_body XML-RPC request body.
1114 # @param verbose Debugging flag.
1115 # @return Parsed response.
1116
Fredrik Lundhb9056332001-07-11 17:42:21 +00001117 def request(self, host, handler, request_body, verbose=0):
1118 # issue XML-RPC request
1119
1120 h = self.make_connection(host)
1121 if verbose:
1122 h.set_debuglevel(1)
1123
1124 self.send_request(h, handler, request_body)
1125 self.send_host(h, host)
1126 self.send_user_agent(h)
1127 self.send_content(h, request_body)
1128
1129 errcode, errmsg, headers = h.getreply()
1130
1131 if errcode != 200:
1132 raise ProtocolError(
1133 host + handler,
1134 errcode, errmsg,
1135 headers
1136 )
1137
1138 self.verbose = verbose
1139
Fredrik Lundh3d9addd2002-06-27 21:36:21 +00001140 try:
1141 sock = h._conn.sock
1142 except AttributeError:
1143 sock = None
1144
1145 return self._parse_response(h.getfile(), sock)
1146
1147 ##
1148 # Create parser.
1149 #
1150 # @return A 2-tuple containing a parser and a unmarshaller.
Fredrik Lundhb9056332001-07-11 17:42:21 +00001151
Fredrik Lundhc4c062f2001-09-10 19:45:02 +00001152 def getparser(self):
1153 # get parser and unmarshaller
Skip Montanaro174dd222005-05-14 20:54:16 +00001154 return getparser(use_datetime=self._use_datetime)
Fredrik Lundhc4c062f2001-09-10 19:45:02 +00001155
Fredrik Lundh3d9addd2002-06-27 21:36:21 +00001156 ##
Fredrik Lundh1303c7c2002-10-22 18:23:00 +00001157 # Get authorization info from host parameter
1158 # Host may be a string, or a (host, x509-dict) tuple; if a string,
1159 # it is checked for a "user:pw@host" format, and a "Basic
1160 # Authentication" header is added if appropriate.
1161 #
1162 # @param host Host descriptor (URL or (URL, x509 info) tuple).
1163 # @return A 3-tuple containing (actual host, extra headers,
1164 # x509 info). The header and x509 fields may be None.
1165
1166 def get_host_info(self, host):
1167
1168 x509 = {}
Raymond Hettingerfe59dc12005-02-07 15:28:45 +00001169 if isinstance(host, TupleType):
Fredrik Lundh1303c7c2002-10-22 18:23:00 +00001170 host, x509 = host
1171
1172 import urllib
1173 auth, host = urllib.splituser(host)
1174
1175 if auth:
1176 import base64
Fredrik Lundh768c98b2002-11-01 17:14:16 +00001177 auth = base64.encodestring(urllib.unquote(auth))
Neal Norwitzff113342007-04-17 08:42:15 +00001178 auth = "".join(auth.split()) # get rid of whitespace
Fredrik Lundh1303c7c2002-10-22 18:23:00 +00001179 extra_headers = [
1180 ("Authorization", "Basic " + auth)
1181 ]
1182 else:
1183 extra_headers = None
1184
1185 return host, extra_headers, x509
1186
1187 ##
Fredrik Lundh3d9addd2002-06-27 21:36:21 +00001188 # Connect to server.
1189 #
1190 # @param host Target host.
1191 # @return A connection handle.
1192
Fredrik Lundhb9056332001-07-11 17:42:21 +00001193 def make_connection(self, host):
1194 # create a HTTP connection object from a host descriptor
1195 import httplib
Fredrik Lundh1303c7c2002-10-22 18:23:00 +00001196 host, extra_headers, x509 = self.get_host_info(host)
Fredrik Lundhb9056332001-07-11 17:42:21 +00001197 return httplib.HTTP(host)
1198
Fredrik Lundh3d9addd2002-06-27 21:36:21 +00001199 ##
1200 # Send request header.
1201 #
1202 # @param connection Connection handle.
1203 # @param handler Target RPC handler.
1204 # @param request_body XML-RPC body.
1205
Fredrik Lundhb9056332001-07-11 17:42:21 +00001206 def send_request(self, connection, handler, request_body):
1207 connection.putrequest("POST", handler)
1208
Fredrik Lundh3d9addd2002-06-27 21:36:21 +00001209 ##
1210 # Send host name.
1211 #
1212 # @param connection Connection handle.
1213 # @param host Host name.
1214
Fredrik Lundhb9056332001-07-11 17:42:21 +00001215 def send_host(self, connection, host):
Fredrik Lundh1303c7c2002-10-22 18:23:00 +00001216 host, extra_headers, x509 = self.get_host_info(host)
Fredrik Lundhb9056332001-07-11 17:42:21 +00001217 connection.putheader("Host", host)
Fredrik Lundh1303c7c2002-10-22 18:23:00 +00001218 if extra_headers:
Raymond Hettingerfe59dc12005-02-07 15:28:45 +00001219 if isinstance(extra_headers, DictType):
Fredrik Lundh1303c7c2002-10-22 18:23:00 +00001220 extra_headers = extra_headers.items()
1221 for key, value in extra_headers:
1222 connection.putheader(key, value)
Fredrik Lundhb9056332001-07-11 17:42:21 +00001223
Fredrik Lundh3d9addd2002-06-27 21:36:21 +00001224 ##
1225 # Send user-agent identifier.
1226 #
1227 # @param connection Connection handle.
1228
Fredrik Lundhb9056332001-07-11 17:42:21 +00001229 def send_user_agent(self, connection):
1230 connection.putheader("User-Agent", self.user_agent)
1231
Fredrik Lundh3d9addd2002-06-27 21:36:21 +00001232 ##
1233 # Send request body.
1234 #
1235 # @param connection Connection handle.
1236 # @param request_body XML-RPC request body.
1237
Fredrik Lundhb9056332001-07-11 17:42:21 +00001238 def send_content(self, connection, request_body):
1239 connection.putheader("Content-Type", "text/xml")
1240 connection.putheader("Content-Length", str(len(request_body)))
1241 connection.endheaders()
1242 if request_body:
1243 connection.send(request_body)
1244
Fredrik Lundh3d9addd2002-06-27 21:36:21 +00001245 ##
1246 # Parse response.
1247 #
1248 # @param file Stream.
1249 # @return Response tuple and target method.
1250
1251 def parse_response(self, file):
1252 # compatibility interface
1253 return self._parse_response(file, None)
1254
1255 ##
1256 # Parse response (alternate interface). This is similar to the
1257 # parse_response method, but also provides direct access to the
1258 # underlying socket object (where available).
1259 #
1260 # @param file Stream.
1261 # @param sock Socket handle (or None, if the socket object
1262 # could not be accessed).
1263 # @return Response tuple and target method.
1264
1265 def _parse_response(self, file, sock):
1266 # read response from input file/socket, and parse it
Fredrik Lundhb9056332001-07-11 17:42:21 +00001267
Fredrik Lundhc4c062f2001-09-10 19:45:02 +00001268 p, u = self.getparser()
Fredrik Lundhb9056332001-07-11 17:42:21 +00001269
1270 while 1:
Fredrik Lundh3d9addd2002-06-27 21:36:21 +00001271 if sock:
1272 response = sock.recv(1024)
1273 else:
1274 response = file.read(1024)
Fredrik Lundhb9056332001-07-11 17:42:21 +00001275 if not response:
1276 break
1277 if self.verbose:
Guido van Rossumbe19ed72007-02-09 05:37:30 +00001278 print("body:", repr(response))
Fredrik Lundhb9056332001-07-11 17:42:21 +00001279 p.feed(response)
1280
Fredrik Lundh3d9addd2002-06-27 21:36:21 +00001281 file.close()
Fredrik Lundhb9056332001-07-11 17:42:21 +00001282 p.close()
1283
1284 return u.close()
1285
Fredrik Lundh3d9addd2002-06-27 21:36:21 +00001286##
1287# Standard transport class for XML-RPC over HTTPS.
1288
Fredrik Lundhb9056332001-07-11 17:42:21 +00001289class SafeTransport(Transport):
Fredrik Lundhc4c062f2001-09-10 19:45:02 +00001290 """Handles an HTTPS transaction to an XML-RPC server."""
Fredrik Lundhb9056332001-07-11 17:42:21 +00001291
Fredrik Lundh3d9addd2002-06-27 21:36:21 +00001292 # FIXME: mostly untested
1293
Fredrik Lundhb9056332001-07-11 17:42:21 +00001294 def make_connection(self, host):
1295 # create a HTTPS connection object from a host descriptor
1296 # host may be a string, or a (host, x509-dict) tuple
1297 import httplib
Fredrik Lundh1303c7c2002-10-22 18:23:00 +00001298 host, extra_headers, x509 = self.get_host_info(host)
Fredrik Lundhb9056332001-07-11 17:42:21 +00001299 try:
1300 HTTPS = httplib.HTTPS
1301 except AttributeError:
Fredrik Lundh1303c7c2002-10-22 18:23:00 +00001302 raise NotImplementedError(
1303 "your version of httplib doesn't support HTTPS"
1304 )
Fredrik Lundhb9056332001-07-11 17:42:21 +00001305 else:
Guido van Rossum68468eb2003-02-27 20:14:51 +00001306 return HTTPS(host, None, **(x509 or {}))
Fredrik Lundhb9056332001-07-11 17:42:21 +00001307
Fredrik Lundh3d9addd2002-06-27 21:36:21 +00001308##
1309# Standard server proxy. This class establishes a virtual connection
1310# to an XML-RPC server.
1311# <p>
1312# This class is available as ServerProxy and Server. New code should
1313# use ServerProxy, to avoid confusion.
1314#
1315# @def ServerProxy(uri, **options)
1316# @param uri The connection point on the server.
1317# @keyparam transport A transport factory, compatible with the
1318# standard transport class.
1319# @keyparam encoding The default encoding used for 8-bit strings
1320# (default is UTF-8).
1321# @keyparam verbose Use a true value to enable debugging output.
1322# (printed to standard output).
1323# @see Transport
1324
Fredrik Lundhb9056332001-07-11 17:42:21 +00001325class ServerProxy:
1326 """uri [,options] -> a logical connection to an XML-RPC server
1327
1328 uri is the connection point on the server, given as
1329 scheme://host/target.
1330
1331 The standard implementation always supports the "http" scheme. If
1332 SSL socket support is available (Python 2.0), it also supports
1333 "https".
1334
1335 If the target part and the slash preceding it are both omitted,
1336 "/RPC2" is assumed.
1337
1338 The following options can be given as keyword arguments:
1339
1340 transport: a transport factory
1341 encoding: the request encoding (default is UTF-8)
1342
1343 All 8-bit strings passed to the server proxy are assumed to use
1344 the given encoding.
1345 """
1346
Andrew M. Kuchlinga4c2b742003-04-25 00:26:51 +00001347 def __init__(self, uri, transport=None, encoding=None, verbose=0,
Skip Montanaro174dd222005-05-14 20:54:16 +00001348 allow_none=0, use_datetime=0):
Fredrik Lundhb9056332001-07-11 17:42:21 +00001349 # establish a "logical" server connection
1350
1351 # get the url
Fredrik Lundhc4c062f2001-09-10 19:45:02 +00001352 import urllib
Fredrik Lundhb9056332001-07-11 17:42:21 +00001353 type, uri = urllib.splittype(uri)
1354 if type not in ("http", "https"):
1355 raise IOError, "unsupported XML-RPC protocol"
1356 self.__host, self.__handler = urllib.splithost(uri)
1357 if not self.__handler:
1358 self.__handler = "/RPC2"
1359
1360 if transport is None:
1361 if type == "https":
Skip Montanaro174dd222005-05-14 20:54:16 +00001362 transport = SafeTransport(use_datetime=use_datetime)
Fredrik Lundhb9056332001-07-11 17:42:21 +00001363 else:
Skip Montanaro174dd222005-05-14 20:54:16 +00001364 transport = Transport(use_datetime=use_datetime)
Fredrik Lundhb9056332001-07-11 17:42:21 +00001365 self.__transport = transport
1366
1367 self.__encoding = encoding
1368 self.__verbose = verbose
Andrew M. Kuchlinga4c2b742003-04-25 00:26:51 +00001369 self.__allow_none = allow_none
Tim Petersc2659cf2003-05-12 20:19:37 +00001370
Fredrik Lundhb9056332001-07-11 17:42:21 +00001371 def __request(self, methodname, params):
1372 # call a method on the remote server
1373
Andrew M. Kuchlinga4c2b742003-04-25 00:26:51 +00001374 request = dumps(params, methodname, encoding=self.__encoding,
1375 allow_none=self.__allow_none)
Fredrik Lundhb9056332001-07-11 17:42:21 +00001376
1377 response = self.__transport.request(
1378 self.__host,
1379 self.__handler,
1380 request,
1381 verbose=self.__verbose
1382 )
1383
1384 if len(response) == 1:
1385 response = response[0]
1386
1387 return response
1388
1389 def __repr__(self):
1390 return (
Fredrik Lundh78eedce2001-08-23 20:04:33 +00001391 "<ServerProxy for %s%s>" %
Fredrik Lundhb9056332001-07-11 17:42:21 +00001392 (self.__host, self.__handler)
1393 )
1394
1395 __str__ = __repr__
Raymond Hettingercc523fc2003-11-02 09:47:05 +00001396
Fredrik Lundhb9056332001-07-11 17:42:21 +00001397 def __getattr__(self, name):
1398 # magic method dispatcher
1399 return _Method(self.__request, name)
1400
1401 # note: to call a remote object with an non-standard name, use
1402 # result getattr(server, "strange-python-name")(args)
1403
Fredrik Lundh78eedce2001-08-23 20:04:33 +00001404# compatibility
Fredrik Lundh3d9addd2002-06-27 21:36:21 +00001405
Fredrik Lundhb9056332001-07-11 17:42:21 +00001406Server = ServerProxy
1407
1408# --------------------------------------------------------------------
1409# test code
1410
1411if __name__ == "__main__":
1412
1413 # simple test program (from the XML-RPC specification)
1414
Fredrik Lundh78eedce2001-08-23 20:04:33 +00001415 # server = ServerProxy("http://localhost:8000") # local server
Martin v. Löwis12237b32004-08-22 16:04:50 +00001416 server = ServerProxy("http://time.xmlrpc.com/RPC2")
Fredrik Lundhb9056332001-07-11 17:42:21 +00001417
Guido van Rossumbe19ed72007-02-09 05:37:30 +00001418 print(server)
Fredrik Lundhb9056332001-07-11 17:42:21 +00001419
1420 try:
Guido van Rossumbe19ed72007-02-09 05:37:30 +00001421 print(server.currentTime.getCurrentTime())
Guido van Rossumb940e112007-01-10 16:19:56 +00001422 except Error as v:
Guido van Rossumbe19ed72007-02-09 05:37:30 +00001423 print("ERROR", v)
Martin v. Löwis12237b32004-08-22 16:04:50 +00001424
1425 multi = MultiCall(server)
1426 multi.currentTime.getCurrentTime()
1427 multi.currentTime.getCurrentTime()
1428 try:
1429 for response in multi():
Guido van Rossumbe19ed72007-02-09 05:37:30 +00001430 print(response)
Guido van Rossumb940e112007-01-10 16:19:56 +00001431 except Error as v:
Guido van Rossumbe19ed72007-02-09 05:37:30 +00001432 print("ERROR", v)