blob: f0e4f8f0057958c3a0ff6311764fa15a4cb66862 [file] [log] [blame]
Fredrik Lundhb9056332001-07-11 17:42:21 +00001#
2# XML-RPC CLIENT LIBRARY
3# $Id$
4#
Fredrik Lundhc4c062f2001-09-10 19:45:02 +00005# an XML-RPC client interface for Python.
6#
7# the marshalling and response parser code can also be used to
8# implement XML-RPC servers.
9#
10# Notes:
Martin v. Löwis37af9862004-08-20 07:31:37 +000011# this version is designed to work with Python 2.1 or newer.
Fredrik Lundhc4c062f2001-09-10 19:45:02 +000012#
Fredrik Lundhb9056332001-07-11 17:42:21 +000013# History:
14# 1999-01-14 fl Created
15# 1999-01-15 fl Changed dateTime to use localtime
16# 1999-01-16 fl Added Binary/base64 element, default to RPC2 service
17# 1999-01-19 fl Fixed array data element (from Skip Montanaro)
18# 1999-01-21 fl Fixed dateTime constructor, etc.
19# 1999-02-02 fl Added fault handling, handle empty sequences, etc.
20# 1999-02-10 fl Fixed problem with empty responses (from Skip Montanaro)
21# 1999-06-20 fl Speed improvements, pluggable parsers/transports (0.9.8)
22# 2000-11-28 fl Changed boolean to check the truth value of its argument
23# 2001-02-24 fl Added encoding/Unicode/SafeTransport patches
24# 2001-02-26 fl Added compare support to wrappers (0.9.9/1.0b1)
25# 2001-03-28 fl Make sure response tuple is a singleton
26# 2001-03-29 fl Don't require empty params element (from Nicholas Riley)
Fredrik Lundh78eedce2001-08-23 20:04:33 +000027# 2001-06-10 fl Folded in _xmlrpclib accelerator support (1.0b2)
28# 2001-08-20 fl Base xmlrpclib.Error on built-in Exception (from Paul Prescod)
Fredrik Lundhc4c062f2001-09-10 19:45:02 +000029# 2001-09-03 fl Allow Transport subclass to override getparser
30# 2001-09-10 fl Lazy import of urllib, cgi, xmllib (20x import speedup)
Fredrik Lundh1538c232001-10-01 19:42:03 +000031# 2001-10-01 fl Remove containers from memo cache when done with them
32# 2001-10-01 fl Use faster escape method (80% dumps speedup)
Fredrik Lundh3d9addd2002-06-27 21:36:21 +000033# 2001-10-02 fl More dumps microtuning
34# 2001-10-04 fl Make sure import expat gets a parser (from Guido van Rossum)
Skip Montanaro5e9c71b2001-10-10 15:56:34 +000035# 2001-10-10 sm Allow long ints to be passed as ints if they don't overflow
Fredrik Lundh3d9addd2002-06-27 21:36:21 +000036# 2001-10-17 sm Test for int and long overflow (allows use on 64-bit systems)
Fredrik Lundhb6ab93f2001-12-19 21:40:04 +000037# 2001-11-12 fl Use repr() to marshal doubles (from Paul Felix)
Fredrik Lundh3d9addd2002-06-27 21:36:21 +000038# 2002-03-17 fl Avoid buffered read when possible (from James Rucker)
39# 2002-04-07 fl Added pythondoc comments
40# 2002-04-16 fl Added __str__ methods to datetime/binary wrappers
41# 2002-05-15 fl Added error constants (from Andrew Kuchling)
42# 2002-06-27 fl Merged with Python CVS version
Fredrik Lundh1303c7c2002-10-22 18:23:00 +000043# 2002-10-22 fl Added basic authentication (based on code from Phillip Eby)
Martin v. Löwis541342f2003-07-12 07:53:04 +000044# 2003-01-22 sm Add support for the bool type
45# 2003-02-27 gvr Remove apply calls
46# 2003-04-24 sm Use cStringIO if available
47# 2003-04-25 ak Add support for nil
48# 2003-06-15 gn Add support for time.struct_time
49# 2003-07-12 gp Correct marshalling of Faults
Martin v. Löwis45394c22003-10-31 13:49:36 +000050# 2003-10-31 mvl Add multicall support
Martin v. Löwis37af9862004-08-20 07:31:37 +000051# 2004-08-20 mvl Bump minimum supported Python version to 2.1
Fredrik Lundhb9056332001-07-11 17:42:21 +000052#
Fredrik Lundh3d9addd2002-06-27 21:36:21 +000053# Copyright (c) 1999-2002 by Secret Labs AB.
54# Copyright (c) 1999-2002 by Fredrik Lundh.
Fredrik Lundhb9056332001-07-11 17:42:21 +000055#
56# info@pythonware.com
57# http://www.pythonware.com
58#
59# --------------------------------------------------------------------
60# The XML-RPC client interface is
61#
Fredrik Lundh3d9addd2002-06-27 21:36:21 +000062# Copyright (c) 1999-2002 by Secret Labs AB
63# Copyright (c) 1999-2002 by Fredrik Lundh
Fredrik Lundhb9056332001-07-11 17:42:21 +000064#
65# By obtaining, using, and/or copying this software and/or its
66# associated documentation, you agree that you have read, understood,
67# and will comply with the following terms and conditions:
68#
69# Permission to use, copy, modify, and distribute this software and
70# its associated documentation for any purpose and without fee is
71# hereby granted, provided that the above copyright notice appears in
72# all copies, and that both that copyright notice and this permission
73# notice appear in supporting documentation, and that the name of
74# Secret Labs AB or the author not be used in advertising or publicity
75# pertaining to distribution of the software without specific, written
76# prior permission.
77#
78# SECRET LABS AB AND THE AUTHOR DISCLAIMS ALL WARRANTIES WITH REGARD
79# TO THIS SOFTWARE, INCLUDING ALL IMPLIED WARRANTIES OF MERCHANT-
80# ABILITY AND FITNESS. IN NO EVENT SHALL SECRET LABS AB OR THE AUTHOR
81# BE LIABLE FOR ANY SPECIAL, INDIRECT OR CONSEQUENTIAL DAMAGES OR ANY
82# DAMAGES WHATSOEVER RESULTING FROM LOSS OF USE, DATA OR PROFITS,
83# WHETHER IN AN ACTION OF CONTRACT, NEGLIGENCE OR OTHER TORTIOUS
84# ACTION, ARISING OUT OF OR IN CONNECTION WITH THE USE OR PERFORMANCE
85# OF THIS SOFTWARE.
86# --------------------------------------------------------------------
87
88#
Fredrik Lundh3d9addd2002-06-27 21:36:21 +000089# things to look into some day:
Fredrik Lundhb9056332001-07-11 17:42:21 +000090
Fredrik Lundh3d9addd2002-06-27 21:36:21 +000091# TODO: sort out True/False/boolean issues for Python 2.3
Fredrik Lundhb9056332001-07-11 17:42:21 +000092
Fred Drake1b410792001-09-04 18:55:03 +000093"""
94An XML-RPC client interface for Python.
95
96The marshalling and response parser code can also be used to
97implement XML-RPC servers.
98
Fred Drake1b410792001-09-04 18:55:03 +000099Exported exceptions:
100
Fredrik Lundhc4c062f2001-09-10 19:45:02 +0000101 Error Base class for client errors
102 ProtocolError Indicates an HTTP protocol error
103 ResponseError Indicates a broken response package
104 Fault Indicates an XML-RPC fault package
Fred Drake1b410792001-09-04 18:55:03 +0000105
106Exported classes:
107
Fredrik Lundhc4c062f2001-09-10 19:45:02 +0000108 ServerProxy Represents a logical connection to an XML-RPC server
Fred Drake1b410792001-09-04 18:55:03 +0000109
Raymond Hettingercc523fc2003-11-02 09:47:05 +0000110 MultiCall Executor of boxcared xmlrpc requests
Fredrik Lundhc4c062f2001-09-10 19:45:02 +0000111 Boolean boolean wrapper to generate a "boolean" XML-RPC value
112 DateTime dateTime wrapper for an ISO 8601 string or time tuple or
113 localtime integer value to generate a "dateTime.iso8601"
114 XML-RPC value
115 Binary binary data wrapper
Fred Drake1b410792001-09-04 18:55:03 +0000116
Fredrik Lundhc4c062f2001-09-10 19:45:02 +0000117 SlowParser Slow but safe standard parser (based on xmllib)
118 Marshaller Generate an XML-RPC params chunk from a Python data structure
119 Unmarshaller Unmarshal an XML-RPC response from incoming XML event message
120 Transport Handles an HTTP transaction to an XML-RPC server
121 SafeTransport Handles an HTTPS transaction to an XML-RPC server
Fred Drake1b410792001-09-04 18:55:03 +0000122
123Exported constants:
124
Fredrik Lundhc4c062f2001-09-10 19:45:02 +0000125 True
126 False
Fred Drake1b410792001-09-04 18:55:03 +0000127
128Exported functions:
129
Fredrik Lundhc4c062f2001-09-10 19:45:02 +0000130 boolean Convert any Python value to an XML-RPC boolean
131 getparser Create instance of the fastest available parser & attach
132 to an unmarshalling object
133 dumps Convert an argument tuple or a Fault instance to an XML-RPC
134 request (or response, if the methodresponse option is used).
135 loads Convert an XML-RPC packet to unmarshalled data plus a method
136 name (None if not present).
Fred Drake1b410792001-09-04 18:55:03 +0000137"""
138
Neal Norwitzff113342007-04-17 08:42:15 +0000139import re, time, operator
Fredrik Lundh1538c232001-10-01 19:42:03 +0000140
Raymond Hettingerfe59dc12005-02-07 15:28:45 +0000141from types import *
Fredrik Lundhb9056332001-07-11 17:42:21 +0000142
Fredrik Lundh3d9addd2002-06-27 21:36:21 +0000143# --------------------------------------------------------------------
144# Internal stuff
145
Fredrik Lundhb9056332001-07-11 17:42:21 +0000146try:
Fred Drakeba613c32005-02-10 18:33:30 +0000147 import datetime
148except ImportError:
149 datetime = None
150
151try:
Skip Montanaro9a7c96a2003-01-22 18:17:25 +0000152 _bool_is_builtin = False.__class__.__name__ == "bool"
153except NameError:
154 _bool_is_builtin = 0
155
Fredrik Lundhb9056332001-07-11 17:42:21 +0000156def _decode(data, encoding, is8bit=re.compile("[\x80-\xff]").search):
157 # decode non-ascii string (if possible)
Guido van Rossum54ad5232007-05-27 09:17:48 +0000158 if encoding and is8bit(data):
Guido van Rossumef87d6e2007-05-02 19:09:54 +0000159 data = str(data, encoding)
Fredrik Lundhb9056332001-07-11 17:42:21 +0000160 return data
161
Neal Norwitzff113342007-04-17 08:42:15 +0000162def escape(s):
163 s = s.replace("&", "&")
164 s = s.replace("<", "&lt;")
165 return s.replace(">", "&gt;",)
Fredrik Lundh1538c232001-10-01 19:42:03 +0000166
Guido van Rossum54ad5232007-05-27 09:17:48 +0000167def _stringify(string):
168 # convert to 7-bit ascii if possible
169 try:
170 return string.encode("ascii")
171 except UnicodeError:
Fredrik Lundhb9056332001-07-11 17:42:21 +0000172 return string
173
Fredrik Lundh3d9addd2002-06-27 21:36:21 +0000174__version__ = "1.0.1"
175
176# xmlrpc integer limits
Guido van Rossume2a383d2007-01-15 16:59:06 +0000177MAXINT = 2**31-1
178MININT = -2**31
Fredrik Lundh3d9addd2002-06-27 21:36:21 +0000179
180# --------------------------------------------------------------------
181# Error constants (from Dan Libby's specification at
182# http://xmlrpc-epi.sourceforge.net/specs/rfc.fault_codes.php)
183
184# Ranges of errors
185PARSE_ERROR = -32700
186SERVER_ERROR = -32600
187APPLICATION_ERROR = -32500
188SYSTEM_ERROR = -32400
189TRANSPORT_ERROR = -32300
190
191# Specific errors
192NOT_WELLFORMED_ERROR = -32700
193UNSUPPORTED_ENCODING = -32701
194INVALID_ENCODING_CHAR = -32702
195INVALID_XMLRPC = -32600
196METHOD_NOT_FOUND = -32601
197INVALID_METHOD_PARAMS = -32602
198INTERNAL_ERROR = -32603
Fredrik Lundhb9056332001-07-11 17:42:21 +0000199
200# --------------------------------------------------------------------
201# Exceptions
202
Fredrik Lundh3d9addd2002-06-27 21:36:21 +0000203##
204# Base class for all kinds of client-side errors.
205
Fredrik Lundh78eedce2001-08-23 20:04:33 +0000206class Error(Exception):
Fred Drake1b410792001-09-04 18:55:03 +0000207 """Base class for client errors."""
Fredrik Lundhb0e8e9b2001-09-10 21:45:42 +0000208 def __str__(self):
209 return repr(self)
Fredrik Lundhb9056332001-07-11 17:42:21 +0000210
Fredrik Lundh3d9addd2002-06-27 21:36:21 +0000211##
212# Indicates an HTTP-level protocol error. This is raised by the HTTP
213# transport layer, if the server returns an error code other than 200
214# (OK).
215#
216# @param url The target URL.
217# @param errcode The HTTP error code.
218# @param errmsg The HTTP error message.
219# @param headers The HTTP header dictionary.
220
Fredrik Lundhb9056332001-07-11 17:42:21 +0000221class ProtocolError(Error):
Fred Drake1b410792001-09-04 18:55:03 +0000222 """Indicates an HTTP protocol error."""
Fredrik Lundhb9056332001-07-11 17:42:21 +0000223 def __init__(self, url, errcode, errmsg, headers):
Fredrik Lundhb0e8e9b2001-09-10 21:45:42 +0000224 Error.__init__(self)
Fredrik Lundhb9056332001-07-11 17:42:21 +0000225 self.url = url
226 self.errcode = errcode
227 self.errmsg = errmsg
228 self.headers = headers
229 def __repr__(self):
230 return (
231 "<ProtocolError for %s: %s %s>" %
232 (self.url, self.errcode, self.errmsg)
233 )
234
Fredrik Lundh3d9addd2002-06-27 21:36:21 +0000235##
236# Indicates a broken XML-RPC response package. This exception is
237# raised by the unmarshalling layer, if the XML-RPC response is
238# malformed.
239
Fredrik Lundhb9056332001-07-11 17:42:21 +0000240class ResponseError(Error):
Fredrik Lundhc4c062f2001-09-10 19:45:02 +0000241 """Indicates a broken response package."""
Fredrik Lundhb9056332001-07-11 17:42:21 +0000242 pass
243
Fredrik Lundh3d9addd2002-06-27 21:36:21 +0000244##
245# Indicates an XML-RPC fault response package. This exception is
246# raised by the unmarshalling layer, if the XML-RPC response contains
247# a fault string. This exception can also used as a class, to
248# generate a fault XML-RPC message.
249#
250# @param faultCode The XML-RPC fault code.
251# @param faultString The XML-RPC fault string.
252
Fredrik Lundhb9056332001-07-11 17:42:21 +0000253class Fault(Error):
Fredrik Lundhc4c062f2001-09-10 19:45:02 +0000254 """Indicates an XML-RPC fault package."""
Fredrik Lundhb9056332001-07-11 17:42:21 +0000255 def __init__(self, faultCode, faultString, **extra):
Fredrik Lundhb0e8e9b2001-09-10 21:45:42 +0000256 Error.__init__(self)
Fredrik Lundhb9056332001-07-11 17:42:21 +0000257 self.faultCode = faultCode
258 self.faultString = faultString
259 def __repr__(self):
260 return (
261 "<Fault %s: %s>" %
Fredrik Lundhb0e8e9b2001-09-10 21:45:42 +0000262 (self.faultCode, repr(self.faultString))
Fredrik Lundhb9056332001-07-11 17:42:21 +0000263 )
264
265# --------------------------------------------------------------------
266# Special values
267
Fredrik Lundh3d9addd2002-06-27 21:36:21 +0000268##
269# Wrapper for XML-RPC boolean values. Use the xmlrpclib.True and
270# xmlrpclib.False constants, or the xmlrpclib.boolean() function, to
271# generate boolean XML-RPC values.
272#
273# @param value A boolean value. Any true value is interpreted as True,
274# all other values are interpreted as False.
275
Guido van Rossum47b9ff62006-08-24 00:41:19 +0000276boolean = Boolean = bool
277# to avoid breaking code which references xmlrpclib.{True,False}
278True, False = True, False
Fredrik Lundh3d9addd2002-06-27 21:36:21 +0000279
280##
281# Wrapper for XML-RPC DateTime values. This converts a time value to
282# the format used by XML-RPC.
283# <p>
284# The value can be given as a string in the format
285# "yyyymmddThh:mm:ss", as a 9-item time tuple (as returned by
286# time.localtime()), or an integer value (as returned by time.time()).
287# The wrapper uses time.localtime() to convert an integer to a time
288# tuple.
289#
290# @param value The time, given as an ISO 8601 string, a time
291# tuple, or a integer time value.
Fredrik Lundhb9056332001-07-11 17:42:21 +0000292
Fredrik Lundhb9056332001-07-11 17:42:21 +0000293class DateTime:
Fredrik Lundhc4c062f2001-09-10 19:45:02 +0000294 """DateTime wrapper for an ISO 8601 string or time tuple or
295 localtime integer value to generate 'dateTime.iso8601' XML-RPC
296 value.
297 """
Fredrik Lundhb9056332001-07-11 17:42:21 +0000298
299 def __init__(self, value=0):
Guido van Rossum54ad5232007-05-27 09:17:48 +0000300 if not isinstance(value, basestring):
Fred Drakeba613c32005-02-10 18:33:30 +0000301 if datetime and isinstance(value, datetime.datetime):
302 self.value = value.strftime("%Y%m%dT%H:%M:%S")
303 return
Skip Montanaro174dd222005-05-14 20:54:16 +0000304 if datetime and isinstance(value, datetime.date):
305 self.value = value.strftime("%Y%m%dT%H:%M:%S")
306 return
307 if datetime and isinstance(value, datetime.time):
308 today = datetime.datetime.now().strftime("%Y%m%d")
309 self.value = value.strftime(today+"T%H:%M:%S")
310 return
311 if not isinstance(value, (TupleType, time.struct_time)):
Fredrik Lundhb9056332001-07-11 17:42:21 +0000312 if value == 0:
313 value = time.time()
314 value = time.localtime(value)
315 value = time.strftime("%Y%m%dT%H:%M:%S", value)
316 self.value = value
317
Guido van Rossum47b9ff62006-08-24 00:41:19 +0000318 def __eq__(self, other):
Fredrik Lundhb9056332001-07-11 17:42:21 +0000319 if isinstance(other, DateTime):
320 other = other.value
Guido van Rossum47b9ff62006-08-24 00:41:19 +0000321 return self.value == other
322
323 def __ne__(self, other):
324 if isinstance(other, DateTime):
325 other = other.value
326 return self.value != other
Fredrik Lundhb9056332001-07-11 17:42:21 +0000327
Fredrik Lundh3d9addd2002-06-27 21:36:21 +0000328 ##
329 # Get date/time value.
330 #
331 # @return Date/time value, as an ISO 8601 string.
332
333 def __str__(self):
334 return self.value
335
Fredrik Lundhb9056332001-07-11 17:42:21 +0000336 def __repr__(self):
Fredrik Lundh3d9addd2002-06-27 21:36:21 +0000337 return "<DateTime %s at %x>" % (repr(self.value), id(self))
Fredrik Lundhb9056332001-07-11 17:42:21 +0000338
339 def decode(self, data):
Neal Norwitzff113342007-04-17 08:42:15 +0000340 self.value = str(data).strip()
Fredrik Lundhb9056332001-07-11 17:42:21 +0000341
342 def encode(self, out):
343 out.write("<value><dateTime.iso8601>")
344 out.write(self.value)
345 out.write("</dateTime.iso8601></value>\n")
346
Fredrik Lundh3d9addd2002-06-27 21:36:21 +0000347def _datetime(data):
348 # decode xml element contents into a DateTime structure.
Fredrik Lundhb9056332001-07-11 17:42:21 +0000349 value = DateTime()
350 value.decode(data)
351 return value
352
Skip Montanaro174dd222005-05-14 20:54:16 +0000353def _datetime_type(data):
354 t = time.strptime(data, "%Y%m%dT%H:%M:%S")
355 return datetime.datetime(*tuple(t)[:6])
356
Fredrik Lundh3d9addd2002-06-27 21:36:21 +0000357##
358# Wrapper for binary data. This can be used to transport any kind
359# of binary data over XML-RPC, using BASE64 encoding.
360#
361# @param data An 8-bit string containing arbitrary data.
362
Skip Montanarobfcbfa72003-04-24 19:51:31 +0000363import base64
Guido van Rossum68937b42007-05-18 00:51:22 +0000364import io
Skip Montanarobfcbfa72003-04-24 19:51:31 +0000365
Fredrik Lundhb9056332001-07-11 17:42:21 +0000366class Binary:
Fred Drake1b410792001-09-04 18:55:03 +0000367 """Wrapper for binary data."""
Fredrik Lundhb9056332001-07-11 17:42:21 +0000368
369 def __init__(self, data=None):
370 self.data = data
371
Fredrik Lundh3d9addd2002-06-27 21:36:21 +0000372 ##
373 # Get buffer contents.
374 #
375 # @return Buffer contents, as an 8-bit string.
376
377 def __str__(self):
378 return self.data or ""
379
Guido van Rossum47b9ff62006-08-24 00:41:19 +0000380 def __eq__(self, other):
Fredrik Lundhb9056332001-07-11 17:42:21 +0000381 if isinstance(other, Binary):
382 other = other.data
Guido van Rossum47b9ff62006-08-24 00:41:19 +0000383 return self.data == other
384
385 def __ne__(self, other):
386 if isinstance(other, Binary):
387 other = other.data
388 return self.data != other
Fredrik Lundhb9056332001-07-11 17:42:21 +0000389
390 def decode(self, data):
Fredrik Lundhb9056332001-07-11 17:42:21 +0000391 self.data = base64.decodestring(data)
392
393 def encode(self, out):
Fredrik Lundhb9056332001-07-11 17:42:21 +0000394 out.write("<value><base64>\n")
Guido van Rossum68937b42007-05-18 00:51:22 +0000395 base64.encode(io.StringIO(self.data), out)
Fredrik Lundhb9056332001-07-11 17:42:21 +0000396 out.write("</base64></value>\n")
397
Fredrik Lundh3d9addd2002-06-27 21:36:21 +0000398def _binary(data):
399 # decode xml element contents into a Binary structure
Fredrik Lundhb9056332001-07-11 17:42:21 +0000400 value = Binary()
401 value.decode(data)
402 return value
403
Skip Montanaro9a7c96a2003-01-22 18:17:25 +0000404WRAPPERS = (DateTime, Binary)
405if not _bool_is_builtin:
406 WRAPPERS = WRAPPERS + (Boolean,)
Fredrik Lundhb9056332001-07-11 17:42:21 +0000407
408# --------------------------------------------------------------------
409# XML parsers
410
411try:
Fredrik Lundh7069c312004-10-13 06:48:37 +0000412 # optional xmlrpclib accelerator
Fredrik Lundhb9056332001-07-11 17:42:21 +0000413 import _xmlrpclib
414 FastParser = _xmlrpclib.Parser
415 FastUnmarshaller = _xmlrpclib.Unmarshaller
416except (AttributeError, ImportError):
417 FastParser = FastUnmarshaller = None
418
Fredrik Lundh3d9addd2002-06-27 21:36:21 +0000419try:
420 import _xmlrpclib
421 FastMarshaller = _xmlrpclib.Marshaller
422except (AttributeError, ImportError):
423 FastMarshaller = None
424
Fredrik Lundhb9056332001-07-11 17:42:21 +0000425#
426# the SGMLOP parser is about 15x faster than Python's builtin
427# XML parser. SGMLOP sources can be downloaded from:
428#
429# http://www.pythonware.com/products/xml/sgmlop.htm
430#
431
432try:
433 import sgmlop
434 if not hasattr(sgmlop, "XMLParser"):
435 raise ImportError
436except ImportError:
437 SgmlopParser = None # sgmlop accelerator not available
438else:
439 class SgmlopParser:
440 def __init__(self, target):
441
442 # setup callbacks
443 self.finish_starttag = target.start
444 self.finish_endtag = target.end
445 self.handle_data = target.data
446 self.handle_xml = target.xml
447
448 # activate parser
449 self.parser = sgmlop.XMLParser()
450 self.parser.register(self)
451 self.feed = self.parser.feed
452 self.entity = {
453 "amp": "&", "gt": ">", "lt": "<",
454 "apos": "'", "quot": '"'
455 }
456
457 def close(self):
458 try:
459 self.parser.close()
460 finally:
461 self.parser = self.feed = None # nuke circular reference
462
463 def handle_proc(self, tag, attr):
464 m = re.search("encoding\s*=\s*['\"]([^\"']+)[\"']", attr)
465 if m:
466 self.handle_xml(m.group(1), 1)
467
468 def handle_entityref(self, entity):
469 # <string> entity
470 try:
471 self.handle_data(self.entity[entity])
472 except KeyError:
473 self.handle_data("&%s;" % entity)
474
475try:
476 from xml.parsers import expat
Guido van Rossumb8551342001-10-02 18:33:11 +0000477 if not hasattr(expat, "ParserCreate"):
Fredrik Lundh3d9addd2002-06-27 21:36:21 +0000478 raise ImportError
Fredrik Lundhb9056332001-07-11 17:42:21 +0000479except ImportError:
Fredrik Lundh3d9addd2002-06-27 21:36:21 +0000480 ExpatParser = None # expat not available
Fredrik Lundhb9056332001-07-11 17:42:21 +0000481else:
482 class ExpatParser:
Fredrik Lundh3d9addd2002-06-27 21:36:21 +0000483 # fast expat parser for Python 2.0 and later. this is about
484 # 50% slower than sgmlop, on roundtrip testing
Fredrik Lundhb9056332001-07-11 17:42:21 +0000485 def __init__(self, target):
486 self._parser = parser = expat.ParserCreate(None, None)
487 self._target = target
488 parser.StartElementHandler = target.start
489 parser.EndElementHandler = target.end
490 parser.CharacterDataHandler = target.data
491 encoding = None
492 if not parser.returns_unicode:
493 encoding = "utf-8"
494 target.xml(encoding, None)
495
496 def feed(self, data):
497 self._parser.Parse(data, 0)
498
499 def close(self):
500 self._parser.Parse("", 1) # end of data
501 del self._target, self._parser # get rid of circular references
502
Fredrik Lundhb9056332001-07-11 17:42:21 +0000503# --------------------------------------------------------------------
504# XML-RPC marshalling and unmarshalling code
505
Fredrik Lundh3d9addd2002-06-27 21:36:21 +0000506##
507# XML-RPC marshaller.
508#
509# @param encoding Default encoding for 8-bit strings. The default
510# value is None (interpreted as UTF-8).
511# @see dumps
512
Fredrik Lundhb9056332001-07-11 17:42:21 +0000513class Marshaller:
Fred Drake1b410792001-09-04 18:55:03 +0000514 """Generate an XML-RPC params chunk from a Python data structure.
Fredrik Lundhb9056332001-07-11 17:42:21 +0000515
Fredrik Lundhc4c062f2001-09-10 19:45:02 +0000516 Create a Marshaller instance for each set of parameters, and use
517 the "dumps" method to convert your data (represented as a tuple)
518 to an XML-RPC params chunk. To write a fault response, pass a
519 Fault instance instead. You may prefer to use the "dumps" module
520 function for this purpose.
Fred Drake1b410792001-09-04 18:55:03 +0000521 """
Fredrik Lundhb9056332001-07-11 17:42:21 +0000522
523 # by the way, if you don't understand what's going on in here,
524 # that's perfectly ok.
525
Andrew M. Kuchlinga4c2b742003-04-25 00:26:51 +0000526 def __init__(self, encoding=None, allow_none=0):
Fredrik Lundhb9056332001-07-11 17:42:21 +0000527 self.memo = {}
528 self.data = None
529 self.encoding = encoding
Andrew M. Kuchlinga4c2b742003-04-25 00:26:51 +0000530 self.allow_none = allow_none
Tim Petersc2659cf2003-05-12 20:19:37 +0000531
Fredrik Lundhb9056332001-07-11 17:42:21 +0000532 dispatch = {}
533
534 def dumps(self, values):
Fredrik Lundh3d9addd2002-06-27 21:36:21 +0000535 out = []
536 write = out.append
537 dump = self.__dump
Fredrik Lundhb9056332001-07-11 17:42:21 +0000538 if isinstance(values, Fault):
539 # fault instance
540 write("<fault>\n")
Martin v. Löwis541342f2003-07-12 07:53:04 +0000541 dump({'faultCode': values.faultCode,
542 'faultString': values.faultString},
543 write)
Fredrik Lundhb9056332001-07-11 17:42:21 +0000544 write("</fault>\n")
545 else:
546 # parameter block
Fredrik Lundhc266bb02001-08-23 20:13:08 +0000547 # FIXME: the xml-rpc specification allows us to leave out
548 # the entire <params> block if there are no parameters.
549 # however, changing this may break older code (including
550 # old versions of xmlrpclib.py), so this is better left as
551 # is for now. See @XMLRPC3 for more information. /F
Fredrik Lundhb9056332001-07-11 17:42:21 +0000552 write("<params>\n")
553 for v in values:
554 write("<param>\n")
Fredrik Lundh3d9addd2002-06-27 21:36:21 +0000555 dump(v, write)
Fredrik Lundhb9056332001-07-11 17:42:21 +0000556 write("</param>\n")
557 write("</params>\n")
Neal Norwitzff113342007-04-17 08:42:15 +0000558 result = "".join(out)
Fredrik Lundhb9056332001-07-11 17:42:21 +0000559 return result
560
Fredrik Lundh3d9addd2002-06-27 21:36:21 +0000561 def __dump(self, value, write):
Fredrik Lundhb9056332001-07-11 17:42:21 +0000562 try:
563 f = self.dispatch[type(value)]
564 except KeyError:
Thomas Wouters89f507f2006-12-13 04:49:30 +0000565 # check if this object can be marshalled as a structure
566 try:
567 value.__dict__
568 except:
569 raise TypeError, "cannot marshal %s objects" % type(value)
570 # check if this class is a sub-class of a basic type,
571 # because we don't know how to marshal these types
572 # (e.g. a string sub-class)
573 for type_ in type(value).__mro__:
574 if type_ in self.dispatch.keys():
575 raise TypeError, "cannot marshal %s objects" % type(value)
576 # XXX(twouters): using "_arbitrary_instance" as key as a quick-fix
577 # for the p3yk merge, this should probably be fixed more neatly.
578 f = self.dispatch["_arbitrary_instance"]
579 f(self, value, write)
Fredrik Lundhb9056332001-07-11 17:42:21 +0000580
Andrew M. Kuchlinga4c2b742003-04-25 00:26:51 +0000581 def dump_nil (self, value, write):
582 if not self.allow_none:
583 raise TypeError, "cannot marshal None unless allow_none is enabled"
584 write("<value><nil/></value>")
Raymond Hettingerfe59dc12005-02-07 15:28:45 +0000585 dispatch[NoneType] = dump_nil
Tim Petersc2659cf2003-05-12 20:19:37 +0000586
Fredrik Lundh3d9addd2002-06-27 21:36:21 +0000587 def dump_int(self, value, write):
Skip Montanaro5449e082001-10-17 22:53:33 +0000588 # in case ints are > 32 bits
589 if value > MAXINT or value < MININT:
590 raise OverflowError, "int exceeds XML-RPC limits"
Fredrik Lundh3d9addd2002-06-27 21:36:21 +0000591 write("<value><int>")
592 write(str(value))
593 write("</int></value>\n")
Raymond Hettingerfe59dc12005-02-07 15:28:45 +0000594 dispatch[IntType] = dump_int
Fredrik Lundhb9056332001-07-11 17:42:21 +0000595
Skip Montanaro9a7c96a2003-01-22 18:17:25 +0000596 if _bool_is_builtin:
597 def dump_bool(self, value, write):
598 write("<value><boolean>")
599 write(value and "1" or "0")
600 write("</boolean></value>\n")
601 dispatch[bool] = dump_bool
602
Fredrik Lundh3d9addd2002-06-27 21:36:21 +0000603 def dump_long(self, value, write):
Skip Montanaro5449e082001-10-17 22:53:33 +0000604 if value > MAXINT or value < MININT:
605 raise OverflowError, "long int exceeds XML-RPC limits"
Fredrik Lundh3d9addd2002-06-27 21:36:21 +0000606 write("<value><int>")
607 write(str(int(value)))
608 write("</int></value>\n")
Raymond Hettingerfe59dc12005-02-07 15:28:45 +0000609 dispatch[LongType] = dump_long
Skip Montanaro5e9c71b2001-10-10 15:56:34 +0000610
Fredrik Lundh3d9addd2002-06-27 21:36:21 +0000611 def dump_double(self, value, write):
612 write("<value><double>")
613 write(repr(value))
614 write("</double></value>\n")
Raymond Hettingerfe59dc12005-02-07 15:28:45 +0000615 dispatch[FloatType] = dump_double
Fredrik Lundhb9056332001-07-11 17:42:21 +0000616
Fredrik Lundh3d9addd2002-06-27 21:36:21 +0000617 def dump_string(self, value, write, escape=escape):
618 write("<value><string>")
619 write(escape(value))
620 write("</string></value>\n")
Guido van Rossum54ad5232007-05-27 09:17:48 +0000621 dispatch[str8] = dump_string
Fredrik Lundhb9056332001-07-11 17:42:21 +0000622
Guido van Rossum54ad5232007-05-27 09:17:48 +0000623 def dump_unicode(self, value, write, escape=escape):
624 value = value.encode(self.encoding)
625 write("<value><string>")
626 write(escape(value))
627 write("</string></value>\n")
628 dispatch[str] = dump_unicode
Fredrik Lundhb9056332001-07-11 17:42:21 +0000629
Fredrik Lundh3d9addd2002-06-27 21:36:21 +0000630 def dump_array(self, value, write):
631 i = id(value)
Guido van Rossume2b70bc2006-08-18 22:13:04 +0000632 if i in self.memo:
Fredrik Lundh3d9addd2002-06-27 21:36:21 +0000633 raise TypeError, "cannot marshal recursive sequences"
634 self.memo[i] = None
Fredrik Lundh1538c232001-10-01 19:42:03 +0000635 dump = self.__dump
Fredrik Lundhb9056332001-07-11 17:42:21 +0000636 write("<value><array><data>\n")
637 for v in value:
Fredrik Lundh3d9addd2002-06-27 21:36:21 +0000638 dump(v, write)
Fredrik Lundhb9056332001-07-11 17:42:21 +0000639 write("</data></array></value>\n")
Fredrik Lundh3d9addd2002-06-27 21:36:21 +0000640 del self.memo[i]
Raymond Hettingerfe59dc12005-02-07 15:28:45 +0000641 dispatch[TupleType] = dump_array
642 dispatch[ListType] = dump_array
Fredrik Lundhb9056332001-07-11 17:42:21 +0000643
Fredrik Lundh3d9addd2002-06-27 21:36:21 +0000644 def dump_struct(self, value, write, escape=escape):
645 i = id(value)
Guido van Rossume2b70bc2006-08-18 22:13:04 +0000646 if i in self.memo:
Fredrik Lundh3d9addd2002-06-27 21:36:21 +0000647 raise TypeError, "cannot marshal recursive dictionaries"
648 self.memo[i] = None
Fredrik Lundh1538c232001-10-01 19:42:03 +0000649 dump = self.__dump
Fredrik Lundhb9056332001-07-11 17:42:21 +0000650 write("<value><struct>\n")
Andrew M. Kuchling5962f452004-06-05 12:35:58 +0000651 for k, v in value.items():
Fredrik Lundhb9056332001-07-11 17:42:21 +0000652 write("<member>\n")
Guido van Rossum54ad5232007-05-27 09:17:48 +0000653 if isinstance(k, basestring):
654 k = k.encode(self.encoding)
655 else:
656 raise TypeError, "dictionary key must be string"
Fredrik Lundh1538c232001-10-01 19:42:03 +0000657 write("<name>%s</name>\n" % escape(k))
Andrew M. Kuchling5962f452004-06-05 12:35:58 +0000658 dump(v, write)
Fredrik Lundhb9056332001-07-11 17:42:21 +0000659 write("</member>\n")
660 write("</struct></value>\n")
Fredrik Lundh3d9addd2002-06-27 21:36:21 +0000661 del self.memo[i]
Raymond Hettingerfe59dc12005-02-07 15:28:45 +0000662 dispatch[DictType] = dump_struct
Fredrik Lundhb9056332001-07-11 17:42:21 +0000663
Fred Drakeba613c32005-02-10 18:33:30 +0000664 if datetime:
665 def dump_datetime(self, value, write):
666 write("<value><dateTime.iso8601>")
667 write(value.strftime("%Y%m%dT%H:%M:%S"))
668 write("</dateTime.iso8601></value>\n")
669 dispatch[datetime.datetime] = dump_datetime
670
Skip Montanaro174dd222005-05-14 20:54:16 +0000671 def dump_date(self, value, write):
672 write("<value><dateTime.iso8601>")
673 write(value.strftime("%Y%m%dT00:00:00"))
674 write("</dateTime.iso8601></value>\n")
675 dispatch[datetime.date] = dump_date
676
677 def dump_time(self, value, write):
678 write("<value><dateTime.iso8601>")
679 write(datetime.datetime.now().date().strftime("%Y%m%dT"))
680 write(value.strftime("%H:%M:%S"))
681 write("</dateTime.iso8601></value>\n")
682 dispatch[datetime.time] = dump_time
683
Fredrik Lundh3d9addd2002-06-27 21:36:21 +0000684 def dump_instance(self, value, write):
Fredrik Lundhb9056332001-07-11 17:42:21 +0000685 # check for special wrappers
686 if value.__class__ in WRAPPERS:
Fredrik Lundh3d9addd2002-06-27 21:36:21 +0000687 self.write = write
Fredrik Lundhb9056332001-07-11 17:42:21 +0000688 value.encode(self)
Fredrik Lundh3d9addd2002-06-27 21:36:21 +0000689 del self.write
Fredrik Lundhb9056332001-07-11 17:42:21 +0000690 else:
691 # store instance attributes as a struct (really?)
Fredrik Lundh3d9addd2002-06-27 21:36:21 +0000692 self.dump_struct(value.__dict__, write)
Guido van Rossume4dea982006-04-21 13:45:00 +0000693 dispatch[DateTime] = dump_instance
694 dispatch[Binary] = dump_instance
Thomas Wouters89f507f2006-12-13 04:49:30 +0000695 # XXX(twouters): using "_arbitrary_instance" as key as a quick-fix
696 # for the p3yk merge, this should probably be fixed more neatly.
697 dispatch["_arbitrary_instance"] = dump_instance
Fredrik Lundhb9056332001-07-11 17:42:21 +0000698
Fredrik Lundh3d9addd2002-06-27 21:36:21 +0000699##
700# XML-RPC unmarshaller.
701#
702# @see loads
703
Fredrik Lundhb9056332001-07-11 17:42:21 +0000704class Unmarshaller:
Fred Drake1b410792001-09-04 18:55:03 +0000705 """Unmarshal an XML-RPC response, based on incoming XML event
Fredrik Lundhb0e8e9b2001-09-10 21:45:42 +0000706 messages (start, data, end). Call close() to get the resulting
Fred Drake1b410792001-09-04 18:55:03 +0000707 data structure.
Fredrik Lundhb9056332001-07-11 17:42:21 +0000708
Fredrik Lundhb0e8e9b2001-09-10 21:45:42 +0000709 Note that this reader is fairly tolerant, and gladly accepts bogus
710 XML-RPC data without complaining (but not bogus XML).
Fred Drake1b410792001-09-04 18:55:03 +0000711 """
Fredrik Lundhb9056332001-07-11 17:42:21 +0000712
713 # and again, if you don't understand what's going on in here,
714 # that's perfectly ok.
715
Skip Montanaro174dd222005-05-14 20:54:16 +0000716 def __init__(self, use_datetime=0):
Fredrik Lundhb9056332001-07-11 17:42:21 +0000717 self._type = None
718 self._stack = []
719 self._marks = []
720 self._data = []
721 self._methodname = None
722 self._encoding = "utf-8"
723 self.append = self._stack.append
Skip Montanaro174dd222005-05-14 20:54:16 +0000724 self._use_datetime = use_datetime
725 if use_datetime and not datetime:
726 raise ValueError, "the datetime module is not available"
Fredrik Lundhb9056332001-07-11 17:42:21 +0000727
728 def close(self):
729 # return response tuple and target method
730 if self._type is None or self._marks:
731 raise ResponseError()
732 if self._type == "fault":
Guido van Rossum68468eb2003-02-27 20:14:51 +0000733 raise Fault(**self._stack[0])
Fredrik Lundhb9056332001-07-11 17:42:21 +0000734 return tuple(self._stack)
735
736 def getmethodname(self):
737 return self._methodname
738
739 #
740 # event handlers
741
742 def xml(self, encoding, standalone):
743 self._encoding = encoding
744 # FIXME: assert standalone == 1 ???
745
746 def start(self, tag, attrs):
747 # prepare to handle this element
748 if tag == "array" or tag == "struct":
749 self._marks.append(len(self._stack))
750 self._data = []
751 self._value = (tag == "value")
752
753 def data(self, text):
754 self._data.append(text)
755
Neal Norwitzff113342007-04-17 08:42:15 +0000756 def end(self, tag):
Fredrik Lundhb9056332001-07-11 17:42:21 +0000757 # call the appropriate end tag handler
758 try:
759 f = self.dispatch[tag]
760 except KeyError:
761 pass # unknown tag ?
762 else:
Neal Norwitzff113342007-04-17 08:42:15 +0000763 return f(self, "".join(self._data))
Fredrik Lundhb9056332001-07-11 17:42:21 +0000764
765 #
766 # accelerator support
767
768 def end_dispatch(self, tag, data):
769 # dispatch data
770 try:
771 f = self.dispatch[tag]
772 except KeyError:
773 pass # unknown tag ?
774 else:
775 return f(self, data)
776
777 #
778 # element decoders
779
780 dispatch = {}
781
Andrew M. Kuchlinga4c2b742003-04-25 00:26:51 +0000782 def end_nil (self, data):
783 self.append(None)
784 self._value = 0
785 dispatch["nil"] = end_nil
Tim Petersc2659cf2003-05-12 20:19:37 +0000786
Fredrik Lundh1538c232001-10-01 19:42:03 +0000787 def end_boolean(self, data):
Fredrik Lundhb9056332001-07-11 17:42:21 +0000788 if data == "0":
789 self.append(False)
790 elif data == "1":
791 self.append(True)
792 else:
793 raise TypeError, "bad boolean value"
794 self._value = 0
795 dispatch["boolean"] = end_boolean
796
Fredrik Lundh1538c232001-10-01 19:42:03 +0000797 def end_int(self, data):
798 self.append(int(data))
Fredrik Lundhb9056332001-07-11 17:42:21 +0000799 self._value = 0
800 dispatch["i4"] = end_int
801 dispatch["int"] = end_int
802
Fredrik Lundh1538c232001-10-01 19:42:03 +0000803 def end_double(self, data):
804 self.append(float(data))
Fredrik Lundhb9056332001-07-11 17:42:21 +0000805 self._value = 0
806 dispatch["double"] = end_double
807
Fredrik Lundh1538c232001-10-01 19:42:03 +0000808 def end_string(self, data):
Fredrik Lundhb9056332001-07-11 17:42:21 +0000809 if self._encoding:
810 data = _decode(data, self._encoding)
811 self.append(_stringify(data))
812 self._value = 0
813 dispatch["string"] = end_string
814 dispatch["name"] = end_string # struct keys are always strings
815
816 def end_array(self, data):
Raymond Hettinger46ac8eb2002-06-30 03:39:14 +0000817 mark = self._marks.pop()
Fredrik Lundhb9056332001-07-11 17:42:21 +0000818 # map arrays to Python lists
819 self._stack[mark:] = [self._stack[mark:]]
820 self._value = 0
821 dispatch["array"] = end_array
822
823 def end_struct(self, data):
Raymond Hettinger46ac8eb2002-06-30 03:39:14 +0000824 mark = self._marks.pop()
Fredrik Lundhb9056332001-07-11 17:42:21 +0000825 # map structs to Python dictionaries
826 dict = {}
827 items = self._stack[mark:]
828 for i in range(0, len(items), 2):
829 dict[_stringify(items[i])] = items[i+1]
830 self._stack[mark:] = [dict]
831 self._value = 0
832 dispatch["struct"] = end_struct
833
Fredrik Lundh1538c232001-10-01 19:42:03 +0000834 def end_base64(self, data):
Fredrik Lundhb9056332001-07-11 17:42:21 +0000835 value = Binary()
Fredrik Lundh1538c232001-10-01 19:42:03 +0000836 value.decode(data)
Fredrik Lundhb9056332001-07-11 17:42:21 +0000837 self.append(value)
838 self._value = 0
839 dispatch["base64"] = end_base64
840
Fredrik Lundh1538c232001-10-01 19:42:03 +0000841 def end_dateTime(self, data):
Fredrik Lundhb9056332001-07-11 17:42:21 +0000842 value = DateTime()
Fredrik Lundh1538c232001-10-01 19:42:03 +0000843 value.decode(data)
Skip Montanaro174dd222005-05-14 20:54:16 +0000844 if self._use_datetime:
845 value = _datetime_type(data)
Fredrik Lundhb9056332001-07-11 17:42:21 +0000846 self.append(value)
847 dispatch["dateTime.iso8601"] = end_dateTime
848
849 def end_value(self, data):
Fredrik Lundh3d9addd2002-06-27 21:36:21 +0000850 # if we stumble upon a value element with no internal
Fredrik Lundhb9056332001-07-11 17:42:21 +0000851 # elements, treat it as a string element
852 if self._value:
853 self.end_string(data)
854 dispatch["value"] = end_value
855
856 def end_params(self, data):
857 self._type = "params"
858 dispatch["params"] = end_params
859
860 def end_fault(self, data):
861 self._type = "fault"
862 dispatch["fault"] = end_fault
863
Fredrik Lundh1538c232001-10-01 19:42:03 +0000864 def end_methodName(self, data):
Fredrik Lundhb9056332001-07-11 17:42:21 +0000865 if self._encoding:
866 data = _decode(data, self._encoding)
867 self._methodname = data
868 self._type = "methodName" # no params
869 dispatch["methodName"] = end_methodName
870
Martin v. Löwis45394c22003-10-31 13:49:36 +0000871## Multicall support
872#
Fredrik Lundhb9056332001-07-11 17:42:21 +0000873
Martin v. Löwis45394c22003-10-31 13:49:36 +0000874class _MultiCallMethod:
875 # some lesser magic to store calls made to a MultiCall object
876 # for batch execution
877 def __init__(self, call_list, name):
878 self.__call_list = call_list
879 self.__name = name
880 def __getattr__(self, name):
881 return _MultiCallMethod(self.__call_list, "%s.%s" % (self.__name, name))
882 def __call__(self, *args):
883 self.__call_list.append((self.__name, args))
884
Martin v. Löwis12237b32004-08-22 16:04:50 +0000885class MultiCallIterator:
Martin v. Löwis45394c22003-10-31 13:49:36 +0000886 """Iterates over the results of a multicall. Exceptions are
887 thrown in response to xmlrpc faults."""
Raymond Hettingercc523fc2003-11-02 09:47:05 +0000888
Martin v. Löwis12237b32004-08-22 16:04:50 +0000889 def __init__(self, results):
890 self.results = results
891
892 def __getitem__(self, i):
893 item = self.results[i]
894 if type(item) == type({}):
895 raise Fault(item['faultCode'], item['faultString'])
896 elif type(item) == type([]):
897 return item[0]
Martin v. Löwis45394c22003-10-31 13:49:36 +0000898 else:
899 raise ValueError,\
900 "unexpected type in multicall result"
Raymond Hettingercc523fc2003-11-02 09:47:05 +0000901
Martin v. Löwis45394c22003-10-31 13:49:36 +0000902class MultiCall:
903 """server -> a object used to boxcar method calls
904
905 server should be a ServerProxy object.
906
907 Methods can be added to the MultiCall using normal
908 method call syntax e.g.:
909
910 multicall = MultiCall(server_proxy)
911 multicall.add(2,3)
912 multicall.get_address("Guido")
913
914 To execute the multicall, call the MultiCall object e.g.:
915
916 add_result, address = multicall()
917 """
Raymond Hettingercc523fc2003-11-02 09:47:05 +0000918
Martin v. Löwis45394c22003-10-31 13:49:36 +0000919 def __init__(self, server):
920 self.__server = server
921 self.__call_list = []
Raymond Hettingercc523fc2003-11-02 09:47:05 +0000922
Martin v. Löwis45394c22003-10-31 13:49:36 +0000923 def __repr__(self):
924 return "<MultiCall at %x>" % id(self)
Raymond Hettingercc523fc2003-11-02 09:47:05 +0000925
Martin v. Löwis45394c22003-10-31 13:49:36 +0000926 __str__ = __repr__
927
928 def __getattr__(self, name):
929 return _MultiCallMethod(self.__call_list, name)
930
931 def __call__(self):
932 marshalled_list = []
933 for name, args in self.__call_list:
934 marshalled_list.append({'methodName' : name, 'params' : args})
935
936 return MultiCallIterator(self.__server.system.multicall(marshalled_list))
Raymond Hettingercc523fc2003-11-02 09:47:05 +0000937
Fredrik Lundhb9056332001-07-11 17:42:21 +0000938# --------------------------------------------------------------------
939# convenience functions
940
Fredrik Lundh3d9addd2002-06-27 21:36:21 +0000941##
942# Create a parser object, and connect it to an unmarshalling instance.
943# This function picks the fastest available XML parser.
944#
945# return A (parser, unmarshaller) tuple.
946
Skip Montanaro174dd222005-05-14 20:54:16 +0000947def getparser(use_datetime=0):
Fredrik Lundhb9056332001-07-11 17:42:21 +0000948 """getparser() -> parser, unmarshaller
949
Fredrik Lundhb0e8e9b2001-09-10 21:45:42 +0000950 Create an instance of the fastest available parser, and attach it
951 to an unmarshalling object. Return both objects.
Fredrik Lundhb9056332001-07-11 17:42:21 +0000952 """
Skip Montanaro174dd222005-05-14 20:54:16 +0000953 if use_datetime and not datetime:
954 raise ValueError, "the datetime module is not available"
Fredrik Lundhb9056332001-07-11 17:42:21 +0000955 if FastParser and FastUnmarshaller:
Skip Montanaro174dd222005-05-14 20:54:16 +0000956 if use_datetime:
957 mkdatetime = _datetime_type
958 else:
959 mkdatetime = _datetime
960 target = FastUnmarshaller(True, False, _binary, mkdatetime, Fault)
Fredrik Lundhb9056332001-07-11 17:42:21 +0000961 parser = FastParser(target)
962 else:
Skip Montanaro174dd222005-05-14 20:54:16 +0000963 target = Unmarshaller(use_datetime=use_datetime)
Fredrik Lundhb9056332001-07-11 17:42:21 +0000964 if FastParser:
965 parser = FastParser(target)
966 elif SgmlopParser:
967 parser = SgmlopParser(target)
968 elif ExpatParser:
969 parser = ExpatParser(target)
970 else:
971 parser = SlowParser(target)
972 return parser, target
973
Fredrik Lundh3d9addd2002-06-27 21:36:21 +0000974##
975# Convert a Python tuple or a Fault instance to an XML-RPC packet.
976#
977# @def dumps(params, **options)
978# @param params A tuple or Fault instance.
979# @keyparam methodname If given, create a methodCall request for
980# this method name.
981# @keyparam methodresponse If given, create a methodResponse packet.
982# If used with a tuple, the tuple must be a singleton (that is,
983# it must contain exactly one element).
984# @keyparam encoding The packet encoding.
985# @return A string containing marshalled data.
986
Andrew M. Kuchlinga4c2b742003-04-25 00:26:51 +0000987def dumps(params, methodname=None, methodresponse=None, encoding=None,
988 allow_none=0):
Fredrik Lundhb9056332001-07-11 17:42:21 +0000989 """data [,options] -> marshalled data
990
991 Convert an argument tuple or a Fault instance to an XML-RPC
992 request (or response, if the methodresponse option is used).
993
Fredrik Lundhb0e8e9b2001-09-10 21:45:42 +0000994 In addition to the data object, the following options can be given
995 as keyword arguments:
Fredrik Lundhb9056332001-07-11 17:42:21 +0000996
997 methodname: the method name for a methodCall packet
998
999 methodresponse: true to create a methodResponse packet.
1000 If this option is used with a tuple, the tuple must be
1001 a singleton (i.e. it can contain only one element).
1002
1003 encoding: the packet encoding (default is UTF-8)
1004
1005 All 8-bit strings in the data structure are assumed to use the
1006 packet encoding. Unicode strings are automatically converted,
Fredrik Lundhb0e8e9b2001-09-10 21:45:42 +00001007 where necessary.
Fredrik Lundhb9056332001-07-11 17:42:21 +00001008 """
1009
Raymond Hettingerfe59dc12005-02-07 15:28:45 +00001010 assert isinstance(params, TupleType) or isinstance(params, Fault),\
Fredrik Lundhb9056332001-07-11 17:42:21 +00001011 "argument must be tuple or Fault instance"
1012
1013 if isinstance(params, Fault):
1014 methodresponse = 1
Raymond Hettingerfe59dc12005-02-07 15:28:45 +00001015 elif methodresponse and isinstance(params, TupleType):
Fredrik Lundhb9056332001-07-11 17:42:21 +00001016 assert len(params) == 1, "response tuple must be a singleton"
1017
Fredrik Lundh3d9addd2002-06-27 21:36:21 +00001018 if not encoding:
Fredrik Lundhb9056332001-07-11 17:42:21 +00001019 encoding = "utf-8"
1020
Fredrik Lundh3d9addd2002-06-27 21:36:21 +00001021 if FastMarshaller:
1022 m = FastMarshaller(encoding)
1023 else:
Andrew M. Kuchlinga4c2b742003-04-25 00:26:51 +00001024 m = Marshaller(encoding, allow_none)
Fredrik Lundh3d9addd2002-06-27 21:36:21 +00001025
Fredrik Lundhb9056332001-07-11 17:42:21 +00001026 data = m.dumps(params)
1027
1028 if encoding != "utf-8":
Fredrik Lundh3d9addd2002-06-27 21:36:21 +00001029 xmlheader = "<?xml version='1.0' encoding='%s'?>\n" % str(encoding)
Fredrik Lundhb9056332001-07-11 17:42:21 +00001030 else:
1031 xmlheader = "<?xml version='1.0'?>\n" # utf-8 is default
1032
1033 # standard XML-RPC wrappings
Fredrik Lundh3d9addd2002-06-27 21:36:21 +00001034 if methodname:
Fredrik Lundhb9056332001-07-11 17:42:21 +00001035 # a method call
Guido van Rossum54ad5232007-05-27 09:17:48 +00001036 if not isinstance(methodname, basestring):
Fredrik Lundhb9056332001-07-11 17:42:21 +00001037 methodname = methodname.encode(encoding)
1038 data = (
1039 xmlheader,
1040 "<methodCall>\n"
1041 "<methodName>", methodname, "</methodName>\n",
1042 data,
1043 "</methodCall>\n"
1044 )
1045 elif methodresponse:
1046 # a method response, or a fault structure
1047 data = (
1048 xmlheader,
1049 "<methodResponse>\n",
1050 data,
1051 "</methodResponse>\n"
1052 )
1053 else:
1054 return data # return as is
Neal Norwitzff113342007-04-17 08:42:15 +00001055 return "".join(data)
Fredrik Lundhb9056332001-07-11 17:42:21 +00001056
Fredrik Lundh3d9addd2002-06-27 21:36:21 +00001057##
1058# Convert an XML-RPC packet to a Python object. If the XML-RPC packet
1059# represents a fault condition, this function raises a Fault exception.
1060#
1061# @param data An XML-RPC packet, given as an 8-bit string.
Walter Dörwaldf0dfc7a2003-10-20 14:01:56 +00001062# @return A tuple containing the unpacked data, and the method name
Fredrik Lundh3d9addd2002-06-27 21:36:21 +00001063# (None if not present).
1064# @see Fault
1065
Skip Montanaro174dd222005-05-14 20:54:16 +00001066def loads(data, use_datetime=0):
Fredrik Lundhb9056332001-07-11 17:42:21 +00001067 """data -> unmarshalled data, method name
1068
1069 Convert an XML-RPC packet to unmarshalled data plus a method
1070 name (None if not present).
1071
1072 If the XML-RPC packet represents a fault condition, this function
1073 raises a Fault exception.
1074 """
Skip Montanaro174dd222005-05-14 20:54:16 +00001075 p, u = getparser(use_datetime=use_datetime)
Fredrik Lundhb9056332001-07-11 17:42:21 +00001076 p.feed(data)
1077 p.close()
1078 return u.close(), u.getmethodname()
1079
1080
1081# --------------------------------------------------------------------
1082# request dispatcher
1083
1084class _Method:
1085 # some magic to bind an XML-RPC method to an RPC server.
1086 # supports "nested" methods (e.g. examples.getStateName)
1087 def __init__(self, send, name):
1088 self.__send = send
1089 self.__name = name
1090 def __getattr__(self, name):
1091 return _Method(self.__send, "%s.%s" % (self.__name, name))
1092 def __call__(self, *args):
1093 return self.__send(self.__name, args)
1094
Fredrik Lundh3d9addd2002-06-27 21:36:21 +00001095##
1096# Standard transport class for XML-RPC over HTTP.
1097# <p>
1098# You can create custom transports by subclassing this method, and
1099# overriding selected methods.
Fredrik Lundhb9056332001-07-11 17:42:21 +00001100
1101class Transport:
Fredrik Lundhc4c062f2001-09-10 19:45:02 +00001102 """Handles an HTTP transaction to an XML-RPC server."""
Fredrik Lundhb9056332001-07-11 17:42:21 +00001103
1104 # client identifier (may be overridden)
1105 user_agent = "xmlrpclib.py/%s (by www.pythonware.com)" % __version__
1106
Skip Montanaro174dd222005-05-14 20:54:16 +00001107 def __init__(self, use_datetime=0):
1108 self._use_datetime = use_datetime
1109
Fredrik Lundh3d9addd2002-06-27 21:36:21 +00001110 ##
1111 # Send a complete request, and parse the response.
1112 #
1113 # @param host Target host.
1114 # @param handler Target PRC handler.
1115 # @param request_body XML-RPC request body.
1116 # @param verbose Debugging flag.
1117 # @return Parsed response.
1118
Fredrik Lundhb9056332001-07-11 17:42:21 +00001119 def request(self, host, handler, request_body, verbose=0):
1120 # issue XML-RPC request
1121
1122 h = self.make_connection(host)
1123 if verbose:
1124 h.set_debuglevel(1)
1125
1126 self.send_request(h, handler, request_body)
1127 self.send_host(h, host)
1128 self.send_user_agent(h)
1129 self.send_content(h, request_body)
1130
1131 errcode, errmsg, headers = h.getreply()
1132
1133 if errcode != 200:
1134 raise ProtocolError(
1135 host + handler,
1136 errcode, errmsg,
1137 headers
1138 )
1139
1140 self.verbose = verbose
1141
Fredrik Lundh3d9addd2002-06-27 21:36:21 +00001142 try:
1143 sock = h._conn.sock
1144 except AttributeError:
1145 sock = None
1146
1147 return self._parse_response(h.getfile(), sock)
1148
1149 ##
1150 # Create parser.
1151 #
1152 # @return A 2-tuple containing a parser and a unmarshaller.
Fredrik Lundhb9056332001-07-11 17:42:21 +00001153
Fredrik Lundhc4c062f2001-09-10 19:45:02 +00001154 def getparser(self):
1155 # get parser and unmarshaller
Skip Montanaro174dd222005-05-14 20:54:16 +00001156 return getparser(use_datetime=self._use_datetime)
Fredrik Lundhc4c062f2001-09-10 19:45:02 +00001157
Fredrik Lundh3d9addd2002-06-27 21:36:21 +00001158 ##
Fredrik Lundh1303c7c2002-10-22 18:23:00 +00001159 # Get authorization info from host parameter
1160 # Host may be a string, or a (host, x509-dict) tuple; if a string,
1161 # it is checked for a "user:pw@host" format, and a "Basic
1162 # Authentication" header is added if appropriate.
1163 #
1164 # @param host Host descriptor (URL or (URL, x509 info) tuple).
1165 # @return A 3-tuple containing (actual host, extra headers,
1166 # x509 info). The header and x509 fields may be None.
1167
1168 def get_host_info(self, host):
1169
1170 x509 = {}
Raymond Hettingerfe59dc12005-02-07 15:28:45 +00001171 if isinstance(host, TupleType):
Fredrik Lundh1303c7c2002-10-22 18:23:00 +00001172 host, x509 = host
1173
1174 import urllib
1175 auth, host = urllib.splituser(host)
1176
1177 if auth:
1178 import base64
Fredrik Lundh768c98b2002-11-01 17:14:16 +00001179 auth = base64.encodestring(urllib.unquote(auth))
Neal Norwitzff113342007-04-17 08:42:15 +00001180 auth = "".join(auth.split()) # get rid of whitespace
Fredrik Lundh1303c7c2002-10-22 18:23:00 +00001181 extra_headers = [
1182 ("Authorization", "Basic " + auth)
1183 ]
1184 else:
1185 extra_headers = None
1186
1187 return host, extra_headers, x509
1188
1189 ##
Fredrik Lundh3d9addd2002-06-27 21:36:21 +00001190 # Connect to server.
1191 #
1192 # @param host Target host.
1193 # @return A connection handle.
1194
Fredrik Lundhb9056332001-07-11 17:42:21 +00001195 def make_connection(self, host):
1196 # create a HTTP connection object from a host descriptor
1197 import httplib
Fredrik Lundh1303c7c2002-10-22 18:23:00 +00001198 host, extra_headers, x509 = self.get_host_info(host)
Fredrik Lundhb9056332001-07-11 17:42:21 +00001199 return httplib.HTTP(host)
1200
Fredrik Lundh3d9addd2002-06-27 21:36:21 +00001201 ##
1202 # Send request header.
1203 #
1204 # @param connection Connection handle.
1205 # @param handler Target RPC handler.
1206 # @param request_body XML-RPC body.
1207
Fredrik Lundhb9056332001-07-11 17:42:21 +00001208 def send_request(self, connection, handler, request_body):
1209 connection.putrequest("POST", handler)
1210
Fredrik Lundh3d9addd2002-06-27 21:36:21 +00001211 ##
1212 # Send host name.
1213 #
1214 # @param connection Connection handle.
1215 # @param host Host name.
1216
Fredrik Lundhb9056332001-07-11 17:42:21 +00001217 def send_host(self, connection, host):
Fredrik Lundh1303c7c2002-10-22 18:23:00 +00001218 host, extra_headers, x509 = self.get_host_info(host)
Fredrik Lundhb9056332001-07-11 17:42:21 +00001219 connection.putheader("Host", host)
Fredrik Lundh1303c7c2002-10-22 18:23:00 +00001220 if extra_headers:
Raymond Hettingerfe59dc12005-02-07 15:28:45 +00001221 if isinstance(extra_headers, DictType):
Fredrik Lundh1303c7c2002-10-22 18:23:00 +00001222 extra_headers = extra_headers.items()
1223 for key, value in extra_headers:
1224 connection.putheader(key, value)
Fredrik Lundhb9056332001-07-11 17:42:21 +00001225
Fredrik Lundh3d9addd2002-06-27 21:36:21 +00001226 ##
1227 # Send user-agent identifier.
1228 #
1229 # @param connection Connection handle.
1230
Fredrik Lundhb9056332001-07-11 17:42:21 +00001231 def send_user_agent(self, connection):
1232 connection.putheader("User-Agent", self.user_agent)
1233
Fredrik Lundh3d9addd2002-06-27 21:36:21 +00001234 ##
1235 # Send request body.
1236 #
1237 # @param connection Connection handle.
1238 # @param request_body XML-RPC request body.
1239
Fredrik Lundhb9056332001-07-11 17:42:21 +00001240 def send_content(self, connection, request_body):
1241 connection.putheader("Content-Type", "text/xml")
1242 connection.putheader("Content-Length", str(len(request_body)))
1243 connection.endheaders()
1244 if request_body:
1245 connection.send(request_body)
1246
Fredrik Lundh3d9addd2002-06-27 21:36:21 +00001247 ##
1248 # Parse response.
1249 #
1250 # @param file Stream.
1251 # @return Response tuple and target method.
1252
1253 def parse_response(self, file):
1254 # compatibility interface
1255 return self._parse_response(file, None)
1256
1257 ##
1258 # Parse response (alternate interface). This is similar to the
1259 # parse_response method, but also provides direct access to the
1260 # underlying socket object (where available).
1261 #
1262 # @param file Stream.
1263 # @param sock Socket handle (or None, if the socket object
1264 # could not be accessed).
1265 # @return Response tuple and target method.
1266
1267 def _parse_response(self, file, sock):
1268 # read response from input file/socket, and parse it
Fredrik Lundhb9056332001-07-11 17:42:21 +00001269
Fredrik Lundhc4c062f2001-09-10 19:45:02 +00001270 p, u = self.getparser()
Fredrik Lundhb9056332001-07-11 17:42:21 +00001271
1272 while 1:
Fredrik Lundh3d9addd2002-06-27 21:36:21 +00001273 if sock:
1274 response = sock.recv(1024)
1275 else:
1276 response = file.read(1024)
Fredrik Lundhb9056332001-07-11 17:42:21 +00001277 if not response:
1278 break
1279 if self.verbose:
Guido van Rossumbe19ed72007-02-09 05:37:30 +00001280 print("body:", repr(response))
Fredrik Lundhb9056332001-07-11 17:42:21 +00001281 p.feed(response)
1282
Fredrik Lundh3d9addd2002-06-27 21:36:21 +00001283 file.close()
Fredrik Lundhb9056332001-07-11 17:42:21 +00001284 p.close()
1285
1286 return u.close()
1287
Fredrik Lundh3d9addd2002-06-27 21:36:21 +00001288##
1289# Standard transport class for XML-RPC over HTTPS.
1290
Fredrik Lundhb9056332001-07-11 17:42:21 +00001291class SafeTransport(Transport):
Fredrik Lundhc4c062f2001-09-10 19:45:02 +00001292 """Handles an HTTPS transaction to an XML-RPC server."""
Fredrik Lundhb9056332001-07-11 17:42:21 +00001293
Fredrik Lundh3d9addd2002-06-27 21:36:21 +00001294 # FIXME: mostly untested
1295
Fredrik Lundhb9056332001-07-11 17:42:21 +00001296 def make_connection(self, host):
1297 # create a HTTPS connection object from a host descriptor
1298 # host may be a string, or a (host, x509-dict) tuple
1299 import httplib
Fredrik Lundh1303c7c2002-10-22 18:23:00 +00001300 host, extra_headers, x509 = self.get_host_info(host)
Fredrik Lundhb9056332001-07-11 17:42:21 +00001301 try:
1302 HTTPS = httplib.HTTPS
1303 except AttributeError:
Fredrik Lundh1303c7c2002-10-22 18:23:00 +00001304 raise NotImplementedError(
1305 "your version of httplib doesn't support HTTPS"
1306 )
Fredrik Lundhb9056332001-07-11 17:42:21 +00001307 else:
Guido van Rossum68468eb2003-02-27 20:14:51 +00001308 return HTTPS(host, None, **(x509 or {}))
Fredrik Lundhb9056332001-07-11 17:42:21 +00001309
Fredrik Lundh3d9addd2002-06-27 21:36:21 +00001310##
1311# Standard server proxy. This class establishes a virtual connection
1312# to an XML-RPC server.
1313# <p>
1314# This class is available as ServerProxy and Server. New code should
1315# use ServerProxy, to avoid confusion.
1316#
1317# @def ServerProxy(uri, **options)
1318# @param uri The connection point on the server.
1319# @keyparam transport A transport factory, compatible with the
1320# standard transport class.
1321# @keyparam encoding The default encoding used for 8-bit strings
1322# (default is UTF-8).
1323# @keyparam verbose Use a true value to enable debugging output.
1324# (printed to standard output).
1325# @see Transport
1326
Fredrik Lundhb9056332001-07-11 17:42:21 +00001327class ServerProxy:
1328 """uri [,options] -> a logical connection to an XML-RPC server
1329
1330 uri is the connection point on the server, given as
1331 scheme://host/target.
1332
1333 The standard implementation always supports the "http" scheme. If
1334 SSL socket support is available (Python 2.0), it also supports
1335 "https".
1336
1337 If the target part and the slash preceding it are both omitted,
1338 "/RPC2" is assumed.
1339
1340 The following options can be given as keyword arguments:
1341
1342 transport: a transport factory
1343 encoding: the request encoding (default is UTF-8)
1344
1345 All 8-bit strings passed to the server proxy are assumed to use
1346 the given encoding.
1347 """
1348
Andrew M. Kuchlinga4c2b742003-04-25 00:26:51 +00001349 def __init__(self, uri, transport=None, encoding=None, verbose=0,
Skip Montanaro174dd222005-05-14 20:54:16 +00001350 allow_none=0, use_datetime=0):
Fredrik Lundhb9056332001-07-11 17:42:21 +00001351 # establish a "logical" server connection
1352
1353 # get the url
Fredrik Lundhc4c062f2001-09-10 19:45:02 +00001354 import urllib
Fredrik Lundhb9056332001-07-11 17:42:21 +00001355 type, uri = urllib.splittype(uri)
1356 if type not in ("http", "https"):
1357 raise IOError, "unsupported XML-RPC protocol"
1358 self.__host, self.__handler = urllib.splithost(uri)
1359 if not self.__handler:
1360 self.__handler = "/RPC2"
1361
1362 if transport is None:
1363 if type == "https":
Skip Montanaro174dd222005-05-14 20:54:16 +00001364 transport = SafeTransport(use_datetime=use_datetime)
Fredrik Lundhb9056332001-07-11 17:42:21 +00001365 else:
Skip Montanaro174dd222005-05-14 20:54:16 +00001366 transport = Transport(use_datetime=use_datetime)
Fredrik Lundhb9056332001-07-11 17:42:21 +00001367 self.__transport = transport
1368
1369 self.__encoding = encoding
1370 self.__verbose = verbose
Andrew M. Kuchlinga4c2b742003-04-25 00:26:51 +00001371 self.__allow_none = allow_none
Tim Petersc2659cf2003-05-12 20:19:37 +00001372
Fredrik Lundhb9056332001-07-11 17:42:21 +00001373 def __request(self, methodname, params):
1374 # call a method on the remote server
1375
Andrew M. Kuchlinga4c2b742003-04-25 00:26:51 +00001376 request = dumps(params, methodname, encoding=self.__encoding,
1377 allow_none=self.__allow_none)
Fredrik Lundhb9056332001-07-11 17:42:21 +00001378
1379 response = self.__transport.request(
1380 self.__host,
1381 self.__handler,
1382 request,
1383 verbose=self.__verbose
1384 )
1385
1386 if len(response) == 1:
1387 response = response[0]
1388
1389 return response
1390
1391 def __repr__(self):
1392 return (
Fredrik Lundh78eedce2001-08-23 20:04:33 +00001393 "<ServerProxy for %s%s>" %
Fredrik Lundhb9056332001-07-11 17:42:21 +00001394 (self.__host, self.__handler)
1395 )
1396
1397 __str__ = __repr__
Raymond Hettingercc523fc2003-11-02 09:47:05 +00001398
Fredrik Lundhb9056332001-07-11 17:42:21 +00001399 def __getattr__(self, name):
1400 # magic method dispatcher
1401 return _Method(self.__request, name)
1402
1403 # note: to call a remote object with an non-standard name, use
1404 # result getattr(server, "strange-python-name")(args)
1405
Fredrik Lundh78eedce2001-08-23 20:04:33 +00001406# compatibility
Fredrik Lundh3d9addd2002-06-27 21:36:21 +00001407
Fredrik Lundhb9056332001-07-11 17:42:21 +00001408Server = ServerProxy
1409
1410# --------------------------------------------------------------------
1411# test code
1412
1413if __name__ == "__main__":
1414
1415 # simple test program (from the XML-RPC specification)
1416
Fredrik Lundh78eedce2001-08-23 20:04:33 +00001417 # server = ServerProxy("http://localhost:8000") # local server
Martin v. Löwis12237b32004-08-22 16:04:50 +00001418 server = ServerProxy("http://time.xmlrpc.com/RPC2")
Fredrik Lundhb9056332001-07-11 17:42:21 +00001419
Guido van Rossumbe19ed72007-02-09 05:37:30 +00001420 print(server)
Fredrik Lundhb9056332001-07-11 17:42:21 +00001421
1422 try:
Guido van Rossumbe19ed72007-02-09 05:37:30 +00001423 print(server.currentTime.getCurrentTime())
Guido van Rossumb940e112007-01-10 16:19:56 +00001424 except Error as v:
Guido van Rossumbe19ed72007-02-09 05:37:30 +00001425 print("ERROR", v)
Martin v. Löwis12237b32004-08-22 16:04:50 +00001426
1427 multi = MultiCall(server)
1428 multi.currentTime.getCurrentTime()
1429 multi.currentTime.getCurrentTime()
1430 try:
1431 for response in multi():
Guido van Rossumbe19ed72007-02-09 05:37:30 +00001432 print(response)
Guido van Rossumb940e112007-01-10 16:19:56 +00001433 except Error as v:
Guido van Rossumbe19ed72007-02-09 05:37:30 +00001434 print("ERROR", v)