blob: e3895bd33ebac928cce046670e4ae6f3cebefb46 [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
Fred Drake2a2d9702001-10-17 01:51:04 +0000139import re, string, time, operator
Fredrik Lundh1538c232001-10-01 19:42:03 +0000140
Fredrik Lundhb9056332001-07-11 17:42:21 +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:
147 unicode
148except NameError:
149 unicode = None # unicode support not available
150
Skip Montanaro9a7c96a2003-01-22 18:17:25 +0000151try:
152 _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)
158 if unicode and encoding and is8bit(data):
159 data = unicode(data, encoding)
160 return data
161
Fredrik Lundh1538c232001-10-01 19:42:03 +0000162def escape(s, replace=string.replace):
163 s = replace(s, "&", "&")
164 s = replace(s, "<", "&lt;")
165 return replace(s, ">", "&gt;",)
166
Fredrik Lundhb9056332001-07-11 17:42:21 +0000167if unicode:
168 def _stringify(string):
169 # convert to 7-bit ascii if possible
170 try:
171 return str(string)
172 except UnicodeError:
173 return string
174else:
175 def _stringify(string):
176 return string
177
Fredrik Lundh3d9addd2002-06-27 21:36:21 +0000178__version__ = "1.0.1"
179
180# xmlrpc integer limits
181MAXINT = 2L**31-1
182MININT = -2L**31
183
184# --------------------------------------------------------------------
185# Error constants (from Dan Libby's specification at
186# http://xmlrpc-epi.sourceforge.net/specs/rfc.fault_codes.php)
187
188# Ranges of errors
189PARSE_ERROR = -32700
190SERVER_ERROR = -32600
191APPLICATION_ERROR = -32500
192SYSTEM_ERROR = -32400
193TRANSPORT_ERROR = -32300
194
195# Specific errors
196NOT_WELLFORMED_ERROR = -32700
197UNSUPPORTED_ENCODING = -32701
198INVALID_ENCODING_CHAR = -32702
199INVALID_XMLRPC = -32600
200METHOD_NOT_FOUND = -32601
201INVALID_METHOD_PARAMS = -32602
202INTERNAL_ERROR = -32603
Fredrik Lundhb9056332001-07-11 17:42:21 +0000203
204# --------------------------------------------------------------------
205# Exceptions
206
Fredrik Lundh3d9addd2002-06-27 21:36:21 +0000207##
208# Base class for all kinds of client-side errors.
209
Fredrik Lundh78eedce2001-08-23 20:04:33 +0000210class Error(Exception):
Fred Drake1b410792001-09-04 18:55:03 +0000211 """Base class for client errors."""
Fredrik Lundhb0e8e9b2001-09-10 21:45:42 +0000212 def __str__(self):
213 return repr(self)
Fredrik Lundhb9056332001-07-11 17:42:21 +0000214
Fredrik Lundh3d9addd2002-06-27 21:36:21 +0000215##
216# Indicates an HTTP-level protocol error. This is raised by the HTTP
217# transport layer, if the server returns an error code other than 200
218# (OK).
219#
220# @param url The target URL.
221# @param errcode The HTTP error code.
222# @param errmsg The HTTP error message.
223# @param headers The HTTP header dictionary.
224
Fredrik Lundhb9056332001-07-11 17:42:21 +0000225class ProtocolError(Error):
Fred Drake1b410792001-09-04 18:55:03 +0000226 """Indicates an HTTP protocol error."""
Fredrik Lundhb9056332001-07-11 17:42:21 +0000227 def __init__(self, url, errcode, errmsg, headers):
Fredrik Lundhb0e8e9b2001-09-10 21:45:42 +0000228 Error.__init__(self)
Fredrik Lundhb9056332001-07-11 17:42:21 +0000229 self.url = url
230 self.errcode = errcode
231 self.errmsg = errmsg
232 self.headers = headers
233 def __repr__(self):
234 return (
235 "<ProtocolError for %s: %s %s>" %
236 (self.url, self.errcode, self.errmsg)
237 )
238
Fredrik Lundh3d9addd2002-06-27 21:36:21 +0000239##
240# Indicates a broken XML-RPC response package. This exception is
241# raised by the unmarshalling layer, if the XML-RPC response is
242# malformed.
243
Fredrik Lundhb9056332001-07-11 17:42:21 +0000244class ResponseError(Error):
Fredrik Lundhc4c062f2001-09-10 19:45:02 +0000245 """Indicates a broken response package."""
Fredrik Lundhb9056332001-07-11 17:42:21 +0000246 pass
247
Fredrik Lundh3d9addd2002-06-27 21:36:21 +0000248##
249# Indicates an XML-RPC fault response package. This exception is
250# raised by the unmarshalling layer, if the XML-RPC response contains
251# a fault string. This exception can also used as a class, to
252# generate a fault XML-RPC message.
253#
254# @param faultCode The XML-RPC fault code.
255# @param faultString The XML-RPC fault string.
256
Fredrik Lundhb9056332001-07-11 17:42:21 +0000257class Fault(Error):
Fredrik Lundhc4c062f2001-09-10 19:45:02 +0000258 """Indicates an XML-RPC fault package."""
Fredrik Lundhb9056332001-07-11 17:42:21 +0000259 def __init__(self, faultCode, faultString, **extra):
Fredrik Lundhb0e8e9b2001-09-10 21:45:42 +0000260 Error.__init__(self)
Fredrik Lundhb9056332001-07-11 17:42:21 +0000261 self.faultCode = faultCode
262 self.faultString = faultString
263 def __repr__(self):
264 return (
265 "<Fault %s: %s>" %
Fredrik Lundhb0e8e9b2001-09-10 21:45:42 +0000266 (self.faultCode, repr(self.faultString))
Fredrik Lundhb9056332001-07-11 17:42:21 +0000267 )
268
269# --------------------------------------------------------------------
270# Special values
271
Fredrik Lundh3d9addd2002-06-27 21:36:21 +0000272##
273# Wrapper for XML-RPC boolean values. Use the xmlrpclib.True and
274# xmlrpclib.False constants, or the xmlrpclib.boolean() function, to
275# generate boolean XML-RPC values.
276#
277# @param value A boolean value. Any true value is interpreted as True,
278# all other values are interpreted as False.
279
Skip Montanaro9a7c96a2003-01-22 18:17:25 +0000280if _bool_is_builtin:
281 boolean = Boolean = bool
282 # to avoid breaking code which references xmlrpclib.{True,False}
283 True, False = True, False
284else:
285 class Boolean:
286 """Boolean-value wrapper.
Fred Drake1b410792001-09-04 18:55:03 +0000287
Skip Montanaro9a7c96a2003-01-22 18:17:25 +0000288 Use True or False to generate a "boolean" XML-RPC value.
289 """
Fredrik Lundhb9056332001-07-11 17:42:21 +0000290
Skip Montanaro9a7c96a2003-01-22 18:17:25 +0000291 def __init__(self, value = 0):
292 self.value = operator.truth(value)
Fredrik Lundhb9056332001-07-11 17:42:21 +0000293
Skip Montanaro9a7c96a2003-01-22 18:17:25 +0000294 def encode(self, out):
295 out.write("<value><boolean>%d</boolean></value>\n" % self.value)
Fredrik Lundhb9056332001-07-11 17:42:21 +0000296
Skip Montanaro9a7c96a2003-01-22 18:17:25 +0000297 def __cmp__(self, other):
298 if isinstance(other, Boolean):
299 other = other.value
300 return cmp(self.value, other)
Fredrik Lundhb9056332001-07-11 17:42:21 +0000301
Skip Montanaro9a7c96a2003-01-22 18:17:25 +0000302 def __repr__(self):
303 if self.value:
304 return "<Boolean True at %x>" % id(self)
305 else:
306 return "<Boolean False at %x>" % id(self)
Fredrik Lundhb9056332001-07-11 17:42:21 +0000307
Skip Montanaro9a7c96a2003-01-22 18:17:25 +0000308 def __int__(self):
309 return self.value
Fredrik Lundhb9056332001-07-11 17:42:21 +0000310
Skip Montanaro9a7c96a2003-01-22 18:17:25 +0000311 def __nonzero__(self):
312 return self.value
Fredrik Lundhb9056332001-07-11 17:42:21 +0000313
Skip Montanaro9a7c96a2003-01-22 18:17:25 +0000314 True, False = Boolean(1), Boolean(0)
Fredrik Lundhb9056332001-07-11 17:42:21 +0000315
Skip Montanaro9a7c96a2003-01-22 18:17:25 +0000316 ##
317 # Map true or false value to XML-RPC boolean values.
318 #
319 # @def boolean(value)
320 # @param value A boolean value. Any true value is mapped to True,
321 # all other values are mapped to False.
322 # @return xmlrpclib.True or xmlrpclib.False.
323 # @see Boolean
324 # @see True
325 # @see False
Fredrik Lundh3d9addd2002-06-27 21:36:21 +0000326
Skip Montanaro9a7c96a2003-01-22 18:17:25 +0000327 def boolean(value, _truefalse=(False, True)):
328 """Convert any Python value to XML-RPC 'boolean'."""
329 return _truefalse[operator.truth(value)]
Fredrik Lundh3d9addd2002-06-27 21:36:21 +0000330
331##
332# Wrapper for XML-RPC DateTime values. This converts a time value to
333# the format used by XML-RPC.
334# <p>
335# The value can be given as a string in the format
336# "yyyymmddThh:mm:ss", as a 9-item time tuple (as returned by
337# time.localtime()), or an integer value (as returned by time.time()).
338# The wrapper uses time.localtime() to convert an integer to a time
339# tuple.
340#
341# @param value The time, given as an ISO 8601 string, a time
342# tuple, or a integer time value.
Fredrik Lundhb9056332001-07-11 17:42:21 +0000343
Fredrik Lundhb9056332001-07-11 17:42:21 +0000344class DateTime:
Fredrik Lundhc4c062f2001-09-10 19:45:02 +0000345 """DateTime wrapper for an ISO 8601 string or time tuple or
346 localtime integer value to generate 'dateTime.iso8601' XML-RPC
347 value.
348 """
Fredrik Lundhb9056332001-07-11 17:42:21 +0000349
350 def __init__(self, value=0):
Fredrik Lundh78eedce2001-08-23 20:04:33 +0000351 if not isinstance(value, StringType):
Gustavo Niemeyerd5b80902003-06-16 02:49:42 +0000352 if not isinstance(value, (TupleType, time.struct_time)):
Fredrik Lundhb9056332001-07-11 17:42:21 +0000353 if value == 0:
354 value = time.time()
355 value = time.localtime(value)
356 value = time.strftime("%Y%m%dT%H:%M:%S", value)
357 self.value = value
358
359 def __cmp__(self, other):
360 if isinstance(other, DateTime):
361 other = other.value
362 return cmp(self.value, other)
363
Fredrik Lundh3d9addd2002-06-27 21:36:21 +0000364 ##
365 # Get date/time value.
366 #
367 # @return Date/time value, as an ISO 8601 string.
368
369 def __str__(self):
370 return self.value
371
Fredrik Lundhb9056332001-07-11 17:42:21 +0000372 def __repr__(self):
Fredrik Lundh3d9addd2002-06-27 21:36:21 +0000373 return "<DateTime %s at %x>" % (repr(self.value), id(self))
Fredrik Lundhb9056332001-07-11 17:42:21 +0000374
375 def decode(self, data):
376 self.value = string.strip(data)
377
378 def encode(self, out):
379 out.write("<value><dateTime.iso8601>")
380 out.write(self.value)
381 out.write("</dateTime.iso8601></value>\n")
382
Fredrik Lundh3d9addd2002-06-27 21:36:21 +0000383def _datetime(data):
384 # decode xml element contents into a DateTime structure.
Fredrik Lundhb9056332001-07-11 17:42:21 +0000385 value = DateTime()
386 value.decode(data)
387 return value
388
Fredrik Lundh3d9addd2002-06-27 21:36:21 +0000389##
390# Wrapper for binary data. This can be used to transport any kind
391# of binary data over XML-RPC, using BASE64 encoding.
392#
393# @param data An 8-bit string containing arbitrary data.
394
Skip Montanarobfcbfa72003-04-24 19:51:31 +0000395import base64
396try:
397 import cStringIO as StringIO
398except ImportError:
399 import StringIO
400
Fredrik Lundhb9056332001-07-11 17:42:21 +0000401class Binary:
Fred Drake1b410792001-09-04 18:55:03 +0000402 """Wrapper for binary data."""
Fredrik Lundhb9056332001-07-11 17:42:21 +0000403
404 def __init__(self, data=None):
405 self.data = data
406
Fredrik Lundh3d9addd2002-06-27 21:36:21 +0000407 ##
408 # Get buffer contents.
409 #
410 # @return Buffer contents, as an 8-bit string.
411
412 def __str__(self):
413 return self.data or ""
414
Fredrik Lundhb9056332001-07-11 17:42:21 +0000415 def __cmp__(self, other):
416 if isinstance(other, Binary):
417 other = other.data
418 return cmp(self.data, other)
419
420 def decode(self, data):
Fredrik Lundhb9056332001-07-11 17:42:21 +0000421 self.data = base64.decodestring(data)
422
423 def encode(self, out):
Fredrik Lundhb9056332001-07-11 17:42:21 +0000424 out.write("<value><base64>\n")
425 base64.encode(StringIO.StringIO(self.data), out)
426 out.write("</base64></value>\n")
427
Fredrik Lundh3d9addd2002-06-27 21:36:21 +0000428def _binary(data):
429 # decode xml element contents into a Binary structure
Fredrik Lundhb9056332001-07-11 17:42:21 +0000430 value = Binary()
431 value.decode(data)
432 return value
433
Skip Montanaro9a7c96a2003-01-22 18:17:25 +0000434WRAPPERS = (DateTime, Binary)
435if not _bool_is_builtin:
436 WRAPPERS = WRAPPERS + (Boolean,)
Fredrik Lundhb9056332001-07-11 17:42:21 +0000437
438# --------------------------------------------------------------------
439# XML parsers
440
441try:
442 # optional xmlrpclib accelerator. for more information on this
443 # component, contact info@pythonware.com
444 import _xmlrpclib
445 FastParser = _xmlrpclib.Parser
446 FastUnmarshaller = _xmlrpclib.Unmarshaller
447except (AttributeError, ImportError):
448 FastParser = FastUnmarshaller = None
449
Fredrik Lundh3d9addd2002-06-27 21:36:21 +0000450try:
451 import _xmlrpclib
452 FastMarshaller = _xmlrpclib.Marshaller
453except (AttributeError, ImportError):
454 FastMarshaller = None
455
Fredrik Lundhb9056332001-07-11 17:42:21 +0000456#
457# the SGMLOP parser is about 15x faster than Python's builtin
458# XML parser. SGMLOP sources can be downloaded from:
459#
460# http://www.pythonware.com/products/xml/sgmlop.htm
461#
462
463try:
464 import sgmlop
465 if not hasattr(sgmlop, "XMLParser"):
466 raise ImportError
467except ImportError:
468 SgmlopParser = None # sgmlop accelerator not available
469else:
470 class SgmlopParser:
471 def __init__(self, target):
472
473 # setup callbacks
474 self.finish_starttag = target.start
475 self.finish_endtag = target.end
476 self.handle_data = target.data
477 self.handle_xml = target.xml
478
479 # activate parser
480 self.parser = sgmlop.XMLParser()
481 self.parser.register(self)
482 self.feed = self.parser.feed
483 self.entity = {
484 "amp": "&", "gt": ">", "lt": "<",
485 "apos": "'", "quot": '"'
486 }
487
488 def close(self):
489 try:
490 self.parser.close()
491 finally:
492 self.parser = self.feed = None # nuke circular reference
493
494 def handle_proc(self, tag, attr):
495 m = re.search("encoding\s*=\s*['\"]([^\"']+)[\"']", attr)
496 if m:
497 self.handle_xml(m.group(1), 1)
498
499 def handle_entityref(self, entity):
500 # <string> entity
501 try:
502 self.handle_data(self.entity[entity])
503 except KeyError:
504 self.handle_data("&%s;" % entity)
505
506try:
507 from xml.parsers import expat
Guido van Rossumb8551342001-10-02 18:33:11 +0000508 if not hasattr(expat, "ParserCreate"):
Fredrik Lundh3d9addd2002-06-27 21:36:21 +0000509 raise ImportError
Fredrik Lundhb9056332001-07-11 17:42:21 +0000510except ImportError:
Fredrik Lundh3d9addd2002-06-27 21:36:21 +0000511 ExpatParser = None # expat not available
Fredrik Lundhb9056332001-07-11 17:42:21 +0000512else:
513 class ExpatParser:
Fredrik Lundh3d9addd2002-06-27 21:36:21 +0000514 # fast expat parser for Python 2.0 and later. this is about
515 # 50% slower than sgmlop, on roundtrip testing
Fredrik Lundhb9056332001-07-11 17:42:21 +0000516 def __init__(self, target):
517 self._parser = parser = expat.ParserCreate(None, None)
518 self._target = target
519 parser.StartElementHandler = target.start
520 parser.EndElementHandler = target.end
521 parser.CharacterDataHandler = target.data
522 encoding = None
523 if not parser.returns_unicode:
524 encoding = "utf-8"
525 target.xml(encoding, None)
526
527 def feed(self, data):
528 self._parser.Parse(data, 0)
529
530 def close(self):
531 self._parser.Parse("", 1) # end of data
532 del self._target, self._parser # get rid of circular references
533
Fredrik Lundhc4c062f2001-09-10 19:45:02 +0000534class SlowParser:
535 """Default XML parser (based on xmllib.XMLParser)."""
536 # this is about 10 times slower than sgmlop, on roundtrip
537 # testing.
Fredrik Lundhb9056332001-07-11 17:42:21 +0000538 def __init__(self, target):
Fredrik Lundhc4c062f2001-09-10 19:45:02 +0000539 import xmllib # lazy subclassing (!)
540 if xmllib.XMLParser not in SlowParser.__bases__:
541 SlowParser.__bases__ = (xmllib.XMLParser,)
Fredrik Lundhb9056332001-07-11 17:42:21 +0000542 self.handle_xml = target.xml
543 self.unknown_starttag = target.start
544 self.handle_data = target.data
Fredrik Lundh3d9addd2002-06-27 21:36:21 +0000545 self.handle_cdata = target.data
Fredrik Lundhb9056332001-07-11 17:42:21 +0000546 self.unknown_endtag = target.end
Fredrik Lundhb0e8e9b2001-09-10 21:45:42 +0000547 try:
548 xmllib.XMLParser.__init__(self, accept_utf8=1)
549 except TypeError:
550 xmllib.XMLParser.__init__(self) # pre-2.0
Fredrik Lundhb9056332001-07-11 17:42:21 +0000551
552# --------------------------------------------------------------------
553# XML-RPC marshalling and unmarshalling code
554
Fredrik Lundh3d9addd2002-06-27 21:36:21 +0000555##
556# XML-RPC marshaller.
557#
558# @param encoding Default encoding for 8-bit strings. The default
559# value is None (interpreted as UTF-8).
560# @see dumps
561
Fredrik Lundhb9056332001-07-11 17:42:21 +0000562class Marshaller:
Fred Drake1b410792001-09-04 18:55:03 +0000563 """Generate an XML-RPC params chunk from a Python data structure.
Fredrik Lundhb9056332001-07-11 17:42:21 +0000564
Fredrik Lundhc4c062f2001-09-10 19:45:02 +0000565 Create a Marshaller instance for each set of parameters, and use
566 the "dumps" method to convert your data (represented as a tuple)
567 to an XML-RPC params chunk. To write a fault response, pass a
568 Fault instance instead. You may prefer to use the "dumps" module
569 function for this purpose.
Fred Drake1b410792001-09-04 18:55:03 +0000570 """
Fredrik Lundhb9056332001-07-11 17:42:21 +0000571
572 # by the way, if you don't understand what's going on in here,
573 # that's perfectly ok.
574
Andrew M. Kuchlinga4c2b742003-04-25 00:26:51 +0000575 def __init__(self, encoding=None, allow_none=0):
Fredrik Lundhb9056332001-07-11 17:42:21 +0000576 self.memo = {}
577 self.data = None
578 self.encoding = encoding
Andrew M. Kuchlinga4c2b742003-04-25 00:26:51 +0000579 self.allow_none = allow_none
Tim Petersc2659cf2003-05-12 20:19:37 +0000580
Fredrik Lundhb9056332001-07-11 17:42:21 +0000581 dispatch = {}
582
583 def dumps(self, values):
Fredrik Lundh3d9addd2002-06-27 21:36:21 +0000584 out = []
585 write = out.append
586 dump = self.__dump
Fredrik Lundhb9056332001-07-11 17:42:21 +0000587 if isinstance(values, Fault):
588 # fault instance
589 write("<fault>\n")
Martin v. Löwis541342f2003-07-12 07:53:04 +0000590 dump({'faultCode': values.faultCode,
591 'faultString': values.faultString},
592 write)
Fredrik Lundhb9056332001-07-11 17:42:21 +0000593 write("</fault>\n")
594 else:
595 # parameter block
Fredrik Lundhc266bb02001-08-23 20:13:08 +0000596 # FIXME: the xml-rpc specification allows us to leave out
597 # the entire <params> block if there are no parameters.
598 # however, changing this may break older code (including
599 # old versions of xmlrpclib.py), so this is better left as
600 # is for now. See @XMLRPC3 for more information. /F
Fredrik Lundhb9056332001-07-11 17:42:21 +0000601 write("<params>\n")
602 for v in values:
603 write("<param>\n")
Fredrik Lundh3d9addd2002-06-27 21:36:21 +0000604 dump(v, write)
Fredrik Lundhb9056332001-07-11 17:42:21 +0000605 write("</param>\n")
606 write("</params>\n")
Fredrik Lundh3d9addd2002-06-27 21:36:21 +0000607 result = string.join(out, "")
Fredrik Lundhb9056332001-07-11 17:42:21 +0000608 return result
609
Fredrik Lundh3d9addd2002-06-27 21:36:21 +0000610 def __dump(self, value, write):
Fredrik Lundhb9056332001-07-11 17:42:21 +0000611 try:
612 f = self.dispatch[type(value)]
613 except KeyError:
614 raise TypeError, "cannot marshal %s objects" % type(value)
615 else:
Fredrik Lundh3d9addd2002-06-27 21:36:21 +0000616 f(self, value, write)
Fredrik Lundhb9056332001-07-11 17:42:21 +0000617
Andrew M. Kuchlinga4c2b742003-04-25 00:26:51 +0000618 def dump_nil (self, value, write):
619 if not self.allow_none:
620 raise TypeError, "cannot marshal None unless allow_none is enabled"
621 write("<value><nil/></value>")
622 dispatch[NoneType] = dump_nil
Tim Petersc2659cf2003-05-12 20:19:37 +0000623
Fredrik Lundh3d9addd2002-06-27 21:36:21 +0000624 def dump_int(self, value, write):
Skip Montanaro5449e082001-10-17 22:53:33 +0000625 # in case ints are > 32 bits
626 if value > MAXINT or value < MININT:
627 raise OverflowError, "int exceeds XML-RPC limits"
Fredrik Lundh3d9addd2002-06-27 21:36:21 +0000628 write("<value><int>")
629 write(str(value))
630 write("</int></value>\n")
Fredrik Lundhb9056332001-07-11 17:42:21 +0000631 dispatch[IntType] = dump_int
632
Skip Montanaro9a7c96a2003-01-22 18:17:25 +0000633 if _bool_is_builtin:
634 def dump_bool(self, value, write):
635 write("<value><boolean>")
636 write(value and "1" or "0")
637 write("</boolean></value>\n")
638 dispatch[bool] = dump_bool
639
Fredrik Lundh3d9addd2002-06-27 21:36:21 +0000640 def dump_long(self, value, write):
Skip Montanaro5449e082001-10-17 22:53:33 +0000641 if value > MAXINT or value < MININT:
642 raise OverflowError, "long int exceeds XML-RPC limits"
Fredrik Lundh3d9addd2002-06-27 21:36:21 +0000643 write("<value><int>")
644 write(str(int(value)))
645 write("</int></value>\n")
Skip Montanaro5e9c71b2001-10-10 15:56:34 +0000646 dispatch[LongType] = dump_long
647
Fredrik Lundh3d9addd2002-06-27 21:36:21 +0000648 def dump_double(self, value, write):
649 write("<value><double>")
650 write(repr(value))
651 write("</double></value>\n")
Fredrik Lundhb9056332001-07-11 17:42:21 +0000652 dispatch[FloatType] = dump_double
653
Fredrik Lundh3d9addd2002-06-27 21:36:21 +0000654 def dump_string(self, value, write, escape=escape):
655 write("<value><string>")
656 write(escape(value))
657 write("</string></value>\n")
Fredrik Lundhb9056332001-07-11 17:42:21 +0000658 dispatch[StringType] = dump_string
659
660 if unicode:
Fredrik Lundh3d9addd2002-06-27 21:36:21 +0000661 def dump_unicode(self, value, write, escape=escape):
Fredrik Lundhb9056332001-07-11 17:42:21 +0000662 value = value.encode(self.encoding)
Fredrik Lundh3d9addd2002-06-27 21:36:21 +0000663 write("<value><string>")
664 write(escape(value))
665 write("</string></value>\n")
Fredrik Lundhb9056332001-07-11 17:42:21 +0000666 dispatch[UnicodeType] = dump_unicode
667
Fredrik Lundh3d9addd2002-06-27 21:36:21 +0000668 def dump_array(self, value, write):
669 i = id(value)
670 if self.memo.has_key(i):
671 raise TypeError, "cannot marshal recursive sequences"
672 self.memo[i] = None
Fredrik Lundh1538c232001-10-01 19:42:03 +0000673 dump = self.__dump
Fredrik Lundhb9056332001-07-11 17:42:21 +0000674 write("<value><array><data>\n")
675 for v in value:
Fredrik Lundh3d9addd2002-06-27 21:36:21 +0000676 dump(v, write)
Fredrik Lundhb9056332001-07-11 17:42:21 +0000677 write("</data></array></value>\n")
Fredrik Lundh3d9addd2002-06-27 21:36:21 +0000678 del self.memo[i]
Fredrik Lundhb9056332001-07-11 17:42:21 +0000679 dispatch[TupleType] = dump_array
680 dispatch[ListType] = dump_array
681
Fredrik Lundh3d9addd2002-06-27 21:36:21 +0000682 def dump_struct(self, value, write, escape=escape):
683 i = id(value)
684 if self.memo.has_key(i):
685 raise TypeError, "cannot marshal recursive dictionaries"
686 self.memo[i] = None
Fredrik Lundh1538c232001-10-01 19:42:03 +0000687 dump = self.__dump
Fredrik Lundhb9056332001-07-11 17:42:21 +0000688 write("<value><struct>\n")
Andrew M. Kuchling5962f452004-06-05 12:35:58 +0000689 for k, v in value.items():
Fredrik Lundhb9056332001-07-11 17:42:21 +0000690 write("<member>\n")
691 if type(k) is not StringType:
Andrew M. Kuchling5962f452004-06-05 12:35:58 +0000692 if unicode and type(k) is UnicodeType:
693 k = k.encode(self.encoding)
694 else:
695 raise TypeError, "dictionary key must be string"
Fredrik Lundh1538c232001-10-01 19:42:03 +0000696 write("<name>%s</name>\n" % escape(k))
Andrew M. Kuchling5962f452004-06-05 12:35:58 +0000697 dump(v, write)
Fredrik Lundhb9056332001-07-11 17:42:21 +0000698 write("</member>\n")
699 write("</struct></value>\n")
Fredrik Lundh3d9addd2002-06-27 21:36:21 +0000700 del self.memo[i]
Fredrik Lundhb9056332001-07-11 17:42:21 +0000701 dispatch[DictType] = dump_struct
702
Fredrik Lundh3d9addd2002-06-27 21:36:21 +0000703 def dump_instance(self, value, write):
Fredrik Lundhb9056332001-07-11 17:42:21 +0000704 # check for special wrappers
705 if value.__class__ in WRAPPERS:
Fredrik Lundh3d9addd2002-06-27 21:36:21 +0000706 self.write = write
Fredrik Lundhb9056332001-07-11 17:42:21 +0000707 value.encode(self)
Fredrik Lundh3d9addd2002-06-27 21:36:21 +0000708 del self.write
Fredrik Lundhb9056332001-07-11 17:42:21 +0000709 else:
710 # store instance attributes as a struct (really?)
Fredrik Lundh3d9addd2002-06-27 21:36:21 +0000711 self.dump_struct(value.__dict__, write)
Fredrik Lundhb9056332001-07-11 17:42:21 +0000712 dispatch[InstanceType] = dump_instance
713
Fredrik Lundh3d9addd2002-06-27 21:36:21 +0000714##
715# XML-RPC unmarshaller.
716#
717# @see loads
718
Fredrik Lundhb9056332001-07-11 17:42:21 +0000719class Unmarshaller:
Fred Drake1b410792001-09-04 18:55:03 +0000720 """Unmarshal an XML-RPC response, based on incoming XML event
Fredrik Lundhb0e8e9b2001-09-10 21:45:42 +0000721 messages (start, data, end). Call close() to get the resulting
Fred Drake1b410792001-09-04 18:55:03 +0000722 data structure.
Fredrik Lundhb9056332001-07-11 17:42:21 +0000723
Fredrik Lundhb0e8e9b2001-09-10 21:45:42 +0000724 Note that this reader is fairly tolerant, and gladly accepts bogus
725 XML-RPC data without complaining (but not bogus XML).
Fred Drake1b410792001-09-04 18:55:03 +0000726 """
Fredrik Lundhb9056332001-07-11 17:42:21 +0000727
728 # and again, if you don't understand what's going on in here,
729 # that's perfectly ok.
730
731 def __init__(self):
732 self._type = None
733 self._stack = []
734 self._marks = []
735 self._data = []
736 self._methodname = None
737 self._encoding = "utf-8"
738 self.append = self._stack.append
739
740 def close(self):
741 # return response tuple and target method
742 if self._type is None or self._marks:
743 raise ResponseError()
744 if self._type == "fault":
Guido van Rossum68468eb2003-02-27 20:14:51 +0000745 raise Fault(**self._stack[0])
Fredrik Lundhb9056332001-07-11 17:42:21 +0000746 return tuple(self._stack)
747
748 def getmethodname(self):
749 return self._methodname
750
751 #
752 # event handlers
753
754 def xml(self, encoding, standalone):
755 self._encoding = encoding
756 # FIXME: assert standalone == 1 ???
757
758 def start(self, tag, attrs):
759 # prepare to handle this element
760 if tag == "array" or tag == "struct":
761 self._marks.append(len(self._stack))
762 self._data = []
763 self._value = (tag == "value")
764
765 def data(self, text):
766 self._data.append(text)
767
Fredrik Lundh1538c232001-10-01 19:42:03 +0000768 def end(self, tag, join=string.join):
Fredrik Lundhb9056332001-07-11 17:42:21 +0000769 # call the appropriate end tag handler
770 try:
771 f = self.dispatch[tag]
772 except KeyError:
773 pass # unknown tag ?
774 else:
Fredrik Lundh1538c232001-10-01 19:42:03 +0000775 return f(self, join(self._data, ""))
Fredrik Lundhb9056332001-07-11 17:42:21 +0000776
777 #
778 # accelerator support
779
780 def end_dispatch(self, tag, data):
781 # dispatch data
782 try:
783 f = self.dispatch[tag]
784 except KeyError:
785 pass # unknown tag ?
786 else:
787 return f(self, data)
788
789 #
790 # element decoders
791
792 dispatch = {}
793
Andrew M. Kuchlinga4c2b742003-04-25 00:26:51 +0000794 def end_nil (self, data):
795 self.append(None)
796 self._value = 0
797 dispatch["nil"] = end_nil
Tim Petersc2659cf2003-05-12 20:19:37 +0000798
Fredrik Lundh1538c232001-10-01 19:42:03 +0000799 def end_boolean(self, data):
Fredrik Lundhb9056332001-07-11 17:42:21 +0000800 if data == "0":
801 self.append(False)
802 elif data == "1":
803 self.append(True)
804 else:
805 raise TypeError, "bad boolean value"
806 self._value = 0
807 dispatch["boolean"] = end_boolean
808
Fredrik Lundh1538c232001-10-01 19:42:03 +0000809 def end_int(self, data):
810 self.append(int(data))
Fredrik Lundhb9056332001-07-11 17:42:21 +0000811 self._value = 0
812 dispatch["i4"] = end_int
813 dispatch["int"] = end_int
814
Fredrik Lundh1538c232001-10-01 19:42:03 +0000815 def end_double(self, data):
816 self.append(float(data))
Fredrik Lundhb9056332001-07-11 17:42:21 +0000817 self._value = 0
818 dispatch["double"] = end_double
819
Fredrik Lundh1538c232001-10-01 19:42:03 +0000820 def end_string(self, data):
Fredrik Lundhb9056332001-07-11 17:42:21 +0000821 if self._encoding:
822 data = _decode(data, self._encoding)
823 self.append(_stringify(data))
824 self._value = 0
825 dispatch["string"] = end_string
826 dispatch["name"] = end_string # struct keys are always strings
827
828 def end_array(self, data):
Raymond Hettinger46ac8eb2002-06-30 03:39:14 +0000829 mark = self._marks.pop()
Fredrik Lundhb9056332001-07-11 17:42:21 +0000830 # map arrays to Python lists
831 self._stack[mark:] = [self._stack[mark:]]
832 self._value = 0
833 dispatch["array"] = end_array
834
835 def end_struct(self, data):
Raymond Hettinger46ac8eb2002-06-30 03:39:14 +0000836 mark = self._marks.pop()
Fredrik Lundhb9056332001-07-11 17:42:21 +0000837 # map structs to Python dictionaries
838 dict = {}
839 items = self._stack[mark:]
840 for i in range(0, len(items), 2):
841 dict[_stringify(items[i])] = items[i+1]
842 self._stack[mark:] = [dict]
843 self._value = 0
844 dispatch["struct"] = end_struct
845
Fredrik Lundh1538c232001-10-01 19:42:03 +0000846 def end_base64(self, data):
Fredrik Lundhb9056332001-07-11 17:42:21 +0000847 value = Binary()
Fredrik Lundh1538c232001-10-01 19:42:03 +0000848 value.decode(data)
Fredrik Lundhb9056332001-07-11 17:42:21 +0000849 self.append(value)
850 self._value = 0
851 dispatch["base64"] = end_base64
852
Fredrik Lundh1538c232001-10-01 19:42:03 +0000853 def end_dateTime(self, data):
Fredrik Lundhb9056332001-07-11 17:42:21 +0000854 value = DateTime()
Fredrik Lundh1538c232001-10-01 19:42:03 +0000855 value.decode(data)
Fredrik Lundhb9056332001-07-11 17:42:21 +0000856 self.append(value)
857 dispatch["dateTime.iso8601"] = end_dateTime
858
859 def end_value(self, data):
Fredrik Lundh3d9addd2002-06-27 21:36:21 +0000860 # if we stumble upon a value element with no internal
Fredrik Lundhb9056332001-07-11 17:42:21 +0000861 # elements, treat it as a string element
862 if self._value:
863 self.end_string(data)
864 dispatch["value"] = end_value
865
866 def end_params(self, data):
867 self._type = "params"
868 dispatch["params"] = end_params
869
870 def end_fault(self, data):
871 self._type = "fault"
872 dispatch["fault"] = end_fault
873
Fredrik Lundh1538c232001-10-01 19:42:03 +0000874 def end_methodName(self, data):
Fredrik Lundhb9056332001-07-11 17:42:21 +0000875 if self._encoding:
876 data = _decode(data, self._encoding)
877 self._methodname = data
878 self._type = "methodName" # no params
879 dispatch["methodName"] = end_methodName
880
Martin v. Löwis45394c22003-10-31 13:49:36 +0000881## Multicall support
882#
Fredrik Lundhb9056332001-07-11 17:42:21 +0000883
Martin v. Löwis45394c22003-10-31 13:49:36 +0000884class _MultiCallMethod:
885 # some lesser magic to store calls made to a MultiCall object
886 # for batch execution
887 def __init__(self, call_list, name):
888 self.__call_list = call_list
889 self.__name = name
890 def __getattr__(self, name):
891 return _MultiCallMethod(self.__call_list, "%s.%s" % (self.__name, name))
892 def __call__(self, *args):
893 self.__call_list.append((self.__name, args))
894
Martin v. Löwis12237b32004-08-22 16:04:50 +0000895class MultiCallIterator:
Martin v. Löwis45394c22003-10-31 13:49:36 +0000896 """Iterates over the results of a multicall. Exceptions are
897 thrown in response to xmlrpc faults."""
Raymond Hettingercc523fc2003-11-02 09:47:05 +0000898
Martin v. Löwis12237b32004-08-22 16:04:50 +0000899 def __init__(self, results):
900 self.results = results
901
902 def __getitem__(self, i):
903 item = self.results[i]
904 if type(item) == type({}):
905 raise Fault(item['faultCode'], item['faultString'])
906 elif type(item) == type([]):
907 return item[0]
Martin v. Löwis45394c22003-10-31 13:49:36 +0000908 else:
909 raise ValueError,\
910 "unexpected type in multicall result"
Raymond Hettingercc523fc2003-11-02 09:47:05 +0000911
Martin v. Löwis45394c22003-10-31 13:49:36 +0000912class MultiCall:
913 """server -> a object used to boxcar method calls
914
915 server should be a ServerProxy object.
916
917 Methods can be added to the MultiCall using normal
918 method call syntax e.g.:
919
920 multicall = MultiCall(server_proxy)
921 multicall.add(2,3)
922 multicall.get_address("Guido")
923
924 To execute the multicall, call the MultiCall object e.g.:
925
926 add_result, address = multicall()
927 """
Raymond Hettingercc523fc2003-11-02 09:47:05 +0000928
Martin v. Löwis45394c22003-10-31 13:49:36 +0000929 def __init__(self, server):
930 self.__server = server
931 self.__call_list = []
Raymond Hettingercc523fc2003-11-02 09:47:05 +0000932
Martin v. Löwis45394c22003-10-31 13:49:36 +0000933 def __repr__(self):
934 return "<MultiCall at %x>" % id(self)
Raymond Hettingercc523fc2003-11-02 09:47:05 +0000935
Martin v. Löwis45394c22003-10-31 13:49:36 +0000936 __str__ = __repr__
937
938 def __getattr__(self, name):
939 return _MultiCallMethod(self.__call_list, name)
940
941 def __call__(self):
942 marshalled_list = []
943 for name, args in self.__call_list:
944 marshalled_list.append({'methodName' : name, 'params' : args})
945
946 return MultiCallIterator(self.__server.system.multicall(marshalled_list))
Raymond Hettingercc523fc2003-11-02 09:47:05 +0000947
Fredrik Lundhb9056332001-07-11 17:42:21 +0000948# --------------------------------------------------------------------
949# convenience functions
950
Fredrik Lundh3d9addd2002-06-27 21:36:21 +0000951##
952# Create a parser object, and connect it to an unmarshalling instance.
953# This function picks the fastest available XML parser.
954#
955# return A (parser, unmarshaller) tuple.
956
Fredrik Lundhb9056332001-07-11 17:42:21 +0000957def getparser():
958 """getparser() -> parser, unmarshaller
959
Fredrik Lundhb0e8e9b2001-09-10 21:45:42 +0000960 Create an instance of the fastest available parser, and attach it
961 to an unmarshalling object. Return both objects.
Fredrik Lundhb9056332001-07-11 17:42:21 +0000962 """
963 if FastParser and FastUnmarshaller:
Fredrik Lundh3d9addd2002-06-27 21:36:21 +0000964 target = FastUnmarshaller(True, False, _binary, _datetime, Fault)
Fredrik Lundhb9056332001-07-11 17:42:21 +0000965 parser = FastParser(target)
966 else:
967 target = Unmarshaller()
968 if FastParser:
969 parser = FastParser(target)
970 elif SgmlopParser:
971 parser = SgmlopParser(target)
972 elif ExpatParser:
973 parser = ExpatParser(target)
974 else:
975 parser = SlowParser(target)
976 return parser, target
977
Fredrik Lundh3d9addd2002-06-27 21:36:21 +0000978##
979# Convert a Python tuple or a Fault instance to an XML-RPC packet.
980#
981# @def dumps(params, **options)
982# @param params A tuple or Fault instance.
983# @keyparam methodname If given, create a methodCall request for
984# this method name.
985# @keyparam methodresponse If given, create a methodResponse packet.
986# If used with a tuple, the tuple must be a singleton (that is,
987# it must contain exactly one element).
988# @keyparam encoding The packet encoding.
989# @return A string containing marshalled data.
990
Andrew M. Kuchlinga4c2b742003-04-25 00:26:51 +0000991def dumps(params, methodname=None, methodresponse=None, encoding=None,
992 allow_none=0):
Fredrik Lundhb9056332001-07-11 17:42:21 +0000993 """data [,options] -> marshalled data
994
995 Convert an argument tuple or a Fault instance to an XML-RPC
996 request (or response, if the methodresponse option is used).
997
Fredrik Lundhb0e8e9b2001-09-10 21:45:42 +0000998 In addition to the data object, the following options can be given
999 as keyword arguments:
Fredrik Lundhb9056332001-07-11 17:42:21 +00001000
1001 methodname: the method name for a methodCall packet
1002
1003 methodresponse: true to create a methodResponse packet.
1004 If this option is used with a tuple, the tuple must be
1005 a singleton (i.e. it can contain only one element).
1006
1007 encoding: the packet encoding (default is UTF-8)
1008
1009 All 8-bit strings in the data structure are assumed to use the
1010 packet encoding. Unicode strings are automatically converted,
Fredrik Lundhb0e8e9b2001-09-10 21:45:42 +00001011 where necessary.
Fredrik Lundhb9056332001-07-11 17:42:21 +00001012 """
1013
1014 assert isinstance(params, TupleType) or isinstance(params, Fault),\
1015 "argument must be tuple or Fault instance"
1016
1017 if isinstance(params, Fault):
1018 methodresponse = 1
1019 elif methodresponse and isinstance(params, TupleType):
1020 assert len(params) == 1, "response tuple must be a singleton"
1021
Fredrik Lundh3d9addd2002-06-27 21:36:21 +00001022 if not encoding:
Fredrik Lundhb9056332001-07-11 17:42:21 +00001023 encoding = "utf-8"
1024
Fredrik Lundh3d9addd2002-06-27 21:36:21 +00001025 if FastMarshaller:
1026 m = FastMarshaller(encoding)
1027 else:
Andrew M. Kuchlinga4c2b742003-04-25 00:26:51 +00001028 m = Marshaller(encoding, allow_none)
Fredrik Lundh3d9addd2002-06-27 21:36:21 +00001029
Fredrik Lundhb9056332001-07-11 17:42:21 +00001030 data = m.dumps(params)
1031
1032 if encoding != "utf-8":
Fredrik Lundh3d9addd2002-06-27 21:36:21 +00001033 xmlheader = "<?xml version='1.0' encoding='%s'?>\n" % str(encoding)
Fredrik Lundhb9056332001-07-11 17:42:21 +00001034 else:
1035 xmlheader = "<?xml version='1.0'?>\n" # utf-8 is default
1036
1037 # standard XML-RPC wrappings
Fredrik Lundh3d9addd2002-06-27 21:36:21 +00001038 if methodname:
Fredrik Lundhb9056332001-07-11 17:42:21 +00001039 # a method call
1040 if not isinstance(methodname, StringType):
1041 methodname = methodname.encode(encoding)
1042 data = (
1043 xmlheader,
1044 "<methodCall>\n"
1045 "<methodName>", methodname, "</methodName>\n",
1046 data,
1047 "</methodCall>\n"
1048 )
1049 elif methodresponse:
1050 # a method response, or a fault structure
1051 data = (
1052 xmlheader,
1053 "<methodResponse>\n",
1054 data,
1055 "</methodResponse>\n"
1056 )
1057 else:
1058 return data # return as is
1059 return string.join(data, "")
1060
Fredrik Lundh3d9addd2002-06-27 21:36:21 +00001061##
1062# Convert an XML-RPC packet to a Python object. If the XML-RPC packet
1063# represents a fault condition, this function raises a Fault exception.
1064#
1065# @param data An XML-RPC packet, given as an 8-bit string.
Walter Dörwaldf0dfc7a2003-10-20 14:01:56 +00001066# @return A tuple containing the unpacked data, and the method name
Fredrik Lundh3d9addd2002-06-27 21:36:21 +00001067# (None if not present).
1068# @see Fault
1069
Fredrik Lundhb9056332001-07-11 17:42:21 +00001070def loads(data):
1071 """data -> unmarshalled data, method name
1072
1073 Convert an XML-RPC packet to unmarshalled data plus a method
1074 name (None if not present).
1075
1076 If the XML-RPC packet represents a fault condition, this function
1077 raises a Fault exception.
1078 """
1079 p, u = getparser()
1080 p.feed(data)
1081 p.close()
1082 return u.close(), u.getmethodname()
1083
1084
1085# --------------------------------------------------------------------
1086# request dispatcher
1087
1088class _Method:
1089 # some magic to bind an XML-RPC method to an RPC server.
1090 # supports "nested" methods (e.g. examples.getStateName)
1091 def __init__(self, send, name):
1092 self.__send = send
1093 self.__name = name
1094 def __getattr__(self, name):
1095 return _Method(self.__send, "%s.%s" % (self.__name, name))
1096 def __call__(self, *args):
1097 return self.__send(self.__name, args)
1098
Fredrik Lundh3d9addd2002-06-27 21:36:21 +00001099##
1100# Standard transport class for XML-RPC over HTTP.
1101# <p>
1102# You can create custom transports by subclassing this method, and
1103# overriding selected methods.
Fredrik Lundhb9056332001-07-11 17:42:21 +00001104
1105class Transport:
Fredrik Lundhc4c062f2001-09-10 19:45:02 +00001106 """Handles an HTTP transaction to an XML-RPC server."""
Fredrik Lundhb9056332001-07-11 17:42:21 +00001107
1108 # client identifier (may be overridden)
1109 user_agent = "xmlrpclib.py/%s (by www.pythonware.com)" % __version__
1110
Fredrik Lundh3d9addd2002-06-27 21:36:21 +00001111 ##
1112 # Send a complete request, and parse the response.
1113 #
1114 # @param host Target host.
1115 # @param handler Target PRC handler.
1116 # @param request_body XML-RPC request body.
1117 # @param verbose Debugging flag.
1118 # @return Parsed response.
1119
Fredrik Lundhb9056332001-07-11 17:42:21 +00001120 def request(self, host, handler, request_body, verbose=0):
1121 # issue XML-RPC request
1122
1123 h = self.make_connection(host)
1124 if verbose:
1125 h.set_debuglevel(1)
1126
1127 self.send_request(h, handler, request_body)
1128 self.send_host(h, host)
1129 self.send_user_agent(h)
1130 self.send_content(h, request_body)
1131
1132 errcode, errmsg, headers = h.getreply()
1133
1134 if errcode != 200:
1135 raise ProtocolError(
1136 host + handler,
1137 errcode, errmsg,
1138 headers
1139 )
1140
1141 self.verbose = verbose
1142
Fredrik Lundh3d9addd2002-06-27 21:36:21 +00001143 try:
1144 sock = h._conn.sock
1145 except AttributeError:
1146 sock = None
1147
1148 return self._parse_response(h.getfile(), sock)
1149
1150 ##
1151 # Create parser.
1152 #
1153 # @return A 2-tuple containing a parser and a unmarshaller.
Fredrik Lundhb9056332001-07-11 17:42:21 +00001154
Fredrik Lundhc4c062f2001-09-10 19:45:02 +00001155 def getparser(self):
1156 # get parser and unmarshaller
1157 return getparser()
1158
Fredrik Lundh3d9addd2002-06-27 21:36:21 +00001159 ##
Fredrik Lundh1303c7c2002-10-22 18:23:00 +00001160 # Get authorization info from host parameter
1161 # Host may be a string, or a (host, x509-dict) tuple; if a string,
1162 # it is checked for a "user:pw@host" format, and a "Basic
1163 # Authentication" header is added if appropriate.
1164 #
1165 # @param host Host descriptor (URL or (URL, x509 info) tuple).
1166 # @return A 3-tuple containing (actual host, extra headers,
1167 # x509 info). The header and x509 fields may be None.
1168
1169 def get_host_info(self, host):
1170
1171 x509 = {}
1172 if isinstance(host, TupleType):
1173 host, x509 = host
1174
1175 import urllib
1176 auth, host = urllib.splituser(host)
1177
1178 if auth:
1179 import base64
Fredrik Lundh768c98b2002-11-01 17:14:16 +00001180 auth = base64.encodestring(urllib.unquote(auth))
Fredrik Lundh1303c7c2002-10-22 18:23:00 +00001181 auth = string.join(string.split(auth), "") # get rid of whitespace
1182 extra_headers = [
1183 ("Authorization", "Basic " + auth)
1184 ]
1185 else:
1186 extra_headers = None
1187
1188 return host, extra_headers, x509
1189
1190 ##
Fredrik Lundh3d9addd2002-06-27 21:36:21 +00001191 # Connect to server.
1192 #
1193 # @param host Target host.
1194 # @return A connection handle.
1195
Fredrik Lundhb9056332001-07-11 17:42:21 +00001196 def make_connection(self, host):
1197 # create a HTTP connection object from a host descriptor
1198 import httplib
Fredrik Lundh1303c7c2002-10-22 18:23:00 +00001199 host, extra_headers, x509 = self.get_host_info(host)
Fredrik Lundhb9056332001-07-11 17:42:21 +00001200 return httplib.HTTP(host)
1201
Fredrik Lundh3d9addd2002-06-27 21:36:21 +00001202 ##
1203 # Send request header.
1204 #
1205 # @param connection Connection handle.
1206 # @param handler Target RPC handler.
1207 # @param request_body XML-RPC body.
1208
Fredrik Lundhb9056332001-07-11 17:42:21 +00001209 def send_request(self, connection, handler, request_body):
1210 connection.putrequest("POST", handler)
1211
Fredrik Lundh3d9addd2002-06-27 21:36:21 +00001212 ##
1213 # Send host name.
1214 #
1215 # @param connection Connection handle.
1216 # @param host Host name.
1217
Fredrik Lundhb9056332001-07-11 17:42:21 +00001218 def send_host(self, connection, host):
Fredrik Lundh1303c7c2002-10-22 18:23:00 +00001219 host, extra_headers, x509 = self.get_host_info(host)
Fredrik Lundhb9056332001-07-11 17:42:21 +00001220 connection.putheader("Host", host)
Fredrik Lundh1303c7c2002-10-22 18:23:00 +00001221 if extra_headers:
1222 if isinstance(extra_headers, DictType):
1223 extra_headers = extra_headers.items()
1224 for key, value in extra_headers:
1225 connection.putheader(key, value)
Fredrik Lundhb9056332001-07-11 17:42:21 +00001226
Fredrik Lundh3d9addd2002-06-27 21:36:21 +00001227 ##
1228 # Send user-agent identifier.
1229 #
1230 # @param connection Connection handle.
1231
Fredrik Lundhb9056332001-07-11 17:42:21 +00001232 def send_user_agent(self, connection):
1233 connection.putheader("User-Agent", self.user_agent)
1234
Fredrik Lundh3d9addd2002-06-27 21:36:21 +00001235 ##
1236 # Send request body.
1237 #
1238 # @param connection Connection handle.
1239 # @param request_body XML-RPC request body.
1240
Fredrik Lundhb9056332001-07-11 17:42:21 +00001241 def send_content(self, connection, request_body):
1242 connection.putheader("Content-Type", "text/xml")
1243 connection.putheader("Content-Length", str(len(request_body)))
1244 connection.endheaders()
1245 if request_body:
1246 connection.send(request_body)
1247
Fredrik Lundh3d9addd2002-06-27 21:36:21 +00001248 ##
1249 # Parse response.
1250 #
1251 # @param file Stream.
1252 # @return Response tuple and target method.
1253
1254 def parse_response(self, file):
1255 # compatibility interface
1256 return self._parse_response(file, None)
1257
1258 ##
1259 # Parse response (alternate interface). This is similar to the
1260 # parse_response method, but also provides direct access to the
1261 # underlying socket object (where available).
1262 #
1263 # @param file Stream.
1264 # @param sock Socket handle (or None, if the socket object
1265 # could not be accessed).
1266 # @return Response tuple and target method.
1267
1268 def _parse_response(self, file, sock):
1269 # read response from input file/socket, and parse it
Fredrik Lundhb9056332001-07-11 17:42:21 +00001270
Fredrik Lundhc4c062f2001-09-10 19:45:02 +00001271 p, u = self.getparser()
Fredrik Lundhb9056332001-07-11 17:42:21 +00001272
1273 while 1:
Fredrik Lundh3d9addd2002-06-27 21:36:21 +00001274 if sock:
1275 response = sock.recv(1024)
1276 else:
1277 response = file.read(1024)
Fredrik Lundhb9056332001-07-11 17:42:21 +00001278 if not response:
1279 break
1280 if self.verbose:
1281 print "body:", repr(response)
1282 p.feed(response)
1283
Fredrik Lundh3d9addd2002-06-27 21:36:21 +00001284 file.close()
Fredrik Lundhb9056332001-07-11 17:42:21 +00001285 p.close()
1286
1287 return u.close()
1288
Fredrik Lundh3d9addd2002-06-27 21:36:21 +00001289##
1290# Standard transport class for XML-RPC over HTTPS.
1291
Fredrik Lundhb9056332001-07-11 17:42:21 +00001292class SafeTransport(Transport):
Fredrik Lundhc4c062f2001-09-10 19:45:02 +00001293 """Handles an HTTPS transaction to an XML-RPC server."""
Fredrik Lundhb9056332001-07-11 17:42:21 +00001294
Fredrik Lundh3d9addd2002-06-27 21:36:21 +00001295 # FIXME: mostly untested
1296
Fredrik Lundhb9056332001-07-11 17:42:21 +00001297 def make_connection(self, host):
1298 # create a HTTPS connection object from a host descriptor
1299 # host may be a string, or a (host, x509-dict) tuple
1300 import httplib
Fredrik Lundh1303c7c2002-10-22 18:23:00 +00001301 host, extra_headers, x509 = self.get_host_info(host)
Fredrik Lundhb9056332001-07-11 17:42:21 +00001302 try:
1303 HTTPS = httplib.HTTPS
1304 except AttributeError:
Fredrik Lundh1303c7c2002-10-22 18:23:00 +00001305 raise NotImplementedError(
1306 "your version of httplib doesn't support HTTPS"
1307 )
Fredrik Lundhb9056332001-07-11 17:42:21 +00001308 else:
Guido van Rossum68468eb2003-02-27 20:14:51 +00001309 return HTTPS(host, None, **(x509 or {}))
Fredrik Lundhb9056332001-07-11 17:42:21 +00001310
Fredrik Lundh3d9addd2002-06-27 21:36:21 +00001311##
1312# Standard server proxy. This class establishes a virtual connection
1313# to an XML-RPC server.
1314# <p>
1315# This class is available as ServerProxy and Server. New code should
1316# use ServerProxy, to avoid confusion.
1317#
1318# @def ServerProxy(uri, **options)
1319# @param uri The connection point on the server.
1320# @keyparam transport A transport factory, compatible with the
1321# standard transport class.
1322# @keyparam encoding The default encoding used for 8-bit strings
1323# (default is UTF-8).
1324# @keyparam verbose Use a true value to enable debugging output.
1325# (printed to standard output).
1326# @see Transport
1327
Fredrik Lundhb9056332001-07-11 17:42:21 +00001328class ServerProxy:
1329 """uri [,options] -> a logical connection to an XML-RPC server
1330
1331 uri is the connection point on the server, given as
1332 scheme://host/target.
1333
1334 The standard implementation always supports the "http" scheme. If
1335 SSL socket support is available (Python 2.0), it also supports
1336 "https".
1337
1338 If the target part and the slash preceding it are both omitted,
1339 "/RPC2" is assumed.
1340
1341 The following options can be given as keyword arguments:
1342
1343 transport: a transport factory
1344 encoding: the request encoding (default is UTF-8)
1345
1346 All 8-bit strings passed to the server proxy are assumed to use
1347 the given encoding.
1348 """
1349
Andrew M. Kuchlinga4c2b742003-04-25 00:26:51 +00001350 def __init__(self, uri, transport=None, encoding=None, verbose=0,
1351 allow_none=0):
Fredrik Lundhb9056332001-07-11 17:42:21 +00001352 # establish a "logical" server connection
1353
1354 # get the url
Fredrik Lundhc4c062f2001-09-10 19:45:02 +00001355 import urllib
Fredrik Lundhb9056332001-07-11 17:42:21 +00001356 type, uri = urllib.splittype(uri)
1357 if type not in ("http", "https"):
1358 raise IOError, "unsupported XML-RPC protocol"
1359 self.__host, self.__handler = urllib.splithost(uri)
1360 if not self.__handler:
1361 self.__handler = "/RPC2"
1362
1363 if transport is None:
1364 if type == "https":
1365 transport = SafeTransport()
1366 else:
1367 transport = Transport()
1368 self.__transport = transport
1369
1370 self.__encoding = encoding
1371 self.__verbose = verbose
Andrew M. Kuchlinga4c2b742003-04-25 00:26:51 +00001372 self.__allow_none = allow_none
Tim Petersc2659cf2003-05-12 20:19:37 +00001373
Fredrik Lundhb9056332001-07-11 17:42:21 +00001374 def __request(self, methodname, params):
1375 # call a method on the remote server
1376
Andrew M. Kuchlinga4c2b742003-04-25 00:26:51 +00001377 request = dumps(params, methodname, encoding=self.__encoding,
1378 allow_none=self.__allow_none)
Fredrik Lundhb9056332001-07-11 17:42:21 +00001379
1380 response = self.__transport.request(
1381 self.__host,
1382 self.__handler,
1383 request,
1384 verbose=self.__verbose
1385 )
1386
1387 if len(response) == 1:
1388 response = response[0]
1389
1390 return response
1391
1392 def __repr__(self):
1393 return (
Fredrik Lundh78eedce2001-08-23 20:04:33 +00001394 "<ServerProxy for %s%s>" %
Fredrik Lundhb9056332001-07-11 17:42:21 +00001395 (self.__host, self.__handler)
1396 )
1397
1398 __str__ = __repr__
Raymond Hettingercc523fc2003-11-02 09:47:05 +00001399
Fredrik Lundhb9056332001-07-11 17:42:21 +00001400 def __getattr__(self, name):
1401 # magic method dispatcher
1402 return _Method(self.__request, name)
1403
1404 # note: to call a remote object with an non-standard name, use
1405 # result getattr(server, "strange-python-name")(args)
1406
Fredrik Lundh78eedce2001-08-23 20:04:33 +00001407# compatibility
Fredrik Lundh3d9addd2002-06-27 21:36:21 +00001408
Fredrik Lundhb9056332001-07-11 17:42:21 +00001409Server = ServerProxy
1410
1411# --------------------------------------------------------------------
1412# test code
1413
1414if __name__ == "__main__":
1415
1416 # simple test program (from the XML-RPC specification)
1417
Fredrik Lundh78eedce2001-08-23 20:04:33 +00001418 # server = ServerProxy("http://localhost:8000") # local server
Martin v. Löwis12237b32004-08-22 16:04:50 +00001419 server = ServerProxy("http://time.xmlrpc.com/RPC2")
Fredrik Lundhb9056332001-07-11 17:42:21 +00001420
1421 print server
1422
1423 try:
Martin v. Löwis12237b32004-08-22 16:04:50 +00001424 print server.currentTime.getCurrentTime()
1425 except Error, v:
1426 print "ERROR", v
1427
1428 multi = MultiCall(server)
1429 multi.currentTime.getCurrentTime()
1430 multi.currentTime.getCurrentTime()
1431 try:
1432 for response in multi():
1433 print response
Fredrik Lundhb9056332001-07-11 17:42:21 +00001434 except Error, v:
1435 print "ERROR", v