blob: b0e748528302628c65e166f3432a226469ead1e1 [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
Serhiy Storchaka9a118f12016-04-17 09:37:36 +0300361# tuple, or an 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")
Serhiy Storchaka2f173fe2016-01-18 19:35:23 +0200396 elif isinstance(other, basestring):
Andrew M. Kuchling085f75a2008-02-23 16:23:05 +0000397 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):
Serhiy Storchaka1aa2c0f2015-04-10 13:24:10 +0300561 try:
562 parser = self._parser
563 except AttributeError:
564 pass
565 else:
566 del self._target, self._parser # get rid of circular references
567 parser.Parse("", 1) # end of data
Fredrik Lundhb9056332001-07-11 17:42:21 +0000568
Fredrik Lundhc4c062f2001-09-10 19:45:02 +0000569class SlowParser:
570 """Default XML parser (based on xmllib.XMLParser)."""
Georg Brandle6632b42009-06-04 08:58:32 +0000571 # this is the slowest parser.
Fredrik Lundhb9056332001-07-11 17:42:21 +0000572 def __init__(self, target):
Fredrik Lundhc4c062f2001-09-10 19:45:02 +0000573 import xmllib # lazy subclassing (!)
574 if xmllib.XMLParser not in SlowParser.__bases__:
575 SlowParser.__bases__ = (xmllib.XMLParser,)
Fredrik Lundhb9056332001-07-11 17:42:21 +0000576 self.handle_xml = target.xml
577 self.unknown_starttag = target.start
578 self.handle_data = target.data
Fredrik Lundh3d9addd2002-06-27 21:36:21 +0000579 self.handle_cdata = target.data
Fredrik Lundhb9056332001-07-11 17:42:21 +0000580 self.unknown_endtag = target.end
Fredrik Lundhb0e8e9b2001-09-10 21:45:42 +0000581 try:
582 xmllib.XMLParser.__init__(self, accept_utf8=1)
583 except TypeError:
584 xmllib.XMLParser.__init__(self) # pre-2.0
Fredrik Lundhb9056332001-07-11 17:42:21 +0000585
586# --------------------------------------------------------------------
587# XML-RPC marshalling and unmarshalling code
588
Fredrik Lundh3d9addd2002-06-27 21:36:21 +0000589##
590# XML-RPC marshaller.
591#
592# @param encoding Default encoding for 8-bit strings. The default
593# value is None (interpreted as UTF-8).
594# @see dumps
595
Fredrik Lundhb9056332001-07-11 17:42:21 +0000596class Marshaller:
Fred Drake1b410792001-09-04 18:55:03 +0000597 """Generate an XML-RPC params chunk from a Python data structure.
Fredrik Lundhb9056332001-07-11 17:42:21 +0000598
Fredrik Lundhc4c062f2001-09-10 19:45:02 +0000599 Create a Marshaller instance for each set of parameters, and use
600 the "dumps" method to convert your data (represented as a tuple)
601 to an XML-RPC params chunk. To write a fault response, pass a
602 Fault instance instead. You may prefer to use the "dumps" module
603 function for this purpose.
Fred Drake1b410792001-09-04 18:55:03 +0000604 """
Fredrik Lundhb9056332001-07-11 17:42:21 +0000605
606 # by the way, if you don't understand what's going on in here,
607 # that's perfectly ok.
608
Andrew M. Kuchlinga4c2b742003-04-25 00:26:51 +0000609 def __init__(self, encoding=None, allow_none=0):
Fredrik Lundhb9056332001-07-11 17:42:21 +0000610 self.memo = {}
611 self.data = None
612 self.encoding = encoding
Andrew M. Kuchlinga4c2b742003-04-25 00:26:51 +0000613 self.allow_none = allow_none
Tim Petersc2659cf2003-05-12 20:19:37 +0000614
Fredrik Lundhb9056332001-07-11 17:42:21 +0000615 dispatch = {}
616
617 def dumps(self, values):
Fredrik Lundh3d9addd2002-06-27 21:36:21 +0000618 out = []
619 write = out.append
620 dump = self.__dump
Fredrik Lundhb9056332001-07-11 17:42:21 +0000621 if isinstance(values, Fault):
622 # fault instance
623 write("<fault>\n")
Martin v. Löwis541342f2003-07-12 07:53:04 +0000624 dump({'faultCode': values.faultCode,
625 'faultString': values.faultString},
626 write)
Fredrik Lundhb9056332001-07-11 17:42:21 +0000627 write("</fault>\n")
628 else:
629 # parameter block
Fredrik Lundhc266bb02001-08-23 20:13:08 +0000630 # FIXME: the xml-rpc specification allows us to leave out
631 # the entire <params> block if there are no parameters.
632 # however, changing this may break older code (including
633 # old versions of xmlrpclib.py), so this is better left as
634 # is for now. See @XMLRPC3 for more information. /F
Fredrik Lundhb9056332001-07-11 17:42:21 +0000635 write("<params>\n")
636 for v in values:
637 write("<param>\n")
Fredrik Lundh3d9addd2002-06-27 21:36:21 +0000638 dump(v, write)
Fredrik Lundhb9056332001-07-11 17:42:21 +0000639 write("</param>\n")
640 write("</params>\n")
Fredrik Lundh3d9addd2002-06-27 21:36:21 +0000641 result = string.join(out, "")
Fredrik Lundhb9056332001-07-11 17:42:21 +0000642 return result
643
Fredrik Lundh3d9addd2002-06-27 21:36:21 +0000644 def __dump(self, value, write):
Fredrik Lundhb9056332001-07-11 17:42:21 +0000645 try:
646 f = self.dispatch[type(value)]
647 except KeyError:
Martin v. Löwis07529352006-11-19 18:51:54 +0000648 # check if this object can be marshalled as a structure
649 try:
650 value.__dict__
651 except:
652 raise TypeError, "cannot marshal %s objects" % type(value)
653 # check if this class is a sub-class of a basic type,
654 # because we don't know how to marshal these types
655 # (e.g. a string sub-class)
656 for type_ in type(value).__mro__:
657 if type_ in self.dispatch.keys():
658 raise TypeError, "cannot marshal %s objects" % type(value)
659 f = self.dispatch[InstanceType]
660 f(self, value, write)
Fredrik Lundhb9056332001-07-11 17:42:21 +0000661
Andrew M. Kuchlinga4c2b742003-04-25 00:26:51 +0000662 def dump_nil (self, value, write):
663 if not self.allow_none:
664 raise TypeError, "cannot marshal None unless allow_none is enabled"
665 write("<value><nil/></value>")
Raymond Hettingerfe59dc12005-02-07 15:28:45 +0000666 dispatch[NoneType] = dump_nil
Tim Petersc2659cf2003-05-12 20:19:37 +0000667
Fredrik Lundh3d9addd2002-06-27 21:36:21 +0000668 def dump_int(self, value, write):
Skip Montanaro5449e082001-10-17 22:53:33 +0000669 # in case ints are > 32 bits
670 if value > MAXINT or value < MININT:
671 raise OverflowError, "int exceeds XML-RPC limits"
Fredrik Lundh3d9addd2002-06-27 21:36:21 +0000672 write("<value><int>")
673 write(str(value))
674 write("</int></value>\n")
Raymond Hettingerfe59dc12005-02-07 15:28:45 +0000675 dispatch[IntType] = dump_int
Fredrik Lundhb9056332001-07-11 17:42:21 +0000676
Skip Montanaro9a7c96a2003-01-22 18:17:25 +0000677 if _bool_is_builtin:
678 def dump_bool(self, value, write):
679 write("<value><boolean>")
680 write(value and "1" or "0")
681 write("</boolean></value>\n")
682 dispatch[bool] = dump_bool
683
Fredrik Lundh3d9addd2002-06-27 21:36:21 +0000684 def dump_long(self, value, write):
Skip Montanaro5449e082001-10-17 22:53:33 +0000685 if value > MAXINT or value < MININT:
686 raise OverflowError, "long int exceeds XML-RPC limits"
Fredrik Lundh3d9addd2002-06-27 21:36:21 +0000687 write("<value><int>")
688 write(str(int(value)))
689 write("</int></value>\n")
Raymond Hettingerfe59dc12005-02-07 15:28:45 +0000690 dispatch[LongType] = dump_long
Skip Montanaro5e9c71b2001-10-10 15:56:34 +0000691
Fredrik Lundh3d9addd2002-06-27 21:36:21 +0000692 def dump_double(self, value, write):
693 write("<value><double>")
694 write(repr(value))
695 write("</double></value>\n")
Raymond Hettingerfe59dc12005-02-07 15:28:45 +0000696 dispatch[FloatType] = dump_double
Fredrik Lundhb9056332001-07-11 17:42:21 +0000697
Fredrik Lundh3d9addd2002-06-27 21:36:21 +0000698 def dump_string(self, value, write, escape=escape):
699 write("<value><string>")
700 write(escape(value))
701 write("</string></value>\n")
Raymond Hettingerfe59dc12005-02-07 15:28:45 +0000702 dispatch[StringType] = dump_string
Fredrik Lundhb9056332001-07-11 17:42:21 +0000703
704 if unicode:
Fredrik Lundh3d9addd2002-06-27 21:36:21 +0000705 def dump_unicode(self, value, write, escape=escape):
Fredrik Lundh3d9addd2002-06-27 21:36:21 +0000706 write("<value><string>")
Serhiy Storchaka9b5177c2016-01-20 10:33:51 +0200707 write(escape(value).encode(self.encoding, 'xmlcharrefreplace'))
Fredrik Lundh3d9addd2002-06-27 21:36:21 +0000708 write("</string></value>\n")
Raymond Hettingerfe59dc12005-02-07 15:28:45 +0000709 dispatch[UnicodeType] = dump_unicode
Fredrik Lundhb9056332001-07-11 17:42:21 +0000710
Fredrik Lundh3d9addd2002-06-27 21:36:21 +0000711 def dump_array(self, value, write):
712 i = id(value)
Brett Cannon814820b2008-08-04 00:50:11 +0000713 if i in self.memo:
Fredrik Lundh3d9addd2002-06-27 21:36:21 +0000714 raise TypeError, "cannot marshal recursive sequences"
715 self.memo[i] = None
Fredrik Lundh1538c232001-10-01 19:42:03 +0000716 dump = self.__dump
Fredrik Lundhb9056332001-07-11 17:42:21 +0000717 write("<value><array><data>\n")
718 for v in value:
Fredrik Lundh3d9addd2002-06-27 21:36:21 +0000719 dump(v, write)
Fredrik Lundhb9056332001-07-11 17:42:21 +0000720 write("</data></array></value>\n")
Fredrik Lundh3d9addd2002-06-27 21:36:21 +0000721 del self.memo[i]
Raymond Hettingerfe59dc12005-02-07 15:28:45 +0000722 dispatch[TupleType] = dump_array
723 dispatch[ListType] = dump_array
Fredrik Lundhb9056332001-07-11 17:42:21 +0000724
Fredrik Lundh3d9addd2002-06-27 21:36:21 +0000725 def dump_struct(self, value, write, escape=escape):
726 i = id(value)
Brett Cannon814820b2008-08-04 00:50:11 +0000727 if i in self.memo:
Fredrik Lundh3d9addd2002-06-27 21:36:21 +0000728 raise TypeError, "cannot marshal recursive dictionaries"
729 self.memo[i] = None
Fredrik Lundh1538c232001-10-01 19:42:03 +0000730 dump = self.__dump
Fredrik Lundhb9056332001-07-11 17:42:21 +0000731 write("<value><struct>\n")
Andrew M. Kuchling5962f452004-06-05 12:35:58 +0000732 for k, v in value.items():
Fredrik Lundhb9056332001-07-11 17:42:21 +0000733 write("<member>\n")
Serhiy Storchaka9b5177c2016-01-20 10:33:51 +0200734 if type(k) is StringType:
735 k = escape(k)
736 elif unicode and type(k) is UnicodeType:
737 k = escape(k).encode(self.encoding, 'xmlcharrefreplace')
738 else:
739 raise TypeError, "dictionary key must be string"
740 write("<name>%s</name>\n" % k)
Andrew M. Kuchling5962f452004-06-05 12:35:58 +0000741 dump(v, write)
Fredrik Lundhb9056332001-07-11 17:42:21 +0000742 write("</member>\n")
743 write("</struct></value>\n")
Fredrik Lundh3d9addd2002-06-27 21:36:21 +0000744 del self.memo[i]
Raymond Hettingerfe59dc12005-02-07 15:28:45 +0000745 dispatch[DictType] = dump_struct
Fredrik Lundhb9056332001-07-11 17:42:21 +0000746
Fred Drakeba613c32005-02-10 18:33:30 +0000747 if datetime:
748 def dump_datetime(self, value, write):
749 write("<value><dateTime.iso8601>")
Skip Montanarob131f042008-04-18 20:35:46 +0000750 write(_strftime(value))
Fred Drakeba613c32005-02-10 18:33:30 +0000751 write("</dateTime.iso8601></value>\n")
752 dispatch[datetime.datetime] = dump_datetime
753
Fredrik Lundh3d9addd2002-06-27 21:36:21 +0000754 def dump_instance(self, value, write):
Fredrik Lundhb9056332001-07-11 17:42:21 +0000755 # check for special wrappers
756 if value.__class__ in WRAPPERS:
Fredrik Lundh3d9addd2002-06-27 21:36:21 +0000757 self.write = write
Fredrik Lundhb9056332001-07-11 17:42:21 +0000758 value.encode(self)
Fredrik Lundh3d9addd2002-06-27 21:36:21 +0000759 del self.write
Fredrik Lundhb9056332001-07-11 17:42:21 +0000760 else:
761 # store instance attributes as a struct (really?)
Fredrik Lundh3d9addd2002-06-27 21:36:21 +0000762 self.dump_struct(value.__dict__, write)
Fredrik Lundhb9056332001-07-11 17:42:21 +0000763 dispatch[InstanceType] = dump_instance
764
Fredrik Lundh3d9addd2002-06-27 21:36:21 +0000765##
766# XML-RPC unmarshaller.
767#
768# @see loads
769
Fredrik Lundhb9056332001-07-11 17:42:21 +0000770class Unmarshaller:
Fred Drake1b410792001-09-04 18:55:03 +0000771 """Unmarshal an XML-RPC response, based on incoming XML event
Fredrik Lundhb0e8e9b2001-09-10 21:45:42 +0000772 messages (start, data, end). Call close() to get the resulting
Fred Drake1b410792001-09-04 18:55:03 +0000773 data structure.
Fredrik Lundhb9056332001-07-11 17:42:21 +0000774
Fredrik Lundhb0e8e9b2001-09-10 21:45:42 +0000775 Note that this reader is fairly tolerant, and gladly accepts bogus
776 XML-RPC data without complaining (but not bogus XML).
Fred Drake1b410792001-09-04 18:55:03 +0000777 """
Fredrik Lundhb9056332001-07-11 17:42:21 +0000778
779 # and again, if you don't understand what's going on in here,
780 # that's perfectly ok.
781
Skip Montanaro174dd222005-05-14 20:54:16 +0000782 def __init__(self, use_datetime=0):
Fredrik Lundhb9056332001-07-11 17:42:21 +0000783 self._type = None
784 self._stack = []
785 self._marks = []
786 self._data = []
Serhiy Storchaka073418a2016-05-04 11:28:09 +0300787 self._value = False
Fredrik Lundhb9056332001-07-11 17:42:21 +0000788 self._methodname = None
789 self._encoding = "utf-8"
790 self.append = self._stack.append
Skip Montanaro174dd222005-05-14 20:54:16 +0000791 self._use_datetime = use_datetime
792 if use_datetime and not datetime:
793 raise ValueError, "the datetime module is not available"
Fredrik Lundhb9056332001-07-11 17:42:21 +0000794
795 def close(self):
796 # return response tuple and target method
797 if self._type is None or self._marks:
798 raise ResponseError()
799 if self._type == "fault":
Guido van Rossum68468eb2003-02-27 20:14:51 +0000800 raise Fault(**self._stack[0])
Fredrik Lundhb9056332001-07-11 17:42:21 +0000801 return tuple(self._stack)
802
803 def getmethodname(self):
804 return self._methodname
805
806 #
807 # event handlers
808
809 def xml(self, encoding, standalone):
810 self._encoding = encoding
811 # FIXME: assert standalone == 1 ???
812
813 def start(self, tag, attrs):
814 # prepare to handle this element
815 if tag == "array" or tag == "struct":
816 self._marks.append(len(self._stack))
817 self._data = []
Serhiy Storchaka073418a2016-05-04 11:28:09 +0300818 if self._value and tag not in self.dispatch:
819 raise ResponseError("unknown tag %r" % tag)
Fredrik Lundhb9056332001-07-11 17:42:21 +0000820 self._value = (tag == "value")
821
822 def data(self, text):
823 self._data.append(text)
824
Fredrik Lundh1538c232001-10-01 19:42:03 +0000825 def end(self, tag, join=string.join):
Fredrik Lundhb9056332001-07-11 17:42:21 +0000826 # call the appropriate end tag handler
827 try:
828 f = self.dispatch[tag]
829 except KeyError:
830 pass # unknown tag ?
831 else:
Fredrik Lundh1538c232001-10-01 19:42:03 +0000832 return f(self, join(self._data, ""))
Fredrik Lundhb9056332001-07-11 17:42:21 +0000833
834 #
835 # accelerator support
836
837 def end_dispatch(self, tag, data):
838 # dispatch data
839 try:
840 f = self.dispatch[tag]
841 except KeyError:
842 pass # unknown tag ?
843 else:
844 return f(self, data)
845
846 #
847 # element decoders
848
849 dispatch = {}
850
Andrew M. Kuchlinga4c2b742003-04-25 00:26:51 +0000851 def end_nil (self, data):
852 self.append(None)
853 self._value = 0
854 dispatch["nil"] = end_nil
Tim Petersc2659cf2003-05-12 20:19:37 +0000855
Fredrik Lundh1538c232001-10-01 19:42:03 +0000856 def end_boolean(self, data):
Fredrik Lundhb9056332001-07-11 17:42:21 +0000857 if data == "0":
858 self.append(False)
859 elif data == "1":
860 self.append(True)
861 else:
862 raise TypeError, "bad boolean value"
863 self._value = 0
864 dispatch["boolean"] = end_boolean
865
Fredrik Lundh1538c232001-10-01 19:42:03 +0000866 def end_int(self, data):
867 self.append(int(data))
Fredrik Lundhb9056332001-07-11 17:42:21 +0000868 self._value = 0
869 dispatch["i4"] = end_int
Georg Brandld0b592f2008-05-29 07:45:26 +0000870 dispatch["i8"] = end_int
Fredrik Lundhb9056332001-07-11 17:42:21 +0000871 dispatch["int"] = end_int
872
Fredrik Lundh1538c232001-10-01 19:42:03 +0000873 def end_double(self, data):
874 self.append(float(data))
Fredrik Lundhb9056332001-07-11 17:42:21 +0000875 self._value = 0
876 dispatch["double"] = end_double
877
Fredrik Lundh1538c232001-10-01 19:42:03 +0000878 def end_string(self, data):
Fredrik Lundhb9056332001-07-11 17:42:21 +0000879 if self._encoding:
880 data = _decode(data, self._encoding)
881 self.append(_stringify(data))
882 self._value = 0
883 dispatch["string"] = end_string
884 dispatch["name"] = end_string # struct keys are always strings
885
886 def end_array(self, data):
Raymond Hettinger46ac8eb2002-06-30 03:39:14 +0000887 mark = self._marks.pop()
Fredrik Lundhb9056332001-07-11 17:42:21 +0000888 # map arrays to Python lists
889 self._stack[mark:] = [self._stack[mark:]]
890 self._value = 0
891 dispatch["array"] = end_array
892
893 def end_struct(self, data):
Raymond Hettinger46ac8eb2002-06-30 03:39:14 +0000894 mark = self._marks.pop()
Fredrik Lundhb9056332001-07-11 17:42:21 +0000895 # map structs to Python dictionaries
896 dict = {}
897 items = self._stack[mark:]
898 for i in range(0, len(items), 2):
899 dict[_stringify(items[i])] = items[i+1]
900 self._stack[mark:] = [dict]
901 self._value = 0
902 dispatch["struct"] = end_struct
903
Fredrik Lundh1538c232001-10-01 19:42:03 +0000904 def end_base64(self, data):
Fredrik Lundhb9056332001-07-11 17:42:21 +0000905 value = Binary()
Fredrik Lundh1538c232001-10-01 19:42:03 +0000906 value.decode(data)
Fredrik Lundhb9056332001-07-11 17:42:21 +0000907 self.append(value)
908 self._value = 0
909 dispatch["base64"] = end_base64
910
Fredrik Lundh1538c232001-10-01 19:42:03 +0000911 def end_dateTime(self, data):
Fredrik Lundhb9056332001-07-11 17:42:21 +0000912 value = DateTime()
Fredrik Lundh1538c232001-10-01 19:42:03 +0000913 value.decode(data)
Skip Montanaro174dd222005-05-14 20:54:16 +0000914 if self._use_datetime:
915 value = _datetime_type(data)
Fredrik Lundhb9056332001-07-11 17:42:21 +0000916 self.append(value)
917 dispatch["dateTime.iso8601"] = end_dateTime
918
919 def end_value(self, data):
Fredrik Lundh3d9addd2002-06-27 21:36:21 +0000920 # if we stumble upon a value element with no internal
Fredrik Lundhb9056332001-07-11 17:42:21 +0000921 # elements, treat it as a string element
922 if self._value:
923 self.end_string(data)
924 dispatch["value"] = end_value
925
926 def end_params(self, data):
927 self._type = "params"
928 dispatch["params"] = end_params
929
930 def end_fault(self, data):
931 self._type = "fault"
932 dispatch["fault"] = end_fault
933
Fredrik Lundh1538c232001-10-01 19:42:03 +0000934 def end_methodName(self, data):
Fredrik Lundhb9056332001-07-11 17:42:21 +0000935 if self._encoding:
936 data = _decode(data, self._encoding)
937 self._methodname = data
938 self._type = "methodName" # no params
939 dispatch["methodName"] = end_methodName
940
Martin v. Löwis45394c22003-10-31 13:49:36 +0000941## Multicall support
942#
Fredrik Lundhb9056332001-07-11 17:42:21 +0000943
Martin v. Löwis45394c22003-10-31 13:49:36 +0000944class _MultiCallMethod:
945 # some lesser magic to store calls made to a MultiCall object
946 # for batch execution
947 def __init__(self, call_list, name):
948 self.__call_list = call_list
949 self.__name = name
950 def __getattr__(self, name):
951 return _MultiCallMethod(self.__call_list, "%s.%s" % (self.__name, name))
952 def __call__(self, *args):
953 self.__call_list.append((self.__name, args))
954
Martin v. Löwis12237b32004-08-22 16:04:50 +0000955class MultiCallIterator:
Martin v. Löwis45394c22003-10-31 13:49:36 +0000956 """Iterates over the results of a multicall. Exceptions are
Andrew Svetlov4bb142b2012-12-18 21:27:37 +0200957 raised in response to xmlrpc faults."""
Raymond Hettingercc523fc2003-11-02 09:47:05 +0000958
Martin v. Löwis12237b32004-08-22 16:04:50 +0000959 def __init__(self, results):
960 self.results = results
961
962 def __getitem__(self, i):
963 item = self.results[i]
964 if type(item) == type({}):
965 raise Fault(item['faultCode'], item['faultString'])
966 elif type(item) == type([]):
967 return item[0]
Martin v. Löwis45394c22003-10-31 13:49:36 +0000968 else:
969 raise ValueError,\
970 "unexpected type in multicall result"
Raymond Hettingercc523fc2003-11-02 09:47:05 +0000971
Martin v. Löwis45394c22003-10-31 13:49:36 +0000972class MultiCall:
Martin Panterb362f752015-11-02 03:37:02 +0000973 """server -> an object used to boxcar method calls
Martin v. Löwis45394c22003-10-31 13:49:36 +0000974
975 server should be a ServerProxy object.
976
977 Methods can be added to the MultiCall using normal
978 method call syntax e.g.:
979
980 multicall = MultiCall(server_proxy)
981 multicall.add(2,3)
982 multicall.get_address("Guido")
983
984 To execute the multicall, call the MultiCall object e.g.:
985
986 add_result, address = multicall()
987 """
Raymond Hettingercc523fc2003-11-02 09:47:05 +0000988
Martin v. Löwis45394c22003-10-31 13:49:36 +0000989 def __init__(self, server):
990 self.__server = server
991 self.__call_list = []
Raymond Hettingercc523fc2003-11-02 09:47:05 +0000992
Martin v. Löwis45394c22003-10-31 13:49:36 +0000993 def __repr__(self):
994 return "<MultiCall at %x>" % id(self)
Raymond Hettingercc523fc2003-11-02 09:47:05 +0000995
Martin v. Löwis45394c22003-10-31 13:49:36 +0000996 __str__ = __repr__
997
998 def __getattr__(self, name):
999 return _MultiCallMethod(self.__call_list, name)
1000
1001 def __call__(self):
1002 marshalled_list = []
1003 for name, args in self.__call_list:
1004 marshalled_list.append({'methodName' : name, 'params' : args})
1005
1006 return MultiCallIterator(self.__server.system.multicall(marshalled_list))
Raymond Hettingercc523fc2003-11-02 09:47:05 +00001007
Fredrik Lundhb9056332001-07-11 17:42:21 +00001008# --------------------------------------------------------------------
1009# convenience functions
1010
Fredrik Lundh3d9addd2002-06-27 21:36:21 +00001011##
1012# Create a parser object, and connect it to an unmarshalling instance.
1013# This function picks the fastest available XML parser.
1014#
1015# return A (parser, unmarshaller) tuple.
1016
Skip Montanaro174dd222005-05-14 20:54:16 +00001017def getparser(use_datetime=0):
Fredrik Lundhb9056332001-07-11 17:42:21 +00001018 """getparser() -> parser, unmarshaller
1019
Fredrik Lundhb0e8e9b2001-09-10 21:45:42 +00001020 Create an instance of the fastest available parser, and attach it
1021 to an unmarshalling object. Return both objects.
Fredrik Lundhb9056332001-07-11 17:42:21 +00001022 """
Skip Montanaro174dd222005-05-14 20:54:16 +00001023 if use_datetime and not datetime:
1024 raise ValueError, "the datetime module is not available"
Fredrik Lundhb9056332001-07-11 17:42:21 +00001025 if FastParser and FastUnmarshaller:
Skip Montanaro174dd222005-05-14 20:54:16 +00001026 if use_datetime:
1027 mkdatetime = _datetime_type
1028 else:
1029 mkdatetime = _datetime
1030 target = FastUnmarshaller(True, False, _binary, mkdatetime, Fault)
Fredrik Lundhb9056332001-07-11 17:42:21 +00001031 parser = FastParser(target)
1032 else:
Skip Montanaro174dd222005-05-14 20:54:16 +00001033 target = Unmarshaller(use_datetime=use_datetime)
Fredrik Lundhb9056332001-07-11 17:42:21 +00001034 if FastParser:
1035 parser = FastParser(target)
Fredrik Lundhb9056332001-07-11 17:42:21 +00001036 elif ExpatParser:
1037 parser = ExpatParser(target)
1038 else:
1039 parser = SlowParser(target)
1040 return parser, target
1041
Fredrik Lundh3d9addd2002-06-27 21:36:21 +00001042##
1043# Convert a Python tuple or a Fault instance to an XML-RPC packet.
1044#
1045# @def dumps(params, **options)
1046# @param params A tuple or Fault instance.
1047# @keyparam methodname If given, create a methodCall request for
1048# this method name.
1049# @keyparam methodresponse If given, create a methodResponse packet.
1050# If used with a tuple, the tuple must be a singleton (that is,
1051# it must contain exactly one element).
1052# @keyparam encoding The packet encoding.
1053# @return A string containing marshalled data.
1054
Andrew M. Kuchlinga4c2b742003-04-25 00:26:51 +00001055def dumps(params, methodname=None, methodresponse=None, encoding=None,
1056 allow_none=0):
Fredrik Lundhb9056332001-07-11 17:42:21 +00001057 """data [,options] -> marshalled data
1058
1059 Convert an argument tuple or a Fault instance to an XML-RPC
1060 request (or response, if the methodresponse option is used).
1061
Fredrik Lundhb0e8e9b2001-09-10 21:45:42 +00001062 In addition to the data object, the following options can be given
1063 as keyword arguments:
Fredrik Lundhb9056332001-07-11 17:42:21 +00001064
1065 methodname: the method name for a methodCall packet
1066
1067 methodresponse: true to create a methodResponse packet.
1068 If this option is used with a tuple, the tuple must be
1069 a singleton (i.e. it can contain only one element).
1070
1071 encoding: the packet encoding (default is UTF-8)
1072
1073 All 8-bit strings in the data structure are assumed to use the
1074 packet encoding. Unicode strings are automatically converted,
Fredrik Lundhb0e8e9b2001-09-10 21:45:42 +00001075 where necessary.
Fredrik Lundhb9056332001-07-11 17:42:21 +00001076 """
1077
Raymond Hettingerfe59dc12005-02-07 15:28:45 +00001078 assert isinstance(params, TupleType) or isinstance(params, Fault),\
Fredrik Lundhb9056332001-07-11 17:42:21 +00001079 "argument must be tuple or Fault instance"
1080
1081 if isinstance(params, Fault):
1082 methodresponse = 1
Raymond Hettingerfe59dc12005-02-07 15:28:45 +00001083 elif methodresponse and isinstance(params, TupleType):
Fredrik Lundhb9056332001-07-11 17:42:21 +00001084 assert len(params) == 1, "response tuple must be a singleton"
1085
Fredrik Lundh3d9addd2002-06-27 21:36:21 +00001086 if not encoding:
Fredrik Lundhb9056332001-07-11 17:42:21 +00001087 encoding = "utf-8"
1088
Fredrik Lundh3d9addd2002-06-27 21:36:21 +00001089 if FastMarshaller:
1090 m = FastMarshaller(encoding)
1091 else:
Andrew M. Kuchlinga4c2b742003-04-25 00:26:51 +00001092 m = Marshaller(encoding, allow_none)
Fredrik Lundh3d9addd2002-06-27 21:36:21 +00001093
Fredrik Lundhb9056332001-07-11 17:42:21 +00001094 data = m.dumps(params)
1095
1096 if encoding != "utf-8":
Fredrik Lundh3d9addd2002-06-27 21:36:21 +00001097 xmlheader = "<?xml version='1.0' encoding='%s'?>\n" % str(encoding)
Fredrik Lundhb9056332001-07-11 17:42:21 +00001098 else:
1099 xmlheader = "<?xml version='1.0'?>\n" # utf-8 is default
1100
1101 # standard XML-RPC wrappings
Fredrik Lundh3d9addd2002-06-27 21:36:21 +00001102 if methodname:
Fredrik Lundhb9056332001-07-11 17:42:21 +00001103 # a method call
Raymond Hettingerfe59dc12005-02-07 15:28:45 +00001104 if not isinstance(methodname, StringType):
Serhiy Storchaka9b5177c2016-01-20 10:33:51 +02001105 methodname = methodname.encode(encoding, 'xmlcharrefreplace')
Fredrik Lundhb9056332001-07-11 17:42:21 +00001106 data = (
1107 xmlheader,
1108 "<methodCall>\n"
1109 "<methodName>", methodname, "</methodName>\n",
1110 data,
1111 "</methodCall>\n"
1112 )
1113 elif methodresponse:
1114 # a method response, or a fault structure
1115 data = (
1116 xmlheader,
1117 "<methodResponse>\n",
1118 data,
1119 "</methodResponse>\n"
1120 )
1121 else:
1122 return data # return as is
1123 return string.join(data, "")
1124
Fredrik Lundh3d9addd2002-06-27 21:36:21 +00001125##
1126# Convert an XML-RPC packet to a Python object. If the XML-RPC packet
1127# represents a fault condition, this function raises a Fault exception.
1128#
1129# @param data An XML-RPC packet, given as an 8-bit string.
Walter Dörwaldf0dfc7a2003-10-20 14:01:56 +00001130# @return A tuple containing the unpacked data, and the method name
Fredrik Lundh3d9addd2002-06-27 21:36:21 +00001131# (None if not present).
1132# @see Fault
1133
Skip Montanaro174dd222005-05-14 20:54:16 +00001134def loads(data, use_datetime=0):
Fredrik Lundhb9056332001-07-11 17:42:21 +00001135 """data -> unmarshalled data, method name
1136
1137 Convert an XML-RPC packet to unmarshalled data plus a method
1138 name (None if not present).
1139
1140 If the XML-RPC packet represents a fault condition, this function
1141 raises a Fault exception.
1142 """
Skip Montanaro174dd222005-05-14 20:54:16 +00001143 p, u = getparser(use_datetime=use_datetime)
Fredrik Lundhb9056332001-07-11 17:42:21 +00001144 p.feed(data)
1145 p.close()
1146 return u.close(), u.getmethodname()
1147
Kristján Valur Jónssone0078602009-06-28 21:04:17 +00001148##
1149# Encode a string using the gzip content encoding such as specified by the
1150# Content-Encoding: gzip
1151# in the HTTP header, as described in RFC 1952
1152#
1153# @param data the unencoded data
1154# @return the encoded data
1155
1156def gzip_encode(data):
1157 """data -> gzip encoded data
1158
1159 Encode data using the gzip content encoding as described in RFC 1952
1160 """
Kristján Valur Jónsson6d755902009-07-19 22:14:00 +00001161 if not gzip:
1162 raise NotImplementedError
Kristján Valur Jónssone0078602009-06-28 21:04:17 +00001163 f = StringIO.StringIO()
1164 gzf = gzip.GzipFile(mode="wb", fileobj=f, compresslevel=1)
1165 gzf.write(data)
1166 gzf.close()
1167 encoded = f.getvalue()
1168 f.close()
1169 return encoded
1170
1171##
1172# Decode a string using the gzip content encoding such as specified by the
1173# Content-Encoding: gzip
1174# in the HTTP header, as described in RFC 1952
1175#
1176# @param data The encoded data
Benjamin Peterson9e8f5232014-12-05 20:15:15 -05001177# @keyparam max_decode Maximum bytes to decode (20MB default), use negative
1178# values for unlimited decoding
Kristján Valur Jónssone0078602009-06-28 21:04:17 +00001179# @return the unencoded data
1180# @raises ValueError if data is not correctly coded.
Benjamin Peterson9e8f5232014-12-05 20:15:15 -05001181# @raises ValueError if max gzipped payload length exceeded
Kristján Valur Jónssone0078602009-06-28 21:04:17 +00001182
Benjamin Peterson9e8f5232014-12-05 20:15:15 -05001183def gzip_decode(data, max_decode=20971520):
Kristján Valur Jónssone0078602009-06-28 21:04:17 +00001184 """gzip encoded data -> unencoded data
1185
1186 Decode data using the gzip content encoding as described in RFC 1952
1187 """
Kristján Valur Jónsson6d755902009-07-19 22:14:00 +00001188 if not gzip:
1189 raise NotImplementedError
Kristján Valur Jónssone0078602009-06-28 21:04:17 +00001190 f = StringIO.StringIO(data)
1191 gzf = gzip.GzipFile(mode="rb", fileobj=f)
1192 try:
Benjamin Peterson9e8f5232014-12-05 20:15:15 -05001193 if max_decode < 0: # no limit
1194 decoded = gzf.read()
1195 else:
1196 decoded = gzf.read(max_decode + 1)
Kristján Valur Jónssone0078602009-06-28 21:04:17 +00001197 except IOError:
1198 raise ValueError("invalid data")
1199 f.close()
1200 gzf.close()
Benjamin Peterson9e8f5232014-12-05 20:15:15 -05001201 if max_decode >= 0 and len(decoded) > max_decode:
1202 raise ValueError("max gzipped payload length exceeded")
Kristján Valur Jónssone0078602009-06-28 21:04:17 +00001203 return decoded
1204
1205##
1206# Return a decoded file-like object for the gzip encoding
1207# as described in RFC 1952.
1208#
1209# @param response A stream supporting a read() method
1210# @return a file-like object that the decoded data can be read() from
1211
Kristján Valur Jónssonc0379432009-07-19 22:35:44 +00001212class GzipDecodedResponse(gzip.GzipFile if gzip else object):
Kristján Valur Jónssone0078602009-06-28 21:04:17 +00001213 """a file-like object to decode a response encoded with the gzip
1214 method, as described in RFC 1952.
1215 """
1216 def __init__(self, response):
1217 #response doesn't support tell() and read(), required by
1218 #GzipFile
Kristján Valur Jónsson6d755902009-07-19 22:14:00 +00001219 if not gzip:
1220 raise NotImplementedError
Kristján Valur Jónssone0078602009-06-28 21:04:17 +00001221 self.stringio = StringIO.StringIO(response.read())
1222 gzip.GzipFile.__init__(self, mode="rb", fileobj=self.stringio)
1223
1224 def close(self):
Serhiy Storchaka1aa2c0f2015-04-10 13:24:10 +03001225 try:
1226 gzip.GzipFile.close(self)
1227 finally:
1228 self.stringio.close()
Kristján Valur Jónssone0078602009-06-28 21:04:17 +00001229
Fredrik Lundhb9056332001-07-11 17:42:21 +00001230
1231# --------------------------------------------------------------------
1232# request dispatcher
1233
1234class _Method:
1235 # some magic to bind an XML-RPC method to an RPC server.
1236 # supports "nested" methods (e.g. examples.getStateName)
1237 def __init__(self, send, name):
1238 self.__send = send
1239 self.__name = name
1240 def __getattr__(self, name):
1241 return _Method(self.__send, "%s.%s" % (self.__name, name))
1242 def __call__(self, *args):
1243 return self.__send(self.__name, args)
1244
Fredrik Lundh3d9addd2002-06-27 21:36:21 +00001245##
1246# Standard transport class for XML-RPC over HTTP.
1247# <p>
1248# You can create custom transports by subclassing this method, and
1249# overriding selected methods.
Fredrik Lundhb9056332001-07-11 17:42:21 +00001250
1251class Transport:
Fredrik Lundhc4c062f2001-09-10 19:45:02 +00001252 """Handles an HTTP transaction to an XML-RPC server."""
Fredrik Lundhb9056332001-07-11 17:42:21 +00001253
1254 # client identifier (may be overridden)
1255 user_agent = "xmlrpclib.py/%s (by www.pythonware.com)" % __version__
1256
Kristján Valur Jónssone0078602009-06-28 21:04:17 +00001257 #if true, we'll request gzip encoding
1258 accept_gzip_encoding = True
1259
1260 # if positive, encode request using gzip if it exceeds this threshold
1261 # note that many server will get confused, so only use it if you know
1262 # that they can decode such a request
1263 encode_threshold = None #None = don't encode
1264
Skip Montanaro174dd222005-05-14 20:54:16 +00001265 def __init__(self, use_datetime=0):
1266 self._use_datetime = use_datetime
Kristján Valur Jónssone0078602009-06-28 21:04:17 +00001267 self._connection = (None, None)
1268 self._extra_headers = []
1269 ##
1270 # Send a complete request, and parse the response.
1271 # Retry request if a cached connection has disconnected.
1272 #
1273 # @param host Target host.
1274 # @param handler Target PRC handler.
1275 # @param request_body XML-RPC request body.
1276 # @param verbose Debugging flag.
1277 # @return Parsed response.
1278
1279 def request(self, host, handler, request_body, verbose=0):
1280 #retry request once if cached connection has gone cold
1281 for i in (0, 1):
1282 try:
1283 return self.single_request(host, handler, request_body, verbose)
Kristján Valur Jónssonf83648e2009-07-03 23:26:02 +00001284 except socket.error, e:
Victor Stinner8b0d84e2010-07-24 02:51:49 +00001285 if i or e.errno not in (errno.ECONNRESET, errno.ECONNABORTED, errno.EPIPE):
Kristján Valur Jónssonf83648e2009-07-03 23:26:02 +00001286 raise
Kristján Valur Jónssone5445da2009-07-03 23:29:50 +00001287 except httplib.BadStatusLine: #close after we sent request
Kristján Valur Jónssonf83648e2009-07-03 23:26:02 +00001288 if i:
Kristján Valur Jónssone0078602009-06-28 21:04:17 +00001289 raise
Skip Montanaro174dd222005-05-14 20:54:16 +00001290
Fredrik Lundh3d9addd2002-06-27 21:36:21 +00001291 ##
1292 # Send a complete request, and parse the response.
1293 #
1294 # @param host Target host.
1295 # @param handler Target PRC handler.
1296 # @param request_body XML-RPC request body.
1297 # @param verbose Debugging flag.
1298 # @return Parsed response.
1299
Kristján Valur Jónssone0078602009-06-28 21:04:17 +00001300 def single_request(self, host, handler, request_body, verbose=0):
Fredrik Lundhb9056332001-07-11 17:42:21 +00001301 # issue XML-RPC request
1302
1303 h = self.make_connection(host)
1304 if verbose:
1305 h.set_debuglevel(1)
1306
Kristján Valur Jónssone0078602009-06-28 21:04:17 +00001307 try:
1308 self.send_request(h, handler, request_body)
1309 self.send_host(h, host)
1310 self.send_user_agent(h)
1311 self.send_content(h, request_body)
Fredrik Lundhb9056332001-07-11 17:42:21 +00001312
Kristján Valur Jónssone0078602009-06-28 21:04:17 +00001313 response = h.getresponse(buffering=True)
1314 if response.status == 200:
1315 self.verbose = verbose
1316 return self.parse_response(response)
1317 except Fault:
1318 raise
1319 except Exception:
1320 # All unexpected errors leave connection in
1321 # a strange state, so we clear it.
1322 self.close()
1323 raise
Fredrik Lundhb9056332001-07-11 17:42:21 +00001324
Kristján Valur Jónssone0078602009-06-28 21:04:17 +00001325 #discard any response data and raise exception
1326 if (response.getheader("content-length", 0)):
1327 response.read()
1328 raise ProtocolError(
1329 host + handler,
1330 response.status, response.reason,
1331 response.msg,
1332 )
Fredrik Lundh3d9addd2002-06-27 21:36:21 +00001333
1334 ##
1335 # Create parser.
1336 #
Martin Panterb362f752015-11-02 03:37:02 +00001337 # @return A 2-tuple containing a parser and an unmarshaller.
Fredrik Lundhb9056332001-07-11 17:42:21 +00001338
Fredrik Lundhc4c062f2001-09-10 19:45:02 +00001339 def getparser(self):
1340 # get parser and unmarshaller
Skip Montanaro174dd222005-05-14 20:54:16 +00001341 return getparser(use_datetime=self._use_datetime)
Fredrik Lundhc4c062f2001-09-10 19:45:02 +00001342
Fredrik Lundh3d9addd2002-06-27 21:36:21 +00001343 ##
Fredrik Lundh1303c7c2002-10-22 18:23:00 +00001344 # Get authorization info from host parameter
1345 # Host may be a string, or a (host, x509-dict) tuple; if a string,
1346 # it is checked for a "user:pw@host" format, and a "Basic
1347 # Authentication" header is added if appropriate.
1348 #
1349 # @param host Host descriptor (URL or (URL, x509 info) tuple).
1350 # @return A 3-tuple containing (actual host, extra headers,
1351 # x509 info). The header and x509 fields may be None.
1352
1353 def get_host_info(self, host):
1354
1355 x509 = {}
Raymond Hettingerfe59dc12005-02-07 15:28:45 +00001356 if isinstance(host, TupleType):
Fredrik Lundh1303c7c2002-10-22 18:23:00 +00001357 host, x509 = host
1358
1359 import urllib
1360 auth, host = urllib.splituser(host)
1361
1362 if auth:
1363 import base64
Fredrik Lundh768c98b2002-11-01 17:14:16 +00001364 auth = base64.encodestring(urllib.unquote(auth))
Fredrik Lundh1303c7c2002-10-22 18:23:00 +00001365 auth = string.join(string.split(auth), "") # get rid of whitespace
1366 extra_headers = [
1367 ("Authorization", "Basic " + auth)
1368 ]
1369 else:
1370 extra_headers = None
1371
1372 return host, extra_headers, x509
1373
1374 ##
Fredrik Lundh3d9addd2002-06-27 21:36:21 +00001375 # Connect to server.
1376 #
1377 # @param host Target host.
1378 # @return A connection handle.
1379
Fredrik Lundhb9056332001-07-11 17:42:21 +00001380 def make_connection(self, host):
Kristján Valur Jónssone0078602009-06-28 21:04:17 +00001381 #return an existing connection if possible. This allows
1382 #HTTP/1.1 keep-alive.
1383 if self._connection and host == self._connection[0]:
1384 return self._connection[1]
1385
Fredrik Lundhb9056332001-07-11 17:42:21 +00001386 # create a HTTP connection object from a host descriptor
Kristján Valur Jónssone0078602009-06-28 21:04:17 +00001387 chost, self._extra_headers, x509 = self.get_host_info(host)
1388 #store the host argument along with the connection object
1389 self._connection = host, httplib.HTTPConnection(chost)
1390 return self._connection[1]
1391
1392 ##
1393 # Clear any cached connection object.
1394 # Used in the event of socket errors.
1395 #
1396 def close(self):
Serhiy Storchaka1aa2c0f2015-04-10 13:24:10 +03001397 host, connection = self._connection
1398 if connection:
Kristján Valur Jónssone0078602009-06-28 21:04:17 +00001399 self._connection = (None, None)
Serhiy Storchaka1aa2c0f2015-04-10 13:24:10 +03001400 connection.close()
Fredrik Lundhb9056332001-07-11 17:42:21 +00001401
Fredrik Lundh3d9addd2002-06-27 21:36:21 +00001402 ##
1403 # Send request header.
1404 #
1405 # @param connection Connection handle.
1406 # @param handler Target RPC handler.
1407 # @param request_body XML-RPC body.
1408
Fredrik Lundhb9056332001-07-11 17:42:21 +00001409 def send_request(self, connection, handler, request_body):
Kristján Valur Jónsson6d755902009-07-19 22:14:00 +00001410 if (self.accept_gzip_encoding and gzip):
Kristján Valur Jónssone0078602009-06-28 21:04:17 +00001411 connection.putrequest("POST", handler, skip_accept_encoding=True)
1412 connection.putheader("Accept-Encoding", "gzip")
1413 else:
1414 connection.putrequest("POST", handler)
Fredrik Lundhb9056332001-07-11 17:42:21 +00001415
Fredrik Lundh3d9addd2002-06-27 21:36:21 +00001416 ##
1417 # Send host name.
1418 #
1419 # @param connection Connection handle.
1420 # @param host Host name.
Kristján Valur Jónssone0078602009-06-28 21:04:17 +00001421 #
1422 # Note: This function doesn't actually add the "Host"
1423 # header anymore, it is done as part of the connection.putrequest() in
1424 # send_request() above.
Fredrik Lundh3d9addd2002-06-27 21:36:21 +00001425
Fredrik Lundhb9056332001-07-11 17:42:21 +00001426 def send_host(self, connection, host):
Kristján Valur Jónssone0078602009-06-28 21:04:17 +00001427 extra_headers = self._extra_headers
Fredrik Lundh1303c7c2002-10-22 18:23:00 +00001428 if extra_headers:
Raymond Hettingerfe59dc12005-02-07 15:28:45 +00001429 if isinstance(extra_headers, DictType):
Fredrik Lundh1303c7c2002-10-22 18:23:00 +00001430 extra_headers = extra_headers.items()
1431 for key, value in extra_headers:
1432 connection.putheader(key, value)
Fredrik Lundhb9056332001-07-11 17:42:21 +00001433
Fredrik Lundh3d9addd2002-06-27 21:36:21 +00001434 ##
1435 # Send user-agent identifier.
1436 #
1437 # @param connection Connection handle.
1438
Fredrik Lundhb9056332001-07-11 17:42:21 +00001439 def send_user_agent(self, connection):
1440 connection.putheader("User-Agent", self.user_agent)
1441
Fredrik Lundh3d9addd2002-06-27 21:36:21 +00001442 ##
1443 # Send request body.
1444 #
1445 # @param connection Connection handle.
1446 # @param request_body XML-RPC request body.
1447
Fredrik Lundhb9056332001-07-11 17:42:21 +00001448 def send_content(self, connection, request_body):
1449 connection.putheader("Content-Type", "text/xml")
Kristján Valur Jónssone0078602009-06-28 21:04:17 +00001450
1451 #optionally encode the request
1452 if (self.encode_threshold is not None and
Kristján Valur Jónsson6d755902009-07-19 22:14:00 +00001453 self.encode_threshold < len(request_body) and
1454 gzip):
Kristján Valur Jónssone0078602009-06-28 21:04:17 +00001455 connection.putheader("Content-Encoding", "gzip")
1456 request_body = gzip_encode(request_body)
1457
Fredrik Lundhb9056332001-07-11 17:42:21 +00001458 connection.putheader("Content-Length", str(len(request_body)))
Jeremy Hylton0381f482008-11-29 01:09:35 +00001459 connection.endheaders(request_body)
Fredrik Lundhb9056332001-07-11 17:42:21 +00001460
Fredrik Lundh3d9addd2002-06-27 21:36:21 +00001461 ##
1462 # Parse response.
1463 #
1464 # @param file Stream.
1465 # @return Response tuple and target method.
1466
Kristján Valur Jónssone0078602009-06-28 21:04:17 +00001467 def parse_response(self, response):
1468 # read response data from httpresponse, and parse it
Senthil Kumarand8e4edd2010-12-07 07:10:04 +00001469
1470 # Check for new http response object, else it is a file object
1471 if hasattr(response,'getheader'):
1472 if response.getheader("Content-Encoding", "") == "gzip":
1473 stream = GzipDecodedResponse(response)
1474 else:
1475 stream = response
Kristján Valur Jónssone0078602009-06-28 21:04:17 +00001476 else:
1477 stream = response
Fredrik Lundhb9056332001-07-11 17:42:21 +00001478
Fredrik Lundhc4c062f2001-09-10 19:45:02 +00001479 p, u = self.getparser()
Fredrik Lundhb9056332001-07-11 17:42:21 +00001480
1481 while 1:
Kristján Valur Jónssone0078602009-06-28 21:04:17 +00001482 data = stream.read(1024)
1483 if not data:
Fredrik Lundhb9056332001-07-11 17:42:21 +00001484 break
1485 if self.verbose:
Kristján Valur Jónssone0078602009-06-28 21:04:17 +00001486 print "body:", repr(data)
1487 p.feed(data)
Fredrik Lundhb9056332001-07-11 17:42:21 +00001488
Kristján Valur Jónssone0078602009-06-28 21:04:17 +00001489 if stream is not response:
1490 stream.close()
Fredrik Lundhb9056332001-07-11 17:42:21 +00001491 p.close()
1492
1493 return u.close()
1494
Fredrik Lundh3d9addd2002-06-27 21:36:21 +00001495##
1496# Standard transport class for XML-RPC over HTTPS.
1497
Fredrik Lundhb9056332001-07-11 17:42:21 +00001498class SafeTransport(Transport):
Fredrik Lundhc4c062f2001-09-10 19:45:02 +00001499 """Handles an HTTPS transaction to an XML-RPC server."""
Fredrik Lundhb9056332001-07-11 17:42:21 +00001500
Benjamin Petersonefa3cf82014-11-29 22:55:35 -05001501 def __init__(self, use_datetime=0, context=None):
1502 Transport.__init__(self, use_datetime=use_datetime)
1503 self.context = context
1504
Fredrik Lundh3d9addd2002-06-27 21:36:21 +00001505 # FIXME: mostly untested
1506
Fredrik Lundhb9056332001-07-11 17:42:21 +00001507 def make_connection(self, host):
Kristján Valur Jónssone0078602009-06-28 21:04:17 +00001508 if self._connection and host == self._connection[0]:
1509 return self._connection[1]
Fredrik Lundhb9056332001-07-11 17:42:21 +00001510 # create a HTTPS connection object from a host descriptor
1511 # host may be a string, or a (host, x509-dict) tuple
Fredrik Lundhb9056332001-07-11 17:42:21 +00001512 try:
Kristján Valur Jónssone0078602009-06-28 21:04:17 +00001513 HTTPS = httplib.HTTPSConnection
Fredrik Lundhb9056332001-07-11 17:42:21 +00001514 except AttributeError:
Fredrik Lundh1303c7c2002-10-22 18:23:00 +00001515 raise NotImplementedError(
1516 "your version of httplib doesn't support HTTPS"
1517 )
Fredrik Lundhb9056332001-07-11 17:42:21 +00001518 else:
Kristján Valur Jónssone0078602009-06-28 21:04:17 +00001519 chost, self._extra_headers, x509 = self.get_host_info(host)
Benjamin Petersonee443142014-11-29 23:06:06 -05001520 self._connection = host, HTTPS(chost, None, context=self.context, **(x509 or {}))
Kristján Valur Jónssone0078602009-06-28 21:04:17 +00001521 return self._connection[1]
Fredrik Lundhb9056332001-07-11 17:42:21 +00001522
Fredrik Lundh3d9addd2002-06-27 21:36:21 +00001523##
1524# Standard server proxy. This class establishes a virtual connection
1525# to an XML-RPC server.
1526# <p>
1527# This class is available as ServerProxy and Server. New code should
1528# use ServerProxy, to avoid confusion.
1529#
1530# @def ServerProxy(uri, **options)
1531# @param uri The connection point on the server.
1532# @keyparam transport A transport factory, compatible with the
1533# standard transport class.
1534# @keyparam encoding The default encoding used for 8-bit strings
1535# (default is UTF-8).
1536# @keyparam verbose Use a true value to enable debugging output.
1537# (printed to standard output).
1538# @see Transport
1539
Fredrik Lundhb9056332001-07-11 17:42:21 +00001540class ServerProxy:
1541 """uri [,options] -> a logical connection to an XML-RPC server
1542
1543 uri is the connection point on the server, given as
1544 scheme://host/target.
1545
1546 The standard implementation always supports the "http" scheme. If
1547 SSL socket support is available (Python 2.0), it also supports
1548 "https".
1549
1550 If the target part and the slash preceding it are both omitted,
1551 "/RPC2" is assumed.
1552
1553 The following options can be given as keyword arguments:
1554
1555 transport: a transport factory
1556 encoding: the request encoding (default is UTF-8)
1557
1558 All 8-bit strings passed to the server proxy are assumed to use
1559 the given encoding.
1560 """
1561
Andrew M. Kuchlinga4c2b742003-04-25 00:26:51 +00001562 def __init__(self, uri, transport=None, encoding=None, verbose=0,
Benjamin Petersonefa3cf82014-11-29 22:55:35 -05001563 allow_none=0, use_datetime=0, context=None):
Fredrik Lundhb9056332001-07-11 17:42:21 +00001564 # establish a "logical" server connection
1565
Serhiy Storchaka2f173fe2016-01-18 19:35:23 +02001566 if unicode and isinstance(uri, unicode):
Victor Stinner51b71982011-09-23 01:15:32 +02001567 uri = uri.encode('ISO-8859-1')
1568
Fredrik Lundhb9056332001-07-11 17:42:21 +00001569 # get the url
Fredrik Lundhc4c062f2001-09-10 19:45:02 +00001570 import urllib
Fredrik Lundhb9056332001-07-11 17:42:21 +00001571 type, uri = urllib.splittype(uri)
1572 if type not in ("http", "https"):
1573 raise IOError, "unsupported XML-RPC protocol"
1574 self.__host, self.__handler = urllib.splithost(uri)
1575 if not self.__handler:
1576 self.__handler = "/RPC2"
1577
1578 if transport is None:
1579 if type == "https":
Benjamin Petersonefa3cf82014-11-29 22:55:35 -05001580 transport = SafeTransport(use_datetime=use_datetime, context=context)
Fredrik Lundhb9056332001-07-11 17:42:21 +00001581 else:
Skip Montanaro174dd222005-05-14 20:54:16 +00001582 transport = Transport(use_datetime=use_datetime)
Fredrik Lundhb9056332001-07-11 17:42:21 +00001583 self.__transport = transport
1584
1585 self.__encoding = encoding
1586 self.__verbose = verbose
Andrew M. Kuchlinga4c2b742003-04-25 00:26:51 +00001587 self.__allow_none = allow_none
Tim Petersc2659cf2003-05-12 20:19:37 +00001588
Kristján Valur Jónssone0078602009-06-28 21:04:17 +00001589 def __close(self):
1590 self.__transport.close()
1591
Fredrik Lundhb9056332001-07-11 17:42:21 +00001592 def __request(self, methodname, params):
1593 # call a method on the remote server
1594
Andrew M. Kuchlinga4c2b742003-04-25 00:26:51 +00001595 request = dumps(params, methodname, encoding=self.__encoding,
1596 allow_none=self.__allow_none)
Fredrik Lundhb9056332001-07-11 17:42:21 +00001597
1598 response = self.__transport.request(
1599 self.__host,
1600 self.__handler,
1601 request,
1602 verbose=self.__verbose
1603 )
1604
1605 if len(response) == 1:
1606 response = response[0]
1607
1608 return response
1609
1610 def __repr__(self):
1611 return (
Fredrik Lundh78eedce2001-08-23 20:04:33 +00001612 "<ServerProxy for %s%s>" %
Fredrik Lundhb9056332001-07-11 17:42:21 +00001613 (self.__host, self.__handler)
1614 )
1615
1616 __str__ = __repr__
Raymond Hettingercc523fc2003-11-02 09:47:05 +00001617
Fredrik Lundhb9056332001-07-11 17:42:21 +00001618 def __getattr__(self, name):
1619 # magic method dispatcher
1620 return _Method(self.__request, name)
1621
Serhiy Storchaka9a118f12016-04-17 09:37:36 +03001622 # note: to call a remote object with a non-standard name, use
Fredrik Lundhb9056332001-07-11 17:42:21 +00001623 # result getattr(server, "strange-python-name")(args)
1624
Kristján Valur Jónssone0078602009-06-28 21:04:17 +00001625 def __call__(self, attr):
1626 """A workaround to get special attributes on the ServerProxy
1627 without interfering with the magic __getattr__
1628 """
1629 if attr == "close":
1630 return self.__close
1631 elif attr == "transport":
1632 return self.__transport
1633 raise AttributeError("Attribute %r not found" % (attr,))
1634
Fredrik Lundh78eedce2001-08-23 20:04:33 +00001635# compatibility
Fredrik Lundh3d9addd2002-06-27 21:36:21 +00001636
Fredrik Lundhb9056332001-07-11 17:42:21 +00001637Server = ServerProxy
1638
1639# --------------------------------------------------------------------
1640# test code
1641
1642if __name__ == "__main__":
1643
Senthil Kumaran01109732014-01-12 16:04:08 -08001644 server = ServerProxy("http://localhost:8000")
Fredrik Lundhb9056332001-07-11 17:42:21 +00001645
1646 print server
1647
Martin v. Löwis12237b32004-08-22 16:04:50 +00001648 multi = MultiCall(server)
Senthil Kumaran01109732014-01-12 16:04:08 -08001649 multi.pow(2, 9)
1650 multi.add(5, 1)
1651 multi.add(24, 11)
Martin v. Löwis12237b32004-08-22 16:04:50 +00001652 try:
1653 for response in multi():
1654 print response
Fredrik Lundhb9056332001-07-11 17:42:21 +00001655 except Error, v:
1656 print "ERROR", v