blob: 340dc51eb6aebd3fe55627c261ef9df32b2a67a5 [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
Benjamin Peterson9e8f5232014-12-05 20:15:15 -050052# 2014-12-02 ch/doko Add workaround for gzip bomb vulnerability
Fredrik Lundhb9056332001-07-11 17:42:21 +000053#
Fredrik Lundh3d9addd2002-06-27 21:36:21 +000054# Copyright (c) 1999-2002 by Secret Labs AB.
55# Copyright (c) 1999-2002 by Fredrik Lundh.
Fredrik Lundhb9056332001-07-11 17:42:21 +000056#
57# info@pythonware.com
58# http://www.pythonware.com
59#
60# --------------------------------------------------------------------
61# The XML-RPC client interface is
62#
Fredrik Lundh3d9addd2002-06-27 21:36:21 +000063# Copyright (c) 1999-2002 by Secret Labs AB
64# Copyright (c) 1999-2002 by Fredrik Lundh
Fredrik Lundhb9056332001-07-11 17:42:21 +000065#
66# By obtaining, using, and/or copying this software and/or its
67# associated documentation, you agree that you have read, understood,
68# and will comply with the following terms and conditions:
69#
70# Permission to use, copy, modify, and distribute this software and
71# its associated documentation for any purpose and without fee is
72# hereby granted, provided that the above copyright notice appears in
73# all copies, and that both that copyright notice and this permission
74# notice appear in supporting documentation, and that the name of
75# Secret Labs AB or the author not be used in advertising or publicity
76# pertaining to distribution of the software without specific, written
77# prior permission.
78#
79# SECRET LABS AB AND THE AUTHOR DISCLAIMS ALL WARRANTIES WITH REGARD
80# TO THIS SOFTWARE, INCLUDING ALL IMPLIED WARRANTIES OF MERCHANT-
81# ABILITY AND FITNESS. IN NO EVENT SHALL SECRET LABS AB OR THE AUTHOR
82# BE LIABLE FOR ANY SPECIAL, INDIRECT OR CONSEQUENTIAL DAMAGES OR ANY
83# DAMAGES WHATSOEVER RESULTING FROM LOSS OF USE, DATA OR PROFITS,
84# WHETHER IN AN ACTION OF CONTRACT, NEGLIGENCE OR OTHER TORTIOUS
85# ACTION, ARISING OUT OF OR IN CONNECTION WITH THE USE OR PERFORMANCE
86# OF THIS SOFTWARE.
87# --------------------------------------------------------------------
88
89#
Fredrik Lundh3d9addd2002-06-27 21:36:21 +000090# things to look into some day:
Fredrik Lundhb9056332001-07-11 17:42:21 +000091
Fredrik Lundh3d9addd2002-06-27 21:36:21 +000092# TODO: sort out True/False/boolean issues for Python 2.3
Fredrik Lundhb9056332001-07-11 17:42:21 +000093
Fred Drake1b410792001-09-04 18:55:03 +000094"""
95An XML-RPC client interface for Python.
96
97The marshalling and response parser code can also be used to
98implement XML-RPC servers.
99
Fred Drake1b410792001-09-04 18:55:03 +0000100Exported exceptions:
101
Fredrik Lundhc4c062f2001-09-10 19:45:02 +0000102 Error Base class for client errors
103 ProtocolError Indicates an HTTP protocol error
104 ResponseError Indicates a broken response package
105 Fault Indicates an XML-RPC fault package
Fred Drake1b410792001-09-04 18:55:03 +0000106
107Exported classes:
108
Fredrik Lundhc4c062f2001-09-10 19:45:02 +0000109 ServerProxy Represents a logical connection to an XML-RPC server
Fred Drake1b410792001-09-04 18:55:03 +0000110
Raymond Hettingercc523fc2003-11-02 09:47:05 +0000111 MultiCall Executor of boxcared xmlrpc requests
Fredrik Lundhc4c062f2001-09-10 19:45:02 +0000112 Boolean boolean wrapper to generate a "boolean" XML-RPC value
113 DateTime dateTime wrapper for an ISO 8601 string or time tuple or
114 localtime integer value to generate a "dateTime.iso8601"
115 XML-RPC value
116 Binary binary data wrapper
Fred Drake1b410792001-09-04 18:55:03 +0000117
Fredrik Lundhc4c062f2001-09-10 19:45:02 +0000118 SlowParser Slow but safe standard parser (based on xmllib)
119 Marshaller Generate an XML-RPC params chunk from a Python data structure
120 Unmarshaller Unmarshal an XML-RPC response from incoming XML event message
121 Transport Handles an HTTP transaction to an XML-RPC server
122 SafeTransport Handles an HTTPS transaction to an XML-RPC server
Fred Drake1b410792001-09-04 18:55:03 +0000123
124Exported constants:
125
Fredrik Lundhc4c062f2001-09-10 19:45:02 +0000126 True
127 False
Fred Drake1b410792001-09-04 18:55:03 +0000128
129Exported functions:
130
Fredrik Lundhc4c062f2001-09-10 19:45:02 +0000131 boolean Convert any Python value to an XML-RPC boolean
132 getparser Create instance of the fastest available parser & attach
133 to an unmarshalling object
134 dumps Convert an argument tuple or a Fault instance to an XML-RPC
135 request (or response, if the methodresponse option is used).
136 loads Convert an XML-RPC packet to unmarshalled data plus a method
137 name (None if not present).
Fred Drake1b410792001-09-04 18:55:03 +0000138"""
139
Fred Drake2a2d9702001-10-17 01:51:04 +0000140import re, string, time, operator
Fredrik Lundh1538c232001-10-01 19:42:03 +0000141
Raymond Hettingerfe59dc12005-02-07 15:28:45 +0000142from types import *
Kristján Valur Jónssone0078602009-06-28 21:04:17 +0000143import socket
144import errno
145import httplib
Kristján Valur Jónsson6d755902009-07-19 22:14:00 +0000146try:
147 import gzip
148except ImportError:
149 gzip = None #python can be built without zlib/gzip support
Fredrik Lundhb9056332001-07-11 17:42:21 +0000150
Fredrik Lundh3d9addd2002-06-27 21:36:21 +0000151# --------------------------------------------------------------------
152# Internal stuff
153
Fredrik Lundhb9056332001-07-11 17:42:21 +0000154try:
155 unicode
156except NameError:
157 unicode = None # unicode support not available
158
Skip Montanaro9a7c96a2003-01-22 18:17:25 +0000159try:
Fred Drakeba613c32005-02-10 18:33:30 +0000160 import datetime
161except ImportError:
162 datetime = None
163
164try:
Skip Montanaro9a7c96a2003-01-22 18:17:25 +0000165 _bool_is_builtin = False.__class__.__name__ == "bool"
166except NameError:
167 _bool_is_builtin = 0
168
Fredrik Lundhb9056332001-07-11 17:42:21 +0000169def _decode(data, encoding, is8bit=re.compile("[\x80-\xff]").search):
170 # decode non-ascii string (if possible)
171 if unicode and encoding and is8bit(data):
172 data = unicode(data, encoding)
173 return data
174
Fredrik Lundh1538c232001-10-01 19:42:03 +0000175def escape(s, replace=string.replace):
176 s = replace(s, "&", "&")
177 s = replace(s, "<", "&lt;")
178 return replace(s, ">", "&gt;",)
179
Fredrik Lundhb9056332001-07-11 17:42:21 +0000180if unicode:
181 def _stringify(string):
182 # convert to 7-bit ascii if possible
183 try:
Fred Drake22c07062005-02-11 17:59:08 +0000184 return string.encode("ascii")
Fredrik Lundhb9056332001-07-11 17:42:21 +0000185 except UnicodeError:
186 return string
187else:
188 def _stringify(string):
189 return string
190
Fredrik Lundh3d9addd2002-06-27 21:36:21 +0000191__version__ = "1.0.1"
192
193# xmlrpc integer limits
194MAXINT = 2L**31-1
195MININT = -2L**31
196
197# --------------------------------------------------------------------
198# Error constants (from Dan Libby's specification at
199# http://xmlrpc-epi.sourceforge.net/specs/rfc.fault_codes.php)
200
201# Ranges of errors
202PARSE_ERROR = -32700
203SERVER_ERROR = -32600
204APPLICATION_ERROR = -32500
205SYSTEM_ERROR = -32400
206TRANSPORT_ERROR = -32300
207
208# Specific errors
209NOT_WELLFORMED_ERROR = -32700
210UNSUPPORTED_ENCODING = -32701
211INVALID_ENCODING_CHAR = -32702
212INVALID_XMLRPC = -32600
213METHOD_NOT_FOUND = -32601
214INVALID_METHOD_PARAMS = -32602
215INTERNAL_ERROR = -32603
Fredrik Lundhb9056332001-07-11 17:42:21 +0000216
217# --------------------------------------------------------------------
218# Exceptions
219
Fredrik Lundh3d9addd2002-06-27 21:36:21 +0000220##
221# Base class for all kinds of client-side errors.
222
Fredrik Lundh78eedce2001-08-23 20:04:33 +0000223class Error(Exception):
Fred Drake1b410792001-09-04 18:55:03 +0000224 """Base class for client errors."""
Fredrik Lundhb0e8e9b2001-09-10 21:45:42 +0000225 def __str__(self):
226 return repr(self)
Fredrik Lundhb9056332001-07-11 17:42:21 +0000227
Fredrik Lundh3d9addd2002-06-27 21:36:21 +0000228##
229# Indicates an HTTP-level protocol error. This is raised by the HTTP
230# transport layer, if the server returns an error code other than 200
231# (OK).
232#
233# @param url The target URL.
234# @param errcode The HTTP error code.
235# @param errmsg The HTTP error message.
236# @param headers The HTTP header dictionary.
237
Fredrik Lundhb9056332001-07-11 17:42:21 +0000238class ProtocolError(Error):
Fred Drake1b410792001-09-04 18:55:03 +0000239 """Indicates an HTTP protocol error."""
Fredrik Lundhb9056332001-07-11 17:42:21 +0000240 def __init__(self, url, errcode, errmsg, headers):
Fredrik Lundhb0e8e9b2001-09-10 21:45:42 +0000241 Error.__init__(self)
Fredrik Lundhb9056332001-07-11 17:42:21 +0000242 self.url = url
243 self.errcode = errcode
244 self.errmsg = errmsg
245 self.headers = headers
246 def __repr__(self):
247 return (
248 "<ProtocolError for %s: %s %s>" %
249 (self.url, self.errcode, self.errmsg)
250 )
251
Fredrik Lundh3d9addd2002-06-27 21:36:21 +0000252##
253# Indicates a broken XML-RPC response package. This exception is
254# raised by the unmarshalling layer, if the XML-RPC response is
255# malformed.
256
Fredrik Lundhb9056332001-07-11 17:42:21 +0000257class ResponseError(Error):
Fredrik Lundhc4c062f2001-09-10 19:45:02 +0000258 """Indicates a broken response package."""
Fredrik Lundhb9056332001-07-11 17:42:21 +0000259 pass
260
Fredrik Lundh3d9addd2002-06-27 21:36:21 +0000261##
262# Indicates an XML-RPC fault response package. This exception is
263# raised by the unmarshalling layer, if the XML-RPC response contains
264# a fault string. This exception can also used as a class, to
265# generate a fault XML-RPC message.
266#
267# @param faultCode The XML-RPC fault code.
268# @param faultString The XML-RPC fault string.
269
Fredrik Lundhb9056332001-07-11 17:42:21 +0000270class Fault(Error):
Fredrik Lundhc4c062f2001-09-10 19:45:02 +0000271 """Indicates an XML-RPC fault package."""
Fredrik Lundhb9056332001-07-11 17:42:21 +0000272 def __init__(self, faultCode, faultString, **extra):
Fredrik Lundhb0e8e9b2001-09-10 21:45:42 +0000273 Error.__init__(self)
Fredrik Lundhb9056332001-07-11 17:42:21 +0000274 self.faultCode = faultCode
275 self.faultString = faultString
276 def __repr__(self):
277 return (
278 "<Fault %s: %s>" %
Fredrik Lundhb0e8e9b2001-09-10 21:45:42 +0000279 (self.faultCode, repr(self.faultString))
Fredrik Lundhb9056332001-07-11 17:42:21 +0000280 )
281
282# --------------------------------------------------------------------
283# Special values
284
Fredrik Lundh3d9addd2002-06-27 21:36:21 +0000285##
286# Wrapper for XML-RPC boolean values. Use the xmlrpclib.True and
287# xmlrpclib.False constants, or the xmlrpclib.boolean() function, to
288# generate boolean XML-RPC values.
289#
290# @param value A boolean value. Any true value is interpreted as True,
291# all other values are interpreted as False.
292
Brett Cannon814820b2008-08-04 00:50:11 +0000293from sys import modules
294mod_dict = modules[__name__].__dict__
Skip Montanaro9a7c96a2003-01-22 18:17:25 +0000295if _bool_is_builtin:
296 boolean = Boolean = bool
297 # to avoid breaking code which references xmlrpclib.{True,False}
Brett Cannon814820b2008-08-04 00:50:11 +0000298 mod_dict['True'] = True
299 mod_dict['False'] = False
Skip Montanaro9a7c96a2003-01-22 18:17:25 +0000300else:
301 class Boolean:
302 """Boolean-value wrapper.
Fred Drake1b410792001-09-04 18:55:03 +0000303
Skip Montanaro9a7c96a2003-01-22 18:17:25 +0000304 Use True or False to generate a "boolean" XML-RPC value.
305 """
Fredrik Lundhb9056332001-07-11 17:42:21 +0000306
Skip Montanaro9a7c96a2003-01-22 18:17:25 +0000307 def __init__(self, value = 0):
308 self.value = operator.truth(value)
Fredrik Lundhb9056332001-07-11 17:42:21 +0000309
Skip Montanaro9a7c96a2003-01-22 18:17:25 +0000310 def encode(self, out):
311 out.write("<value><boolean>%d</boolean></value>\n" % self.value)
Fredrik Lundhb9056332001-07-11 17:42:21 +0000312
Skip Montanaro9a7c96a2003-01-22 18:17:25 +0000313 def __cmp__(self, other):
314 if isinstance(other, Boolean):
315 other = other.value
316 return cmp(self.value, other)
Fredrik Lundhb9056332001-07-11 17:42:21 +0000317
Skip Montanaro9a7c96a2003-01-22 18:17:25 +0000318 def __repr__(self):
319 if self.value:
320 return "<Boolean True at %x>" % id(self)
321 else:
322 return "<Boolean False at %x>" % id(self)
Fredrik Lundhb9056332001-07-11 17:42:21 +0000323
Skip Montanaro9a7c96a2003-01-22 18:17:25 +0000324 def __int__(self):
325 return self.value
Fredrik Lundhb9056332001-07-11 17:42:21 +0000326
Skip Montanaro9a7c96a2003-01-22 18:17:25 +0000327 def __nonzero__(self):
328 return self.value
Fredrik Lundhb9056332001-07-11 17:42:21 +0000329
Brett Cannon814820b2008-08-04 00:50:11 +0000330 mod_dict['True'] = Boolean(1)
331 mod_dict['False'] = Boolean(0)
Fredrik Lundhb9056332001-07-11 17:42:21 +0000332
Skip Montanaro9a7c96a2003-01-22 18:17:25 +0000333 ##
334 # Map true or false value to XML-RPC boolean values.
335 #
336 # @def boolean(value)
337 # @param value A boolean value. Any true value is mapped to True,
338 # all other values are mapped to False.
339 # @return xmlrpclib.True or xmlrpclib.False.
340 # @see Boolean
341 # @see True
342 # @see False
Fredrik Lundh3d9addd2002-06-27 21:36:21 +0000343
Skip Montanaro9a7c96a2003-01-22 18:17:25 +0000344 def boolean(value, _truefalse=(False, True)):
345 """Convert any Python value to XML-RPC 'boolean'."""
346 return _truefalse[operator.truth(value)]
Fredrik Lundh3d9addd2002-06-27 21:36:21 +0000347
Brett Cannon814820b2008-08-04 00:50:11 +0000348del modules, mod_dict
349
Fredrik Lundh3d9addd2002-06-27 21:36:21 +0000350##
351# Wrapper for XML-RPC DateTime values. This converts a time value to
352# the format used by XML-RPC.
353# <p>
354# The value can be given as a string in the format
355# "yyyymmddThh:mm:ss", as a 9-item time tuple (as returned by
356# time.localtime()), or an integer value (as returned by time.time()).
357# The wrapper uses time.localtime() to convert an integer to a time
358# tuple.
359#
360# @param value The time, given as an ISO 8601 string, a time
361# tuple, or a integer time value.
Fredrik Lundhb9056332001-07-11 17:42:21 +0000362
Skip Montanarob131f042008-04-18 20:35:46 +0000363def _strftime(value):
364 if datetime:
365 if isinstance(value, datetime.datetime):
366 return "%04d%02d%02dT%02d:%02d:%02d" % (
367 value.year, value.month, value.day,
368 value.hour, value.minute, value.second)
369
370 if not isinstance(value, (TupleType, time.struct_time)):
371 if value == 0:
372 value = time.time()
373 value = time.localtime(value)
374
375 return "%04d%02d%02dT%02d:%02d:%02d" % value[:6]
376
Fredrik Lundhb9056332001-07-11 17:42:21 +0000377class DateTime:
Fredrik Lundhc4c062f2001-09-10 19:45:02 +0000378 """DateTime wrapper for an ISO 8601 string or time tuple or
379 localtime integer value to generate 'dateTime.iso8601' XML-RPC
380 value.
381 """
Fredrik Lundhb9056332001-07-11 17:42:21 +0000382
383 def __init__(self, value=0):
Skip Montanarob131f042008-04-18 20:35:46 +0000384 if isinstance(value, StringType):
385 self.value = value
386 else:
387 self.value = _strftime(value)
Fredrik Lundhb9056332001-07-11 17:42:21 +0000388
Andrew M. Kuchling085f75a2008-02-23 16:23:05 +0000389 def make_comparable(self, other):
Fredrik Lundhb9056332001-07-11 17:42:21 +0000390 if isinstance(other, DateTime):
Andrew M. Kuchling085f75a2008-02-23 16:23:05 +0000391 s = self.value
392 o = other.value
393 elif datetime and isinstance(other, datetime.datetime):
394 s = self.value
395 o = other.strftime("%Y%m%dT%H:%M:%S")
396 elif isinstance(other, (str, unicode)):
397 s = self.value
398 o = other
399 elif hasattr(other, "timetuple"):
400 s = self.timetuple()
401 o = other.timetuple()
402 else:
403 otype = (hasattr(other, "__class__")
404 and other.__class__.__name__
405 or type(other))
406 raise TypeError("Can't compare %s and %s" %
407 (self.__class__.__name__, otype))
408 return s, o
409
410 def __lt__(self, other):
411 s, o = self.make_comparable(other)
412 return s < o
413
414 def __le__(self, other):
415 s, o = self.make_comparable(other)
416 return s <= o
417
418 def __gt__(self, other):
419 s, o = self.make_comparable(other)
420 return s > o
421
422 def __ge__(self, other):
423 s, o = self.make_comparable(other)
424 return s >= o
425
426 def __eq__(self, other):
427 s, o = self.make_comparable(other)
428 return s == o
429
430 def __ne__(self, other):
431 s, o = self.make_comparable(other)
432 return s != o
433
434 def timetuple(self):
435 return time.strptime(self.value, "%Y%m%dT%H:%M:%S")
436
437 def __cmp__(self, other):
438 s, o = self.make_comparable(other)
439 return cmp(s, o)
Fredrik Lundhb9056332001-07-11 17:42:21 +0000440
Fredrik Lundh3d9addd2002-06-27 21:36:21 +0000441 ##
442 # Get date/time value.
443 #
444 # @return Date/time value, as an ISO 8601 string.
445
446 def __str__(self):
447 return self.value
448
Fredrik Lundhb9056332001-07-11 17:42:21 +0000449 def __repr__(self):
Fredrik Lundh3d9addd2002-06-27 21:36:21 +0000450 return "<DateTime %s at %x>" % (repr(self.value), id(self))
Fredrik Lundhb9056332001-07-11 17:42:21 +0000451
452 def decode(self, data):
Andrew M. Kuchlingbdb39012005-12-04 19:11:17 +0000453 data = str(data)
Fredrik Lundhb9056332001-07-11 17:42:21 +0000454 self.value = string.strip(data)
455
456 def encode(self, out):
457 out.write("<value><dateTime.iso8601>")
458 out.write(self.value)
459 out.write("</dateTime.iso8601></value>\n")
460
Fredrik Lundh3d9addd2002-06-27 21:36:21 +0000461def _datetime(data):
462 # decode xml element contents into a DateTime structure.
Fredrik Lundhb9056332001-07-11 17:42:21 +0000463 value = DateTime()
464 value.decode(data)
465 return value
466
Skip Montanaro174dd222005-05-14 20:54:16 +0000467def _datetime_type(data):
468 t = time.strptime(data, "%Y%m%dT%H:%M:%S")
469 return datetime.datetime(*tuple(t)[:6])
470
Fredrik Lundh3d9addd2002-06-27 21:36:21 +0000471##
472# Wrapper for binary data. This can be used to transport any kind
473# of binary data over XML-RPC, using BASE64 encoding.
474#
475# @param data An 8-bit string containing arbitrary data.
476
Skip Montanarobfcbfa72003-04-24 19:51:31 +0000477import base64
478try:
479 import cStringIO as StringIO
480except ImportError:
481 import StringIO
482
Fredrik Lundhb9056332001-07-11 17:42:21 +0000483class Binary:
Fred Drake1b410792001-09-04 18:55:03 +0000484 """Wrapper for binary data."""
Fredrik Lundhb9056332001-07-11 17:42:21 +0000485
486 def __init__(self, data=None):
487 self.data = data
488
Fredrik Lundh3d9addd2002-06-27 21:36:21 +0000489 ##
490 # Get buffer contents.
491 #
492 # @return Buffer contents, as an 8-bit string.
493
494 def __str__(self):
495 return self.data or ""
496
Fredrik Lundhb9056332001-07-11 17:42:21 +0000497 def __cmp__(self, other):
498 if isinstance(other, Binary):
499 other = other.data
500 return cmp(self.data, other)
501
502 def decode(self, data):
Fredrik Lundhb9056332001-07-11 17:42:21 +0000503 self.data = base64.decodestring(data)
504
505 def encode(self, out):
Fredrik Lundhb9056332001-07-11 17:42:21 +0000506 out.write("<value><base64>\n")
507 base64.encode(StringIO.StringIO(self.data), out)
508 out.write("</base64></value>\n")
509
Fredrik Lundh3d9addd2002-06-27 21:36:21 +0000510def _binary(data):
511 # decode xml element contents into a Binary structure
Fredrik Lundhb9056332001-07-11 17:42:21 +0000512 value = Binary()
513 value.decode(data)
514 return value
515
Skip Montanaro9a7c96a2003-01-22 18:17:25 +0000516WRAPPERS = (DateTime, Binary)
517if not _bool_is_builtin:
518 WRAPPERS = WRAPPERS + (Boolean,)
Fredrik Lundhb9056332001-07-11 17:42:21 +0000519
520# --------------------------------------------------------------------
521# XML parsers
522
523try:
Fredrik Lundh7069c312004-10-13 06:48:37 +0000524 # optional xmlrpclib accelerator
Fredrik Lundhb9056332001-07-11 17:42:21 +0000525 import _xmlrpclib
526 FastParser = _xmlrpclib.Parser
527 FastUnmarshaller = _xmlrpclib.Unmarshaller
528except (AttributeError, ImportError):
529 FastParser = FastUnmarshaller = None
530
Fredrik Lundh3d9addd2002-06-27 21:36:21 +0000531try:
532 import _xmlrpclib
533 FastMarshaller = _xmlrpclib.Marshaller
534except (AttributeError, ImportError):
535 FastMarshaller = None
536
Fredrik Lundhb9056332001-07-11 17:42:21 +0000537try:
538 from xml.parsers import expat
Guido van Rossumb8551342001-10-02 18:33:11 +0000539 if not hasattr(expat, "ParserCreate"):
Fredrik Lundh3d9addd2002-06-27 21:36:21 +0000540 raise ImportError
Fredrik Lundhb9056332001-07-11 17:42:21 +0000541except ImportError:
Fredrik Lundh3d9addd2002-06-27 21:36:21 +0000542 ExpatParser = None # expat not available
Fredrik Lundhb9056332001-07-11 17:42:21 +0000543else:
544 class ExpatParser:
Georg Brandle6632b42009-06-04 08:58:32 +0000545 # fast expat parser for Python 2.0 and later.
Fredrik Lundhb9056332001-07-11 17:42:21 +0000546 def __init__(self, target):
547 self._parser = parser = expat.ParserCreate(None, None)
548 self._target = target
549 parser.StartElementHandler = target.start
550 parser.EndElementHandler = target.end
551 parser.CharacterDataHandler = target.data
552 encoding = None
553 if not parser.returns_unicode:
554 encoding = "utf-8"
555 target.xml(encoding, None)
556
557 def feed(self, data):
558 self._parser.Parse(data, 0)
559
560 def close(self):
561 self._parser.Parse("", 1) # end of data
562 del self._target, self._parser # get rid of circular references
563
Fredrik Lundhc4c062f2001-09-10 19:45:02 +0000564class SlowParser:
565 """Default XML parser (based on xmllib.XMLParser)."""
Georg Brandle6632b42009-06-04 08:58:32 +0000566 # this is the slowest parser.
Fredrik Lundhb9056332001-07-11 17:42:21 +0000567 def __init__(self, target):
Fredrik Lundhc4c062f2001-09-10 19:45:02 +0000568 import xmllib # lazy subclassing (!)
569 if xmllib.XMLParser not in SlowParser.__bases__:
570 SlowParser.__bases__ = (xmllib.XMLParser,)
Fredrik Lundhb9056332001-07-11 17:42:21 +0000571 self.handle_xml = target.xml
572 self.unknown_starttag = target.start
573 self.handle_data = target.data
Fredrik Lundh3d9addd2002-06-27 21:36:21 +0000574 self.handle_cdata = target.data
Fredrik Lundhb9056332001-07-11 17:42:21 +0000575 self.unknown_endtag = target.end
Fredrik Lundhb0e8e9b2001-09-10 21:45:42 +0000576 try:
577 xmllib.XMLParser.__init__(self, accept_utf8=1)
578 except TypeError:
579 xmllib.XMLParser.__init__(self) # pre-2.0
Fredrik Lundhb9056332001-07-11 17:42:21 +0000580
581# --------------------------------------------------------------------
582# XML-RPC marshalling and unmarshalling code
583
Fredrik Lundh3d9addd2002-06-27 21:36:21 +0000584##
585# XML-RPC marshaller.
586#
587# @param encoding Default encoding for 8-bit strings. The default
588# value is None (interpreted as UTF-8).
589# @see dumps
590
Fredrik Lundhb9056332001-07-11 17:42:21 +0000591class Marshaller:
Fred Drake1b410792001-09-04 18:55:03 +0000592 """Generate an XML-RPC params chunk from a Python data structure.
Fredrik Lundhb9056332001-07-11 17:42:21 +0000593
Fredrik Lundhc4c062f2001-09-10 19:45:02 +0000594 Create a Marshaller instance for each set of parameters, and use
595 the "dumps" method to convert your data (represented as a tuple)
596 to an XML-RPC params chunk. To write a fault response, pass a
597 Fault instance instead. You may prefer to use the "dumps" module
598 function for this purpose.
Fred Drake1b410792001-09-04 18:55:03 +0000599 """
Fredrik Lundhb9056332001-07-11 17:42:21 +0000600
601 # by the way, if you don't understand what's going on in here,
602 # that's perfectly ok.
603
Andrew M. Kuchlinga4c2b742003-04-25 00:26:51 +0000604 def __init__(self, encoding=None, allow_none=0):
Fredrik Lundhb9056332001-07-11 17:42:21 +0000605 self.memo = {}
606 self.data = None
607 self.encoding = encoding
Andrew M. Kuchlinga4c2b742003-04-25 00:26:51 +0000608 self.allow_none = allow_none
Tim Petersc2659cf2003-05-12 20:19:37 +0000609
Fredrik Lundhb9056332001-07-11 17:42:21 +0000610 dispatch = {}
611
612 def dumps(self, values):
Fredrik Lundh3d9addd2002-06-27 21:36:21 +0000613 out = []
614 write = out.append
615 dump = self.__dump
Fredrik Lundhb9056332001-07-11 17:42:21 +0000616 if isinstance(values, Fault):
617 # fault instance
618 write("<fault>\n")
Martin v. Löwis541342f2003-07-12 07:53:04 +0000619 dump({'faultCode': values.faultCode,
620 'faultString': values.faultString},
621 write)
Fredrik Lundhb9056332001-07-11 17:42:21 +0000622 write("</fault>\n")
623 else:
624 # parameter block
Fredrik Lundhc266bb02001-08-23 20:13:08 +0000625 # FIXME: the xml-rpc specification allows us to leave out
626 # the entire <params> block if there are no parameters.
627 # however, changing this may break older code (including
628 # old versions of xmlrpclib.py), so this is better left as
629 # is for now. See @XMLRPC3 for more information. /F
Fredrik Lundhb9056332001-07-11 17:42:21 +0000630 write("<params>\n")
631 for v in values:
632 write("<param>\n")
Fredrik Lundh3d9addd2002-06-27 21:36:21 +0000633 dump(v, write)
Fredrik Lundhb9056332001-07-11 17:42:21 +0000634 write("</param>\n")
635 write("</params>\n")
Fredrik Lundh3d9addd2002-06-27 21:36:21 +0000636 result = string.join(out, "")
Fredrik Lundhb9056332001-07-11 17:42:21 +0000637 return result
638
Fredrik Lundh3d9addd2002-06-27 21:36:21 +0000639 def __dump(self, value, write):
Fredrik Lundhb9056332001-07-11 17:42:21 +0000640 try:
641 f = self.dispatch[type(value)]
642 except KeyError:
Martin v. Löwis07529352006-11-19 18:51:54 +0000643 # check if this object can be marshalled as a structure
644 try:
645 value.__dict__
646 except:
647 raise TypeError, "cannot marshal %s objects" % type(value)
648 # check if this class is a sub-class of a basic type,
649 # because we don't know how to marshal these types
650 # (e.g. a string sub-class)
651 for type_ in type(value).__mro__:
652 if type_ in self.dispatch.keys():
653 raise TypeError, "cannot marshal %s objects" % type(value)
654 f = self.dispatch[InstanceType]
655 f(self, value, write)
Fredrik Lundhb9056332001-07-11 17:42:21 +0000656
Andrew M. Kuchlinga4c2b742003-04-25 00:26:51 +0000657 def dump_nil (self, value, write):
658 if not self.allow_none:
659 raise TypeError, "cannot marshal None unless allow_none is enabled"
660 write("<value><nil/></value>")
Raymond Hettingerfe59dc12005-02-07 15:28:45 +0000661 dispatch[NoneType] = dump_nil
Tim Petersc2659cf2003-05-12 20:19:37 +0000662
Fredrik Lundh3d9addd2002-06-27 21:36:21 +0000663 def dump_int(self, value, write):
Skip Montanaro5449e082001-10-17 22:53:33 +0000664 # in case ints are > 32 bits
665 if value > MAXINT or value < MININT:
666 raise OverflowError, "int exceeds XML-RPC limits"
Fredrik Lundh3d9addd2002-06-27 21:36:21 +0000667 write("<value><int>")
668 write(str(value))
669 write("</int></value>\n")
Raymond Hettingerfe59dc12005-02-07 15:28:45 +0000670 dispatch[IntType] = dump_int
Fredrik Lundhb9056332001-07-11 17:42:21 +0000671
Skip Montanaro9a7c96a2003-01-22 18:17:25 +0000672 if _bool_is_builtin:
673 def dump_bool(self, value, write):
674 write("<value><boolean>")
675 write(value and "1" or "0")
676 write("</boolean></value>\n")
677 dispatch[bool] = dump_bool
678
Fredrik Lundh3d9addd2002-06-27 21:36:21 +0000679 def dump_long(self, value, write):
Skip Montanaro5449e082001-10-17 22:53:33 +0000680 if value > MAXINT or value < MININT:
681 raise OverflowError, "long int exceeds XML-RPC limits"
Fredrik Lundh3d9addd2002-06-27 21:36:21 +0000682 write("<value><int>")
683 write(str(int(value)))
684 write("</int></value>\n")
Raymond Hettingerfe59dc12005-02-07 15:28:45 +0000685 dispatch[LongType] = dump_long
Skip Montanaro5e9c71b2001-10-10 15:56:34 +0000686
Fredrik Lundh3d9addd2002-06-27 21:36:21 +0000687 def dump_double(self, value, write):
688 write("<value><double>")
689 write(repr(value))
690 write("</double></value>\n")
Raymond Hettingerfe59dc12005-02-07 15:28:45 +0000691 dispatch[FloatType] = dump_double
Fredrik Lundhb9056332001-07-11 17:42:21 +0000692
Fredrik Lundh3d9addd2002-06-27 21:36:21 +0000693 def dump_string(self, value, write, escape=escape):
694 write("<value><string>")
695 write(escape(value))
696 write("</string></value>\n")
Raymond Hettingerfe59dc12005-02-07 15:28:45 +0000697 dispatch[StringType] = dump_string
Fredrik Lundhb9056332001-07-11 17:42:21 +0000698
699 if unicode:
Fredrik Lundh3d9addd2002-06-27 21:36:21 +0000700 def dump_unicode(self, value, write, escape=escape):
Fredrik Lundhb9056332001-07-11 17:42:21 +0000701 value = value.encode(self.encoding)
Fredrik Lundh3d9addd2002-06-27 21:36:21 +0000702 write("<value><string>")
703 write(escape(value))
704 write("</string></value>\n")
Raymond Hettingerfe59dc12005-02-07 15:28:45 +0000705 dispatch[UnicodeType] = dump_unicode
Fredrik Lundhb9056332001-07-11 17:42:21 +0000706
Fredrik Lundh3d9addd2002-06-27 21:36:21 +0000707 def dump_array(self, value, write):
708 i = id(value)
Brett Cannon814820b2008-08-04 00:50:11 +0000709 if i in self.memo:
Fredrik Lundh3d9addd2002-06-27 21:36:21 +0000710 raise TypeError, "cannot marshal recursive sequences"
711 self.memo[i] = None
Fredrik Lundh1538c232001-10-01 19:42:03 +0000712 dump = self.__dump
Fredrik Lundhb9056332001-07-11 17:42:21 +0000713 write("<value><array><data>\n")
714 for v in value:
Fredrik Lundh3d9addd2002-06-27 21:36:21 +0000715 dump(v, write)
Fredrik Lundhb9056332001-07-11 17:42:21 +0000716 write("</data></array></value>\n")
Fredrik Lundh3d9addd2002-06-27 21:36:21 +0000717 del self.memo[i]
Raymond Hettingerfe59dc12005-02-07 15:28:45 +0000718 dispatch[TupleType] = dump_array
719 dispatch[ListType] = dump_array
Fredrik Lundhb9056332001-07-11 17:42:21 +0000720
Fredrik Lundh3d9addd2002-06-27 21:36:21 +0000721 def dump_struct(self, value, write, escape=escape):
722 i = id(value)
Brett Cannon814820b2008-08-04 00:50:11 +0000723 if i in self.memo:
Fredrik Lundh3d9addd2002-06-27 21:36:21 +0000724 raise TypeError, "cannot marshal recursive dictionaries"
725 self.memo[i] = None
Fredrik Lundh1538c232001-10-01 19:42:03 +0000726 dump = self.__dump
Fredrik Lundhb9056332001-07-11 17:42:21 +0000727 write("<value><struct>\n")
Andrew M. Kuchling5962f452004-06-05 12:35:58 +0000728 for k, v in value.items():
Fredrik Lundhb9056332001-07-11 17:42:21 +0000729 write("<member>\n")
Raymond Hettingerfe59dc12005-02-07 15:28:45 +0000730 if type(k) is not StringType:
731 if unicode and type(k) is UnicodeType:
Andrew M. Kuchling5962f452004-06-05 12:35:58 +0000732 k = k.encode(self.encoding)
733 else:
734 raise TypeError, "dictionary key must be string"
Fredrik Lundh1538c232001-10-01 19:42:03 +0000735 write("<name>%s</name>\n" % escape(k))
Andrew M. Kuchling5962f452004-06-05 12:35:58 +0000736 dump(v, write)
Fredrik Lundhb9056332001-07-11 17:42:21 +0000737 write("</member>\n")
738 write("</struct></value>\n")
Fredrik Lundh3d9addd2002-06-27 21:36:21 +0000739 del self.memo[i]
Raymond Hettingerfe59dc12005-02-07 15:28:45 +0000740 dispatch[DictType] = dump_struct
Fredrik Lundhb9056332001-07-11 17:42:21 +0000741
Fred Drakeba613c32005-02-10 18:33:30 +0000742 if datetime:
743 def dump_datetime(self, value, write):
744 write("<value><dateTime.iso8601>")
Skip Montanarob131f042008-04-18 20:35:46 +0000745 write(_strftime(value))
Fred Drakeba613c32005-02-10 18:33:30 +0000746 write("</dateTime.iso8601></value>\n")
747 dispatch[datetime.datetime] = dump_datetime
748
Fredrik Lundh3d9addd2002-06-27 21:36:21 +0000749 def dump_instance(self, value, write):
Fredrik Lundhb9056332001-07-11 17:42:21 +0000750 # check for special wrappers
751 if value.__class__ in WRAPPERS:
Fredrik Lundh3d9addd2002-06-27 21:36:21 +0000752 self.write = write
Fredrik Lundhb9056332001-07-11 17:42:21 +0000753 value.encode(self)
Fredrik Lundh3d9addd2002-06-27 21:36:21 +0000754 del self.write
Fredrik Lundhb9056332001-07-11 17:42:21 +0000755 else:
756 # store instance attributes as a struct (really?)
Fredrik Lundh3d9addd2002-06-27 21:36:21 +0000757 self.dump_struct(value.__dict__, write)
Fredrik Lundhb9056332001-07-11 17:42:21 +0000758 dispatch[InstanceType] = dump_instance
759
Fredrik Lundh3d9addd2002-06-27 21:36:21 +0000760##
761# XML-RPC unmarshaller.
762#
763# @see loads
764
Fredrik Lundhb9056332001-07-11 17:42:21 +0000765class Unmarshaller:
Fred Drake1b410792001-09-04 18:55:03 +0000766 """Unmarshal an XML-RPC response, based on incoming XML event
Fredrik Lundhb0e8e9b2001-09-10 21:45:42 +0000767 messages (start, data, end). Call close() to get the resulting
Fred Drake1b410792001-09-04 18:55:03 +0000768 data structure.
Fredrik Lundhb9056332001-07-11 17:42:21 +0000769
Fredrik Lundhb0e8e9b2001-09-10 21:45:42 +0000770 Note that this reader is fairly tolerant, and gladly accepts bogus
771 XML-RPC data without complaining (but not bogus XML).
Fred Drake1b410792001-09-04 18:55:03 +0000772 """
Fredrik Lundhb9056332001-07-11 17:42:21 +0000773
774 # and again, if you don't understand what's going on in here,
775 # that's perfectly ok.
776
Skip Montanaro174dd222005-05-14 20:54:16 +0000777 def __init__(self, use_datetime=0):
Fredrik Lundhb9056332001-07-11 17:42:21 +0000778 self._type = None
779 self._stack = []
780 self._marks = []
781 self._data = []
782 self._methodname = None
783 self._encoding = "utf-8"
784 self.append = self._stack.append
Skip Montanaro174dd222005-05-14 20:54:16 +0000785 self._use_datetime = use_datetime
786 if use_datetime and not datetime:
787 raise ValueError, "the datetime module is not available"
Fredrik Lundhb9056332001-07-11 17:42:21 +0000788
789 def close(self):
790 # return response tuple and target method
791 if self._type is None or self._marks:
792 raise ResponseError()
793 if self._type == "fault":
Guido van Rossum68468eb2003-02-27 20:14:51 +0000794 raise Fault(**self._stack[0])
Fredrik Lundhb9056332001-07-11 17:42:21 +0000795 return tuple(self._stack)
796
797 def getmethodname(self):
798 return self._methodname
799
800 #
801 # event handlers
802
803 def xml(self, encoding, standalone):
804 self._encoding = encoding
805 # FIXME: assert standalone == 1 ???
806
807 def start(self, tag, attrs):
808 # prepare to handle this element
809 if tag == "array" or tag == "struct":
810 self._marks.append(len(self._stack))
811 self._data = []
812 self._value = (tag == "value")
813
814 def data(self, text):
815 self._data.append(text)
816
Fredrik Lundh1538c232001-10-01 19:42:03 +0000817 def end(self, tag, join=string.join):
Fredrik Lundhb9056332001-07-11 17:42:21 +0000818 # call the appropriate end tag handler
819 try:
820 f = self.dispatch[tag]
821 except KeyError:
822 pass # unknown tag ?
823 else:
Fredrik Lundh1538c232001-10-01 19:42:03 +0000824 return f(self, join(self._data, ""))
Fredrik Lundhb9056332001-07-11 17:42:21 +0000825
826 #
827 # accelerator support
828
829 def end_dispatch(self, tag, data):
830 # dispatch data
831 try:
832 f = self.dispatch[tag]
833 except KeyError:
834 pass # unknown tag ?
835 else:
836 return f(self, data)
837
838 #
839 # element decoders
840
841 dispatch = {}
842
Andrew M. Kuchlinga4c2b742003-04-25 00:26:51 +0000843 def end_nil (self, data):
844 self.append(None)
845 self._value = 0
846 dispatch["nil"] = end_nil
Tim Petersc2659cf2003-05-12 20:19:37 +0000847
Fredrik Lundh1538c232001-10-01 19:42:03 +0000848 def end_boolean(self, data):
Fredrik Lundhb9056332001-07-11 17:42:21 +0000849 if data == "0":
850 self.append(False)
851 elif data == "1":
852 self.append(True)
853 else:
854 raise TypeError, "bad boolean value"
855 self._value = 0
856 dispatch["boolean"] = end_boolean
857
Fredrik Lundh1538c232001-10-01 19:42:03 +0000858 def end_int(self, data):
859 self.append(int(data))
Fredrik Lundhb9056332001-07-11 17:42:21 +0000860 self._value = 0
861 dispatch["i4"] = end_int
Georg Brandld0b592f2008-05-29 07:45:26 +0000862 dispatch["i8"] = end_int
Fredrik Lundhb9056332001-07-11 17:42:21 +0000863 dispatch["int"] = end_int
864
Fredrik Lundh1538c232001-10-01 19:42:03 +0000865 def end_double(self, data):
866 self.append(float(data))
Fredrik Lundhb9056332001-07-11 17:42:21 +0000867 self._value = 0
868 dispatch["double"] = end_double
869
Fredrik Lundh1538c232001-10-01 19:42:03 +0000870 def end_string(self, data):
Fredrik Lundhb9056332001-07-11 17:42:21 +0000871 if self._encoding:
872 data = _decode(data, self._encoding)
873 self.append(_stringify(data))
874 self._value = 0
875 dispatch["string"] = end_string
876 dispatch["name"] = end_string # struct keys are always strings
877
878 def end_array(self, data):
Raymond Hettinger46ac8eb2002-06-30 03:39:14 +0000879 mark = self._marks.pop()
Fredrik Lundhb9056332001-07-11 17:42:21 +0000880 # map arrays to Python lists
881 self._stack[mark:] = [self._stack[mark:]]
882 self._value = 0
883 dispatch["array"] = end_array
884
885 def end_struct(self, data):
Raymond Hettinger46ac8eb2002-06-30 03:39:14 +0000886 mark = self._marks.pop()
Fredrik Lundhb9056332001-07-11 17:42:21 +0000887 # map structs to Python dictionaries
888 dict = {}
889 items = self._stack[mark:]
890 for i in range(0, len(items), 2):
891 dict[_stringify(items[i])] = items[i+1]
892 self._stack[mark:] = [dict]
893 self._value = 0
894 dispatch["struct"] = end_struct
895
Fredrik Lundh1538c232001-10-01 19:42:03 +0000896 def end_base64(self, data):
Fredrik Lundhb9056332001-07-11 17:42:21 +0000897 value = Binary()
Fredrik Lundh1538c232001-10-01 19:42:03 +0000898 value.decode(data)
Fredrik Lundhb9056332001-07-11 17:42:21 +0000899 self.append(value)
900 self._value = 0
901 dispatch["base64"] = end_base64
902
Fredrik Lundh1538c232001-10-01 19:42:03 +0000903 def end_dateTime(self, data):
Fredrik Lundhb9056332001-07-11 17:42:21 +0000904 value = DateTime()
Fredrik Lundh1538c232001-10-01 19:42:03 +0000905 value.decode(data)
Skip Montanaro174dd222005-05-14 20:54:16 +0000906 if self._use_datetime:
907 value = _datetime_type(data)
Fredrik Lundhb9056332001-07-11 17:42:21 +0000908 self.append(value)
909 dispatch["dateTime.iso8601"] = end_dateTime
910
911 def end_value(self, data):
Fredrik Lundh3d9addd2002-06-27 21:36:21 +0000912 # if we stumble upon a value element with no internal
Fredrik Lundhb9056332001-07-11 17:42:21 +0000913 # elements, treat it as a string element
914 if self._value:
915 self.end_string(data)
916 dispatch["value"] = end_value
917
918 def end_params(self, data):
919 self._type = "params"
920 dispatch["params"] = end_params
921
922 def end_fault(self, data):
923 self._type = "fault"
924 dispatch["fault"] = end_fault
925
Fredrik Lundh1538c232001-10-01 19:42:03 +0000926 def end_methodName(self, data):
Fredrik Lundhb9056332001-07-11 17:42:21 +0000927 if self._encoding:
928 data = _decode(data, self._encoding)
929 self._methodname = data
930 self._type = "methodName" # no params
931 dispatch["methodName"] = end_methodName
932
Martin v. Löwis45394c22003-10-31 13:49:36 +0000933## Multicall support
934#
Fredrik Lundhb9056332001-07-11 17:42:21 +0000935
Martin v. Löwis45394c22003-10-31 13:49:36 +0000936class _MultiCallMethod:
937 # some lesser magic to store calls made to a MultiCall object
938 # for batch execution
939 def __init__(self, call_list, name):
940 self.__call_list = call_list
941 self.__name = name
942 def __getattr__(self, name):
943 return _MultiCallMethod(self.__call_list, "%s.%s" % (self.__name, name))
944 def __call__(self, *args):
945 self.__call_list.append((self.__name, args))
946
Martin v. Löwis12237b32004-08-22 16:04:50 +0000947class MultiCallIterator:
Martin v. Löwis45394c22003-10-31 13:49:36 +0000948 """Iterates over the results of a multicall. Exceptions are
Andrew Svetlov4bb142b2012-12-18 21:27:37 +0200949 raised in response to xmlrpc faults."""
Raymond Hettingercc523fc2003-11-02 09:47:05 +0000950
Martin v. Löwis12237b32004-08-22 16:04:50 +0000951 def __init__(self, results):
952 self.results = results
953
954 def __getitem__(self, i):
955 item = self.results[i]
956 if type(item) == type({}):
957 raise Fault(item['faultCode'], item['faultString'])
958 elif type(item) == type([]):
959 return item[0]
Martin v. Löwis45394c22003-10-31 13:49:36 +0000960 else:
961 raise ValueError,\
962 "unexpected type in multicall result"
Raymond Hettingercc523fc2003-11-02 09:47:05 +0000963
Martin v. Löwis45394c22003-10-31 13:49:36 +0000964class MultiCall:
965 """server -> a object used to boxcar method calls
966
967 server should be a ServerProxy object.
968
969 Methods can be added to the MultiCall using normal
970 method call syntax e.g.:
971
972 multicall = MultiCall(server_proxy)
973 multicall.add(2,3)
974 multicall.get_address("Guido")
975
976 To execute the multicall, call the MultiCall object e.g.:
977
978 add_result, address = multicall()
979 """
Raymond Hettingercc523fc2003-11-02 09:47:05 +0000980
Martin v. Löwis45394c22003-10-31 13:49:36 +0000981 def __init__(self, server):
982 self.__server = server
983 self.__call_list = []
Raymond Hettingercc523fc2003-11-02 09:47:05 +0000984
Martin v. Löwis45394c22003-10-31 13:49:36 +0000985 def __repr__(self):
986 return "<MultiCall at %x>" % id(self)
Raymond Hettingercc523fc2003-11-02 09:47:05 +0000987
Martin v. Löwis45394c22003-10-31 13:49:36 +0000988 __str__ = __repr__
989
990 def __getattr__(self, name):
991 return _MultiCallMethod(self.__call_list, name)
992
993 def __call__(self):
994 marshalled_list = []
995 for name, args in self.__call_list:
996 marshalled_list.append({'methodName' : name, 'params' : args})
997
998 return MultiCallIterator(self.__server.system.multicall(marshalled_list))
Raymond Hettingercc523fc2003-11-02 09:47:05 +0000999
Fredrik Lundhb9056332001-07-11 17:42:21 +00001000# --------------------------------------------------------------------
1001# convenience functions
1002
Fredrik Lundh3d9addd2002-06-27 21:36:21 +00001003##
1004# Create a parser object, and connect it to an unmarshalling instance.
1005# This function picks the fastest available XML parser.
1006#
1007# return A (parser, unmarshaller) tuple.
1008
Skip Montanaro174dd222005-05-14 20:54:16 +00001009def getparser(use_datetime=0):
Fredrik Lundhb9056332001-07-11 17:42:21 +00001010 """getparser() -> parser, unmarshaller
1011
Fredrik Lundhb0e8e9b2001-09-10 21:45:42 +00001012 Create an instance of the fastest available parser, and attach it
1013 to an unmarshalling object. Return both objects.
Fredrik Lundhb9056332001-07-11 17:42:21 +00001014 """
Skip Montanaro174dd222005-05-14 20:54:16 +00001015 if use_datetime and not datetime:
1016 raise ValueError, "the datetime module is not available"
Fredrik Lundhb9056332001-07-11 17:42:21 +00001017 if FastParser and FastUnmarshaller:
Skip Montanaro174dd222005-05-14 20:54:16 +00001018 if use_datetime:
1019 mkdatetime = _datetime_type
1020 else:
1021 mkdatetime = _datetime
1022 target = FastUnmarshaller(True, False, _binary, mkdatetime, Fault)
Fredrik Lundhb9056332001-07-11 17:42:21 +00001023 parser = FastParser(target)
1024 else:
Skip Montanaro174dd222005-05-14 20:54:16 +00001025 target = Unmarshaller(use_datetime=use_datetime)
Fredrik Lundhb9056332001-07-11 17:42:21 +00001026 if FastParser:
1027 parser = FastParser(target)
Fredrik Lundhb9056332001-07-11 17:42:21 +00001028 elif ExpatParser:
1029 parser = ExpatParser(target)
1030 else:
1031 parser = SlowParser(target)
1032 return parser, target
1033
Fredrik Lundh3d9addd2002-06-27 21:36:21 +00001034##
1035# Convert a Python tuple or a Fault instance to an XML-RPC packet.
1036#
1037# @def dumps(params, **options)
1038# @param params A tuple or Fault instance.
1039# @keyparam methodname If given, create a methodCall request for
1040# this method name.
1041# @keyparam methodresponse If given, create a methodResponse packet.
1042# If used with a tuple, the tuple must be a singleton (that is,
1043# it must contain exactly one element).
1044# @keyparam encoding The packet encoding.
1045# @return A string containing marshalled data.
1046
Andrew M. Kuchlinga4c2b742003-04-25 00:26:51 +00001047def dumps(params, methodname=None, methodresponse=None, encoding=None,
1048 allow_none=0):
Fredrik Lundhb9056332001-07-11 17:42:21 +00001049 """data [,options] -> marshalled data
1050
1051 Convert an argument tuple or a Fault instance to an XML-RPC
1052 request (or response, if the methodresponse option is used).
1053
Fredrik Lundhb0e8e9b2001-09-10 21:45:42 +00001054 In addition to the data object, the following options can be given
1055 as keyword arguments:
Fredrik Lundhb9056332001-07-11 17:42:21 +00001056
1057 methodname: the method name for a methodCall packet
1058
1059 methodresponse: true to create a methodResponse packet.
1060 If this option is used with a tuple, the tuple must be
1061 a singleton (i.e. it can contain only one element).
1062
1063 encoding: the packet encoding (default is UTF-8)
1064
1065 All 8-bit strings in the data structure are assumed to use the
1066 packet encoding. Unicode strings are automatically converted,
Fredrik Lundhb0e8e9b2001-09-10 21:45:42 +00001067 where necessary.
Fredrik Lundhb9056332001-07-11 17:42:21 +00001068 """
1069
Raymond Hettingerfe59dc12005-02-07 15:28:45 +00001070 assert isinstance(params, TupleType) or isinstance(params, Fault),\
Fredrik Lundhb9056332001-07-11 17:42:21 +00001071 "argument must be tuple or Fault instance"
1072
1073 if isinstance(params, Fault):
1074 methodresponse = 1
Raymond Hettingerfe59dc12005-02-07 15:28:45 +00001075 elif methodresponse and isinstance(params, TupleType):
Fredrik Lundhb9056332001-07-11 17:42:21 +00001076 assert len(params) == 1, "response tuple must be a singleton"
1077
Fredrik Lundh3d9addd2002-06-27 21:36:21 +00001078 if not encoding:
Fredrik Lundhb9056332001-07-11 17:42:21 +00001079 encoding = "utf-8"
1080
Fredrik Lundh3d9addd2002-06-27 21:36:21 +00001081 if FastMarshaller:
1082 m = FastMarshaller(encoding)
1083 else:
Andrew M. Kuchlinga4c2b742003-04-25 00:26:51 +00001084 m = Marshaller(encoding, allow_none)
Fredrik Lundh3d9addd2002-06-27 21:36:21 +00001085
Fredrik Lundhb9056332001-07-11 17:42:21 +00001086 data = m.dumps(params)
1087
1088 if encoding != "utf-8":
Fredrik Lundh3d9addd2002-06-27 21:36:21 +00001089 xmlheader = "<?xml version='1.0' encoding='%s'?>\n" % str(encoding)
Fredrik Lundhb9056332001-07-11 17:42:21 +00001090 else:
1091 xmlheader = "<?xml version='1.0'?>\n" # utf-8 is default
1092
1093 # standard XML-RPC wrappings
Fredrik Lundh3d9addd2002-06-27 21:36:21 +00001094 if methodname:
Fredrik Lundhb9056332001-07-11 17:42:21 +00001095 # a method call
Raymond Hettingerfe59dc12005-02-07 15:28:45 +00001096 if not isinstance(methodname, StringType):
Fredrik Lundhb9056332001-07-11 17:42:21 +00001097 methodname = methodname.encode(encoding)
1098 data = (
1099 xmlheader,
1100 "<methodCall>\n"
1101 "<methodName>", methodname, "</methodName>\n",
1102 data,
1103 "</methodCall>\n"
1104 )
1105 elif methodresponse:
1106 # a method response, or a fault structure
1107 data = (
1108 xmlheader,
1109 "<methodResponse>\n",
1110 data,
1111 "</methodResponse>\n"
1112 )
1113 else:
1114 return data # return as is
1115 return string.join(data, "")
1116
Fredrik Lundh3d9addd2002-06-27 21:36:21 +00001117##
1118# Convert an XML-RPC packet to a Python object. If the XML-RPC packet
1119# represents a fault condition, this function raises a Fault exception.
1120#
1121# @param data An XML-RPC packet, given as an 8-bit string.
Walter Dörwaldf0dfc7a2003-10-20 14:01:56 +00001122# @return A tuple containing the unpacked data, and the method name
Fredrik Lundh3d9addd2002-06-27 21:36:21 +00001123# (None if not present).
1124# @see Fault
1125
Skip Montanaro174dd222005-05-14 20:54:16 +00001126def loads(data, use_datetime=0):
Fredrik Lundhb9056332001-07-11 17:42:21 +00001127 """data -> unmarshalled data, method name
1128
1129 Convert an XML-RPC packet to unmarshalled data plus a method
1130 name (None if not present).
1131
1132 If the XML-RPC packet represents a fault condition, this function
1133 raises a Fault exception.
1134 """
Skip Montanaro174dd222005-05-14 20:54:16 +00001135 p, u = getparser(use_datetime=use_datetime)
Fredrik Lundhb9056332001-07-11 17:42:21 +00001136 p.feed(data)
1137 p.close()
1138 return u.close(), u.getmethodname()
1139
Kristján Valur Jónssone0078602009-06-28 21:04:17 +00001140##
1141# Encode a string using the gzip content encoding such as specified by the
1142# Content-Encoding: gzip
1143# in the HTTP header, as described in RFC 1952
1144#
1145# @param data the unencoded data
1146# @return the encoded data
1147
1148def gzip_encode(data):
1149 """data -> gzip encoded data
1150
1151 Encode data using the gzip content encoding as described in RFC 1952
1152 """
Kristján Valur Jónsson6d755902009-07-19 22:14:00 +00001153 if not gzip:
1154 raise NotImplementedError
Kristján Valur Jónssone0078602009-06-28 21:04:17 +00001155 f = StringIO.StringIO()
1156 gzf = gzip.GzipFile(mode="wb", fileobj=f, compresslevel=1)
1157 gzf.write(data)
1158 gzf.close()
1159 encoded = f.getvalue()
1160 f.close()
1161 return encoded
1162
1163##
1164# Decode a string using the gzip content encoding such as specified by the
1165# Content-Encoding: gzip
1166# in the HTTP header, as described in RFC 1952
1167#
1168# @param data The encoded data
Benjamin Peterson9e8f5232014-12-05 20:15:15 -05001169# @keyparam max_decode Maximum bytes to decode (20MB default), use negative
1170# values for unlimited decoding
Kristján Valur Jónssone0078602009-06-28 21:04:17 +00001171# @return the unencoded data
1172# @raises ValueError if data is not correctly coded.
Benjamin Peterson9e8f5232014-12-05 20:15:15 -05001173# @raises ValueError if max gzipped payload length exceeded
Kristján Valur Jónssone0078602009-06-28 21:04:17 +00001174
Benjamin Peterson9e8f5232014-12-05 20:15:15 -05001175def gzip_decode(data, max_decode=20971520):
Kristján Valur Jónssone0078602009-06-28 21:04:17 +00001176 """gzip encoded data -> unencoded data
1177
1178 Decode data using the gzip content encoding as described in RFC 1952
1179 """
Kristján Valur Jónsson6d755902009-07-19 22:14:00 +00001180 if not gzip:
1181 raise NotImplementedError
Kristján Valur Jónssone0078602009-06-28 21:04:17 +00001182 f = StringIO.StringIO(data)
1183 gzf = gzip.GzipFile(mode="rb", fileobj=f)
1184 try:
Benjamin Peterson9e8f5232014-12-05 20:15:15 -05001185 if max_decode < 0: # no limit
1186 decoded = gzf.read()
1187 else:
1188 decoded = gzf.read(max_decode + 1)
Kristján Valur Jónssone0078602009-06-28 21:04:17 +00001189 except IOError:
1190 raise ValueError("invalid data")
1191 f.close()
1192 gzf.close()
Benjamin Peterson9e8f5232014-12-05 20:15:15 -05001193 if max_decode >= 0 and len(decoded) > max_decode:
1194 raise ValueError("max gzipped payload length exceeded")
Kristján Valur Jónssone0078602009-06-28 21:04:17 +00001195 return decoded
1196
1197##
1198# Return a decoded file-like object for the gzip encoding
1199# as described in RFC 1952.
1200#
1201# @param response A stream supporting a read() method
1202# @return a file-like object that the decoded data can be read() from
1203
Kristján Valur Jónssonc0379432009-07-19 22:35:44 +00001204class GzipDecodedResponse(gzip.GzipFile if gzip else object):
Kristján Valur Jónssone0078602009-06-28 21:04:17 +00001205 """a file-like object to decode a response encoded with the gzip
1206 method, as described in RFC 1952.
1207 """
1208 def __init__(self, response):
1209 #response doesn't support tell() and read(), required by
1210 #GzipFile
Kristján Valur Jónsson6d755902009-07-19 22:14:00 +00001211 if not gzip:
1212 raise NotImplementedError
Kristján Valur Jónssone0078602009-06-28 21:04:17 +00001213 self.stringio = StringIO.StringIO(response.read())
1214 gzip.GzipFile.__init__(self, mode="rb", fileobj=self.stringio)
1215
1216 def close(self):
1217 gzip.GzipFile.close(self)
1218 self.stringio.close()
1219
Fredrik Lundhb9056332001-07-11 17:42:21 +00001220
1221# --------------------------------------------------------------------
1222# request dispatcher
1223
1224class _Method:
1225 # some magic to bind an XML-RPC method to an RPC server.
1226 # supports "nested" methods (e.g. examples.getStateName)
1227 def __init__(self, send, name):
1228 self.__send = send
1229 self.__name = name
1230 def __getattr__(self, name):
1231 return _Method(self.__send, "%s.%s" % (self.__name, name))
1232 def __call__(self, *args):
1233 return self.__send(self.__name, args)
1234
Fredrik Lundh3d9addd2002-06-27 21:36:21 +00001235##
1236# Standard transport class for XML-RPC over HTTP.
1237# <p>
1238# You can create custom transports by subclassing this method, and
1239# overriding selected methods.
Fredrik Lundhb9056332001-07-11 17:42:21 +00001240
1241class Transport:
Fredrik Lundhc4c062f2001-09-10 19:45:02 +00001242 """Handles an HTTP transaction to an XML-RPC server."""
Fredrik Lundhb9056332001-07-11 17:42:21 +00001243
1244 # client identifier (may be overridden)
1245 user_agent = "xmlrpclib.py/%s (by www.pythonware.com)" % __version__
1246
Kristján Valur Jónssone0078602009-06-28 21:04:17 +00001247 #if true, we'll request gzip encoding
1248 accept_gzip_encoding = True
1249
1250 # if positive, encode request using gzip if it exceeds this threshold
1251 # note that many server will get confused, so only use it if you know
1252 # that they can decode such a request
1253 encode_threshold = None #None = don't encode
1254
Skip Montanaro174dd222005-05-14 20:54:16 +00001255 def __init__(self, use_datetime=0):
1256 self._use_datetime = use_datetime
Kristján Valur Jónssone0078602009-06-28 21:04:17 +00001257 self._connection = (None, None)
1258 self._extra_headers = []
1259 ##
1260 # Send a complete request, and parse the response.
1261 # Retry request if a cached connection has disconnected.
1262 #
1263 # @param host Target host.
1264 # @param handler Target PRC handler.
1265 # @param request_body XML-RPC request body.
1266 # @param verbose Debugging flag.
1267 # @return Parsed response.
1268
1269 def request(self, host, handler, request_body, verbose=0):
1270 #retry request once if cached connection has gone cold
1271 for i in (0, 1):
1272 try:
1273 return self.single_request(host, handler, request_body, verbose)
Kristján Valur Jónssonf83648e2009-07-03 23:26:02 +00001274 except socket.error, e:
Victor Stinner8b0d84e2010-07-24 02:51:49 +00001275 if i or e.errno not in (errno.ECONNRESET, errno.ECONNABORTED, errno.EPIPE):
Kristján Valur Jónssonf83648e2009-07-03 23:26:02 +00001276 raise
Kristján Valur Jónssone5445da2009-07-03 23:29:50 +00001277 except httplib.BadStatusLine: #close after we sent request
Kristján Valur Jónssonf83648e2009-07-03 23:26:02 +00001278 if i:
Kristján Valur Jónssone0078602009-06-28 21:04:17 +00001279 raise
Skip Montanaro174dd222005-05-14 20:54:16 +00001280
Fredrik Lundh3d9addd2002-06-27 21:36:21 +00001281 ##
1282 # Send a complete request, and parse the response.
1283 #
1284 # @param host Target host.
1285 # @param handler Target PRC handler.
1286 # @param request_body XML-RPC request body.
1287 # @param verbose Debugging flag.
1288 # @return Parsed response.
1289
Kristján Valur Jónssone0078602009-06-28 21:04:17 +00001290 def single_request(self, host, handler, request_body, verbose=0):
Fredrik Lundhb9056332001-07-11 17:42:21 +00001291 # issue XML-RPC request
1292
1293 h = self.make_connection(host)
1294 if verbose:
1295 h.set_debuglevel(1)
1296
Kristján Valur Jónssone0078602009-06-28 21:04:17 +00001297 try:
1298 self.send_request(h, handler, request_body)
1299 self.send_host(h, host)
1300 self.send_user_agent(h)
1301 self.send_content(h, request_body)
Fredrik Lundhb9056332001-07-11 17:42:21 +00001302
Kristján Valur Jónssone0078602009-06-28 21:04:17 +00001303 response = h.getresponse(buffering=True)
1304 if response.status == 200:
1305 self.verbose = verbose
1306 return self.parse_response(response)
1307 except Fault:
1308 raise
1309 except Exception:
1310 # All unexpected errors leave connection in
1311 # a strange state, so we clear it.
1312 self.close()
1313 raise
Fredrik Lundhb9056332001-07-11 17:42:21 +00001314
Kristján Valur Jónssone0078602009-06-28 21:04:17 +00001315 #discard any response data and raise exception
1316 if (response.getheader("content-length", 0)):
1317 response.read()
1318 raise ProtocolError(
1319 host + handler,
1320 response.status, response.reason,
1321 response.msg,
1322 )
Fredrik Lundh3d9addd2002-06-27 21:36:21 +00001323
1324 ##
1325 # Create parser.
1326 #
1327 # @return A 2-tuple containing a parser and a unmarshaller.
Fredrik Lundhb9056332001-07-11 17:42:21 +00001328
Fredrik Lundhc4c062f2001-09-10 19:45:02 +00001329 def getparser(self):
1330 # get parser and unmarshaller
Skip Montanaro174dd222005-05-14 20:54:16 +00001331 return getparser(use_datetime=self._use_datetime)
Fredrik Lundhc4c062f2001-09-10 19:45:02 +00001332
Fredrik Lundh3d9addd2002-06-27 21:36:21 +00001333 ##
Fredrik Lundh1303c7c2002-10-22 18:23:00 +00001334 # Get authorization info from host parameter
1335 # Host may be a string, or a (host, x509-dict) tuple; if a string,
1336 # it is checked for a "user:pw@host" format, and a "Basic
1337 # Authentication" header is added if appropriate.
1338 #
1339 # @param host Host descriptor (URL or (URL, x509 info) tuple).
1340 # @return A 3-tuple containing (actual host, extra headers,
1341 # x509 info). The header and x509 fields may be None.
1342
1343 def get_host_info(self, host):
1344
1345 x509 = {}
Raymond Hettingerfe59dc12005-02-07 15:28:45 +00001346 if isinstance(host, TupleType):
Fredrik Lundh1303c7c2002-10-22 18:23:00 +00001347 host, x509 = host
1348
1349 import urllib
1350 auth, host = urllib.splituser(host)
1351
1352 if auth:
1353 import base64
Fredrik Lundh768c98b2002-11-01 17:14:16 +00001354 auth = base64.encodestring(urllib.unquote(auth))
Fredrik Lundh1303c7c2002-10-22 18:23:00 +00001355 auth = string.join(string.split(auth), "") # get rid of whitespace
1356 extra_headers = [
1357 ("Authorization", "Basic " + auth)
1358 ]
1359 else:
1360 extra_headers = None
1361
1362 return host, extra_headers, x509
1363
1364 ##
Fredrik Lundh3d9addd2002-06-27 21:36:21 +00001365 # Connect to server.
1366 #
1367 # @param host Target host.
1368 # @return A connection handle.
1369
Fredrik Lundhb9056332001-07-11 17:42:21 +00001370 def make_connection(self, host):
Kristján Valur Jónssone0078602009-06-28 21:04:17 +00001371 #return an existing connection if possible. This allows
1372 #HTTP/1.1 keep-alive.
1373 if self._connection and host == self._connection[0]:
1374 return self._connection[1]
1375
Fredrik Lundhb9056332001-07-11 17:42:21 +00001376 # create a HTTP connection object from a host descriptor
Kristján Valur Jónssone0078602009-06-28 21:04:17 +00001377 chost, self._extra_headers, x509 = self.get_host_info(host)
1378 #store the host argument along with the connection object
1379 self._connection = host, httplib.HTTPConnection(chost)
1380 return self._connection[1]
1381
1382 ##
1383 # Clear any cached connection object.
1384 # Used in the event of socket errors.
1385 #
1386 def close(self):
1387 if self._connection[1]:
1388 self._connection[1].close()
1389 self._connection = (None, None)
Fredrik Lundhb9056332001-07-11 17:42:21 +00001390
Fredrik Lundh3d9addd2002-06-27 21:36:21 +00001391 ##
1392 # Send request header.
1393 #
1394 # @param connection Connection handle.
1395 # @param handler Target RPC handler.
1396 # @param request_body XML-RPC body.
1397
Fredrik Lundhb9056332001-07-11 17:42:21 +00001398 def send_request(self, connection, handler, request_body):
Kristján Valur Jónsson6d755902009-07-19 22:14:00 +00001399 if (self.accept_gzip_encoding and gzip):
Kristján Valur Jónssone0078602009-06-28 21:04:17 +00001400 connection.putrequest("POST", handler, skip_accept_encoding=True)
1401 connection.putheader("Accept-Encoding", "gzip")
1402 else:
1403 connection.putrequest("POST", handler)
Fredrik Lundhb9056332001-07-11 17:42:21 +00001404
Fredrik Lundh3d9addd2002-06-27 21:36:21 +00001405 ##
1406 # Send host name.
1407 #
1408 # @param connection Connection handle.
1409 # @param host Host name.
Kristján Valur Jónssone0078602009-06-28 21:04:17 +00001410 #
1411 # Note: This function doesn't actually add the "Host"
1412 # header anymore, it is done as part of the connection.putrequest() in
1413 # send_request() above.
Fredrik Lundh3d9addd2002-06-27 21:36:21 +00001414
Fredrik Lundhb9056332001-07-11 17:42:21 +00001415 def send_host(self, connection, host):
Kristján Valur Jónssone0078602009-06-28 21:04:17 +00001416 extra_headers = self._extra_headers
Fredrik Lundh1303c7c2002-10-22 18:23:00 +00001417 if extra_headers:
Raymond Hettingerfe59dc12005-02-07 15:28:45 +00001418 if isinstance(extra_headers, DictType):
Fredrik Lundh1303c7c2002-10-22 18:23:00 +00001419 extra_headers = extra_headers.items()
1420 for key, value in extra_headers:
1421 connection.putheader(key, value)
Fredrik Lundhb9056332001-07-11 17:42:21 +00001422
Fredrik Lundh3d9addd2002-06-27 21:36:21 +00001423 ##
1424 # Send user-agent identifier.
1425 #
1426 # @param connection Connection handle.
1427
Fredrik Lundhb9056332001-07-11 17:42:21 +00001428 def send_user_agent(self, connection):
1429 connection.putheader("User-Agent", self.user_agent)
1430
Fredrik Lundh3d9addd2002-06-27 21:36:21 +00001431 ##
1432 # Send request body.
1433 #
1434 # @param connection Connection handle.
1435 # @param request_body XML-RPC request body.
1436
Fredrik Lundhb9056332001-07-11 17:42:21 +00001437 def send_content(self, connection, request_body):
1438 connection.putheader("Content-Type", "text/xml")
Kristján Valur Jónssone0078602009-06-28 21:04:17 +00001439
1440 #optionally encode the request
1441 if (self.encode_threshold is not None and
Kristján Valur Jónsson6d755902009-07-19 22:14:00 +00001442 self.encode_threshold < len(request_body) and
1443 gzip):
Kristján Valur Jónssone0078602009-06-28 21:04:17 +00001444 connection.putheader("Content-Encoding", "gzip")
1445 request_body = gzip_encode(request_body)
1446
Fredrik Lundhb9056332001-07-11 17:42:21 +00001447 connection.putheader("Content-Length", str(len(request_body)))
Jeremy Hylton0381f482008-11-29 01:09:35 +00001448 connection.endheaders(request_body)
Fredrik Lundhb9056332001-07-11 17:42:21 +00001449
Fredrik Lundh3d9addd2002-06-27 21:36:21 +00001450 ##
1451 # Parse response.
1452 #
1453 # @param file Stream.
1454 # @return Response tuple and target method.
1455
Kristján Valur Jónssone0078602009-06-28 21:04:17 +00001456 def parse_response(self, response):
1457 # read response data from httpresponse, and parse it
Senthil Kumarand8e4edd2010-12-07 07:10:04 +00001458
1459 # Check for new http response object, else it is a file object
1460 if hasattr(response,'getheader'):
1461 if response.getheader("Content-Encoding", "") == "gzip":
1462 stream = GzipDecodedResponse(response)
1463 else:
1464 stream = response
Kristján Valur Jónssone0078602009-06-28 21:04:17 +00001465 else:
1466 stream = response
Fredrik Lundhb9056332001-07-11 17:42:21 +00001467
Fredrik Lundhc4c062f2001-09-10 19:45:02 +00001468 p, u = self.getparser()
Fredrik Lundhb9056332001-07-11 17:42:21 +00001469
1470 while 1:
Kristján Valur Jónssone0078602009-06-28 21:04:17 +00001471 data = stream.read(1024)
1472 if not data:
Fredrik Lundhb9056332001-07-11 17:42:21 +00001473 break
1474 if self.verbose:
Kristján Valur Jónssone0078602009-06-28 21:04:17 +00001475 print "body:", repr(data)
1476 p.feed(data)
Fredrik Lundhb9056332001-07-11 17:42:21 +00001477
Kristján Valur Jónssone0078602009-06-28 21:04:17 +00001478 if stream is not response:
1479 stream.close()
Fredrik Lundhb9056332001-07-11 17:42:21 +00001480 p.close()
1481
1482 return u.close()
1483
Fredrik Lundh3d9addd2002-06-27 21:36:21 +00001484##
1485# Standard transport class for XML-RPC over HTTPS.
1486
Fredrik Lundhb9056332001-07-11 17:42:21 +00001487class SafeTransport(Transport):
Fredrik Lundhc4c062f2001-09-10 19:45:02 +00001488 """Handles an HTTPS transaction to an XML-RPC server."""
Fredrik Lundhb9056332001-07-11 17:42:21 +00001489
Benjamin Petersonefa3cf82014-11-29 22:55:35 -05001490 def __init__(self, use_datetime=0, context=None):
1491 Transport.__init__(self, use_datetime=use_datetime)
1492 self.context = context
1493
Fredrik Lundh3d9addd2002-06-27 21:36:21 +00001494 # FIXME: mostly untested
1495
Fredrik Lundhb9056332001-07-11 17:42:21 +00001496 def make_connection(self, host):
Kristján Valur Jónssone0078602009-06-28 21:04:17 +00001497 if self._connection and host == self._connection[0]:
1498 return self._connection[1]
Fredrik Lundhb9056332001-07-11 17:42:21 +00001499 # create a HTTPS connection object from a host descriptor
1500 # host may be a string, or a (host, x509-dict) tuple
Fredrik Lundhb9056332001-07-11 17:42:21 +00001501 try:
Kristján Valur Jónssone0078602009-06-28 21:04:17 +00001502 HTTPS = httplib.HTTPSConnection
Fredrik Lundhb9056332001-07-11 17:42:21 +00001503 except AttributeError:
Fredrik Lundh1303c7c2002-10-22 18:23:00 +00001504 raise NotImplementedError(
1505 "your version of httplib doesn't support HTTPS"
1506 )
Fredrik Lundhb9056332001-07-11 17:42:21 +00001507 else:
Kristján Valur Jónssone0078602009-06-28 21:04:17 +00001508 chost, self._extra_headers, x509 = self.get_host_info(host)
Benjamin Petersonee443142014-11-29 23:06:06 -05001509 self._connection = host, HTTPS(chost, None, context=self.context, **(x509 or {}))
Kristján Valur Jónssone0078602009-06-28 21:04:17 +00001510 return self._connection[1]
Fredrik Lundhb9056332001-07-11 17:42:21 +00001511
Fredrik Lundh3d9addd2002-06-27 21:36:21 +00001512##
1513# Standard server proxy. This class establishes a virtual connection
1514# to an XML-RPC server.
1515# <p>
1516# This class is available as ServerProxy and Server. New code should
1517# use ServerProxy, to avoid confusion.
1518#
1519# @def ServerProxy(uri, **options)
1520# @param uri The connection point on the server.
1521# @keyparam transport A transport factory, compatible with the
1522# standard transport class.
1523# @keyparam encoding The default encoding used for 8-bit strings
1524# (default is UTF-8).
1525# @keyparam verbose Use a true value to enable debugging output.
1526# (printed to standard output).
1527# @see Transport
1528
Fredrik Lundhb9056332001-07-11 17:42:21 +00001529class ServerProxy:
1530 """uri [,options] -> a logical connection to an XML-RPC server
1531
1532 uri is the connection point on the server, given as
1533 scheme://host/target.
1534
1535 The standard implementation always supports the "http" scheme. If
1536 SSL socket support is available (Python 2.0), it also supports
1537 "https".
1538
1539 If the target part and the slash preceding it are both omitted,
1540 "/RPC2" is assumed.
1541
1542 The following options can be given as keyword arguments:
1543
1544 transport: a transport factory
1545 encoding: the request encoding (default is UTF-8)
1546
1547 All 8-bit strings passed to the server proxy are assumed to use
1548 the given encoding.
1549 """
1550
Andrew M. Kuchlinga4c2b742003-04-25 00:26:51 +00001551 def __init__(self, uri, transport=None, encoding=None, verbose=0,
Benjamin Petersonefa3cf82014-11-29 22:55:35 -05001552 allow_none=0, use_datetime=0, context=None):
Fredrik Lundhb9056332001-07-11 17:42:21 +00001553 # establish a "logical" server connection
1554
Victor Stinner51b71982011-09-23 01:15:32 +02001555 if isinstance(uri, unicode):
1556 uri = uri.encode('ISO-8859-1')
1557
Fredrik Lundhb9056332001-07-11 17:42:21 +00001558 # get the url
Fredrik Lundhc4c062f2001-09-10 19:45:02 +00001559 import urllib
Fredrik Lundhb9056332001-07-11 17:42:21 +00001560 type, uri = urllib.splittype(uri)
1561 if type not in ("http", "https"):
1562 raise IOError, "unsupported XML-RPC protocol"
1563 self.__host, self.__handler = urllib.splithost(uri)
1564 if not self.__handler:
1565 self.__handler = "/RPC2"
1566
1567 if transport is None:
1568 if type == "https":
Benjamin Petersonefa3cf82014-11-29 22:55:35 -05001569 transport = SafeTransport(use_datetime=use_datetime, context=context)
Fredrik Lundhb9056332001-07-11 17:42:21 +00001570 else:
Skip Montanaro174dd222005-05-14 20:54:16 +00001571 transport = Transport(use_datetime=use_datetime)
Fredrik Lundhb9056332001-07-11 17:42:21 +00001572 self.__transport = transport
1573
1574 self.__encoding = encoding
1575 self.__verbose = verbose
Andrew M. Kuchlinga4c2b742003-04-25 00:26:51 +00001576 self.__allow_none = allow_none
Tim Petersc2659cf2003-05-12 20:19:37 +00001577
Kristján Valur Jónssone0078602009-06-28 21:04:17 +00001578 def __close(self):
1579 self.__transport.close()
1580
Fredrik Lundhb9056332001-07-11 17:42:21 +00001581 def __request(self, methodname, params):
1582 # call a method on the remote server
1583
Andrew M. Kuchlinga4c2b742003-04-25 00:26:51 +00001584 request = dumps(params, methodname, encoding=self.__encoding,
1585 allow_none=self.__allow_none)
Fredrik Lundhb9056332001-07-11 17:42:21 +00001586
1587 response = self.__transport.request(
1588 self.__host,
1589 self.__handler,
1590 request,
1591 verbose=self.__verbose
1592 )
1593
1594 if len(response) == 1:
1595 response = response[0]
1596
1597 return response
1598
1599 def __repr__(self):
1600 return (
Fredrik Lundh78eedce2001-08-23 20:04:33 +00001601 "<ServerProxy for %s%s>" %
Fredrik Lundhb9056332001-07-11 17:42:21 +00001602 (self.__host, self.__handler)
1603 )
1604
1605 __str__ = __repr__
Raymond Hettingercc523fc2003-11-02 09:47:05 +00001606
Fredrik Lundhb9056332001-07-11 17:42:21 +00001607 def __getattr__(self, name):
1608 # magic method dispatcher
1609 return _Method(self.__request, name)
1610
1611 # note: to call a remote object with an non-standard name, use
1612 # result getattr(server, "strange-python-name")(args)
1613
Kristján Valur Jónssone0078602009-06-28 21:04:17 +00001614 def __call__(self, attr):
1615 """A workaround to get special attributes on the ServerProxy
1616 without interfering with the magic __getattr__
1617 """
1618 if attr == "close":
1619 return self.__close
1620 elif attr == "transport":
1621 return self.__transport
1622 raise AttributeError("Attribute %r not found" % (attr,))
1623
Fredrik Lundh78eedce2001-08-23 20:04:33 +00001624# compatibility
Fredrik Lundh3d9addd2002-06-27 21:36:21 +00001625
Fredrik Lundhb9056332001-07-11 17:42:21 +00001626Server = ServerProxy
1627
1628# --------------------------------------------------------------------
1629# test code
1630
1631if __name__ == "__main__":
1632
Senthil Kumaran01109732014-01-12 16:04:08 -08001633 server = ServerProxy("http://localhost:8000")
Fredrik Lundhb9056332001-07-11 17:42:21 +00001634
1635 print server
1636
Martin v. Löwis12237b32004-08-22 16:04:50 +00001637 multi = MultiCall(server)
Senthil Kumaran01109732014-01-12 16:04:08 -08001638 multi.pow(2, 9)
1639 multi.add(5, 1)
1640 multi.add(24, 11)
Martin v. Löwis12237b32004-08-22 16:04:50 +00001641 try:
1642 for response in multi():
1643 print response
Fredrik Lundhb9056332001-07-11 17:42:21 +00001644 except Error, v:
1645 print "ERROR", v