blob: f600de7f245ea47151c8a71d029fe9b7cef9621b [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
Georg Brandld0b592f2008-05-29 07:45:26 +0000900 dispatch["i8"] = end_int
Fredrik Lundhb9056332001-07-11 17:42:21 +0000901 dispatch["int"] = end_int
902
Fredrik Lundh1538c232001-10-01 19:42:03 +0000903 def end_double(self, data):
904 self.append(float(data))
Fredrik Lundhb9056332001-07-11 17:42:21 +0000905 self._value = 0
906 dispatch["double"] = end_double
907
Fredrik Lundh1538c232001-10-01 19:42:03 +0000908 def end_string(self, data):
Fredrik Lundhb9056332001-07-11 17:42:21 +0000909 if self._encoding:
910 data = _decode(data, self._encoding)
911 self.append(_stringify(data))
912 self._value = 0
913 dispatch["string"] = end_string
914 dispatch["name"] = end_string # struct keys are always strings
915
916 def end_array(self, data):
Raymond Hettinger46ac8eb2002-06-30 03:39:14 +0000917 mark = self._marks.pop()
Fredrik Lundhb9056332001-07-11 17:42:21 +0000918 # map arrays to Python lists
919 self._stack[mark:] = [self._stack[mark:]]
920 self._value = 0
921 dispatch["array"] = end_array
922
923 def end_struct(self, data):
Raymond Hettinger46ac8eb2002-06-30 03:39:14 +0000924 mark = self._marks.pop()
Fredrik Lundhb9056332001-07-11 17:42:21 +0000925 # map structs to Python dictionaries
926 dict = {}
927 items = self._stack[mark:]
928 for i in range(0, len(items), 2):
929 dict[_stringify(items[i])] = items[i+1]
930 self._stack[mark:] = [dict]
931 self._value = 0
932 dispatch["struct"] = end_struct
933
Fredrik Lundh1538c232001-10-01 19:42:03 +0000934 def end_base64(self, data):
Fredrik Lundhb9056332001-07-11 17:42:21 +0000935 value = Binary()
Fredrik Lundh1538c232001-10-01 19:42:03 +0000936 value.decode(data)
Fredrik Lundhb9056332001-07-11 17:42:21 +0000937 self.append(value)
938 self._value = 0
939 dispatch["base64"] = end_base64
940
Fredrik Lundh1538c232001-10-01 19:42:03 +0000941 def end_dateTime(self, data):
Fredrik Lundhb9056332001-07-11 17:42:21 +0000942 value = DateTime()
Fredrik Lundh1538c232001-10-01 19:42:03 +0000943 value.decode(data)
Skip Montanaro174dd222005-05-14 20:54:16 +0000944 if self._use_datetime:
945 value = _datetime_type(data)
Fredrik Lundhb9056332001-07-11 17:42:21 +0000946 self.append(value)
947 dispatch["dateTime.iso8601"] = end_dateTime
948
949 def end_value(self, data):
Fredrik Lundh3d9addd2002-06-27 21:36:21 +0000950 # if we stumble upon a value element with no internal
Fredrik Lundhb9056332001-07-11 17:42:21 +0000951 # elements, treat it as a string element
952 if self._value:
953 self.end_string(data)
954 dispatch["value"] = end_value
955
956 def end_params(self, data):
957 self._type = "params"
958 dispatch["params"] = end_params
959
960 def end_fault(self, data):
961 self._type = "fault"
962 dispatch["fault"] = end_fault
963
Fredrik Lundh1538c232001-10-01 19:42:03 +0000964 def end_methodName(self, data):
Fredrik Lundhb9056332001-07-11 17:42:21 +0000965 if self._encoding:
966 data = _decode(data, self._encoding)
967 self._methodname = data
968 self._type = "methodName" # no params
969 dispatch["methodName"] = end_methodName
970
Martin v. Löwis45394c22003-10-31 13:49:36 +0000971## Multicall support
972#
Fredrik Lundhb9056332001-07-11 17:42:21 +0000973
Martin v. Löwis45394c22003-10-31 13:49:36 +0000974class _MultiCallMethod:
975 # some lesser magic to store calls made to a MultiCall object
976 # for batch execution
977 def __init__(self, call_list, name):
978 self.__call_list = call_list
979 self.__name = name
980 def __getattr__(self, name):
981 return _MultiCallMethod(self.__call_list, "%s.%s" % (self.__name, name))
982 def __call__(self, *args):
983 self.__call_list.append((self.__name, args))
984
Martin v. Löwis12237b32004-08-22 16:04:50 +0000985class MultiCallIterator:
Martin v. Löwis45394c22003-10-31 13:49:36 +0000986 """Iterates over the results of a multicall. Exceptions are
987 thrown in response to xmlrpc faults."""
Raymond Hettingercc523fc2003-11-02 09:47:05 +0000988
Martin v. Löwis12237b32004-08-22 16:04:50 +0000989 def __init__(self, results):
990 self.results = results
991
992 def __getitem__(self, i):
993 item = self.results[i]
994 if type(item) == type({}):
995 raise Fault(item['faultCode'], item['faultString'])
996 elif type(item) == type([]):
997 return item[0]
Martin v. Löwis45394c22003-10-31 13:49:36 +0000998 else:
999 raise ValueError,\
1000 "unexpected type in multicall result"
Raymond Hettingercc523fc2003-11-02 09:47:05 +00001001
Martin v. Löwis45394c22003-10-31 13:49:36 +00001002class MultiCall:
1003 """server -> a object used to boxcar method calls
1004
1005 server should be a ServerProxy object.
1006
1007 Methods can be added to the MultiCall using normal
1008 method call syntax e.g.:
1009
1010 multicall = MultiCall(server_proxy)
1011 multicall.add(2,3)
1012 multicall.get_address("Guido")
1013
1014 To execute the multicall, call the MultiCall object e.g.:
1015
1016 add_result, address = multicall()
1017 """
Raymond Hettingercc523fc2003-11-02 09:47:05 +00001018
Martin v. Löwis45394c22003-10-31 13:49:36 +00001019 def __init__(self, server):
1020 self.__server = server
1021 self.__call_list = []
Raymond Hettingercc523fc2003-11-02 09:47:05 +00001022
Martin v. Löwis45394c22003-10-31 13:49:36 +00001023 def __repr__(self):
1024 return "<MultiCall at %x>" % id(self)
Raymond Hettingercc523fc2003-11-02 09:47:05 +00001025
Martin v. Löwis45394c22003-10-31 13:49:36 +00001026 __str__ = __repr__
1027
1028 def __getattr__(self, name):
1029 return _MultiCallMethod(self.__call_list, name)
1030
1031 def __call__(self):
1032 marshalled_list = []
1033 for name, args in self.__call_list:
1034 marshalled_list.append({'methodName' : name, 'params' : args})
1035
1036 return MultiCallIterator(self.__server.system.multicall(marshalled_list))
Raymond Hettingercc523fc2003-11-02 09:47:05 +00001037
Fredrik Lundhb9056332001-07-11 17:42:21 +00001038# --------------------------------------------------------------------
1039# convenience functions
1040
Fredrik Lundh3d9addd2002-06-27 21:36:21 +00001041##
1042# Create a parser object, and connect it to an unmarshalling instance.
1043# This function picks the fastest available XML parser.
1044#
1045# return A (parser, unmarshaller) tuple.
1046
Skip Montanaro174dd222005-05-14 20:54:16 +00001047def getparser(use_datetime=0):
Fredrik Lundhb9056332001-07-11 17:42:21 +00001048 """getparser() -> parser, unmarshaller
1049
Fredrik Lundhb0e8e9b2001-09-10 21:45:42 +00001050 Create an instance of the fastest available parser, and attach it
1051 to an unmarshalling object. Return both objects.
Fredrik Lundhb9056332001-07-11 17:42:21 +00001052 """
Skip Montanaro174dd222005-05-14 20:54:16 +00001053 if use_datetime and not datetime:
1054 raise ValueError, "the datetime module is not available"
Fredrik Lundhb9056332001-07-11 17:42:21 +00001055 if FastParser and FastUnmarshaller:
Skip Montanaro174dd222005-05-14 20:54:16 +00001056 if use_datetime:
1057 mkdatetime = _datetime_type
1058 else:
1059 mkdatetime = _datetime
1060 target = FastUnmarshaller(True, False, _binary, mkdatetime, Fault)
Fredrik Lundhb9056332001-07-11 17:42:21 +00001061 parser = FastParser(target)
1062 else:
Skip Montanaro174dd222005-05-14 20:54:16 +00001063 target = Unmarshaller(use_datetime=use_datetime)
Fredrik Lundhb9056332001-07-11 17:42:21 +00001064 if FastParser:
1065 parser = FastParser(target)
1066 elif SgmlopParser:
1067 parser = SgmlopParser(target)
1068 elif ExpatParser:
1069 parser = ExpatParser(target)
1070 else:
1071 parser = SlowParser(target)
1072 return parser, target
1073
Fredrik Lundh3d9addd2002-06-27 21:36:21 +00001074##
1075# Convert a Python tuple or a Fault instance to an XML-RPC packet.
1076#
1077# @def dumps(params, **options)
1078# @param params A tuple or Fault instance.
1079# @keyparam methodname If given, create a methodCall request for
1080# this method name.
1081# @keyparam methodresponse If given, create a methodResponse packet.
1082# If used with a tuple, the tuple must be a singleton (that is,
1083# it must contain exactly one element).
1084# @keyparam encoding The packet encoding.
1085# @return A string containing marshalled data.
1086
Andrew M. Kuchlinga4c2b742003-04-25 00:26:51 +00001087def dumps(params, methodname=None, methodresponse=None, encoding=None,
1088 allow_none=0):
Fredrik Lundhb9056332001-07-11 17:42:21 +00001089 """data [,options] -> marshalled data
1090
1091 Convert an argument tuple or a Fault instance to an XML-RPC
1092 request (or response, if the methodresponse option is used).
1093
Fredrik Lundhb0e8e9b2001-09-10 21:45:42 +00001094 In addition to the data object, the following options can be given
1095 as keyword arguments:
Fredrik Lundhb9056332001-07-11 17:42:21 +00001096
1097 methodname: the method name for a methodCall packet
1098
1099 methodresponse: true to create a methodResponse packet.
1100 If this option is used with a tuple, the tuple must be
1101 a singleton (i.e. it can contain only one element).
1102
1103 encoding: the packet encoding (default is UTF-8)
1104
1105 All 8-bit strings in the data structure are assumed to use the
1106 packet encoding. Unicode strings are automatically converted,
Fredrik Lundhb0e8e9b2001-09-10 21:45:42 +00001107 where necessary.
Fredrik Lundhb9056332001-07-11 17:42:21 +00001108 """
1109
Raymond Hettingerfe59dc12005-02-07 15:28:45 +00001110 assert isinstance(params, TupleType) or isinstance(params, Fault),\
Fredrik Lundhb9056332001-07-11 17:42:21 +00001111 "argument must be tuple or Fault instance"
1112
1113 if isinstance(params, Fault):
1114 methodresponse = 1
Raymond Hettingerfe59dc12005-02-07 15:28:45 +00001115 elif methodresponse and isinstance(params, TupleType):
Fredrik Lundhb9056332001-07-11 17:42:21 +00001116 assert len(params) == 1, "response tuple must be a singleton"
1117
Fredrik Lundh3d9addd2002-06-27 21:36:21 +00001118 if not encoding:
Fredrik Lundhb9056332001-07-11 17:42:21 +00001119 encoding = "utf-8"
1120
Fredrik Lundh3d9addd2002-06-27 21:36:21 +00001121 if FastMarshaller:
1122 m = FastMarshaller(encoding)
1123 else:
Andrew M. Kuchlinga4c2b742003-04-25 00:26:51 +00001124 m = Marshaller(encoding, allow_none)
Fredrik Lundh3d9addd2002-06-27 21:36:21 +00001125
Fredrik Lundhb9056332001-07-11 17:42:21 +00001126 data = m.dumps(params)
1127
1128 if encoding != "utf-8":
Fredrik Lundh3d9addd2002-06-27 21:36:21 +00001129 xmlheader = "<?xml version='1.0' encoding='%s'?>\n" % str(encoding)
Fredrik Lundhb9056332001-07-11 17:42:21 +00001130 else:
1131 xmlheader = "<?xml version='1.0'?>\n" # utf-8 is default
1132
1133 # standard XML-RPC wrappings
Fredrik Lundh3d9addd2002-06-27 21:36:21 +00001134 if methodname:
Fredrik Lundhb9056332001-07-11 17:42:21 +00001135 # a method call
Raymond Hettingerfe59dc12005-02-07 15:28:45 +00001136 if not isinstance(methodname, StringType):
Fredrik Lundhb9056332001-07-11 17:42:21 +00001137 methodname = methodname.encode(encoding)
1138 data = (
1139 xmlheader,
1140 "<methodCall>\n"
1141 "<methodName>", methodname, "</methodName>\n",
1142 data,
1143 "</methodCall>\n"
1144 )
1145 elif methodresponse:
1146 # a method response, or a fault structure
1147 data = (
1148 xmlheader,
1149 "<methodResponse>\n",
1150 data,
1151 "</methodResponse>\n"
1152 )
1153 else:
1154 return data # return as is
1155 return string.join(data, "")
1156
Fredrik Lundh3d9addd2002-06-27 21:36:21 +00001157##
1158# Convert an XML-RPC packet to a Python object. If the XML-RPC packet
1159# represents a fault condition, this function raises a Fault exception.
1160#
1161# @param data An XML-RPC packet, given as an 8-bit string.
Walter Dörwaldf0dfc7a2003-10-20 14:01:56 +00001162# @return A tuple containing the unpacked data, and the method name
Fredrik Lundh3d9addd2002-06-27 21:36:21 +00001163# (None if not present).
1164# @see Fault
1165
Skip Montanaro174dd222005-05-14 20:54:16 +00001166def loads(data, use_datetime=0):
Fredrik Lundhb9056332001-07-11 17:42:21 +00001167 """data -> unmarshalled data, method name
1168
1169 Convert an XML-RPC packet to unmarshalled data plus a method
1170 name (None if not present).
1171
1172 If the XML-RPC packet represents a fault condition, this function
1173 raises a Fault exception.
1174 """
Skip Montanaro174dd222005-05-14 20:54:16 +00001175 p, u = getparser(use_datetime=use_datetime)
Fredrik Lundhb9056332001-07-11 17:42:21 +00001176 p.feed(data)
1177 p.close()
1178 return u.close(), u.getmethodname()
1179
1180
1181# --------------------------------------------------------------------
1182# request dispatcher
1183
1184class _Method:
1185 # some magic to bind an XML-RPC method to an RPC server.
1186 # supports "nested" methods (e.g. examples.getStateName)
1187 def __init__(self, send, name):
1188 self.__send = send
1189 self.__name = name
1190 def __getattr__(self, name):
1191 return _Method(self.__send, "%s.%s" % (self.__name, name))
1192 def __call__(self, *args):
1193 return self.__send(self.__name, args)
1194
Fredrik Lundh3d9addd2002-06-27 21:36:21 +00001195##
1196# Standard transport class for XML-RPC over HTTP.
1197# <p>
1198# You can create custom transports by subclassing this method, and
1199# overriding selected methods.
Fredrik Lundhb9056332001-07-11 17:42:21 +00001200
1201class Transport:
Fredrik Lundhc4c062f2001-09-10 19:45:02 +00001202 """Handles an HTTP transaction to an XML-RPC server."""
Fredrik Lundhb9056332001-07-11 17:42:21 +00001203
1204 # client identifier (may be overridden)
1205 user_agent = "xmlrpclib.py/%s (by www.pythonware.com)" % __version__
1206
Skip Montanaro174dd222005-05-14 20:54:16 +00001207 def __init__(self, use_datetime=0):
1208 self._use_datetime = use_datetime
1209
Fredrik Lundh3d9addd2002-06-27 21:36:21 +00001210 ##
1211 # Send a complete request, and parse the response.
1212 #
1213 # @param host Target host.
1214 # @param handler Target PRC handler.
1215 # @param request_body XML-RPC request body.
1216 # @param verbose Debugging flag.
1217 # @return Parsed response.
1218
Fredrik Lundhb9056332001-07-11 17:42:21 +00001219 def request(self, host, handler, request_body, verbose=0):
1220 # issue XML-RPC request
1221
1222 h = self.make_connection(host)
1223 if verbose:
1224 h.set_debuglevel(1)
1225
1226 self.send_request(h, handler, request_body)
1227 self.send_host(h, host)
1228 self.send_user_agent(h)
1229 self.send_content(h, request_body)
1230
1231 errcode, errmsg, headers = h.getreply()
1232
1233 if errcode != 200:
1234 raise ProtocolError(
1235 host + handler,
1236 errcode, errmsg,
1237 headers
1238 )
1239
1240 self.verbose = verbose
1241
Fredrik Lundh3d9addd2002-06-27 21:36:21 +00001242 try:
1243 sock = h._conn.sock
1244 except AttributeError:
1245 sock = None
1246
1247 return self._parse_response(h.getfile(), sock)
1248
1249 ##
1250 # Create parser.
1251 #
1252 # @return A 2-tuple containing a parser and a unmarshaller.
Fredrik Lundhb9056332001-07-11 17:42:21 +00001253
Fredrik Lundhc4c062f2001-09-10 19:45:02 +00001254 def getparser(self):
1255 # get parser and unmarshaller
Skip Montanaro174dd222005-05-14 20:54:16 +00001256 return getparser(use_datetime=self._use_datetime)
Fredrik Lundhc4c062f2001-09-10 19:45:02 +00001257
Fredrik Lundh3d9addd2002-06-27 21:36:21 +00001258 ##
Fredrik Lundh1303c7c2002-10-22 18:23:00 +00001259 # Get authorization info from host parameter
1260 # Host may be a string, or a (host, x509-dict) tuple; if a string,
1261 # it is checked for a "user:pw@host" format, and a "Basic
1262 # Authentication" header is added if appropriate.
1263 #
1264 # @param host Host descriptor (URL or (URL, x509 info) tuple).
1265 # @return A 3-tuple containing (actual host, extra headers,
1266 # x509 info). The header and x509 fields may be None.
1267
1268 def get_host_info(self, host):
1269
1270 x509 = {}
Raymond Hettingerfe59dc12005-02-07 15:28:45 +00001271 if isinstance(host, TupleType):
Fredrik Lundh1303c7c2002-10-22 18:23:00 +00001272 host, x509 = host
1273
1274 import urllib
1275 auth, host = urllib.splituser(host)
1276
1277 if auth:
1278 import base64
Fredrik Lundh768c98b2002-11-01 17:14:16 +00001279 auth = base64.encodestring(urllib.unquote(auth))
Fredrik Lundh1303c7c2002-10-22 18:23:00 +00001280 auth = string.join(string.split(auth), "") # get rid of whitespace
1281 extra_headers = [
1282 ("Authorization", "Basic " + auth)
1283 ]
1284 else:
1285 extra_headers = None
1286
1287 return host, extra_headers, x509
1288
1289 ##
Fredrik Lundh3d9addd2002-06-27 21:36:21 +00001290 # Connect to server.
1291 #
1292 # @param host Target host.
1293 # @return A connection handle.
1294
Fredrik Lundhb9056332001-07-11 17:42:21 +00001295 def make_connection(self, host):
1296 # create a HTTP connection object from a host descriptor
1297 import httplib
Fredrik Lundh1303c7c2002-10-22 18:23:00 +00001298 host, extra_headers, x509 = self.get_host_info(host)
Fredrik Lundhb9056332001-07-11 17:42:21 +00001299 return httplib.HTTP(host)
1300
Fredrik Lundh3d9addd2002-06-27 21:36:21 +00001301 ##
1302 # Send request header.
1303 #
1304 # @param connection Connection handle.
1305 # @param handler Target RPC handler.
1306 # @param request_body XML-RPC body.
1307
Fredrik Lundhb9056332001-07-11 17:42:21 +00001308 def send_request(self, connection, handler, request_body):
1309 connection.putrequest("POST", handler)
1310
Fredrik Lundh3d9addd2002-06-27 21:36:21 +00001311 ##
1312 # Send host name.
1313 #
1314 # @param connection Connection handle.
1315 # @param host Host name.
1316
Fredrik Lundhb9056332001-07-11 17:42:21 +00001317 def send_host(self, connection, host):
Fredrik Lundh1303c7c2002-10-22 18:23:00 +00001318 host, extra_headers, x509 = self.get_host_info(host)
Fredrik Lundhb9056332001-07-11 17:42:21 +00001319 connection.putheader("Host", host)
Fredrik Lundh1303c7c2002-10-22 18:23:00 +00001320 if extra_headers:
Raymond Hettingerfe59dc12005-02-07 15:28:45 +00001321 if isinstance(extra_headers, DictType):
Fredrik Lundh1303c7c2002-10-22 18:23:00 +00001322 extra_headers = extra_headers.items()
1323 for key, value in extra_headers:
1324 connection.putheader(key, value)
Fredrik Lundhb9056332001-07-11 17:42:21 +00001325
Fredrik Lundh3d9addd2002-06-27 21:36:21 +00001326 ##
1327 # Send user-agent identifier.
1328 #
1329 # @param connection Connection handle.
1330
Fredrik Lundhb9056332001-07-11 17:42:21 +00001331 def send_user_agent(self, connection):
1332 connection.putheader("User-Agent", self.user_agent)
1333
Fredrik Lundh3d9addd2002-06-27 21:36:21 +00001334 ##
1335 # Send request body.
1336 #
1337 # @param connection Connection handle.
1338 # @param request_body XML-RPC request body.
1339
Fredrik Lundhb9056332001-07-11 17:42:21 +00001340 def send_content(self, connection, request_body):
1341 connection.putheader("Content-Type", "text/xml")
1342 connection.putheader("Content-Length", str(len(request_body)))
1343 connection.endheaders()
1344 if request_body:
1345 connection.send(request_body)
1346
Fredrik Lundh3d9addd2002-06-27 21:36:21 +00001347 ##
1348 # Parse response.
1349 #
1350 # @param file Stream.
1351 # @return Response tuple and target method.
1352
1353 def parse_response(self, file):
1354 # compatibility interface
1355 return self._parse_response(file, None)
1356
1357 ##
1358 # Parse response (alternate interface). This is similar to the
1359 # parse_response method, but also provides direct access to the
1360 # underlying socket object (where available).
1361 #
1362 # @param file Stream.
1363 # @param sock Socket handle (or None, if the socket object
1364 # could not be accessed).
1365 # @return Response tuple and target method.
1366
1367 def _parse_response(self, file, sock):
1368 # read response from input file/socket, and parse it
Fredrik Lundhb9056332001-07-11 17:42:21 +00001369
Fredrik Lundhc4c062f2001-09-10 19:45:02 +00001370 p, u = self.getparser()
Fredrik Lundhb9056332001-07-11 17:42:21 +00001371
1372 while 1:
Fredrik Lundh3d9addd2002-06-27 21:36:21 +00001373 if sock:
1374 response = sock.recv(1024)
1375 else:
1376 response = file.read(1024)
Fredrik Lundhb9056332001-07-11 17:42:21 +00001377 if not response:
1378 break
1379 if self.verbose:
1380 print "body:", repr(response)
1381 p.feed(response)
1382
Fredrik Lundh3d9addd2002-06-27 21:36:21 +00001383 file.close()
Fredrik Lundhb9056332001-07-11 17:42:21 +00001384 p.close()
1385
1386 return u.close()
1387
Fredrik Lundh3d9addd2002-06-27 21:36:21 +00001388##
1389# Standard transport class for XML-RPC over HTTPS.
1390
Fredrik Lundhb9056332001-07-11 17:42:21 +00001391class SafeTransport(Transport):
Fredrik Lundhc4c062f2001-09-10 19:45:02 +00001392 """Handles an HTTPS transaction to an XML-RPC server."""
Fredrik Lundhb9056332001-07-11 17:42:21 +00001393
Fredrik Lundh3d9addd2002-06-27 21:36:21 +00001394 # FIXME: mostly untested
1395
Fredrik Lundhb9056332001-07-11 17:42:21 +00001396 def make_connection(self, host):
1397 # create a HTTPS connection object from a host descriptor
1398 # host may be a string, or a (host, x509-dict) tuple
1399 import httplib
Fredrik Lundh1303c7c2002-10-22 18:23:00 +00001400 host, extra_headers, x509 = self.get_host_info(host)
Fredrik Lundhb9056332001-07-11 17:42:21 +00001401 try:
1402 HTTPS = httplib.HTTPS
1403 except AttributeError:
Fredrik Lundh1303c7c2002-10-22 18:23:00 +00001404 raise NotImplementedError(
1405 "your version of httplib doesn't support HTTPS"
1406 )
Fredrik Lundhb9056332001-07-11 17:42:21 +00001407 else:
Guido van Rossum68468eb2003-02-27 20:14:51 +00001408 return HTTPS(host, None, **(x509 or {}))
Fredrik Lundhb9056332001-07-11 17:42:21 +00001409
Fredrik Lundh3d9addd2002-06-27 21:36:21 +00001410##
1411# Standard server proxy. This class establishes a virtual connection
1412# to an XML-RPC server.
1413# <p>
1414# This class is available as ServerProxy and Server. New code should
1415# use ServerProxy, to avoid confusion.
1416#
1417# @def ServerProxy(uri, **options)
1418# @param uri The connection point on the server.
1419# @keyparam transport A transport factory, compatible with the
1420# standard transport class.
1421# @keyparam encoding The default encoding used for 8-bit strings
1422# (default is UTF-8).
1423# @keyparam verbose Use a true value to enable debugging output.
1424# (printed to standard output).
1425# @see Transport
1426
Fredrik Lundhb9056332001-07-11 17:42:21 +00001427class ServerProxy:
1428 """uri [,options] -> a logical connection to an XML-RPC server
1429
1430 uri is the connection point on the server, given as
1431 scheme://host/target.
1432
1433 The standard implementation always supports the "http" scheme. If
1434 SSL socket support is available (Python 2.0), it also supports
1435 "https".
1436
1437 If the target part and the slash preceding it are both omitted,
1438 "/RPC2" is assumed.
1439
1440 The following options can be given as keyword arguments:
1441
1442 transport: a transport factory
1443 encoding: the request encoding (default is UTF-8)
1444
1445 All 8-bit strings passed to the server proxy are assumed to use
1446 the given encoding.
1447 """
1448
Andrew M. Kuchlinga4c2b742003-04-25 00:26:51 +00001449 def __init__(self, uri, transport=None, encoding=None, verbose=0,
Skip Montanaro174dd222005-05-14 20:54:16 +00001450 allow_none=0, use_datetime=0):
Fredrik Lundhb9056332001-07-11 17:42:21 +00001451 # establish a "logical" server connection
1452
1453 # get the url
Fredrik Lundhc4c062f2001-09-10 19:45:02 +00001454 import urllib
Fredrik Lundhb9056332001-07-11 17:42:21 +00001455 type, uri = urllib.splittype(uri)
1456 if type not in ("http", "https"):
1457 raise IOError, "unsupported XML-RPC protocol"
1458 self.__host, self.__handler = urllib.splithost(uri)
1459 if not self.__handler:
1460 self.__handler = "/RPC2"
1461
1462 if transport is None:
1463 if type == "https":
Skip Montanaro174dd222005-05-14 20:54:16 +00001464 transport = SafeTransport(use_datetime=use_datetime)
Fredrik Lundhb9056332001-07-11 17:42:21 +00001465 else:
Skip Montanaro174dd222005-05-14 20:54:16 +00001466 transport = Transport(use_datetime=use_datetime)
Fredrik Lundhb9056332001-07-11 17:42:21 +00001467 self.__transport = transport
1468
1469 self.__encoding = encoding
1470 self.__verbose = verbose
Andrew M. Kuchlinga4c2b742003-04-25 00:26:51 +00001471 self.__allow_none = allow_none
Tim Petersc2659cf2003-05-12 20:19:37 +00001472
Fredrik Lundhb9056332001-07-11 17:42:21 +00001473 def __request(self, methodname, params):
1474 # call a method on the remote server
1475
Andrew M. Kuchlinga4c2b742003-04-25 00:26:51 +00001476 request = dumps(params, methodname, encoding=self.__encoding,
1477 allow_none=self.__allow_none)
Fredrik Lundhb9056332001-07-11 17:42:21 +00001478
1479 response = self.__transport.request(
1480 self.__host,
1481 self.__handler,
1482 request,
1483 verbose=self.__verbose
1484 )
1485
1486 if len(response) == 1:
1487 response = response[0]
1488
1489 return response
1490
1491 def __repr__(self):
1492 return (
Fredrik Lundh78eedce2001-08-23 20:04:33 +00001493 "<ServerProxy for %s%s>" %
Fredrik Lundhb9056332001-07-11 17:42:21 +00001494 (self.__host, self.__handler)
1495 )
1496
1497 __str__ = __repr__
Raymond Hettingercc523fc2003-11-02 09:47:05 +00001498
Fredrik Lundhb9056332001-07-11 17:42:21 +00001499 def __getattr__(self, name):
1500 # magic method dispatcher
1501 return _Method(self.__request, name)
1502
1503 # note: to call a remote object with an non-standard name, use
1504 # result getattr(server, "strange-python-name")(args)
1505
Fredrik Lundh78eedce2001-08-23 20:04:33 +00001506# compatibility
Fredrik Lundh3d9addd2002-06-27 21:36:21 +00001507
Fredrik Lundhb9056332001-07-11 17:42:21 +00001508Server = ServerProxy
1509
1510# --------------------------------------------------------------------
1511# test code
1512
1513if __name__ == "__main__":
1514
1515 # simple test program (from the XML-RPC specification)
1516
Fredrik Lundh78eedce2001-08-23 20:04:33 +00001517 # server = ServerProxy("http://localhost:8000") # local server
Martin v. Löwis12237b32004-08-22 16:04:50 +00001518 server = ServerProxy("http://time.xmlrpc.com/RPC2")
Fredrik Lundhb9056332001-07-11 17:42:21 +00001519
1520 print server
1521
1522 try:
Martin v. Löwis12237b32004-08-22 16:04:50 +00001523 print server.currentTime.getCurrentTime()
1524 except Error, v:
1525 print "ERROR", v
1526
1527 multi = MultiCall(server)
1528 multi.currentTime.getCurrentTime()
1529 multi.currentTime.getCurrentTime()
1530 try:
1531 for response in multi():
1532 print response
Fredrik Lundhb9056332001-07-11 17:42:21 +00001533 except Error, v:
1534 print "ERROR", v