blob: 83b3a04666787a4efa91d289ffeb89a1cf14533b [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
Fredrik Lundh3d9addd2002-06-27 21:36:21 +0000141# --------------------------------------------------------------------
142# Internal stuff
143
Fredrik Lundhb9056332001-07-11 17:42:21 +0000144try:
Fred Drakeba613c32005-02-10 18:33:30 +0000145 import datetime
146except ImportError:
147 datetime = None
148
149try:
Skip Montanaro9a7c96a2003-01-22 18:17:25 +0000150 _bool_is_builtin = False.__class__.__name__ == "bool"
151except NameError:
152 _bool_is_builtin = 0
153
Fredrik Lundhb9056332001-07-11 17:42:21 +0000154def _decode(data, encoding, is8bit=re.compile("[\x80-\xff]").search):
155 # decode non-ascii string (if possible)
Guido van Rossum54ad5232007-05-27 09:17:48 +0000156 if encoding and is8bit(data):
Guido van Rossumef87d6e2007-05-02 19:09:54 +0000157 data = str(data, encoding)
Fredrik Lundhb9056332001-07-11 17:42:21 +0000158 return data
159
Neal Norwitzff113342007-04-17 08:42:15 +0000160def escape(s):
161 s = s.replace("&", "&")
162 s = s.replace("<", "&lt;")
163 return s.replace(">", "&gt;",)
Fredrik Lundh1538c232001-10-01 19:42:03 +0000164
Guido van Rossum54ad5232007-05-27 09:17:48 +0000165def _stringify(string):
166 # convert to 7-bit ascii if possible
167 try:
Brett Cannon2dbde5e2007-07-30 03:50:35 +0000168 return string.decode("ascii")
Guido van Rossum54ad5232007-05-27 09:17:48 +0000169 except UnicodeError:
Fredrik Lundhb9056332001-07-11 17:42:21 +0000170 return string
171
Fredrik Lundh3d9addd2002-06-27 21:36:21 +0000172__version__ = "1.0.1"
173
174# xmlrpc integer limits
Guido van Rossume2a383d2007-01-15 16:59:06 +0000175MAXINT = 2**31-1
176MININT = -2**31
Fredrik Lundh3d9addd2002-06-27 21:36:21 +0000177
178# --------------------------------------------------------------------
179# Error constants (from Dan Libby's specification at
180# http://xmlrpc-epi.sourceforge.net/specs/rfc.fault_codes.php)
181
182# Ranges of errors
183PARSE_ERROR = -32700
184SERVER_ERROR = -32600
185APPLICATION_ERROR = -32500
186SYSTEM_ERROR = -32400
187TRANSPORT_ERROR = -32300
188
189# Specific errors
190NOT_WELLFORMED_ERROR = -32700
191UNSUPPORTED_ENCODING = -32701
192INVALID_ENCODING_CHAR = -32702
193INVALID_XMLRPC = -32600
194METHOD_NOT_FOUND = -32601
195INVALID_METHOD_PARAMS = -32602
196INTERNAL_ERROR = -32603
Fredrik Lundhb9056332001-07-11 17:42:21 +0000197
198# --------------------------------------------------------------------
199# Exceptions
200
Fredrik Lundh3d9addd2002-06-27 21:36:21 +0000201##
202# Base class for all kinds of client-side errors.
203
Fredrik Lundh78eedce2001-08-23 20:04:33 +0000204class Error(Exception):
Fred Drake1b410792001-09-04 18:55:03 +0000205 """Base class for client errors."""
Fredrik Lundhb0e8e9b2001-09-10 21:45:42 +0000206 def __str__(self):
207 return repr(self)
Fredrik Lundhb9056332001-07-11 17:42:21 +0000208
Fredrik Lundh3d9addd2002-06-27 21:36:21 +0000209##
210# Indicates an HTTP-level protocol error. This is raised by the HTTP
211# transport layer, if the server returns an error code other than 200
212# (OK).
213#
214# @param url The target URL.
215# @param errcode The HTTP error code.
216# @param errmsg The HTTP error message.
217# @param headers The HTTP header dictionary.
218
Fredrik Lundhb9056332001-07-11 17:42:21 +0000219class ProtocolError(Error):
Fred Drake1b410792001-09-04 18:55:03 +0000220 """Indicates an HTTP protocol error."""
Fredrik Lundhb9056332001-07-11 17:42:21 +0000221 def __init__(self, url, errcode, errmsg, headers):
Fredrik Lundhb0e8e9b2001-09-10 21:45:42 +0000222 Error.__init__(self)
Fredrik Lundhb9056332001-07-11 17:42:21 +0000223 self.url = url
224 self.errcode = errcode
225 self.errmsg = errmsg
226 self.headers = headers
227 def __repr__(self):
228 return (
229 "<ProtocolError for %s: %s %s>" %
230 (self.url, self.errcode, self.errmsg)
231 )
232
Fredrik Lundh3d9addd2002-06-27 21:36:21 +0000233##
234# Indicates a broken XML-RPC response package. This exception is
235# raised by the unmarshalling layer, if the XML-RPC response is
236# malformed.
237
Fredrik Lundhb9056332001-07-11 17:42:21 +0000238class ResponseError(Error):
Fredrik Lundhc4c062f2001-09-10 19:45:02 +0000239 """Indicates a broken response package."""
Fredrik Lundhb9056332001-07-11 17:42:21 +0000240 pass
241
Fredrik Lundh3d9addd2002-06-27 21:36:21 +0000242##
243# Indicates an XML-RPC fault response package. This exception is
244# raised by the unmarshalling layer, if the XML-RPC response contains
245# a fault string. This exception can also used as a class, to
246# generate a fault XML-RPC message.
247#
248# @param faultCode The XML-RPC fault code.
249# @param faultString The XML-RPC fault string.
250
Fredrik Lundhb9056332001-07-11 17:42:21 +0000251class Fault(Error):
Fredrik Lundhc4c062f2001-09-10 19:45:02 +0000252 """Indicates an XML-RPC fault package."""
Fredrik Lundhb9056332001-07-11 17:42:21 +0000253 def __init__(self, faultCode, faultString, **extra):
Fredrik Lundhb0e8e9b2001-09-10 21:45:42 +0000254 Error.__init__(self)
Fredrik Lundhb9056332001-07-11 17:42:21 +0000255 self.faultCode = faultCode
256 self.faultString = faultString
257 def __repr__(self):
258 return (
259 "<Fault %s: %s>" %
Fredrik Lundhb0e8e9b2001-09-10 21:45:42 +0000260 (self.faultCode, repr(self.faultString))
Fredrik Lundhb9056332001-07-11 17:42:21 +0000261 )
262
263# --------------------------------------------------------------------
264# Special values
265
Fredrik Lundh3d9addd2002-06-27 21:36:21 +0000266##
267# Wrapper for XML-RPC boolean values. Use the xmlrpclib.True and
268# xmlrpclib.False constants, or the xmlrpclib.boolean() function, to
269# generate boolean XML-RPC values.
270#
271# @param value A boolean value. Any true value is interpreted as True,
272# all other values are interpreted as False.
273
Guido van Rossum47b9ff62006-08-24 00:41:19 +0000274boolean = Boolean = bool
Fredrik Lundh3d9addd2002-06-27 21:36:21 +0000275
276##
277# Wrapper for XML-RPC DateTime values. This converts a time value to
278# the format used by XML-RPC.
279# <p>
280# The value can be given as a string in the format
281# "yyyymmddThh:mm:ss", as a 9-item time tuple (as returned by
282# time.localtime()), or an integer value (as returned by time.time()).
283# The wrapper uses time.localtime() to convert an integer to a time
284# tuple.
285#
286# @param value The time, given as an ISO 8601 string, a time
287# tuple, or a integer time value.
Fredrik Lundhb9056332001-07-11 17:42:21 +0000288
Fredrik Lundhb9056332001-07-11 17:42:21 +0000289class DateTime:
Fredrik Lundhc4c062f2001-09-10 19:45:02 +0000290 """DateTime wrapper for an ISO 8601 string or time tuple or
291 localtime integer value to generate 'dateTime.iso8601' XML-RPC
292 value.
293 """
Fredrik Lundhb9056332001-07-11 17:42:21 +0000294
295 def __init__(self, value=0):
Guido van Rossum54ad5232007-05-27 09:17:48 +0000296 if not isinstance(value, basestring):
Fred Drakeba613c32005-02-10 18:33:30 +0000297 if datetime and isinstance(value, datetime.datetime):
298 self.value = value.strftime("%Y%m%dT%H:%M:%S")
299 return
Skip Montanaro174dd222005-05-14 20:54:16 +0000300 if datetime and isinstance(value, datetime.date):
301 self.value = value.strftime("%Y%m%dT%H:%M:%S")
302 return
303 if datetime and isinstance(value, datetime.time):
304 today = datetime.datetime.now().strftime("%Y%m%d")
305 self.value = value.strftime(today+"T%H:%M:%S")
306 return
Guido van Rossum13257902007-06-07 23:15:56 +0000307 if not isinstance(value, (tuple, time.struct_time)):
Fredrik Lundhb9056332001-07-11 17:42:21 +0000308 if value == 0:
309 value = time.time()
310 value = time.localtime(value)
311 value = time.strftime("%Y%m%dT%H:%M:%S", value)
312 self.value = value
313
Guido van Rossum47b9ff62006-08-24 00:41:19 +0000314 def __eq__(self, other):
Fredrik Lundhb9056332001-07-11 17:42:21 +0000315 if isinstance(other, DateTime):
316 other = other.value
Guido van Rossum47b9ff62006-08-24 00:41:19 +0000317 return self.value == other
318
319 def __ne__(self, other):
320 if isinstance(other, DateTime):
321 other = other.value
322 return self.value != other
Fredrik Lundhb9056332001-07-11 17:42:21 +0000323
Fredrik Lundh3d9addd2002-06-27 21:36:21 +0000324 ##
325 # Get date/time value.
326 #
327 # @return Date/time value, as an ISO 8601 string.
328
329 def __str__(self):
330 return self.value
331
Fredrik Lundhb9056332001-07-11 17:42:21 +0000332 def __repr__(self):
Fredrik Lundh3d9addd2002-06-27 21:36:21 +0000333 return "<DateTime %s at %x>" % (repr(self.value), id(self))
Fredrik Lundhb9056332001-07-11 17:42:21 +0000334
335 def decode(self, data):
Neal Norwitzff113342007-04-17 08:42:15 +0000336 self.value = str(data).strip()
Fredrik Lundhb9056332001-07-11 17:42:21 +0000337
338 def encode(self, out):
339 out.write("<value><dateTime.iso8601>")
340 out.write(self.value)
341 out.write("</dateTime.iso8601></value>\n")
342
Fredrik Lundh3d9addd2002-06-27 21:36:21 +0000343def _datetime(data):
344 # decode xml element contents into a DateTime structure.
Fredrik Lundhb9056332001-07-11 17:42:21 +0000345 value = DateTime()
346 value.decode(data)
347 return value
348
Skip Montanaro174dd222005-05-14 20:54:16 +0000349def _datetime_type(data):
350 t = time.strptime(data, "%Y%m%dT%H:%M:%S")
351 return datetime.datetime(*tuple(t)[:6])
352
Fredrik Lundh3d9addd2002-06-27 21:36:21 +0000353##
354# Wrapper for binary data. This can be used to transport any kind
355# of binary data over XML-RPC, using BASE64 encoding.
356#
357# @param data An 8-bit string containing arbitrary data.
358
Skip Montanarobfcbfa72003-04-24 19:51:31 +0000359import base64
Guido van Rossum68937b42007-05-18 00:51:22 +0000360import io
Skip Montanarobfcbfa72003-04-24 19:51:31 +0000361
Fredrik Lundhb9056332001-07-11 17:42:21 +0000362class Binary:
Fred Drake1b410792001-09-04 18:55:03 +0000363 """Wrapper for binary data."""
Fredrik Lundhb9056332001-07-11 17:42:21 +0000364
365 def __init__(self, data=None):
366 self.data = data
367
Fredrik Lundh3d9addd2002-06-27 21:36:21 +0000368 ##
369 # Get buffer contents.
370 #
371 # @return Buffer contents, as an 8-bit string.
372
373 def __str__(self):
374 return self.data or ""
375
Guido van Rossum47b9ff62006-08-24 00:41:19 +0000376 def __eq__(self, other):
Fredrik Lundhb9056332001-07-11 17:42:21 +0000377 if isinstance(other, Binary):
378 other = other.data
Guido van Rossum47b9ff62006-08-24 00:41:19 +0000379 return self.data == other
380
381 def __ne__(self, other):
382 if isinstance(other, Binary):
383 other = other.data
384 return self.data != other
Fredrik Lundhb9056332001-07-11 17:42:21 +0000385
386 def decode(self, data):
Brett Cannon2dbde5e2007-07-30 03:50:35 +0000387 self.data = str8(base64.decodestring(data))
Fredrik Lundhb9056332001-07-11 17:42:21 +0000388
389 def encode(self, out):
Fredrik Lundhb9056332001-07-11 17:42:21 +0000390 out.write("<value><base64>\n")
Brett Cannon2dbde5e2007-07-30 03:50:35 +0000391 encoded = base64.encodestring(self.data)
392 out.write(encoded.decode('ascii'))
393 out.write('\n')
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
Fredrik Lundhb9056332001-07-11 17:42:21 +0000490 target.xml(encoding, None)
491
492 def feed(self, data):
493 self._parser.Parse(data, 0)
494
495 def close(self):
496 self._parser.Parse("", 1) # end of data
497 del self._target, self._parser # get rid of circular references
498
Fredrik Lundhb9056332001-07-11 17:42:21 +0000499# --------------------------------------------------------------------
500# XML-RPC marshalling and unmarshalling code
501
Fredrik Lundh3d9addd2002-06-27 21:36:21 +0000502##
503# XML-RPC marshaller.
504#
505# @param encoding Default encoding for 8-bit strings. The default
506# value is None (interpreted as UTF-8).
507# @see dumps
508
Fredrik Lundhb9056332001-07-11 17:42:21 +0000509class Marshaller:
Fred Drake1b410792001-09-04 18:55:03 +0000510 """Generate an XML-RPC params chunk from a Python data structure.
Fredrik Lundhb9056332001-07-11 17:42:21 +0000511
Fredrik Lundhc4c062f2001-09-10 19:45:02 +0000512 Create a Marshaller instance for each set of parameters, and use
513 the "dumps" method to convert your data (represented as a tuple)
514 to an XML-RPC params chunk. To write a fault response, pass a
515 Fault instance instead. You may prefer to use the "dumps" module
516 function for this purpose.
Fred Drake1b410792001-09-04 18:55:03 +0000517 """
Fredrik Lundhb9056332001-07-11 17:42:21 +0000518
519 # by the way, if you don't understand what's going on in here,
520 # that's perfectly ok.
521
Andrew M. Kuchlinga4c2b742003-04-25 00:26:51 +0000522 def __init__(self, encoding=None, allow_none=0):
Fredrik Lundhb9056332001-07-11 17:42:21 +0000523 self.memo = {}
524 self.data = None
525 self.encoding = encoding
Andrew M. Kuchlinga4c2b742003-04-25 00:26:51 +0000526 self.allow_none = allow_none
Tim Petersc2659cf2003-05-12 20:19:37 +0000527
Fredrik Lundhb9056332001-07-11 17:42:21 +0000528 dispatch = {}
529
530 def dumps(self, values):
Fredrik Lundh3d9addd2002-06-27 21:36:21 +0000531 out = []
532 write = out.append
533 dump = self.__dump
Fredrik Lundhb9056332001-07-11 17:42:21 +0000534 if isinstance(values, Fault):
535 # fault instance
536 write("<fault>\n")
Martin v. Löwis541342f2003-07-12 07:53:04 +0000537 dump({'faultCode': values.faultCode,
538 'faultString': values.faultString},
539 write)
Fredrik Lundhb9056332001-07-11 17:42:21 +0000540 write("</fault>\n")
541 else:
542 # parameter block
Fredrik Lundhc266bb02001-08-23 20:13:08 +0000543 # FIXME: the xml-rpc specification allows us to leave out
544 # the entire <params> block if there are no parameters.
545 # however, changing this may break older code (including
546 # old versions of xmlrpclib.py), so this is better left as
547 # is for now. See @XMLRPC3 for more information. /F
Fredrik Lundhb9056332001-07-11 17:42:21 +0000548 write("<params>\n")
549 for v in values:
550 write("<param>\n")
Fredrik Lundh3d9addd2002-06-27 21:36:21 +0000551 dump(v, write)
Fredrik Lundhb9056332001-07-11 17:42:21 +0000552 write("</param>\n")
553 write("</params>\n")
Neal Norwitzff113342007-04-17 08:42:15 +0000554 result = "".join(out)
Fredrik Lundhb9056332001-07-11 17:42:21 +0000555 return result
556
Fredrik Lundh3d9addd2002-06-27 21:36:21 +0000557 def __dump(self, value, write):
Fredrik Lundhb9056332001-07-11 17:42:21 +0000558 try:
559 f = self.dispatch[type(value)]
560 except KeyError:
Thomas Wouters89f507f2006-12-13 04:49:30 +0000561 # check if this object can be marshalled as a structure
562 try:
563 value.__dict__
564 except:
565 raise TypeError, "cannot marshal %s objects" % type(value)
566 # check if this class is a sub-class of a basic type,
567 # because we don't know how to marshal these types
568 # (e.g. a string sub-class)
569 for type_ in type(value).__mro__:
570 if type_ in self.dispatch.keys():
571 raise TypeError, "cannot marshal %s objects" % type(value)
572 # XXX(twouters): using "_arbitrary_instance" as key as a quick-fix
573 # for the p3yk merge, this should probably be fixed more neatly.
574 f = self.dispatch["_arbitrary_instance"]
575 f(self, value, write)
Fredrik Lundhb9056332001-07-11 17:42:21 +0000576
Andrew M. Kuchlinga4c2b742003-04-25 00:26:51 +0000577 def dump_nil (self, value, write):
578 if not self.allow_none:
579 raise TypeError, "cannot marshal None unless allow_none is enabled"
580 write("<value><nil/></value>")
Guido van Rossum13257902007-06-07 23:15:56 +0000581 dispatch[type(None)] = dump_nil
Tim Petersc2659cf2003-05-12 20:19:37 +0000582
Fredrik Lundh3d9addd2002-06-27 21:36:21 +0000583 def dump_int(self, value, write):
Skip Montanaro5449e082001-10-17 22:53:33 +0000584 # in case ints are > 32 bits
585 if value > MAXINT or value < MININT:
586 raise OverflowError, "int exceeds XML-RPC limits"
Fredrik Lundh3d9addd2002-06-27 21:36:21 +0000587 write("<value><int>")
588 write(str(value))
589 write("</int></value>\n")
Guido van Rossum13257902007-06-07 23:15:56 +0000590 #dispatch[int] = dump_int
Fredrik Lundhb9056332001-07-11 17:42:21 +0000591
Skip Montanaro9a7c96a2003-01-22 18:17:25 +0000592 if _bool_is_builtin:
593 def dump_bool(self, value, write):
594 write("<value><boolean>")
595 write(value and "1" or "0")
596 write("</boolean></value>\n")
597 dispatch[bool] = dump_bool
598
Fredrik Lundh3d9addd2002-06-27 21:36:21 +0000599 def dump_long(self, value, write):
Skip Montanaro5449e082001-10-17 22:53:33 +0000600 if value > MAXINT or value < MININT:
601 raise OverflowError, "long int exceeds XML-RPC limits"
Fredrik Lundh3d9addd2002-06-27 21:36:21 +0000602 write("<value><int>")
603 write(str(int(value)))
604 write("</int></value>\n")
Guido van Rossum13257902007-06-07 23:15:56 +0000605 dispatch[int] = dump_long
Skip Montanaro5e9c71b2001-10-10 15:56:34 +0000606
Fredrik Lundh3d9addd2002-06-27 21:36:21 +0000607 def dump_double(self, value, write):
608 write("<value><double>")
609 write(repr(value))
610 write("</double></value>\n")
Guido van Rossum13257902007-06-07 23:15:56 +0000611 dispatch[float] = dump_double
Fredrik Lundhb9056332001-07-11 17:42:21 +0000612
Fredrik Lundh3d9addd2002-06-27 21:36:21 +0000613 def dump_string(self, value, write, escape=escape):
614 write("<value><string>")
615 write(escape(value))
616 write("</string></value>\n")
Guido van Rossum54ad5232007-05-27 09:17:48 +0000617 dispatch[str8] = dump_string
Fredrik Lundhb9056332001-07-11 17:42:21 +0000618
Guido van Rossum54ad5232007-05-27 09:17:48 +0000619 def dump_unicode(self, value, write, escape=escape):
Guido van Rossum54ad5232007-05-27 09:17:48 +0000620 write("<value><string>")
621 write(escape(value))
622 write("</string></value>\n")
623 dispatch[str] = dump_unicode
Fredrik Lundhb9056332001-07-11 17:42:21 +0000624
Fredrik Lundh3d9addd2002-06-27 21:36:21 +0000625 def dump_array(self, value, write):
626 i = id(value)
Guido van Rossume2b70bc2006-08-18 22:13:04 +0000627 if i in self.memo:
Fredrik Lundh3d9addd2002-06-27 21:36:21 +0000628 raise TypeError, "cannot marshal recursive sequences"
629 self.memo[i] = None
Fredrik Lundh1538c232001-10-01 19:42:03 +0000630 dump = self.__dump
Fredrik Lundhb9056332001-07-11 17:42:21 +0000631 write("<value><array><data>\n")
632 for v in value:
Fredrik Lundh3d9addd2002-06-27 21:36:21 +0000633 dump(v, write)
Fredrik Lundhb9056332001-07-11 17:42:21 +0000634 write("</data></array></value>\n")
Fredrik Lundh3d9addd2002-06-27 21:36:21 +0000635 del self.memo[i]
Guido van Rossum13257902007-06-07 23:15:56 +0000636 dispatch[tuple] = dump_array
637 dispatch[list] = dump_array
Fredrik Lundhb9056332001-07-11 17:42:21 +0000638
Fredrik Lundh3d9addd2002-06-27 21:36:21 +0000639 def dump_struct(self, value, write, escape=escape):
640 i = id(value)
Guido van Rossume2b70bc2006-08-18 22:13:04 +0000641 if i in self.memo:
Fredrik Lundh3d9addd2002-06-27 21:36:21 +0000642 raise TypeError, "cannot marshal recursive dictionaries"
643 self.memo[i] = None
Fredrik Lundh1538c232001-10-01 19:42:03 +0000644 dump = self.__dump
Fredrik Lundhb9056332001-07-11 17:42:21 +0000645 write("<value><struct>\n")
Andrew M. Kuchling5962f452004-06-05 12:35:58 +0000646 for k, v in value.items():
Fredrik Lundhb9056332001-07-11 17:42:21 +0000647 write("<member>\n")
Brett Cannon2dbde5e2007-07-30 03:50:35 +0000648 if not isinstance(k, basestring):
Guido van Rossum54ad5232007-05-27 09:17:48 +0000649 raise TypeError, "dictionary key must be string"
Fredrik Lundh1538c232001-10-01 19:42:03 +0000650 write("<name>%s</name>\n" % escape(k))
Andrew M. Kuchling5962f452004-06-05 12:35:58 +0000651 dump(v, write)
Fredrik Lundhb9056332001-07-11 17:42:21 +0000652 write("</member>\n")
653 write("</struct></value>\n")
Fredrik Lundh3d9addd2002-06-27 21:36:21 +0000654 del self.memo[i]
Guido van Rossum13257902007-06-07 23:15:56 +0000655 dispatch[dict] = dump_struct
Fredrik Lundhb9056332001-07-11 17:42:21 +0000656
Fred Drakeba613c32005-02-10 18:33:30 +0000657 if datetime:
658 def dump_datetime(self, value, write):
659 write("<value><dateTime.iso8601>")
660 write(value.strftime("%Y%m%dT%H:%M:%S"))
661 write("</dateTime.iso8601></value>\n")
662 dispatch[datetime.datetime] = dump_datetime
663
Skip Montanaro174dd222005-05-14 20:54:16 +0000664 def dump_date(self, value, write):
665 write("<value><dateTime.iso8601>")
666 write(value.strftime("%Y%m%dT00:00:00"))
667 write("</dateTime.iso8601></value>\n")
668 dispatch[datetime.date] = dump_date
669
670 def dump_time(self, value, write):
671 write("<value><dateTime.iso8601>")
672 write(datetime.datetime.now().date().strftime("%Y%m%dT"))
673 write(value.strftime("%H:%M:%S"))
674 write("</dateTime.iso8601></value>\n")
675 dispatch[datetime.time] = dump_time
676
Fredrik Lundh3d9addd2002-06-27 21:36:21 +0000677 def dump_instance(self, value, write):
Fredrik Lundhb9056332001-07-11 17:42:21 +0000678 # check for special wrappers
679 if value.__class__ in WRAPPERS:
Fredrik Lundh3d9addd2002-06-27 21:36:21 +0000680 self.write = write
Fredrik Lundhb9056332001-07-11 17:42:21 +0000681 value.encode(self)
Fredrik Lundh3d9addd2002-06-27 21:36:21 +0000682 del self.write
Fredrik Lundhb9056332001-07-11 17:42:21 +0000683 else:
684 # store instance attributes as a struct (really?)
Fredrik Lundh3d9addd2002-06-27 21:36:21 +0000685 self.dump_struct(value.__dict__, write)
Guido van Rossume4dea982006-04-21 13:45:00 +0000686 dispatch[DateTime] = dump_instance
687 dispatch[Binary] = dump_instance
Thomas Wouters89f507f2006-12-13 04:49:30 +0000688 # XXX(twouters): using "_arbitrary_instance" as key as a quick-fix
689 # for the p3yk merge, this should probably be fixed more neatly.
690 dispatch["_arbitrary_instance"] = dump_instance
Fredrik Lundhb9056332001-07-11 17:42:21 +0000691
Fredrik Lundh3d9addd2002-06-27 21:36:21 +0000692##
693# XML-RPC unmarshaller.
694#
695# @see loads
696
Fredrik Lundhb9056332001-07-11 17:42:21 +0000697class Unmarshaller:
Fred Drake1b410792001-09-04 18:55:03 +0000698 """Unmarshal an XML-RPC response, based on incoming XML event
Fredrik Lundhb0e8e9b2001-09-10 21:45:42 +0000699 messages (start, data, end). Call close() to get the resulting
Fred Drake1b410792001-09-04 18:55:03 +0000700 data structure.
Fredrik Lundhb9056332001-07-11 17:42:21 +0000701
Fredrik Lundhb0e8e9b2001-09-10 21:45:42 +0000702 Note that this reader is fairly tolerant, and gladly accepts bogus
703 XML-RPC data without complaining (but not bogus XML).
Fred Drake1b410792001-09-04 18:55:03 +0000704 """
Fredrik Lundhb9056332001-07-11 17:42:21 +0000705
706 # and again, if you don't understand what's going on in here,
707 # that's perfectly ok.
708
Skip Montanaro174dd222005-05-14 20:54:16 +0000709 def __init__(self, use_datetime=0):
Fredrik Lundhb9056332001-07-11 17:42:21 +0000710 self._type = None
711 self._stack = []
712 self._marks = []
713 self._data = []
714 self._methodname = None
715 self._encoding = "utf-8"
716 self.append = self._stack.append
Skip Montanaro174dd222005-05-14 20:54:16 +0000717 self._use_datetime = use_datetime
718 if use_datetime and not datetime:
719 raise ValueError, "the datetime module is not available"
Fredrik Lundhb9056332001-07-11 17:42:21 +0000720
721 def close(self):
722 # return response tuple and target method
723 if self._type is None or self._marks:
724 raise ResponseError()
725 if self._type == "fault":
Guido van Rossum68468eb2003-02-27 20:14:51 +0000726 raise Fault(**self._stack[0])
Fredrik Lundhb9056332001-07-11 17:42:21 +0000727 return tuple(self._stack)
728
729 def getmethodname(self):
730 return self._methodname
731
732 #
733 # event handlers
734
735 def xml(self, encoding, standalone):
736 self._encoding = encoding
737 # FIXME: assert standalone == 1 ???
738
739 def start(self, tag, attrs):
740 # prepare to handle this element
741 if tag == "array" or tag == "struct":
742 self._marks.append(len(self._stack))
743 self._data = []
744 self._value = (tag == "value")
745
746 def data(self, text):
747 self._data.append(text)
748
Neal Norwitzff113342007-04-17 08:42:15 +0000749 def end(self, tag):
Fredrik Lundhb9056332001-07-11 17:42:21 +0000750 # call the appropriate end tag handler
751 try:
752 f = self.dispatch[tag]
753 except KeyError:
754 pass # unknown tag ?
755 else:
Neal Norwitzff113342007-04-17 08:42:15 +0000756 return f(self, "".join(self._data))
Fredrik Lundhb9056332001-07-11 17:42:21 +0000757
758 #
759 # accelerator support
760
761 def end_dispatch(self, tag, data):
762 # dispatch data
763 try:
764 f = self.dispatch[tag]
765 except KeyError:
766 pass # unknown tag ?
767 else:
768 return f(self, data)
769
770 #
771 # element decoders
772
773 dispatch = {}
774
Andrew M. Kuchlinga4c2b742003-04-25 00:26:51 +0000775 def end_nil (self, data):
776 self.append(None)
777 self._value = 0
778 dispatch["nil"] = end_nil
Tim Petersc2659cf2003-05-12 20:19:37 +0000779
Fredrik Lundh1538c232001-10-01 19:42:03 +0000780 def end_boolean(self, data):
Fredrik Lundhb9056332001-07-11 17:42:21 +0000781 if data == "0":
782 self.append(False)
783 elif data == "1":
784 self.append(True)
785 else:
786 raise TypeError, "bad boolean value"
787 self._value = 0
788 dispatch["boolean"] = end_boolean
789
Fredrik Lundh1538c232001-10-01 19:42:03 +0000790 def end_int(self, data):
791 self.append(int(data))
Fredrik Lundhb9056332001-07-11 17:42:21 +0000792 self._value = 0
793 dispatch["i4"] = end_int
794 dispatch["int"] = end_int
795
Fredrik Lundh1538c232001-10-01 19:42:03 +0000796 def end_double(self, data):
797 self.append(float(data))
Fredrik Lundhb9056332001-07-11 17:42:21 +0000798 self._value = 0
799 dispatch["double"] = end_double
800
Fredrik Lundh1538c232001-10-01 19:42:03 +0000801 def end_string(self, data):
Fredrik Lundhb9056332001-07-11 17:42:21 +0000802 if self._encoding:
803 data = _decode(data, self._encoding)
804 self.append(_stringify(data))
805 self._value = 0
806 dispatch["string"] = end_string
807 dispatch["name"] = end_string # struct keys are always strings
808
809 def end_array(self, data):
Raymond Hettinger46ac8eb2002-06-30 03:39:14 +0000810 mark = self._marks.pop()
Fredrik Lundhb9056332001-07-11 17:42:21 +0000811 # map arrays to Python lists
812 self._stack[mark:] = [self._stack[mark:]]
813 self._value = 0
814 dispatch["array"] = end_array
815
816 def end_struct(self, data):
Raymond Hettinger46ac8eb2002-06-30 03:39:14 +0000817 mark = self._marks.pop()
Fredrik Lundhb9056332001-07-11 17:42:21 +0000818 # map structs to Python dictionaries
819 dict = {}
820 items = self._stack[mark:]
821 for i in range(0, len(items), 2):
822 dict[_stringify(items[i])] = items[i+1]
823 self._stack[mark:] = [dict]
824 self._value = 0
825 dispatch["struct"] = end_struct
826
Fredrik Lundh1538c232001-10-01 19:42:03 +0000827 def end_base64(self, data):
Fredrik Lundhb9056332001-07-11 17:42:21 +0000828 value = Binary()
Fredrik Lundh1538c232001-10-01 19:42:03 +0000829 value.decode(data)
Fredrik Lundhb9056332001-07-11 17:42:21 +0000830 self.append(value)
831 self._value = 0
832 dispatch["base64"] = end_base64
833
Fredrik Lundh1538c232001-10-01 19:42:03 +0000834 def end_dateTime(self, data):
Fredrik Lundhb9056332001-07-11 17:42:21 +0000835 value = DateTime()
Fredrik Lundh1538c232001-10-01 19:42:03 +0000836 value.decode(data)
Skip Montanaro174dd222005-05-14 20:54:16 +0000837 if self._use_datetime:
838 value = _datetime_type(data)
Fredrik Lundhb9056332001-07-11 17:42:21 +0000839 self.append(value)
840 dispatch["dateTime.iso8601"] = end_dateTime
841
842 def end_value(self, data):
Fredrik Lundh3d9addd2002-06-27 21:36:21 +0000843 # if we stumble upon a value element with no internal
Fredrik Lundhb9056332001-07-11 17:42:21 +0000844 # elements, treat it as a string element
845 if self._value:
846 self.end_string(data)
847 dispatch["value"] = end_value
848
849 def end_params(self, data):
850 self._type = "params"
851 dispatch["params"] = end_params
852
853 def end_fault(self, data):
854 self._type = "fault"
855 dispatch["fault"] = end_fault
856
Fredrik Lundh1538c232001-10-01 19:42:03 +0000857 def end_methodName(self, data):
Fredrik Lundhb9056332001-07-11 17:42:21 +0000858 if self._encoding:
859 data = _decode(data, self._encoding)
860 self._methodname = data
861 self._type = "methodName" # no params
862 dispatch["methodName"] = end_methodName
863
Martin v. Löwis45394c22003-10-31 13:49:36 +0000864## Multicall support
865#
Fredrik Lundhb9056332001-07-11 17:42:21 +0000866
Martin v. Löwis45394c22003-10-31 13:49:36 +0000867class _MultiCallMethod:
868 # some lesser magic to store calls made to a MultiCall object
869 # for batch execution
870 def __init__(self, call_list, name):
871 self.__call_list = call_list
872 self.__name = name
873 def __getattr__(self, name):
874 return _MultiCallMethod(self.__call_list, "%s.%s" % (self.__name, name))
875 def __call__(self, *args):
876 self.__call_list.append((self.__name, args))
877
Martin v. Löwis12237b32004-08-22 16:04:50 +0000878class MultiCallIterator:
Martin v. Löwis45394c22003-10-31 13:49:36 +0000879 """Iterates over the results of a multicall. Exceptions are
880 thrown in response to xmlrpc faults."""
Raymond Hettingercc523fc2003-11-02 09:47:05 +0000881
Martin v. Löwis12237b32004-08-22 16:04:50 +0000882 def __init__(self, results):
883 self.results = results
884
885 def __getitem__(self, i):
886 item = self.results[i]
887 if type(item) == type({}):
888 raise Fault(item['faultCode'], item['faultString'])
889 elif type(item) == type([]):
890 return item[0]
Martin v. Löwis45394c22003-10-31 13:49:36 +0000891 else:
892 raise ValueError,\
893 "unexpected type in multicall result"
Raymond Hettingercc523fc2003-11-02 09:47:05 +0000894
Martin v. Löwis45394c22003-10-31 13:49:36 +0000895class MultiCall:
896 """server -> a object used to boxcar method calls
897
898 server should be a ServerProxy object.
899
900 Methods can be added to the MultiCall using normal
901 method call syntax e.g.:
902
903 multicall = MultiCall(server_proxy)
904 multicall.add(2,3)
905 multicall.get_address("Guido")
906
907 To execute the multicall, call the MultiCall object e.g.:
908
909 add_result, address = multicall()
910 """
Raymond Hettingercc523fc2003-11-02 09:47:05 +0000911
Martin v. Löwis45394c22003-10-31 13:49:36 +0000912 def __init__(self, server):
913 self.__server = server
914 self.__call_list = []
Raymond Hettingercc523fc2003-11-02 09:47:05 +0000915
Martin v. Löwis45394c22003-10-31 13:49:36 +0000916 def __repr__(self):
917 return "<MultiCall at %x>" % id(self)
Raymond Hettingercc523fc2003-11-02 09:47:05 +0000918
Martin v. Löwis45394c22003-10-31 13:49:36 +0000919 __str__ = __repr__
920
921 def __getattr__(self, name):
922 return _MultiCallMethod(self.__call_list, name)
923
924 def __call__(self):
925 marshalled_list = []
926 for name, args in self.__call_list:
927 marshalled_list.append({'methodName' : name, 'params' : args})
928
929 return MultiCallIterator(self.__server.system.multicall(marshalled_list))
Raymond Hettingercc523fc2003-11-02 09:47:05 +0000930
Fredrik Lundhb9056332001-07-11 17:42:21 +0000931# --------------------------------------------------------------------
932# convenience functions
933
Fredrik Lundh3d9addd2002-06-27 21:36:21 +0000934##
935# Create a parser object, and connect it to an unmarshalling instance.
936# This function picks the fastest available XML parser.
937#
938# return A (parser, unmarshaller) tuple.
939
Skip Montanaro174dd222005-05-14 20:54:16 +0000940def getparser(use_datetime=0):
Fredrik Lundhb9056332001-07-11 17:42:21 +0000941 """getparser() -> parser, unmarshaller
942
Fredrik Lundhb0e8e9b2001-09-10 21:45:42 +0000943 Create an instance of the fastest available parser, and attach it
944 to an unmarshalling object. Return both objects.
Fredrik Lundhb9056332001-07-11 17:42:21 +0000945 """
Skip Montanaro174dd222005-05-14 20:54:16 +0000946 if use_datetime and not datetime:
947 raise ValueError, "the datetime module is not available"
Fredrik Lundhb9056332001-07-11 17:42:21 +0000948 if FastParser and FastUnmarshaller:
Skip Montanaro174dd222005-05-14 20:54:16 +0000949 if use_datetime:
950 mkdatetime = _datetime_type
951 else:
952 mkdatetime = _datetime
953 target = FastUnmarshaller(True, False, _binary, mkdatetime, Fault)
Fredrik Lundhb9056332001-07-11 17:42:21 +0000954 parser = FastParser(target)
955 else:
Skip Montanaro174dd222005-05-14 20:54:16 +0000956 target = Unmarshaller(use_datetime=use_datetime)
Fredrik Lundhb9056332001-07-11 17:42:21 +0000957 if FastParser:
958 parser = FastParser(target)
959 elif SgmlopParser:
960 parser = SgmlopParser(target)
961 elif ExpatParser:
962 parser = ExpatParser(target)
963 else:
964 parser = SlowParser(target)
965 return parser, target
966
Fredrik Lundh3d9addd2002-06-27 21:36:21 +0000967##
968# Convert a Python tuple or a Fault instance to an XML-RPC packet.
969#
970# @def dumps(params, **options)
971# @param params A tuple or Fault instance.
972# @keyparam methodname If given, create a methodCall request for
973# this method name.
974# @keyparam methodresponse If given, create a methodResponse packet.
975# If used with a tuple, the tuple must be a singleton (that is,
976# it must contain exactly one element).
977# @keyparam encoding The packet encoding.
978# @return A string containing marshalled data.
979
Andrew M. Kuchlinga4c2b742003-04-25 00:26:51 +0000980def dumps(params, methodname=None, methodresponse=None, encoding=None,
981 allow_none=0):
Fredrik Lundhb9056332001-07-11 17:42:21 +0000982 """data [,options] -> marshalled data
983
984 Convert an argument tuple or a Fault instance to an XML-RPC
985 request (or response, if the methodresponse option is used).
986
Fredrik Lundhb0e8e9b2001-09-10 21:45:42 +0000987 In addition to the data object, the following options can be given
988 as keyword arguments:
Fredrik Lundhb9056332001-07-11 17:42:21 +0000989
990 methodname: the method name for a methodCall packet
991
992 methodresponse: true to create a methodResponse packet.
993 If this option is used with a tuple, the tuple must be
994 a singleton (i.e. it can contain only one element).
995
996 encoding: the packet encoding (default is UTF-8)
997
998 All 8-bit strings in the data structure are assumed to use the
999 packet encoding. Unicode strings are automatically converted,
Fredrik Lundhb0e8e9b2001-09-10 21:45:42 +00001000 where necessary.
Fredrik Lundhb9056332001-07-11 17:42:21 +00001001 """
1002
Guido van Rossum13257902007-06-07 23:15:56 +00001003 assert isinstance(params, (tuple, Fault)), "argument must be tuple or Fault instance"
Fredrik Lundhb9056332001-07-11 17:42:21 +00001004 if isinstance(params, Fault):
1005 methodresponse = 1
Guido van Rossum13257902007-06-07 23:15:56 +00001006 elif methodresponse and isinstance(params, tuple):
Fredrik Lundhb9056332001-07-11 17:42:21 +00001007 assert len(params) == 1, "response tuple must be a singleton"
1008
Fredrik Lundh3d9addd2002-06-27 21:36:21 +00001009 if not encoding:
Fredrik Lundhb9056332001-07-11 17:42:21 +00001010 encoding = "utf-8"
1011
Fredrik Lundh3d9addd2002-06-27 21:36:21 +00001012 if FastMarshaller:
1013 m = FastMarshaller(encoding)
1014 else:
Andrew M. Kuchlinga4c2b742003-04-25 00:26:51 +00001015 m = Marshaller(encoding, allow_none)
Fredrik Lundh3d9addd2002-06-27 21:36:21 +00001016
Fredrik Lundhb9056332001-07-11 17:42:21 +00001017 data = m.dumps(params)
1018
1019 if encoding != "utf-8":
Fredrik Lundh3d9addd2002-06-27 21:36:21 +00001020 xmlheader = "<?xml version='1.0' encoding='%s'?>\n" % str(encoding)
Fredrik Lundhb9056332001-07-11 17:42:21 +00001021 else:
1022 xmlheader = "<?xml version='1.0'?>\n" # utf-8 is default
1023
1024 # standard XML-RPC wrappings
Fredrik Lundh3d9addd2002-06-27 21:36:21 +00001025 if methodname:
Fredrik Lundhb9056332001-07-11 17:42:21 +00001026 # a method call
Guido van Rossum54ad5232007-05-27 09:17:48 +00001027 if not isinstance(methodname, basestring):
Fredrik Lundhb9056332001-07-11 17:42:21 +00001028 methodname = methodname.encode(encoding)
1029 data = (
1030 xmlheader,
1031 "<methodCall>\n"
1032 "<methodName>", methodname, "</methodName>\n",
1033 data,
1034 "</methodCall>\n"
1035 )
1036 elif methodresponse:
1037 # a method response, or a fault structure
1038 data = (
1039 xmlheader,
1040 "<methodResponse>\n",
1041 data,
1042 "</methodResponse>\n"
1043 )
1044 else:
1045 return data # return as is
Neal Norwitzff113342007-04-17 08:42:15 +00001046 return "".join(data)
Fredrik Lundhb9056332001-07-11 17:42:21 +00001047
Fredrik Lundh3d9addd2002-06-27 21:36:21 +00001048##
1049# Convert an XML-RPC packet to a Python object. If the XML-RPC packet
1050# represents a fault condition, this function raises a Fault exception.
1051#
1052# @param data An XML-RPC packet, given as an 8-bit string.
Walter Dörwaldf0dfc7a2003-10-20 14:01:56 +00001053# @return A tuple containing the unpacked data, and the method name
Fredrik Lundh3d9addd2002-06-27 21:36:21 +00001054# (None if not present).
1055# @see Fault
1056
Skip Montanaro174dd222005-05-14 20:54:16 +00001057def loads(data, use_datetime=0):
Fredrik Lundhb9056332001-07-11 17:42:21 +00001058 """data -> unmarshalled data, method name
1059
1060 Convert an XML-RPC packet to unmarshalled data plus a method
1061 name (None if not present).
1062
1063 If the XML-RPC packet represents a fault condition, this function
1064 raises a Fault exception.
1065 """
Skip Montanaro174dd222005-05-14 20:54:16 +00001066 p, u = getparser(use_datetime=use_datetime)
Fredrik Lundhb9056332001-07-11 17:42:21 +00001067 p.feed(data)
1068 p.close()
1069 return u.close(), u.getmethodname()
1070
1071
1072# --------------------------------------------------------------------
1073# request dispatcher
1074
1075class _Method:
1076 # some magic to bind an XML-RPC method to an RPC server.
1077 # supports "nested" methods (e.g. examples.getStateName)
1078 def __init__(self, send, name):
1079 self.__send = send
1080 self.__name = name
1081 def __getattr__(self, name):
1082 return _Method(self.__send, "%s.%s" % (self.__name, name))
1083 def __call__(self, *args):
1084 return self.__send(self.__name, args)
1085
Fredrik Lundh3d9addd2002-06-27 21:36:21 +00001086##
1087# Standard transport class for XML-RPC over HTTP.
1088# <p>
1089# You can create custom transports by subclassing this method, and
1090# overriding selected methods.
Fredrik Lundhb9056332001-07-11 17:42:21 +00001091
1092class Transport:
Fredrik Lundhc4c062f2001-09-10 19:45:02 +00001093 """Handles an HTTP transaction to an XML-RPC server."""
Fredrik Lundhb9056332001-07-11 17:42:21 +00001094
1095 # client identifier (may be overridden)
1096 user_agent = "xmlrpclib.py/%s (by www.pythonware.com)" % __version__
1097
Skip Montanaro174dd222005-05-14 20:54:16 +00001098 def __init__(self, use_datetime=0):
1099 self._use_datetime = use_datetime
1100
Fredrik Lundh3d9addd2002-06-27 21:36:21 +00001101 ##
1102 # Send a complete request, and parse the response.
1103 #
1104 # @param host Target host.
1105 # @param handler Target PRC handler.
1106 # @param request_body XML-RPC request body.
1107 # @param verbose Debugging flag.
1108 # @return Parsed response.
1109
Fredrik Lundhb9056332001-07-11 17:42:21 +00001110 def request(self, host, handler, request_body, verbose=0):
1111 # issue XML-RPC request
1112
1113 h = self.make_connection(host)
1114 if verbose:
1115 h.set_debuglevel(1)
1116
1117 self.send_request(h, handler, request_body)
1118 self.send_host(h, host)
1119 self.send_user_agent(h)
1120 self.send_content(h, request_body)
1121
1122 errcode, errmsg, headers = h.getreply()
1123
1124 if errcode != 200:
1125 raise ProtocolError(
1126 host + handler,
1127 errcode, errmsg,
1128 headers
1129 )
1130
1131 self.verbose = verbose
1132
Fredrik Lundh3d9addd2002-06-27 21:36:21 +00001133 try:
1134 sock = h._conn.sock
1135 except AttributeError:
1136 sock = None
1137
1138 return self._parse_response(h.getfile(), sock)
1139
1140 ##
1141 # Create parser.
1142 #
1143 # @return A 2-tuple containing a parser and a unmarshaller.
Fredrik Lundhb9056332001-07-11 17:42:21 +00001144
Fredrik Lundhc4c062f2001-09-10 19:45:02 +00001145 def getparser(self):
1146 # get parser and unmarshaller
Skip Montanaro174dd222005-05-14 20:54:16 +00001147 return getparser(use_datetime=self._use_datetime)
Fredrik Lundhc4c062f2001-09-10 19:45:02 +00001148
Fredrik Lundh3d9addd2002-06-27 21:36:21 +00001149 ##
Fredrik Lundh1303c7c2002-10-22 18:23:00 +00001150 # Get authorization info from host parameter
1151 # Host may be a string, or a (host, x509-dict) tuple; if a string,
1152 # it is checked for a "user:pw@host" format, and a "Basic
1153 # Authentication" header is added if appropriate.
1154 #
1155 # @param host Host descriptor (URL or (URL, x509 info) tuple).
1156 # @return A 3-tuple containing (actual host, extra headers,
1157 # x509 info). The header and x509 fields may be None.
1158
1159 def get_host_info(self, host):
1160
1161 x509 = {}
Guido van Rossum13257902007-06-07 23:15:56 +00001162 if isinstance(host, tuple):
Fredrik Lundh1303c7c2002-10-22 18:23:00 +00001163 host, x509 = host
1164
1165 import urllib
1166 auth, host = urllib.splituser(host)
1167
1168 if auth:
1169 import base64
Fredrik Lundh768c98b2002-11-01 17:14:16 +00001170 auth = base64.encodestring(urllib.unquote(auth))
Neal Norwitzff113342007-04-17 08:42:15 +00001171 auth = "".join(auth.split()) # get rid of whitespace
Fredrik Lundh1303c7c2002-10-22 18:23:00 +00001172 extra_headers = [
1173 ("Authorization", "Basic " + auth)
1174 ]
1175 else:
1176 extra_headers = None
1177
1178 return host, extra_headers, x509
1179
1180 ##
Fredrik Lundh3d9addd2002-06-27 21:36:21 +00001181 # Connect to server.
1182 #
1183 # @param host Target host.
1184 # @return A connection handle.
1185
Fredrik Lundhb9056332001-07-11 17:42:21 +00001186 def make_connection(self, host):
1187 # create a HTTP connection object from a host descriptor
1188 import httplib
Fredrik Lundh1303c7c2002-10-22 18:23:00 +00001189 host, extra_headers, x509 = self.get_host_info(host)
Fredrik Lundhb9056332001-07-11 17:42:21 +00001190 return httplib.HTTP(host)
1191
Fredrik Lundh3d9addd2002-06-27 21:36:21 +00001192 ##
1193 # Send request header.
1194 #
1195 # @param connection Connection handle.
1196 # @param handler Target RPC handler.
1197 # @param request_body XML-RPC body.
1198
Fredrik Lundhb9056332001-07-11 17:42:21 +00001199 def send_request(self, connection, handler, request_body):
1200 connection.putrequest("POST", handler)
1201
Fredrik Lundh3d9addd2002-06-27 21:36:21 +00001202 ##
1203 # Send host name.
1204 #
1205 # @param connection Connection handle.
1206 # @param host Host name.
1207
Fredrik Lundhb9056332001-07-11 17:42:21 +00001208 def send_host(self, connection, host):
Fredrik Lundh1303c7c2002-10-22 18:23:00 +00001209 host, extra_headers, x509 = self.get_host_info(host)
Fredrik Lundhb9056332001-07-11 17:42:21 +00001210 connection.putheader("Host", host)
Fredrik Lundh1303c7c2002-10-22 18:23:00 +00001211 if extra_headers:
Guido van Rossum13257902007-06-07 23:15:56 +00001212 if isinstance(extra_headers, dict):
Fredrik Lundh1303c7c2002-10-22 18:23:00 +00001213 extra_headers = extra_headers.items()
1214 for key, value in extra_headers:
1215 connection.putheader(key, value)
Fredrik Lundhb9056332001-07-11 17:42:21 +00001216
Fredrik Lundh3d9addd2002-06-27 21:36:21 +00001217 ##
1218 # Send user-agent identifier.
1219 #
1220 # @param connection Connection handle.
1221
Fredrik Lundhb9056332001-07-11 17:42:21 +00001222 def send_user_agent(self, connection):
1223 connection.putheader("User-Agent", self.user_agent)
1224
Fredrik Lundh3d9addd2002-06-27 21:36:21 +00001225 ##
1226 # Send request body.
1227 #
1228 # @param connection Connection handle.
1229 # @param request_body XML-RPC request body.
1230
Fredrik Lundhb9056332001-07-11 17:42:21 +00001231 def send_content(self, connection, request_body):
1232 connection.putheader("Content-Type", "text/xml")
1233 connection.putheader("Content-Length", str(len(request_body)))
1234 connection.endheaders()
1235 if request_body:
1236 connection.send(request_body)
1237
Fredrik Lundh3d9addd2002-06-27 21:36:21 +00001238 ##
1239 # Parse response.
1240 #
1241 # @param file Stream.
1242 # @return Response tuple and target method.
1243
1244 def parse_response(self, file):
1245 # compatibility interface
1246 return self._parse_response(file, None)
1247
1248 ##
1249 # Parse response (alternate interface). This is similar to the
1250 # parse_response method, but also provides direct access to the
1251 # underlying socket object (where available).
1252 #
1253 # @param file Stream.
1254 # @param sock Socket handle (or None, if the socket object
1255 # could not be accessed).
1256 # @return Response tuple and target method.
1257
1258 def _parse_response(self, file, sock):
1259 # read response from input file/socket, and parse it
Fredrik Lundhb9056332001-07-11 17:42:21 +00001260
Fredrik Lundhc4c062f2001-09-10 19:45:02 +00001261 p, u = self.getparser()
Fredrik Lundhb9056332001-07-11 17:42:21 +00001262
1263 while 1:
Fredrik Lundh3d9addd2002-06-27 21:36:21 +00001264 if sock:
1265 response = sock.recv(1024)
1266 else:
1267 response = file.read(1024)
Fredrik Lundhb9056332001-07-11 17:42:21 +00001268 if not response:
1269 break
1270 if self.verbose:
Guido van Rossumbe19ed72007-02-09 05:37:30 +00001271 print("body:", repr(response))
Fredrik Lundhb9056332001-07-11 17:42:21 +00001272 p.feed(response)
1273
Fredrik Lundh3d9addd2002-06-27 21:36:21 +00001274 file.close()
Fredrik Lundhb9056332001-07-11 17:42:21 +00001275 p.close()
1276
1277 return u.close()
1278
Fredrik Lundh3d9addd2002-06-27 21:36:21 +00001279##
1280# Standard transport class for XML-RPC over HTTPS.
1281
Fredrik Lundhb9056332001-07-11 17:42:21 +00001282class SafeTransport(Transport):
Fredrik Lundhc4c062f2001-09-10 19:45:02 +00001283 """Handles an HTTPS transaction to an XML-RPC server."""
Fredrik Lundhb9056332001-07-11 17:42:21 +00001284
Fredrik Lundh3d9addd2002-06-27 21:36:21 +00001285 # FIXME: mostly untested
1286
Fredrik Lundhb9056332001-07-11 17:42:21 +00001287 def make_connection(self, host):
1288 # create a HTTPS connection object from a host descriptor
1289 # host may be a string, or a (host, x509-dict) tuple
1290 import httplib
Fredrik Lundh1303c7c2002-10-22 18:23:00 +00001291 host, extra_headers, x509 = self.get_host_info(host)
Fredrik Lundhb9056332001-07-11 17:42:21 +00001292 try:
1293 HTTPS = httplib.HTTPS
1294 except AttributeError:
Fredrik Lundh1303c7c2002-10-22 18:23:00 +00001295 raise NotImplementedError(
1296 "your version of httplib doesn't support HTTPS"
1297 )
Fredrik Lundhb9056332001-07-11 17:42:21 +00001298 else:
Guido van Rossum68468eb2003-02-27 20:14:51 +00001299 return HTTPS(host, None, **(x509 or {}))
Fredrik Lundhb9056332001-07-11 17:42:21 +00001300
Fredrik Lundh3d9addd2002-06-27 21:36:21 +00001301##
1302# Standard server proxy. This class establishes a virtual connection
1303# to an XML-RPC server.
1304# <p>
1305# This class is available as ServerProxy and Server. New code should
1306# use ServerProxy, to avoid confusion.
1307#
1308# @def ServerProxy(uri, **options)
1309# @param uri The connection point on the server.
1310# @keyparam transport A transport factory, compatible with the
1311# standard transport class.
1312# @keyparam encoding The default encoding used for 8-bit strings
1313# (default is UTF-8).
1314# @keyparam verbose Use a true value to enable debugging output.
1315# (printed to standard output).
1316# @see Transport
1317
Fredrik Lundhb9056332001-07-11 17:42:21 +00001318class ServerProxy:
1319 """uri [,options] -> a logical connection to an XML-RPC server
1320
1321 uri is the connection point on the server, given as
1322 scheme://host/target.
1323
1324 The standard implementation always supports the "http" scheme. If
1325 SSL socket support is available (Python 2.0), it also supports
1326 "https".
1327
1328 If the target part and the slash preceding it are both omitted,
1329 "/RPC2" is assumed.
1330
1331 The following options can be given as keyword arguments:
1332
1333 transport: a transport factory
1334 encoding: the request encoding (default is UTF-8)
1335
1336 All 8-bit strings passed to the server proxy are assumed to use
1337 the given encoding.
1338 """
1339
Andrew M. Kuchlinga4c2b742003-04-25 00:26:51 +00001340 def __init__(self, uri, transport=None, encoding=None, verbose=0,
Skip Montanaro174dd222005-05-14 20:54:16 +00001341 allow_none=0, use_datetime=0):
Fredrik Lundhb9056332001-07-11 17:42:21 +00001342 # establish a "logical" server connection
1343
1344 # get the url
Fredrik Lundhc4c062f2001-09-10 19:45:02 +00001345 import urllib
Fredrik Lundhb9056332001-07-11 17:42:21 +00001346 type, uri = urllib.splittype(uri)
1347 if type not in ("http", "https"):
1348 raise IOError, "unsupported XML-RPC protocol"
1349 self.__host, self.__handler = urllib.splithost(uri)
1350 if not self.__handler:
1351 self.__handler = "/RPC2"
1352
1353 if transport is None:
1354 if type == "https":
Skip Montanaro174dd222005-05-14 20:54:16 +00001355 transport = SafeTransport(use_datetime=use_datetime)
Fredrik Lundhb9056332001-07-11 17:42:21 +00001356 else:
Skip Montanaro174dd222005-05-14 20:54:16 +00001357 transport = Transport(use_datetime=use_datetime)
Fredrik Lundhb9056332001-07-11 17:42:21 +00001358 self.__transport = transport
1359
1360 self.__encoding = encoding
1361 self.__verbose = verbose
Andrew M. Kuchlinga4c2b742003-04-25 00:26:51 +00001362 self.__allow_none = allow_none
Tim Petersc2659cf2003-05-12 20:19:37 +00001363
Fredrik Lundhb9056332001-07-11 17:42:21 +00001364 def __request(self, methodname, params):
1365 # call a method on the remote server
1366
Andrew M. Kuchlinga4c2b742003-04-25 00:26:51 +00001367 request = dumps(params, methodname, encoding=self.__encoding,
1368 allow_none=self.__allow_none)
Fredrik Lundhb9056332001-07-11 17:42:21 +00001369
1370 response = self.__transport.request(
1371 self.__host,
1372 self.__handler,
1373 request,
1374 verbose=self.__verbose
1375 )
1376
1377 if len(response) == 1:
1378 response = response[0]
1379
1380 return response
1381
1382 def __repr__(self):
1383 return (
Fredrik Lundh78eedce2001-08-23 20:04:33 +00001384 "<ServerProxy for %s%s>" %
Fredrik Lundhb9056332001-07-11 17:42:21 +00001385 (self.__host, self.__handler)
1386 )
1387
1388 __str__ = __repr__
Raymond Hettingercc523fc2003-11-02 09:47:05 +00001389
Fredrik Lundhb9056332001-07-11 17:42:21 +00001390 def __getattr__(self, name):
1391 # magic method dispatcher
1392 return _Method(self.__request, name)
1393
1394 # note: to call a remote object with an non-standard name, use
1395 # result getattr(server, "strange-python-name")(args)
1396
Fredrik Lundh78eedce2001-08-23 20:04:33 +00001397# compatibility
Fredrik Lundh3d9addd2002-06-27 21:36:21 +00001398
Fredrik Lundhb9056332001-07-11 17:42:21 +00001399Server = ServerProxy
1400
1401# --------------------------------------------------------------------
1402# test code
1403
1404if __name__ == "__main__":
1405
1406 # simple test program (from the XML-RPC specification)
1407
Fredrik Lundh78eedce2001-08-23 20:04:33 +00001408 # server = ServerProxy("http://localhost:8000") # local server
Martin v. Löwis12237b32004-08-22 16:04:50 +00001409 server = ServerProxy("http://time.xmlrpc.com/RPC2")
Fredrik Lundhb9056332001-07-11 17:42:21 +00001410
Guido van Rossumbe19ed72007-02-09 05:37:30 +00001411 print(server)
Fredrik Lundhb9056332001-07-11 17:42:21 +00001412
1413 try:
Guido van Rossumbe19ed72007-02-09 05:37:30 +00001414 print(server.currentTime.getCurrentTime())
Guido van Rossumb940e112007-01-10 16:19:56 +00001415 except Error as v:
Guido van Rossumbe19ed72007-02-09 05:37:30 +00001416 print("ERROR", v)
Martin v. Löwis12237b32004-08-22 16:04:50 +00001417
1418 multi = MultiCall(server)
1419 multi.currentTime.getCurrentTime()
1420 multi.currentTime.getCurrentTime()
1421 try:
1422 for response in multi():
Guido van Rossumbe19ed72007-02-09 05:37:30 +00001423 print(response)
Guido van Rossumb940e112007-01-10 16:19:56 +00001424 except Error as v:
Guido van Rossumbe19ed72007-02-09 05:37:30 +00001425 print("ERROR", v)