blob: d635a4e011386c8cdfeada27d247c86288acd3f0 [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
Raymond Hettingerfe59dc12005-02-07 15:28:45 +0000141from types import *
Fredrik Lundhb9056332001-07-11 17:42:21 +0000142
Fredrik Lundh3d9addd2002-06-27 21:36:21 +0000143# --------------------------------------------------------------------
144# Internal stuff
145
Fredrik Lundhb9056332001-07-11 17:42:21 +0000146try:
147 unicode
148except NameError:
149 unicode = None # unicode support not available
150
Skip Montanaro9a7c96a2003-01-22 18:17:25 +0000151try:
Fred Drakeba613c32005-02-10 18:33:30 +0000152 import datetime
153except ImportError:
154 datetime = None
155
156try:
Skip Montanaro9a7c96a2003-01-22 18:17:25 +0000157 _bool_is_builtin = False.__class__.__name__ == "bool"
158except NameError:
159 _bool_is_builtin = 0
160
Fredrik Lundhb9056332001-07-11 17:42:21 +0000161def _decode(data, encoding, is8bit=re.compile("[\x80-\xff]").search):
162 # decode non-ascii string (if possible)
163 if unicode and encoding and is8bit(data):
164 data = unicode(data, encoding)
165 return data
166
Fredrik Lundh1538c232001-10-01 19:42:03 +0000167def escape(s, replace=string.replace):
168 s = replace(s, "&", "&")
169 s = replace(s, "<", "&lt;")
170 return replace(s, ">", "&gt;",)
171
Fredrik Lundhb9056332001-07-11 17:42:21 +0000172if unicode:
173 def _stringify(string):
174 # convert to 7-bit ascii if possible
175 try:
Fred Drake22c07062005-02-11 17:59:08 +0000176 return string.encode("ascii")
Fredrik Lundhb9056332001-07-11 17:42:21 +0000177 except UnicodeError:
178 return string
179else:
180 def _stringify(string):
181 return string
182
Fredrik Lundh3d9addd2002-06-27 21:36:21 +0000183__version__ = "1.0.1"
184
185# xmlrpc integer limits
186MAXINT = 2L**31-1
187MININT = -2L**31
188
189# --------------------------------------------------------------------
190# Error constants (from Dan Libby's specification at
191# http://xmlrpc-epi.sourceforge.net/specs/rfc.fault_codes.php)
192
193# Ranges of errors
194PARSE_ERROR = -32700
195SERVER_ERROR = -32600
196APPLICATION_ERROR = -32500
197SYSTEM_ERROR = -32400
198TRANSPORT_ERROR = -32300
199
200# Specific errors
201NOT_WELLFORMED_ERROR = -32700
202UNSUPPORTED_ENCODING = -32701
203INVALID_ENCODING_CHAR = -32702
204INVALID_XMLRPC = -32600
205METHOD_NOT_FOUND = -32601
206INVALID_METHOD_PARAMS = -32602
207INTERNAL_ERROR = -32603
Fredrik Lundhb9056332001-07-11 17:42:21 +0000208
209# --------------------------------------------------------------------
210# Exceptions
211
Fredrik Lundh3d9addd2002-06-27 21:36:21 +0000212##
213# Base class for all kinds of client-side errors.
214
Fredrik Lundh78eedce2001-08-23 20:04:33 +0000215class Error(Exception):
Fred Drake1b410792001-09-04 18:55:03 +0000216 """Base class for client errors."""
Fredrik Lundhb0e8e9b2001-09-10 21:45:42 +0000217 def __str__(self):
218 return repr(self)
Fredrik Lundhb9056332001-07-11 17:42:21 +0000219
Fredrik Lundh3d9addd2002-06-27 21:36:21 +0000220##
221# Indicates an HTTP-level protocol error. This is raised by the HTTP
222# transport layer, if the server returns an error code other than 200
223# (OK).
224#
225# @param url The target URL.
226# @param errcode The HTTP error code.
227# @param errmsg The HTTP error message.
228# @param headers The HTTP header dictionary.
229
Fredrik Lundhb9056332001-07-11 17:42:21 +0000230class ProtocolError(Error):
Fred Drake1b410792001-09-04 18:55:03 +0000231 """Indicates an HTTP protocol error."""
Fredrik Lundhb9056332001-07-11 17:42:21 +0000232 def __init__(self, url, errcode, errmsg, headers):
Fredrik Lundhb0e8e9b2001-09-10 21:45:42 +0000233 Error.__init__(self)
Fredrik Lundhb9056332001-07-11 17:42:21 +0000234 self.url = url
235 self.errcode = errcode
236 self.errmsg = errmsg
237 self.headers = headers
238 def __repr__(self):
239 return (
240 "<ProtocolError for %s: %s %s>" %
241 (self.url, self.errcode, self.errmsg)
242 )
243
Fredrik Lundh3d9addd2002-06-27 21:36:21 +0000244##
245# Indicates a broken XML-RPC response package. This exception is
246# raised by the unmarshalling layer, if the XML-RPC response is
247# malformed.
248
Fredrik Lundhb9056332001-07-11 17:42:21 +0000249class ResponseError(Error):
Fredrik Lundhc4c062f2001-09-10 19:45:02 +0000250 """Indicates a broken response package."""
Fredrik Lundhb9056332001-07-11 17:42:21 +0000251 pass
252
Fredrik Lundh3d9addd2002-06-27 21:36:21 +0000253##
254# Indicates an XML-RPC fault response package. This exception is
255# raised by the unmarshalling layer, if the XML-RPC response contains
256# a fault string. This exception can also used as a class, to
257# generate a fault XML-RPC message.
258#
259# @param faultCode The XML-RPC fault code.
260# @param faultString The XML-RPC fault string.
261
Fredrik Lundhb9056332001-07-11 17:42:21 +0000262class Fault(Error):
Fredrik Lundhc4c062f2001-09-10 19:45:02 +0000263 """Indicates an XML-RPC fault package."""
Fredrik Lundhb9056332001-07-11 17:42:21 +0000264 def __init__(self, faultCode, faultString, **extra):
Fredrik Lundhb0e8e9b2001-09-10 21:45:42 +0000265 Error.__init__(self)
Fredrik Lundhb9056332001-07-11 17:42:21 +0000266 self.faultCode = faultCode
267 self.faultString = faultString
268 def __repr__(self):
269 return (
270 "<Fault %s: %s>" %
Fredrik Lundhb0e8e9b2001-09-10 21:45:42 +0000271 (self.faultCode, repr(self.faultString))
Fredrik Lundhb9056332001-07-11 17:42:21 +0000272 )
273
274# --------------------------------------------------------------------
275# Special values
276
Fredrik Lundh3d9addd2002-06-27 21:36:21 +0000277##
278# Wrapper for XML-RPC boolean values. Use the xmlrpclib.True and
279# xmlrpclib.False constants, or the xmlrpclib.boolean() function, to
280# generate boolean XML-RPC values.
281#
282# @param value A boolean value. Any true value is interpreted as True,
283# all other values are interpreted as False.
284
Skip Montanaro9a7c96a2003-01-22 18:17:25 +0000285if _bool_is_builtin:
286 boolean = Boolean = bool
287 # to avoid breaking code which references xmlrpclib.{True,False}
288 True, False = True, False
289else:
290 class Boolean:
291 """Boolean-value wrapper.
Fred Drake1b410792001-09-04 18:55:03 +0000292
Skip Montanaro9a7c96a2003-01-22 18:17:25 +0000293 Use True or False to generate a "boolean" XML-RPC value.
294 """
Fredrik Lundhb9056332001-07-11 17:42:21 +0000295
Skip Montanaro9a7c96a2003-01-22 18:17:25 +0000296 def __init__(self, value = 0):
297 self.value = operator.truth(value)
Fredrik Lundhb9056332001-07-11 17:42:21 +0000298
Skip Montanaro9a7c96a2003-01-22 18:17:25 +0000299 def encode(self, out):
300 out.write("<value><boolean>%d</boolean></value>\n" % self.value)
Fredrik Lundhb9056332001-07-11 17:42:21 +0000301
Skip Montanaro9a7c96a2003-01-22 18:17:25 +0000302 def __cmp__(self, other):
303 if isinstance(other, Boolean):
304 other = other.value
305 return cmp(self.value, other)
Fredrik Lundhb9056332001-07-11 17:42:21 +0000306
Skip Montanaro9a7c96a2003-01-22 18:17:25 +0000307 def __repr__(self):
308 if self.value:
309 return "<Boolean True at %x>" % id(self)
310 else:
311 return "<Boolean False at %x>" % id(self)
Fredrik Lundhb9056332001-07-11 17:42:21 +0000312
Skip Montanaro9a7c96a2003-01-22 18:17:25 +0000313 def __int__(self):
314 return self.value
Fredrik Lundhb9056332001-07-11 17:42:21 +0000315
Skip Montanaro9a7c96a2003-01-22 18:17:25 +0000316 def __nonzero__(self):
317 return self.value
Fredrik Lundhb9056332001-07-11 17:42:21 +0000318
Skip Montanaro9a7c96a2003-01-22 18:17:25 +0000319 True, False = Boolean(1), Boolean(0)
Fredrik Lundhb9056332001-07-11 17:42:21 +0000320
Skip Montanaro9a7c96a2003-01-22 18:17:25 +0000321 ##
322 # Map true or false value to XML-RPC boolean values.
323 #
324 # @def boolean(value)
325 # @param value A boolean value. Any true value is mapped to True,
326 # all other values are mapped to False.
327 # @return xmlrpclib.True or xmlrpclib.False.
328 # @see Boolean
329 # @see True
330 # @see False
Fredrik Lundh3d9addd2002-06-27 21:36:21 +0000331
Skip Montanaro9a7c96a2003-01-22 18:17:25 +0000332 def boolean(value, _truefalse=(False, True)):
333 """Convert any Python value to XML-RPC 'boolean'."""
334 return _truefalse[operator.truth(value)]
Fredrik Lundh3d9addd2002-06-27 21:36:21 +0000335
336##
337# Wrapper for XML-RPC DateTime values. This converts a time value to
338# the format used by XML-RPC.
339# <p>
340# The value can be given as a string in the format
341# "yyyymmddThh:mm:ss", as a 9-item time tuple (as returned by
342# time.localtime()), or an integer value (as returned by time.time()).
343# The wrapper uses time.localtime() to convert an integer to a time
344# tuple.
345#
346# @param value The time, given as an ISO 8601 string, a time
347# tuple, or a integer time value.
Fredrik Lundhb9056332001-07-11 17:42:21 +0000348
Skip Montanarob131f042008-04-18 20:35:46 +0000349def _strftime(value):
350 if datetime:
351 if isinstance(value, datetime.datetime):
352 return "%04d%02d%02dT%02d:%02d:%02d" % (
353 value.year, value.month, value.day,
354 value.hour, value.minute, value.second)
355
356 if not isinstance(value, (TupleType, time.struct_time)):
357 if value == 0:
358 value = time.time()
359 value = time.localtime(value)
360
361 return "%04d%02d%02dT%02d:%02d:%02d" % value[:6]
362
Fredrik Lundhb9056332001-07-11 17:42:21 +0000363class DateTime:
Fredrik Lundhc4c062f2001-09-10 19:45:02 +0000364 """DateTime wrapper for an ISO 8601 string or time tuple or
365 localtime integer value to generate 'dateTime.iso8601' XML-RPC
366 value.
367 """
Fredrik Lundhb9056332001-07-11 17:42:21 +0000368
369 def __init__(self, value=0):
Skip Montanarob131f042008-04-18 20:35:46 +0000370 if isinstance(value, StringType):
371 self.value = value
372 else:
373 self.value = _strftime(value)
Fredrik Lundhb9056332001-07-11 17:42:21 +0000374
Andrew M. Kuchling085f75a2008-02-23 16:23:05 +0000375 def make_comparable(self, other):
Fredrik Lundhb9056332001-07-11 17:42:21 +0000376 if isinstance(other, DateTime):
Andrew M. Kuchling085f75a2008-02-23 16:23:05 +0000377 s = self.value
378 o = other.value
379 elif datetime and isinstance(other, datetime.datetime):
380 s = self.value
381 o = other.strftime("%Y%m%dT%H:%M:%S")
382 elif isinstance(other, (str, unicode)):
383 s = self.value
384 o = other
385 elif hasattr(other, "timetuple"):
386 s = self.timetuple()
387 o = other.timetuple()
388 else:
389 otype = (hasattr(other, "__class__")
390 and other.__class__.__name__
391 or type(other))
392 raise TypeError("Can't compare %s and %s" %
393 (self.__class__.__name__, otype))
394 return s, o
395
396 def __lt__(self, other):
397 s, o = self.make_comparable(other)
398 return s < o
399
400 def __le__(self, other):
401 s, o = self.make_comparable(other)
402 return s <= o
403
404 def __gt__(self, other):
405 s, o = self.make_comparable(other)
406 return s > o
407
408 def __ge__(self, other):
409 s, o = self.make_comparable(other)
410 return s >= o
411
412 def __eq__(self, other):
413 s, o = self.make_comparable(other)
414 return s == o
415
416 def __ne__(self, other):
417 s, o = self.make_comparable(other)
418 return s != o
419
420 def timetuple(self):
421 return time.strptime(self.value, "%Y%m%dT%H:%M:%S")
422
423 def __cmp__(self, other):
424 s, o = self.make_comparable(other)
425 return cmp(s, o)
Fredrik Lundhb9056332001-07-11 17:42:21 +0000426
Fredrik Lundh3d9addd2002-06-27 21:36:21 +0000427 ##
428 # Get date/time value.
429 #
430 # @return Date/time value, as an ISO 8601 string.
431
432 def __str__(self):
433 return self.value
434
Fredrik Lundhb9056332001-07-11 17:42:21 +0000435 def __repr__(self):
Fredrik Lundh3d9addd2002-06-27 21:36:21 +0000436 return "<DateTime %s at %x>" % (repr(self.value), id(self))
Fredrik Lundhb9056332001-07-11 17:42:21 +0000437
438 def decode(self, data):
Andrew M. Kuchlingbdb39012005-12-04 19:11:17 +0000439 data = str(data)
Fredrik Lundhb9056332001-07-11 17:42:21 +0000440 self.value = string.strip(data)
441
442 def encode(self, out):
443 out.write("<value><dateTime.iso8601>")
444 out.write(self.value)
445 out.write("</dateTime.iso8601></value>\n")
446
Fredrik Lundh3d9addd2002-06-27 21:36:21 +0000447def _datetime(data):
448 # decode xml element contents into a DateTime structure.
Fredrik Lundhb9056332001-07-11 17:42:21 +0000449 value = DateTime()
450 value.decode(data)
451 return value
452
Skip Montanaro174dd222005-05-14 20:54:16 +0000453def _datetime_type(data):
454 t = time.strptime(data, "%Y%m%dT%H:%M:%S")
455 return datetime.datetime(*tuple(t)[:6])
456
Fredrik Lundh3d9addd2002-06-27 21:36:21 +0000457##
458# Wrapper for binary data. This can be used to transport any kind
459# of binary data over XML-RPC, using BASE64 encoding.
460#
461# @param data An 8-bit string containing arbitrary data.
462
Skip Montanarobfcbfa72003-04-24 19:51:31 +0000463import base64
464try:
465 import cStringIO as StringIO
466except ImportError:
467 import StringIO
468
Fredrik Lundhb9056332001-07-11 17:42:21 +0000469class Binary:
Fred Drake1b410792001-09-04 18:55:03 +0000470 """Wrapper for binary data."""
Fredrik Lundhb9056332001-07-11 17:42:21 +0000471
472 def __init__(self, data=None):
473 self.data = data
474
Fredrik Lundh3d9addd2002-06-27 21:36:21 +0000475 ##
476 # Get buffer contents.
477 #
478 # @return Buffer contents, as an 8-bit string.
479
480 def __str__(self):
481 return self.data or ""
482
Fredrik Lundhb9056332001-07-11 17:42:21 +0000483 def __cmp__(self, other):
484 if isinstance(other, Binary):
485 other = other.data
486 return cmp(self.data, other)
487
488 def decode(self, data):
Fredrik Lundhb9056332001-07-11 17:42:21 +0000489 self.data = base64.decodestring(data)
490
491 def encode(self, out):
Fredrik Lundhb9056332001-07-11 17:42:21 +0000492 out.write("<value><base64>\n")
493 base64.encode(StringIO.StringIO(self.data), out)
494 out.write("</base64></value>\n")
495
Fredrik Lundh3d9addd2002-06-27 21:36:21 +0000496def _binary(data):
497 # decode xml element contents into a Binary structure
Fredrik Lundhb9056332001-07-11 17:42:21 +0000498 value = Binary()
499 value.decode(data)
500 return value
501
Skip Montanaro9a7c96a2003-01-22 18:17:25 +0000502WRAPPERS = (DateTime, Binary)
503if not _bool_is_builtin:
504 WRAPPERS = WRAPPERS + (Boolean,)
Fredrik Lundhb9056332001-07-11 17:42:21 +0000505
506# --------------------------------------------------------------------
507# XML parsers
508
509try:
Fredrik Lundh7069c312004-10-13 06:48:37 +0000510 # optional xmlrpclib accelerator
Fredrik Lundhb9056332001-07-11 17:42:21 +0000511 import _xmlrpclib
512 FastParser = _xmlrpclib.Parser
513 FastUnmarshaller = _xmlrpclib.Unmarshaller
514except (AttributeError, ImportError):
515 FastParser = FastUnmarshaller = None
516
Fredrik Lundh3d9addd2002-06-27 21:36:21 +0000517try:
518 import _xmlrpclib
519 FastMarshaller = _xmlrpclib.Marshaller
520except (AttributeError, ImportError):
521 FastMarshaller = None
522
Fredrik Lundhb9056332001-07-11 17:42:21 +0000523#
524# the SGMLOP parser is about 15x faster than Python's builtin
525# XML parser. SGMLOP sources can be downloaded from:
526#
527# http://www.pythonware.com/products/xml/sgmlop.htm
528#
529
530try:
531 import sgmlop
532 if not hasattr(sgmlop, "XMLParser"):
533 raise ImportError
534except ImportError:
535 SgmlopParser = None # sgmlop accelerator not available
536else:
537 class SgmlopParser:
538 def __init__(self, target):
539
540 # setup callbacks
541 self.finish_starttag = target.start
542 self.finish_endtag = target.end
543 self.handle_data = target.data
544 self.handle_xml = target.xml
545
546 # activate parser
547 self.parser = sgmlop.XMLParser()
548 self.parser.register(self)
549 self.feed = self.parser.feed
550 self.entity = {
551 "amp": "&", "gt": ">", "lt": "<",
552 "apos": "'", "quot": '"'
553 }
554
555 def close(self):
556 try:
557 self.parser.close()
558 finally:
559 self.parser = self.feed = None # nuke circular reference
560
561 def handle_proc(self, tag, attr):
562 m = re.search("encoding\s*=\s*['\"]([^\"']+)[\"']", attr)
563 if m:
564 self.handle_xml(m.group(1), 1)
565
566 def handle_entityref(self, entity):
567 # <string> entity
568 try:
569 self.handle_data(self.entity[entity])
570 except KeyError:
571 self.handle_data("&%s;" % entity)
572
573try:
574 from xml.parsers import expat
Guido van Rossumb8551342001-10-02 18:33:11 +0000575 if not hasattr(expat, "ParserCreate"):
Fredrik Lundh3d9addd2002-06-27 21:36:21 +0000576 raise ImportError
Fredrik Lundhb9056332001-07-11 17:42:21 +0000577except ImportError:
Fredrik Lundh3d9addd2002-06-27 21:36:21 +0000578 ExpatParser = None # expat not available
Fredrik Lundhb9056332001-07-11 17:42:21 +0000579else:
580 class ExpatParser:
Fredrik Lundh3d9addd2002-06-27 21:36:21 +0000581 # fast expat parser for Python 2.0 and later. this is about
582 # 50% slower than sgmlop, on roundtrip testing
Fredrik Lundhb9056332001-07-11 17:42:21 +0000583 def __init__(self, target):
584 self._parser = parser = expat.ParserCreate(None, None)
585 self._target = target
586 parser.StartElementHandler = target.start
587 parser.EndElementHandler = target.end
588 parser.CharacterDataHandler = target.data
589 encoding = None
590 if not parser.returns_unicode:
591 encoding = "utf-8"
592 target.xml(encoding, None)
593
594 def feed(self, data):
595 self._parser.Parse(data, 0)
596
597 def close(self):
598 self._parser.Parse("", 1) # end of data
599 del self._target, self._parser # get rid of circular references
600
Fredrik Lundhc4c062f2001-09-10 19:45:02 +0000601class SlowParser:
602 """Default XML parser (based on xmllib.XMLParser)."""
603 # this is about 10 times slower than sgmlop, on roundtrip
604 # testing.
Fredrik Lundhb9056332001-07-11 17:42:21 +0000605 def __init__(self, target):
Fredrik Lundhc4c062f2001-09-10 19:45:02 +0000606 import xmllib # lazy subclassing (!)
607 if xmllib.XMLParser not in SlowParser.__bases__:
608 SlowParser.__bases__ = (xmllib.XMLParser,)
Fredrik Lundhb9056332001-07-11 17:42:21 +0000609 self.handle_xml = target.xml
610 self.unknown_starttag = target.start
611 self.handle_data = target.data
Fredrik Lundh3d9addd2002-06-27 21:36:21 +0000612 self.handle_cdata = target.data
Fredrik Lundhb9056332001-07-11 17:42:21 +0000613 self.unknown_endtag = target.end
Fredrik Lundhb0e8e9b2001-09-10 21:45:42 +0000614 try:
615 xmllib.XMLParser.__init__(self, accept_utf8=1)
616 except TypeError:
617 xmllib.XMLParser.__init__(self) # pre-2.0
Fredrik Lundhb9056332001-07-11 17:42:21 +0000618
619# --------------------------------------------------------------------
620# XML-RPC marshalling and unmarshalling code
621
Fredrik Lundh3d9addd2002-06-27 21:36:21 +0000622##
623# XML-RPC marshaller.
624#
625# @param encoding Default encoding for 8-bit strings. The default
626# value is None (interpreted as UTF-8).
627# @see dumps
628
Fredrik Lundhb9056332001-07-11 17:42:21 +0000629class Marshaller:
Fred Drake1b410792001-09-04 18:55:03 +0000630 """Generate an XML-RPC params chunk from a Python data structure.
Fredrik Lundhb9056332001-07-11 17:42:21 +0000631
Fredrik Lundhc4c062f2001-09-10 19:45:02 +0000632 Create a Marshaller instance for each set of parameters, and use
633 the "dumps" method to convert your data (represented as a tuple)
634 to an XML-RPC params chunk. To write a fault response, pass a
635 Fault instance instead. You may prefer to use the "dumps" module
636 function for this purpose.
Fred Drake1b410792001-09-04 18:55:03 +0000637 """
Fredrik Lundhb9056332001-07-11 17:42:21 +0000638
639 # by the way, if you don't understand what's going on in here,
640 # that's perfectly ok.
641
Andrew M. Kuchlinga4c2b742003-04-25 00:26:51 +0000642 def __init__(self, encoding=None, allow_none=0):
Fredrik Lundhb9056332001-07-11 17:42:21 +0000643 self.memo = {}
644 self.data = None
645 self.encoding = encoding
Andrew M. Kuchlinga4c2b742003-04-25 00:26:51 +0000646 self.allow_none = allow_none
Tim Petersc2659cf2003-05-12 20:19:37 +0000647
Fredrik Lundhb9056332001-07-11 17:42:21 +0000648 dispatch = {}
649
650 def dumps(self, values):
Fredrik Lundh3d9addd2002-06-27 21:36:21 +0000651 out = []
652 write = out.append
653 dump = self.__dump
Fredrik Lundhb9056332001-07-11 17:42:21 +0000654 if isinstance(values, Fault):
655 # fault instance
656 write("<fault>\n")
Martin v. Löwis541342f2003-07-12 07:53:04 +0000657 dump({'faultCode': values.faultCode,
658 'faultString': values.faultString},
659 write)
Fredrik Lundhb9056332001-07-11 17:42:21 +0000660 write("</fault>\n")
661 else:
662 # parameter block
Fredrik Lundhc266bb02001-08-23 20:13:08 +0000663 # FIXME: the xml-rpc specification allows us to leave out
664 # the entire <params> block if there are no parameters.
665 # however, changing this may break older code (including
666 # old versions of xmlrpclib.py), so this is better left as
667 # is for now. See @XMLRPC3 for more information. /F
Fredrik Lundhb9056332001-07-11 17:42:21 +0000668 write("<params>\n")
669 for v in values:
670 write("<param>\n")
Fredrik Lundh3d9addd2002-06-27 21:36:21 +0000671 dump(v, write)
Fredrik Lundhb9056332001-07-11 17:42:21 +0000672 write("</param>\n")
673 write("</params>\n")
Fredrik Lundh3d9addd2002-06-27 21:36:21 +0000674 result = string.join(out, "")
Fredrik Lundhb9056332001-07-11 17:42:21 +0000675 return result
676
Fredrik Lundh3d9addd2002-06-27 21:36:21 +0000677 def __dump(self, value, write):
Fredrik Lundhb9056332001-07-11 17:42:21 +0000678 try:
679 f = self.dispatch[type(value)]
680 except KeyError:
Martin v. Löwis07529352006-11-19 18:51:54 +0000681 # check if this object can be marshalled as a structure
682 try:
683 value.__dict__
684 except:
685 raise TypeError, "cannot marshal %s objects" % type(value)
686 # check if this class is a sub-class of a basic type,
687 # because we don't know how to marshal these types
688 # (e.g. a string sub-class)
689 for type_ in type(value).__mro__:
690 if type_ in self.dispatch.keys():
691 raise TypeError, "cannot marshal %s objects" % type(value)
692 f = self.dispatch[InstanceType]
693 f(self, value, write)
Fredrik Lundhb9056332001-07-11 17:42:21 +0000694
Andrew M. Kuchlinga4c2b742003-04-25 00:26:51 +0000695 def dump_nil (self, value, write):
696 if not self.allow_none:
697 raise TypeError, "cannot marshal None unless allow_none is enabled"
698 write("<value><nil/></value>")
Raymond Hettingerfe59dc12005-02-07 15:28:45 +0000699 dispatch[NoneType] = dump_nil
Tim Petersc2659cf2003-05-12 20:19:37 +0000700
Fredrik Lundh3d9addd2002-06-27 21:36:21 +0000701 def dump_int(self, value, write):
Skip Montanaro5449e082001-10-17 22:53:33 +0000702 # in case ints are > 32 bits
703 if value > MAXINT or value < MININT:
704 raise OverflowError, "int exceeds XML-RPC limits"
Fredrik Lundh3d9addd2002-06-27 21:36:21 +0000705 write("<value><int>")
706 write(str(value))
707 write("</int></value>\n")
Raymond Hettingerfe59dc12005-02-07 15:28:45 +0000708 dispatch[IntType] = dump_int
Fredrik Lundhb9056332001-07-11 17:42:21 +0000709
Skip Montanaro9a7c96a2003-01-22 18:17:25 +0000710 if _bool_is_builtin:
711 def dump_bool(self, value, write):
712 write("<value><boolean>")
713 write(value and "1" or "0")
714 write("</boolean></value>\n")
715 dispatch[bool] = dump_bool
716
Fredrik Lundh3d9addd2002-06-27 21:36:21 +0000717 def dump_long(self, value, write):
Skip Montanaro5449e082001-10-17 22:53:33 +0000718 if value > MAXINT or value < MININT:
719 raise OverflowError, "long int exceeds XML-RPC limits"
Fredrik Lundh3d9addd2002-06-27 21:36:21 +0000720 write("<value><int>")
721 write(str(int(value)))
722 write("</int></value>\n")
Raymond Hettingerfe59dc12005-02-07 15:28:45 +0000723 dispatch[LongType] = dump_long
Skip Montanaro5e9c71b2001-10-10 15:56:34 +0000724
Fredrik Lundh3d9addd2002-06-27 21:36:21 +0000725 def dump_double(self, value, write):
726 write("<value><double>")
727 write(repr(value))
728 write("</double></value>\n")
Raymond Hettingerfe59dc12005-02-07 15:28:45 +0000729 dispatch[FloatType] = dump_double
Fredrik Lundhb9056332001-07-11 17:42:21 +0000730
Fredrik Lundh3d9addd2002-06-27 21:36:21 +0000731 def dump_string(self, value, write, escape=escape):
732 write("<value><string>")
733 write(escape(value))
734 write("</string></value>\n")
Raymond Hettingerfe59dc12005-02-07 15:28:45 +0000735 dispatch[StringType] = dump_string
Fredrik Lundhb9056332001-07-11 17:42:21 +0000736
737 if unicode:
Fredrik Lundh3d9addd2002-06-27 21:36:21 +0000738 def dump_unicode(self, value, write, escape=escape):
Fredrik Lundhb9056332001-07-11 17:42:21 +0000739 value = value.encode(self.encoding)
Fredrik Lundh3d9addd2002-06-27 21:36:21 +0000740 write("<value><string>")
741 write(escape(value))
742 write("</string></value>\n")
Raymond Hettingerfe59dc12005-02-07 15:28:45 +0000743 dispatch[UnicodeType] = dump_unicode
Fredrik Lundhb9056332001-07-11 17:42:21 +0000744
Fredrik Lundh3d9addd2002-06-27 21:36:21 +0000745 def dump_array(self, value, write):
746 i = id(value)
747 if self.memo.has_key(i):
748 raise TypeError, "cannot marshal recursive sequences"
749 self.memo[i] = None
Fredrik Lundh1538c232001-10-01 19:42:03 +0000750 dump = self.__dump
Fredrik Lundhb9056332001-07-11 17:42:21 +0000751 write("<value><array><data>\n")
752 for v in value:
Fredrik Lundh3d9addd2002-06-27 21:36:21 +0000753 dump(v, write)
Fredrik Lundhb9056332001-07-11 17:42:21 +0000754 write("</data></array></value>\n")
Fredrik Lundh3d9addd2002-06-27 21:36:21 +0000755 del self.memo[i]
Raymond Hettingerfe59dc12005-02-07 15:28:45 +0000756 dispatch[TupleType] = dump_array
757 dispatch[ListType] = dump_array
Fredrik Lundhb9056332001-07-11 17:42:21 +0000758
Fredrik Lundh3d9addd2002-06-27 21:36:21 +0000759 def dump_struct(self, value, write, escape=escape):
760 i = id(value)
761 if self.memo.has_key(i):
762 raise TypeError, "cannot marshal recursive dictionaries"
763 self.memo[i] = None
Fredrik Lundh1538c232001-10-01 19:42:03 +0000764 dump = self.__dump
Fredrik Lundhb9056332001-07-11 17:42:21 +0000765 write("<value><struct>\n")
Andrew M. Kuchling5962f452004-06-05 12:35:58 +0000766 for k, v in value.items():
Fredrik Lundhb9056332001-07-11 17:42:21 +0000767 write("<member>\n")
Raymond Hettingerfe59dc12005-02-07 15:28:45 +0000768 if type(k) is not StringType:
769 if unicode and type(k) is UnicodeType:
Andrew M. Kuchling5962f452004-06-05 12:35:58 +0000770 k = k.encode(self.encoding)
771 else:
772 raise TypeError, "dictionary key must be string"
Fredrik Lundh1538c232001-10-01 19:42:03 +0000773 write("<name>%s</name>\n" % escape(k))
Andrew M. Kuchling5962f452004-06-05 12:35:58 +0000774 dump(v, write)
Fredrik Lundhb9056332001-07-11 17:42:21 +0000775 write("</member>\n")
776 write("</struct></value>\n")
Fredrik Lundh3d9addd2002-06-27 21:36:21 +0000777 del self.memo[i]
Raymond Hettingerfe59dc12005-02-07 15:28:45 +0000778 dispatch[DictType] = dump_struct
Fredrik Lundhb9056332001-07-11 17:42:21 +0000779
Fred Drakeba613c32005-02-10 18:33:30 +0000780 if datetime:
781 def dump_datetime(self, value, write):
782 write("<value><dateTime.iso8601>")
Skip Montanarob131f042008-04-18 20:35:46 +0000783 write(_strftime(value))
Fred Drakeba613c32005-02-10 18:33:30 +0000784 write("</dateTime.iso8601></value>\n")
785 dispatch[datetime.datetime] = dump_datetime
786
Fredrik Lundh3d9addd2002-06-27 21:36:21 +0000787 def dump_instance(self, value, write):
Fredrik Lundhb9056332001-07-11 17:42:21 +0000788 # check for special wrappers
789 if value.__class__ in WRAPPERS:
Fredrik Lundh3d9addd2002-06-27 21:36:21 +0000790 self.write = write
Fredrik Lundhb9056332001-07-11 17:42:21 +0000791 value.encode(self)
Fredrik Lundh3d9addd2002-06-27 21:36:21 +0000792 del self.write
Fredrik Lundhb9056332001-07-11 17:42:21 +0000793 else:
794 # store instance attributes as a struct (really?)
Fredrik Lundh3d9addd2002-06-27 21:36:21 +0000795 self.dump_struct(value.__dict__, write)
Fredrik Lundhb9056332001-07-11 17:42:21 +0000796 dispatch[InstanceType] = dump_instance
797
Fredrik Lundh3d9addd2002-06-27 21:36:21 +0000798##
799# XML-RPC unmarshaller.
800#
801# @see loads
802
Fredrik Lundhb9056332001-07-11 17:42:21 +0000803class Unmarshaller:
Fred Drake1b410792001-09-04 18:55:03 +0000804 """Unmarshal an XML-RPC response, based on incoming XML event
Fredrik Lundhb0e8e9b2001-09-10 21:45:42 +0000805 messages (start, data, end). Call close() to get the resulting
Fred Drake1b410792001-09-04 18:55:03 +0000806 data structure.
Fredrik Lundhb9056332001-07-11 17:42:21 +0000807
Fredrik Lundhb0e8e9b2001-09-10 21:45:42 +0000808 Note that this reader is fairly tolerant, and gladly accepts bogus
809 XML-RPC data without complaining (but not bogus XML).
Fred Drake1b410792001-09-04 18:55:03 +0000810 """
Fredrik Lundhb9056332001-07-11 17:42:21 +0000811
812 # and again, if you don't understand what's going on in here,
813 # that's perfectly ok.
814
Skip Montanaro174dd222005-05-14 20:54:16 +0000815 def __init__(self, use_datetime=0):
Fredrik Lundhb9056332001-07-11 17:42:21 +0000816 self._type = None
817 self._stack = []
818 self._marks = []
819 self._data = []
820 self._methodname = None
821 self._encoding = "utf-8"
822 self.append = self._stack.append
Skip Montanaro174dd222005-05-14 20:54:16 +0000823 self._use_datetime = use_datetime
824 if use_datetime and not datetime:
825 raise ValueError, "the datetime module is not available"
Fredrik Lundhb9056332001-07-11 17:42:21 +0000826
827 def close(self):
828 # return response tuple and target method
829 if self._type is None or self._marks:
830 raise ResponseError()
831 if self._type == "fault":
Guido van Rossum68468eb2003-02-27 20:14:51 +0000832 raise Fault(**self._stack[0])
Fredrik Lundhb9056332001-07-11 17:42:21 +0000833 return tuple(self._stack)
834
835 def getmethodname(self):
836 return self._methodname
837
838 #
839 # event handlers
840
841 def xml(self, encoding, standalone):
842 self._encoding = encoding
843 # FIXME: assert standalone == 1 ???
844
845 def start(self, tag, attrs):
846 # prepare to handle this element
847 if tag == "array" or tag == "struct":
848 self._marks.append(len(self._stack))
849 self._data = []
850 self._value = (tag == "value")
851
852 def data(self, text):
853 self._data.append(text)
854
Fredrik Lundh1538c232001-10-01 19:42:03 +0000855 def end(self, tag, join=string.join):
Fredrik Lundhb9056332001-07-11 17:42:21 +0000856 # call the appropriate end tag handler
857 try:
858 f = self.dispatch[tag]
859 except KeyError:
860 pass # unknown tag ?
861 else:
Fredrik Lundh1538c232001-10-01 19:42:03 +0000862 return f(self, join(self._data, ""))
Fredrik Lundhb9056332001-07-11 17:42:21 +0000863
864 #
865 # accelerator support
866
867 def end_dispatch(self, tag, data):
868 # dispatch data
869 try:
870 f = self.dispatch[tag]
871 except KeyError:
872 pass # unknown tag ?
873 else:
874 return f(self, data)
875
876 #
877 # element decoders
878
879 dispatch = {}
880
Andrew M. Kuchlinga4c2b742003-04-25 00:26:51 +0000881 def end_nil (self, data):
882 self.append(None)
883 self._value = 0
884 dispatch["nil"] = end_nil
Tim Petersc2659cf2003-05-12 20:19:37 +0000885
Fredrik Lundh1538c232001-10-01 19:42:03 +0000886 def end_boolean(self, data):
Fredrik Lundhb9056332001-07-11 17:42:21 +0000887 if data == "0":
888 self.append(False)
889 elif data == "1":
890 self.append(True)
891 else:
892 raise TypeError, "bad boolean value"
893 self._value = 0
894 dispatch["boolean"] = end_boolean
895
Fredrik Lundh1538c232001-10-01 19:42:03 +0000896 def end_int(self, data):
897 self.append(int(data))
Fredrik Lundhb9056332001-07-11 17:42:21 +0000898 self._value = 0
899 dispatch["i4"] = end_int
900 dispatch["int"] = end_int
901
Fredrik Lundh1538c232001-10-01 19:42:03 +0000902 def end_double(self, data):
903 self.append(float(data))
Fredrik Lundhb9056332001-07-11 17:42:21 +0000904 self._value = 0
905 dispatch["double"] = end_double
906
Fredrik Lundh1538c232001-10-01 19:42:03 +0000907 def end_string(self, data):
Fredrik Lundhb9056332001-07-11 17:42:21 +0000908 if self._encoding:
909 data = _decode(data, self._encoding)
910 self.append(_stringify(data))
911 self._value = 0
912 dispatch["string"] = end_string
913 dispatch["name"] = end_string # struct keys are always strings
914
915 def end_array(self, data):
Raymond Hettinger46ac8eb2002-06-30 03:39:14 +0000916 mark = self._marks.pop()
Fredrik Lundhb9056332001-07-11 17:42:21 +0000917 # map arrays to Python lists
918 self._stack[mark:] = [self._stack[mark:]]
919 self._value = 0
920 dispatch["array"] = end_array
921
922 def end_struct(self, data):
Raymond Hettinger46ac8eb2002-06-30 03:39:14 +0000923 mark = self._marks.pop()
Fredrik Lundhb9056332001-07-11 17:42:21 +0000924 # map structs to Python dictionaries
925 dict = {}
926 items = self._stack[mark:]
927 for i in range(0, len(items), 2):
928 dict[_stringify(items[i])] = items[i+1]
929 self._stack[mark:] = [dict]
930 self._value = 0
931 dispatch["struct"] = end_struct
932
Fredrik Lundh1538c232001-10-01 19:42:03 +0000933 def end_base64(self, data):
Fredrik Lundhb9056332001-07-11 17:42:21 +0000934 value = Binary()
Fredrik Lundh1538c232001-10-01 19:42:03 +0000935 value.decode(data)
Fredrik Lundhb9056332001-07-11 17:42:21 +0000936 self.append(value)
937 self._value = 0
938 dispatch["base64"] = end_base64
939
Fredrik Lundh1538c232001-10-01 19:42:03 +0000940 def end_dateTime(self, data):
Fredrik Lundhb9056332001-07-11 17:42:21 +0000941 value = DateTime()
Fredrik Lundh1538c232001-10-01 19:42:03 +0000942 value.decode(data)
Skip Montanaro174dd222005-05-14 20:54:16 +0000943 if self._use_datetime:
944 value = _datetime_type(data)
Fredrik Lundhb9056332001-07-11 17:42:21 +0000945 self.append(value)
946 dispatch["dateTime.iso8601"] = end_dateTime
947
948 def end_value(self, data):
Fredrik Lundh3d9addd2002-06-27 21:36:21 +0000949 # if we stumble upon a value element with no internal
Fredrik Lundhb9056332001-07-11 17:42:21 +0000950 # elements, treat it as a string element
951 if self._value:
952 self.end_string(data)
953 dispatch["value"] = end_value
954
955 def end_params(self, data):
956 self._type = "params"
957 dispatch["params"] = end_params
958
959 def end_fault(self, data):
960 self._type = "fault"
961 dispatch["fault"] = end_fault
962
Fredrik Lundh1538c232001-10-01 19:42:03 +0000963 def end_methodName(self, data):
Fredrik Lundhb9056332001-07-11 17:42:21 +0000964 if self._encoding:
965 data = _decode(data, self._encoding)
966 self._methodname = data
967 self._type = "methodName" # no params
968 dispatch["methodName"] = end_methodName
969
Martin v. Löwis45394c22003-10-31 13:49:36 +0000970## Multicall support
971#
Fredrik Lundhb9056332001-07-11 17:42:21 +0000972
Martin v. Löwis45394c22003-10-31 13:49:36 +0000973class _MultiCallMethod:
974 # some lesser magic to store calls made to a MultiCall object
975 # for batch execution
976 def __init__(self, call_list, name):
977 self.__call_list = call_list
978 self.__name = name
979 def __getattr__(self, name):
980 return _MultiCallMethod(self.__call_list, "%s.%s" % (self.__name, name))
981 def __call__(self, *args):
982 self.__call_list.append((self.__name, args))
983
Martin v. Löwis12237b32004-08-22 16:04:50 +0000984class MultiCallIterator:
Martin v. Löwis45394c22003-10-31 13:49:36 +0000985 """Iterates over the results of a multicall. Exceptions are
986 thrown in response to xmlrpc faults."""
Raymond Hettingercc523fc2003-11-02 09:47:05 +0000987
Martin v. Löwis12237b32004-08-22 16:04:50 +0000988 def __init__(self, results):
989 self.results = results
990
991 def __getitem__(self, i):
992 item = self.results[i]
993 if type(item) == type({}):
994 raise Fault(item['faultCode'], item['faultString'])
995 elif type(item) == type([]):
996 return item[0]
Martin v. Löwis45394c22003-10-31 13:49:36 +0000997 else:
998 raise ValueError,\
999 "unexpected type in multicall result"
Raymond Hettingercc523fc2003-11-02 09:47:05 +00001000
Martin v. Löwis45394c22003-10-31 13:49:36 +00001001class MultiCall:
1002 """server -> a object used to boxcar method calls
1003
1004 server should be a ServerProxy object.
1005
1006 Methods can be added to the MultiCall using normal
1007 method call syntax e.g.:
1008
1009 multicall = MultiCall(server_proxy)
1010 multicall.add(2,3)
1011 multicall.get_address("Guido")
1012
1013 To execute the multicall, call the MultiCall object e.g.:
1014
1015 add_result, address = multicall()
1016 """
Raymond Hettingercc523fc2003-11-02 09:47:05 +00001017
Martin v. Löwis45394c22003-10-31 13:49:36 +00001018 def __init__(self, server):
1019 self.__server = server
1020 self.__call_list = []
Raymond Hettingercc523fc2003-11-02 09:47:05 +00001021
Martin v. Löwis45394c22003-10-31 13:49:36 +00001022 def __repr__(self):
1023 return "<MultiCall at %x>" % id(self)
Raymond Hettingercc523fc2003-11-02 09:47:05 +00001024
Martin v. Löwis45394c22003-10-31 13:49:36 +00001025 __str__ = __repr__
1026
1027 def __getattr__(self, name):
1028 return _MultiCallMethod(self.__call_list, name)
1029
1030 def __call__(self):
1031 marshalled_list = []
1032 for name, args in self.__call_list:
1033 marshalled_list.append({'methodName' : name, 'params' : args})
1034
1035 return MultiCallIterator(self.__server.system.multicall(marshalled_list))
Raymond Hettingercc523fc2003-11-02 09:47:05 +00001036
Fredrik Lundhb9056332001-07-11 17:42:21 +00001037# --------------------------------------------------------------------
1038# convenience functions
1039
Fredrik Lundh3d9addd2002-06-27 21:36:21 +00001040##
1041# Create a parser object, and connect it to an unmarshalling instance.
1042# This function picks the fastest available XML parser.
1043#
1044# return A (parser, unmarshaller) tuple.
1045
Skip Montanaro174dd222005-05-14 20:54:16 +00001046def getparser(use_datetime=0):
Fredrik Lundhb9056332001-07-11 17:42:21 +00001047 """getparser() -> parser, unmarshaller
1048
Fredrik Lundhb0e8e9b2001-09-10 21:45:42 +00001049 Create an instance of the fastest available parser, and attach it
1050 to an unmarshalling object. Return both objects.
Fredrik Lundhb9056332001-07-11 17:42:21 +00001051 """
Skip Montanaro174dd222005-05-14 20:54:16 +00001052 if use_datetime and not datetime:
1053 raise ValueError, "the datetime module is not available"
Fredrik Lundhb9056332001-07-11 17:42:21 +00001054 if FastParser and FastUnmarshaller:
Skip Montanaro174dd222005-05-14 20:54:16 +00001055 if use_datetime:
1056 mkdatetime = _datetime_type
1057 else:
1058 mkdatetime = _datetime
1059 target = FastUnmarshaller(True, False, _binary, mkdatetime, Fault)
Fredrik Lundhb9056332001-07-11 17:42:21 +00001060 parser = FastParser(target)
1061 else:
Skip Montanaro174dd222005-05-14 20:54:16 +00001062 target = Unmarshaller(use_datetime=use_datetime)
Fredrik Lundhb9056332001-07-11 17:42:21 +00001063 if FastParser:
1064 parser = FastParser(target)
1065 elif SgmlopParser:
1066 parser = SgmlopParser(target)
1067 elif ExpatParser:
1068 parser = ExpatParser(target)
1069 else:
1070 parser = SlowParser(target)
1071 return parser, target
1072
Fredrik Lundh3d9addd2002-06-27 21:36:21 +00001073##
1074# Convert a Python tuple or a Fault instance to an XML-RPC packet.
1075#
1076# @def dumps(params, **options)
1077# @param params A tuple or Fault instance.
1078# @keyparam methodname If given, create a methodCall request for
1079# this method name.
1080# @keyparam methodresponse If given, create a methodResponse packet.
1081# If used with a tuple, the tuple must be a singleton (that is,
1082# it must contain exactly one element).
1083# @keyparam encoding The packet encoding.
1084# @return A string containing marshalled data.
1085
Andrew M. Kuchlinga4c2b742003-04-25 00:26:51 +00001086def dumps(params, methodname=None, methodresponse=None, encoding=None,
1087 allow_none=0):
Fredrik Lundhb9056332001-07-11 17:42:21 +00001088 """data [,options] -> marshalled data
1089
1090 Convert an argument tuple or a Fault instance to an XML-RPC
1091 request (or response, if the methodresponse option is used).
1092
Fredrik Lundhb0e8e9b2001-09-10 21:45:42 +00001093 In addition to the data object, the following options can be given
1094 as keyword arguments:
Fredrik Lundhb9056332001-07-11 17:42:21 +00001095
1096 methodname: the method name for a methodCall packet
1097
1098 methodresponse: true to create a methodResponse packet.
1099 If this option is used with a tuple, the tuple must be
1100 a singleton (i.e. it can contain only one element).
1101
1102 encoding: the packet encoding (default is UTF-8)
1103
1104 All 8-bit strings in the data structure are assumed to use the
1105 packet encoding. Unicode strings are automatically converted,
Fredrik Lundhb0e8e9b2001-09-10 21:45:42 +00001106 where necessary.
Fredrik Lundhb9056332001-07-11 17:42:21 +00001107 """
1108
Raymond Hettingerfe59dc12005-02-07 15:28:45 +00001109 assert isinstance(params, TupleType) or isinstance(params, Fault),\
Fredrik Lundhb9056332001-07-11 17:42:21 +00001110 "argument must be tuple or Fault instance"
1111
1112 if isinstance(params, Fault):
1113 methodresponse = 1
Raymond Hettingerfe59dc12005-02-07 15:28:45 +00001114 elif methodresponse and isinstance(params, TupleType):
Fredrik Lundhb9056332001-07-11 17:42:21 +00001115 assert len(params) == 1, "response tuple must be a singleton"
1116
Fredrik Lundh3d9addd2002-06-27 21:36:21 +00001117 if not encoding:
Fredrik Lundhb9056332001-07-11 17:42:21 +00001118 encoding = "utf-8"
1119
Fredrik Lundh3d9addd2002-06-27 21:36:21 +00001120 if FastMarshaller:
1121 m = FastMarshaller(encoding)
1122 else:
Andrew M. Kuchlinga4c2b742003-04-25 00:26:51 +00001123 m = Marshaller(encoding, allow_none)
Fredrik Lundh3d9addd2002-06-27 21:36:21 +00001124
Fredrik Lundhb9056332001-07-11 17:42:21 +00001125 data = m.dumps(params)
1126
1127 if encoding != "utf-8":
Fredrik Lundh3d9addd2002-06-27 21:36:21 +00001128 xmlheader = "<?xml version='1.0' encoding='%s'?>\n" % str(encoding)
Fredrik Lundhb9056332001-07-11 17:42:21 +00001129 else:
1130 xmlheader = "<?xml version='1.0'?>\n" # utf-8 is default
1131
1132 # standard XML-RPC wrappings
Fredrik Lundh3d9addd2002-06-27 21:36:21 +00001133 if methodname:
Fredrik Lundhb9056332001-07-11 17:42:21 +00001134 # a method call
Raymond Hettingerfe59dc12005-02-07 15:28:45 +00001135 if not isinstance(methodname, StringType):
Fredrik Lundhb9056332001-07-11 17:42:21 +00001136 methodname = methodname.encode(encoding)
1137 data = (
1138 xmlheader,
1139 "<methodCall>\n"
1140 "<methodName>", methodname, "</methodName>\n",
1141 data,
1142 "</methodCall>\n"
1143 )
1144 elif methodresponse:
1145 # a method response, or a fault structure
1146 data = (
1147 xmlheader,
1148 "<methodResponse>\n",
1149 data,
1150 "</methodResponse>\n"
1151 )
1152 else:
1153 return data # return as is
1154 return string.join(data, "")
1155
Fredrik Lundh3d9addd2002-06-27 21:36:21 +00001156##
1157# Convert an XML-RPC packet to a Python object. If the XML-RPC packet
1158# represents a fault condition, this function raises a Fault exception.
1159#
1160# @param data An XML-RPC packet, given as an 8-bit string.
Walter Dörwaldf0dfc7a2003-10-20 14:01:56 +00001161# @return A tuple containing the unpacked data, and the method name
Fredrik Lundh3d9addd2002-06-27 21:36:21 +00001162# (None if not present).
1163# @see Fault
1164
Skip Montanaro174dd222005-05-14 20:54:16 +00001165def loads(data, use_datetime=0):
Fredrik Lundhb9056332001-07-11 17:42:21 +00001166 """data -> unmarshalled data, method name
1167
1168 Convert an XML-RPC packet to unmarshalled data plus a method
1169 name (None if not present).
1170
1171 If the XML-RPC packet represents a fault condition, this function
1172 raises a Fault exception.
1173 """
Skip Montanaro174dd222005-05-14 20:54:16 +00001174 p, u = getparser(use_datetime=use_datetime)
Fredrik Lundhb9056332001-07-11 17:42:21 +00001175 p.feed(data)
1176 p.close()
1177 return u.close(), u.getmethodname()
1178
1179
1180# --------------------------------------------------------------------
1181# request dispatcher
1182
1183class _Method:
1184 # some magic to bind an XML-RPC method to an RPC server.
1185 # supports "nested" methods (e.g. examples.getStateName)
1186 def __init__(self, send, name):
1187 self.__send = send
1188 self.__name = name
1189 def __getattr__(self, name):
1190 return _Method(self.__send, "%s.%s" % (self.__name, name))
1191 def __call__(self, *args):
1192 return self.__send(self.__name, args)
1193
Fredrik Lundh3d9addd2002-06-27 21:36:21 +00001194##
1195# Standard transport class for XML-RPC over HTTP.
1196# <p>
1197# You can create custom transports by subclassing this method, and
1198# overriding selected methods.
Fredrik Lundhb9056332001-07-11 17:42:21 +00001199
1200class Transport:
Fredrik Lundhc4c062f2001-09-10 19:45:02 +00001201 """Handles an HTTP transaction to an XML-RPC server."""
Fredrik Lundhb9056332001-07-11 17:42:21 +00001202
1203 # client identifier (may be overridden)
1204 user_agent = "xmlrpclib.py/%s (by www.pythonware.com)" % __version__
1205
Skip Montanaro174dd222005-05-14 20:54:16 +00001206 def __init__(self, use_datetime=0):
1207 self._use_datetime = use_datetime
1208
Fredrik Lundh3d9addd2002-06-27 21:36:21 +00001209 ##
1210 # Send a complete request, and parse the response.
1211 #
1212 # @param host Target host.
1213 # @param handler Target PRC handler.
1214 # @param request_body XML-RPC request body.
1215 # @param verbose Debugging flag.
1216 # @return Parsed response.
1217
Fredrik Lundhb9056332001-07-11 17:42:21 +00001218 def request(self, host, handler, request_body, verbose=0):
1219 # issue XML-RPC request
1220
1221 h = self.make_connection(host)
1222 if verbose:
1223 h.set_debuglevel(1)
1224
1225 self.send_request(h, handler, request_body)
1226 self.send_host(h, host)
1227 self.send_user_agent(h)
1228 self.send_content(h, request_body)
1229
1230 errcode, errmsg, headers = h.getreply()
1231
1232 if errcode != 200:
1233 raise ProtocolError(
1234 host + handler,
1235 errcode, errmsg,
1236 headers
1237 )
1238
1239 self.verbose = verbose
1240
Fredrik Lundh3d9addd2002-06-27 21:36:21 +00001241 try:
1242 sock = h._conn.sock
1243 except AttributeError:
1244 sock = None
1245
1246 return self._parse_response(h.getfile(), sock)
1247
1248 ##
1249 # Create parser.
1250 #
1251 # @return A 2-tuple containing a parser and a unmarshaller.
Fredrik Lundhb9056332001-07-11 17:42:21 +00001252
Fredrik Lundhc4c062f2001-09-10 19:45:02 +00001253 def getparser(self):
1254 # get parser and unmarshaller
Skip Montanaro174dd222005-05-14 20:54:16 +00001255 return getparser(use_datetime=self._use_datetime)
Fredrik Lundhc4c062f2001-09-10 19:45:02 +00001256
Fredrik Lundh3d9addd2002-06-27 21:36:21 +00001257 ##
Fredrik Lundh1303c7c2002-10-22 18:23:00 +00001258 # Get authorization info from host parameter
1259 # Host may be a string, or a (host, x509-dict) tuple; if a string,
1260 # it is checked for a "user:pw@host" format, and a "Basic
1261 # Authentication" header is added if appropriate.
1262 #
1263 # @param host Host descriptor (URL or (URL, x509 info) tuple).
1264 # @return A 3-tuple containing (actual host, extra headers,
1265 # x509 info). The header and x509 fields may be None.
1266
1267 def get_host_info(self, host):
1268
1269 x509 = {}
Raymond Hettingerfe59dc12005-02-07 15:28:45 +00001270 if isinstance(host, TupleType):
Fredrik Lundh1303c7c2002-10-22 18:23:00 +00001271 host, x509 = host
1272
1273 import urllib
1274 auth, host = urllib.splituser(host)
1275
1276 if auth:
1277 import base64
Fredrik Lundh768c98b2002-11-01 17:14:16 +00001278 auth = base64.encodestring(urllib.unquote(auth))
Fredrik Lundh1303c7c2002-10-22 18:23:00 +00001279 auth = string.join(string.split(auth), "") # get rid of whitespace
1280 extra_headers = [
1281 ("Authorization", "Basic " + auth)
1282 ]
1283 else:
1284 extra_headers = None
1285
1286 return host, extra_headers, x509
1287
1288 ##
Fredrik Lundh3d9addd2002-06-27 21:36:21 +00001289 # Connect to server.
1290 #
1291 # @param host Target host.
1292 # @return A connection handle.
1293
Fredrik Lundhb9056332001-07-11 17:42:21 +00001294 def make_connection(self, host):
1295 # create a HTTP connection object from a host descriptor
1296 import httplib
Fredrik Lundh1303c7c2002-10-22 18:23:00 +00001297 host, extra_headers, x509 = self.get_host_info(host)
Fredrik Lundhb9056332001-07-11 17:42:21 +00001298 return httplib.HTTP(host)
1299
Fredrik Lundh3d9addd2002-06-27 21:36:21 +00001300 ##
1301 # Send request header.
1302 #
1303 # @param connection Connection handle.
1304 # @param handler Target RPC handler.
1305 # @param request_body XML-RPC body.
1306
Fredrik Lundhb9056332001-07-11 17:42:21 +00001307 def send_request(self, connection, handler, request_body):
1308 connection.putrequest("POST", handler)
1309
Fredrik Lundh3d9addd2002-06-27 21:36:21 +00001310 ##
1311 # Send host name.
1312 #
1313 # @param connection Connection handle.
1314 # @param host Host name.
1315
Fredrik Lundhb9056332001-07-11 17:42:21 +00001316 def send_host(self, connection, host):
Fredrik Lundh1303c7c2002-10-22 18:23:00 +00001317 host, extra_headers, x509 = self.get_host_info(host)
Fredrik Lundhb9056332001-07-11 17:42:21 +00001318 connection.putheader("Host", host)
Fredrik Lundh1303c7c2002-10-22 18:23:00 +00001319 if extra_headers:
Raymond Hettingerfe59dc12005-02-07 15:28:45 +00001320 if isinstance(extra_headers, DictType):
Fredrik Lundh1303c7c2002-10-22 18:23:00 +00001321 extra_headers = extra_headers.items()
1322 for key, value in extra_headers:
1323 connection.putheader(key, value)
Fredrik Lundhb9056332001-07-11 17:42:21 +00001324
Fredrik Lundh3d9addd2002-06-27 21:36:21 +00001325 ##
1326 # Send user-agent identifier.
1327 #
1328 # @param connection Connection handle.
1329
Fredrik Lundhb9056332001-07-11 17:42:21 +00001330 def send_user_agent(self, connection):
1331 connection.putheader("User-Agent", self.user_agent)
1332
Fredrik Lundh3d9addd2002-06-27 21:36:21 +00001333 ##
1334 # Send request body.
1335 #
1336 # @param connection Connection handle.
1337 # @param request_body XML-RPC request body.
1338
Fredrik Lundhb9056332001-07-11 17:42:21 +00001339 def send_content(self, connection, request_body):
1340 connection.putheader("Content-Type", "text/xml")
1341 connection.putheader("Content-Length", str(len(request_body)))
1342 connection.endheaders()
1343 if request_body:
1344 connection.send(request_body)
1345
Fredrik Lundh3d9addd2002-06-27 21:36:21 +00001346 ##
1347 # Parse response.
1348 #
1349 # @param file Stream.
1350 # @return Response tuple and target method.
1351
1352 def parse_response(self, file):
1353 # compatibility interface
1354 return self._parse_response(file, None)
1355
1356 ##
1357 # Parse response (alternate interface). This is similar to the
1358 # parse_response method, but also provides direct access to the
1359 # underlying socket object (where available).
1360 #
1361 # @param file Stream.
1362 # @param sock Socket handle (or None, if the socket object
1363 # could not be accessed).
1364 # @return Response tuple and target method.
1365
1366 def _parse_response(self, file, sock):
1367 # read response from input file/socket, and parse it
Fredrik Lundhb9056332001-07-11 17:42:21 +00001368
Fredrik Lundhc4c062f2001-09-10 19:45:02 +00001369 p, u = self.getparser()
Fredrik Lundhb9056332001-07-11 17:42:21 +00001370
1371 while 1:
Fredrik Lundh3d9addd2002-06-27 21:36:21 +00001372 if sock:
1373 response = sock.recv(1024)
1374 else:
1375 response = file.read(1024)
Fredrik Lundhb9056332001-07-11 17:42:21 +00001376 if not response:
1377 break
1378 if self.verbose:
1379 print "body:", repr(response)
1380 p.feed(response)
1381
Fredrik Lundh3d9addd2002-06-27 21:36:21 +00001382 file.close()
Fredrik Lundhb9056332001-07-11 17:42:21 +00001383 p.close()
1384
1385 return u.close()
1386
Fredrik Lundh3d9addd2002-06-27 21:36:21 +00001387##
1388# Standard transport class for XML-RPC over HTTPS.
1389
Fredrik Lundhb9056332001-07-11 17:42:21 +00001390class SafeTransport(Transport):
Fredrik Lundhc4c062f2001-09-10 19:45:02 +00001391 """Handles an HTTPS transaction to an XML-RPC server."""
Fredrik Lundhb9056332001-07-11 17:42:21 +00001392
Fredrik Lundh3d9addd2002-06-27 21:36:21 +00001393 # FIXME: mostly untested
1394
Fredrik Lundhb9056332001-07-11 17:42:21 +00001395 def make_connection(self, host):
1396 # create a HTTPS connection object from a host descriptor
1397 # host may be a string, or a (host, x509-dict) tuple
1398 import httplib
Fredrik Lundh1303c7c2002-10-22 18:23:00 +00001399 host, extra_headers, x509 = self.get_host_info(host)
Fredrik Lundhb9056332001-07-11 17:42:21 +00001400 try:
1401 HTTPS = httplib.HTTPS
1402 except AttributeError:
Fredrik Lundh1303c7c2002-10-22 18:23:00 +00001403 raise NotImplementedError(
1404 "your version of httplib doesn't support HTTPS"
1405 )
Fredrik Lundhb9056332001-07-11 17:42:21 +00001406 else:
Guido van Rossum68468eb2003-02-27 20:14:51 +00001407 return HTTPS(host, None, **(x509 or {}))
Fredrik Lundhb9056332001-07-11 17:42:21 +00001408
Fredrik Lundh3d9addd2002-06-27 21:36:21 +00001409##
1410# Standard server proxy. This class establishes a virtual connection
1411# to an XML-RPC server.
1412# <p>
1413# This class is available as ServerProxy and Server. New code should
1414# use ServerProxy, to avoid confusion.
1415#
1416# @def ServerProxy(uri, **options)
1417# @param uri The connection point on the server.
1418# @keyparam transport A transport factory, compatible with the
1419# standard transport class.
1420# @keyparam encoding The default encoding used for 8-bit strings
1421# (default is UTF-8).
1422# @keyparam verbose Use a true value to enable debugging output.
1423# (printed to standard output).
1424# @see Transport
1425
Fredrik Lundhb9056332001-07-11 17:42:21 +00001426class ServerProxy:
1427 """uri [,options] -> a logical connection to an XML-RPC server
1428
1429 uri is the connection point on the server, given as
1430 scheme://host/target.
1431
1432 The standard implementation always supports the "http" scheme. If
1433 SSL socket support is available (Python 2.0), it also supports
1434 "https".
1435
1436 If the target part and the slash preceding it are both omitted,
1437 "/RPC2" is assumed.
1438
1439 The following options can be given as keyword arguments:
1440
1441 transport: a transport factory
1442 encoding: the request encoding (default is UTF-8)
1443
1444 All 8-bit strings passed to the server proxy are assumed to use
1445 the given encoding.
1446 """
1447
Andrew M. Kuchlinga4c2b742003-04-25 00:26:51 +00001448 def __init__(self, uri, transport=None, encoding=None, verbose=0,
Skip Montanaro174dd222005-05-14 20:54:16 +00001449 allow_none=0, use_datetime=0):
Fredrik Lundhb9056332001-07-11 17:42:21 +00001450 # establish a "logical" server connection
1451
1452 # get the url
Fredrik Lundhc4c062f2001-09-10 19:45:02 +00001453 import urllib
Fredrik Lundhb9056332001-07-11 17:42:21 +00001454 type, uri = urllib.splittype(uri)
1455 if type not in ("http", "https"):
1456 raise IOError, "unsupported XML-RPC protocol"
1457 self.__host, self.__handler = urllib.splithost(uri)
1458 if not self.__handler:
1459 self.__handler = "/RPC2"
1460
1461 if transport is None:
1462 if type == "https":
Skip Montanaro174dd222005-05-14 20:54:16 +00001463 transport = SafeTransport(use_datetime=use_datetime)
Fredrik Lundhb9056332001-07-11 17:42:21 +00001464 else:
Skip Montanaro174dd222005-05-14 20:54:16 +00001465 transport = Transport(use_datetime=use_datetime)
Fredrik Lundhb9056332001-07-11 17:42:21 +00001466 self.__transport = transport
1467
1468 self.__encoding = encoding
1469 self.__verbose = verbose
Andrew M. Kuchlinga4c2b742003-04-25 00:26:51 +00001470 self.__allow_none = allow_none
Tim Petersc2659cf2003-05-12 20:19:37 +00001471
Fredrik Lundhb9056332001-07-11 17:42:21 +00001472 def __request(self, methodname, params):
1473 # call a method on the remote server
1474
Andrew M. Kuchlinga4c2b742003-04-25 00:26:51 +00001475 request = dumps(params, methodname, encoding=self.__encoding,
1476 allow_none=self.__allow_none)
Fredrik Lundhb9056332001-07-11 17:42:21 +00001477
1478 response = self.__transport.request(
1479 self.__host,
1480 self.__handler,
1481 request,
1482 verbose=self.__verbose
1483 )
1484
1485 if len(response) == 1:
1486 response = response[0]
1487
1488 return response
1489
1490 def __repr__(self):
1491 return (
Fredrik Lundh78eedce2001-08-23 20:04:33 +00001492 "<ServerProxy for %s%s>" %
Fredrik Lundhb9056332001-07-11 17:42:21 +00001493 (self.__host, self.__handler)
1494 )
1495
1496 __str__ = __repr__
Raymond Hettingercc523fc2003-11-02 09:47:05 +00001497
Fredrik Lundhb9056332001-07-11 17:42:21 +00001498 def __getattr__(self, name):
1499 # magic method dispatcher
1500 return _Method(self.__request, name)
1501
1502 # note: to call a remote object with an non-standard name, use
1503 # result getattr(server, "strange-python-name")(args)
1504
Fredrik Lundh78eedce2001-08-23 20:04:33 +00001505# compatibility
Fredrik Lundh3d9addd2002-06-27 21:36:21 +00001506
Fredrik Lundhb9056332001-07-11 17:42:21 +00001507Server = ServerProxy
1508
1509# --------------------------------------------------------------------
1510# test code
1511
1512if __name__ == "__main__":
1513
1514 # simple test program (from the XML-RPC specification)
1515
Fredrik Lundh78eedce2001-08-23 20:04:33 +00001516 # server = ServerProxy("http://localhost:8000") # local server
Martin v. Löwis12237b32004-08-22 16:04:50 +00001517 server = ServerProxy("http://time.xmlrpc.com/RPC2")
Fredrik Lundhb9056332001-07-11 17:42:21 +00001518
1519 print server
1520
1521 try:
Martin v. Löwis12237b32004-08-22 16:04:50 +00001522 print server.currentTime.getCurrentTime()
1523 except Error, v:
1524 print "ERROR", v
1525
1526 multi = MultiCall(server)
1527 multi.currentTime.getCurrentTime()
1528 multi.currentTime.getCurrentTime()
1529 try:
1530 for response in multi():
1531 print response
Fredrik Lundhb9056332001-07-11 17:42:21 +00001532 except Error, v:
1533 print "ERROR", v