blob: 50cedfc4bb191109fe273c286e8d751d02faa177 [file] [log] [blame]
Fredrik Lundhb9056332001-07-11 17:42:21 +00001#
2# XML-RPC CLIENT LIBRARY
3# $Id$
4#
Fredrik Lundhc4c062f2001-09-10 19:45:02 +00005# an XML-RPC client interface for Python.
6#
7# the marshalling and response parser code can also be used to
8# implement XML-RPC servers.
9#
10# Notes:
Martin v. Löwis37af9862004-08-20 07:31:37 +000011# this version is designed to work with Python 2.1 or newer.
Fredrik Lundhc4c062f2001-09-10 19:45:02 +000012#
Fredrik Lundhb9056332001-07-11 17:42:21 +000013# History:
14# 1999-01-14 fl Created
15# 1999-01-15 fl Changed dateTime to use localtime
16# 1999-01-16 fl Added Binary/base64 element, default to RPC2 service
17# 1999-01-19 fl Fixed array data element (from Skip Montanaro)
18# 1999-01-21 fl Fixed dateTime constructor, etc.
19# 1999-02-02 fl Added fault handling, handle empty sequences, etc.
20# 1999-02-10 fl Fixed problem with empty responses (from Skip Montanaro)
21# 1999-06-20 fl Speed improvements, pluggable parsers/transports (0.9.8)
22# 2000-11-28 fl Changed boolean to check the truth value of its argument
23# 2001-02-24 fl Added encoding/Unicode/SafeTransport patches
24# 2001-02-26 fl Added compare support to wrappers (0.9.9/1.0b1)
25# 2001-03-28 fl Make sure response tuple is a singleton
26# 2001-03-29 fl Don't require empty params element (from Nicholas Riley)
Fredrik Lundh78eedce2001-08-23 20:04:33 +000027# 2001-06-10 fl Folded in _xmlrpclib accelerator support (1.0b2)
28# 2001-08-20 fl Base xmlrpclib.Error on built-in Exception (from Paul Prescod)
Fredrik Lundhc4c062f2001-09-10 19:45:02 +000029# 2001-09-03 fl Allow Transport subclass to override getparser
30# 2001-09-10 fl Lazy import of urllib, cgi, xmllib (20x import speedup)
Fredrik Lundh1538c232001-10-01 19:42:03 +000031# 2001-10-01 fl Remove containers from memo cache when done with them
32# 2001-10-01 fl Use faster escape method (80% dumps speedup)
Fredrik Lundh3d9addd2002-06-27 21:36:21 +000033# 2001-10-02 fl More dumps microtuning
34# 2001-10-04 fl Make sure import expat gets a parser (from Guido van Rossum)
Skip Montanaro5e9c71b2001-10-10 15:56:34 +000035# 2001-10-10 sm Allow long ints to be passed as ints if they don't overflow
Fredrik Lundh3d9addd2002-06-27 21:36:21 +000036# 2001-10-17 sm Test for int and long overflow (allows use on 64-bit systems)
Fredrik Lundhb6ab93f2001-12-19 21:40:04 +000037# 2001-11-12 fl Use repr() to marshal doubles (from Paul Felix)
Fredrik Lundh3d9addd2002-06-27 21:36:21 +000038# 2002-03-17 fl Avoid buffered read when possible (from James Rucker)
39# 2002-04-07 fl Added pythondoc comments
40# 2002-04-16 fl Added __str__ methods to datetime/binary wrappers
41# 2002-05-15 fl Added error constants (from Andrew Kuchling)
42# 2002-06-27 fl Merged with Python CVS version
Fredrik Lundh1303c7c2002-10-22 18:23:00 +000043# 2002-10-22 fl Added basic authentication (based on code from Phillip Eby)
Martin v. Löwis541342f2003-07-12 07:53:04 +000044# 2003-01-22 sm Add support for the bool type
45# 2003-02-27 gvr Remove apply calls
46# 2003-04-24 sm Use cStringIO if available
47# 2003-04-25 ak Add support for nil
48# 2003-06-15 gn Add support for time.struct_time
49# 2003-07-12 gp Correct marshalling of Faults
Martin v. Löwis45394c22003-10-31 13:49:36 +000050# 2003-10-31 mvl Add multicall support
Martin v. Löwis37af9862004-08-20 07:31:37 +000051# 2004-08-20 mvl Bump minimum supported Python version to 2.1
Fredrik Lundhb9056332001-07-11 17:42:21 +000052#
Fredrik Lundh3d9addd2002-06-27 21:36:21 +000053# Copyright (c) 1999-2002 by Secret Labs AB.
54# Copyright (c) 1999-2002 by Fredrik Lundh.
Fredrik Lundhb9056332001-07-11 17:42:21 +000055#
56# info@pythonware.com
57# http://www.pythonware.com
58#
59# --------------------------------------------------------------------
60# The XML-RPC client interface is
61#
Fredrik Lundh3d9addd2002-06-27 21:36:21 +000062# Copyright (c) 1999-2002 by Secret Labs AB
63# Copyright (c) 1999-2002 by Fredrik Lundh
Fredrik Lundhb9056332001-07-11 17:42:21 +000064#
65# By obtaining, using, and/or copying this software and/or its
66# associated documentation, you agree that you have read, understood,
67# and will comply with the following terms and conditions:
68#
69# Permission to use, copy, modify, and distribute this software and
70# its associated documentation for any purpose and without fee is
71# hereby granted, provided that the above copyright notice appears in
72# all copies, and that both that copyright notice and this permission
73# notice appear in supporting documentation, and that the name of
74# Secret Labs AB or the author not be used in advertising or publicity
75# pertaining to distribution of the software without specific, written
76# prior permission.
77#
78# SECRET LABS AB AND THE AUTHOR DISCLAIMS ALL WARRANTIES WITH REGARD
79# TO THIS SOFTWARE, INCLUDING ALL IMPLIED WARRANTIES OF MERCHANT-
80# ABILITY AND FITNESS. IN NO EVENT SHALL SECRET LABS AB OR THE AUTHOR
81# BE LIABLE FOR ANY SPECIAL, INDIRECT OR CONSEQUENTIAL DAMAGES OR ANY
82# DAMAGES WHATSOEVER RESULTING FROM LOSS OF USE, DATA OR PROFITS,
83# WHETHER IN AN ACTION OF CONTRACT, NEGLIGENCE OR OTHER TORTIOUS
84# ACTION, ARISING OUT OF OR IN CONNECTION WITH THE USE OR PERFORMANCE
85# OF THIS SOFTWARE.
86# --------------------------------------------------------------------
87
Fred Drake1b410792001-09-04 18:55:03 +000088"""
89An XML-RPC client interface for Python.
90
91The marshalling and response parser code can also be used to
92implement XML-RPC servers.
93
Fred Drake1b410792001-09-04 18:55:03 +000094Exported exceptions:
95
Fredrik Lundhc4c062f2001-09-10 19:45:02 +000096 Error Base class for client errors
97 ProtocolError Indicates an HTTP protocol error
98 ResponseError Indicates a broken response package
99 Fault Indicates an XML-RPC fault package
Fred Drake1b410792001-09-04 18:55:03 +0000100
101Exported classes:
102
Fredrik Lundhc4c062f2001-09-10 19:45:02 +0000103 ServerProxy Represents a logical connection to an XML-RPC server
Fred Drake1b410792001-09-04 18:55:03 +0000104
Raymond Hettingercc523fc2003-11-02 09:47:05 +0000105 MultiCall Executor of boxcared xmlrpc requests
Fredrik Lundhc4c062f2001-09-10 19:45:02 +0000106 DateTime dateTime wrapper for an ISO 8601 string or time tuple or
107 localtime integer value to generate a "dateTime.iso8601"
108 XML-RPC value
109 Binary binary data wrapper
Fred Drake1b410792001-09-04 18:55:03 +0000110
Fredrik Lundhc4c062f2001-09-10 19:45:02 +0000111 Marshaller Generate an XML-RPC params chunk from a Python data structure
112 Unmarshaller Unmarshal an XML-RPC response from incoming XML event message
113 Transport Handles an HTTP transaction to an XML-RPC server
114 SafeTransport Handles an HTTPS transaction to an XML-RPC server
Fred Drake1b410792001-09-04 18:55:03 +0000115
116Exported constants:
117
Florent Xiclunac4fec932011-10-30 20:19:32 +0100118 (none)
Fred Drake1b410792001-09-04 18:55:03 +0000119
120Exported functions:
121
Fredrik Lundhc4c062f2001-09-10 19:45:02 +0000122 getparser Create instance of the fastest available parser & attach
123 to an unmarshalling object
124 dumps Convert an argument tuple or a Fault instance to an XML-RPC
125 request (or response, if the methodresponse option is used).
126 loads Convert an XML-RPC packet to unmarshalled data plus a method
127 name (None if not present).
Fred Drake1b410792001-09-04 18:55:03 +0000128"""
129
Florent Xiclunac4fec932011-10-30 20:19:32 +0100130import base64
Florent Xicluna75861df2011-10-30 20:39:24 +0100131import sys
Florent Xiclunac4fec932011-10-30 20:19:32 +0100132import time
Florent Xiclunab6f019a2011-10-30 23:54:17 +0100133from datetime import datetime
Georg Brandl24420152008-05-26 16:32:26 +0000134import http.client
Ezio Melotti91932da2012-02-24 12:44:04 +0200135import urllib.parse
Georg Brandlcef803f2009-06-04 09:04:53 +0000136from xml.parsers import expat
Kristján Valur Jónsson985fc6a2009-07-01 10:01:31 +0000137import errno
138from io import BytesIO
Kristján Valur Jónssonaefde242009-07-19 22:29:24 +0000139try:
140 import gzip
Brett Cannoncd171c82013-07-04 17:43:24 -0400141except ImportError:
Kristján Valur Jónssonaefde242009-07-19 22:29:24 +0000142 gzip = None #python can be built without zlib/gzip support
Fredrik Lundh1538c232001-10-01 19:42:03 +0000143
Fredrik Lundh3d9addd2002-06-27 21:36:21 +0000144# --------------------------------------------------------------------
145# Internal stuff
146
Neal Norwitzff113342007-04-17 08:42:15 +0000147def escape(s):
148 s = s.replace("&", "&")
149 s = s.replace("<", "&lt;")
150 return s.replace(">", "&gt;",)
Fredrik Lundh1538c232001-10-01 19:42:03 +0000151
Florent Xicluna75861df2011-10-30 20:39:24 +0100152# used in User-Agent header sent
153__version__ = sys.version[:3]
Fredrik Lundh3d9addd2002-06-27 21:36:21 +0000154
155# xmlrpc integer limits
Guido van Rossume2a383d2007-01-15 16:59:06 +0000156MAXINT = 2**31-1
157MININT = -2**31
Fredrik Lundh3d9addd2002-06-27 21:36:21 +0000158
159# --------------------------------------------------------------------
160# Error constants (from Dan Libby's specification at
161# http://xmlrpc-epi.sourceforge.net/specs/rfc.fault_codes.php)
162
163# Ranges of errors
164PARSE_ERROR = -32700
165SERVER_ERROR = -32600
166APPLICATION_ERROR = -32500
167SYSTEM_ERROR = -32400
168TRANSPORT_ERROR = -32300
169
170# Specific errors
171NOT_WELLFORMED_ERROR = -32700
172UNSUPPORTED_ENCODING = -32701
173INVALID_ENCODING_CHAR = -32702
174INVALID_XMLRPC = -32600
175METHOD_NOT_FOUND = -32601
176INVALID_METHOD_PARAMS = -32602
177INTERNAL_ERROR = -32603
Fredrik Lundhb9056332001-07-11 17:42:21 +0000178
179# --------------------------------------------------------------------
180# Exceptions
181
Fredrik Lundh3d9addd2002-06-27 21:36:21 +0000182##
183# Base class for all kinds of client-side errors.
184
Fredrik Lundh78eedce2001-08-23 20:04:33 +0000185class Error(Exception):
Fred Drake1b410792001-09-04 18:55:03 +0000186 """Base class for client errors."""
Fredrik Lundhb0e8e9b2001-09-10 21:45:42 +0000187 def __str__(self):
188 return repr(self)
Fredrik Lundhb9056332001-07-11 17:42:21 +0000189
Fredrik Lundh3d9addd2002-06-27 21:36:21 +0000190##
191# Indicates an HTTP-level protocol error. This is raised by the HTTP
192# transport layer, if the server returns an error code other than 200
193# (OK).
194#
195# @param url The target URL.
196# @param errcode The HTTP error code.
197# @param errmsg The HTTP error message.
198# @param headers The HTTP header dictionary.
199
Fredrik Lundhb9056332001-07-11 17:42:21 +0000200class ProtocolError(Error):
Fred Drake1b410792001-09-04 18:55:03 +0000201 """Indicates an HTTP protocol error."""
Fredrik Lundhb9056332001-07-11 17:42:21 +0000202 def __init__(self, url, errcode, errmsg, headers):
Fredrik Lundhb0e8e9b2001-09-10 21:45:42 +0000203 Error.__init__(self)
Fredrik Lundhb9056332001-07-11 17:42:21 +0000204 self.url = url
205 self.errcode = errcode
206 self.errmsg = errmsg
207 self.headers = headers
208 def __repr__(self):
209 return (
210 "<ProtocolError for %s: %s %s>" %
211 (self.url, self.errcode, self.errmsg)
212 )
213
Fredrik Lundh3d9addd2002-06-27 21:36:21 +0000214##
215# Indicates a broken XML-RPC response package. This exception is
216# raised by the unmarshalling layer, if the XML-RPC response is
217# malformed.
218
Fredrik Lundhb9056332001-07-11 17:42:21 +0000219class ResponseError(Error):
Fredrik Lundhc4c062f2001-09-10 19:45:02 +0000220 """Indicates a broken response package."""
Fredrik Lundhb9056332001-07-11 17:42:21 +0000221 pass
222
Fredrik Lundh3d9addd2002-06-27 21:36:21 +0000223##
224# Indicates an XML-RPC fault response package. This exception is
225# raised by the unmarshalling layer, if the XML-RPC response contains
Florent Xiclunac4fec932011-10-30 20:19:32 +0100226# a fault string. This exception can also be used as a class, to
Fredrik Lundh3d9addd2002-06-27 21:36:21 +0000227# generate a fault XML-RPC message.
228#
229# @param faultCode The XML-RPC fault code.
230# @param faultString The XML-RPC fault string.
231
Fredrik Lundhb9056332001-07-11 17:42:21 +0000232class Fault(Error):
Fredrik Lundhc4c062f2001-09-10 19:45:02 +0000233 """Indicates an XML-RPC fault package."""
Fredrik Lundhb9056332001-07-11 17:42:21 +0000234 def __init__(self, faultCode, faultString, **extra):
Fredrik Lundhb0e8e9b2001-09-10 21:45:42 +0000235 Error.__init__(self)
Fredrik Lundhb9056332001-07-11 17:42:21 +0000236 self.faultCode = faultCode
237 self.faultString = faultString
238 def __repr__(self):
Florent Xiclunac4fec932011-10-30 20:19:32 +0100239 return "<Fault %s: %r>" % (self.faultCode, self.faultString)
Fredrik Lundhb9056332001-07-11 17:42:21 +0000240
241# --------------------------------------------------------------------
242# Special values
243
Fredrik Lundh3d9addd2002-06-27 21:36:21 +0000244##
Georg Brandl38eceaa2008-05-26 11:14:17 +0000245# Backwards compatibility
Fredrik Lundh3d9addd2002-06-27 21:36:21 +0000246
Guido van Rossum47b9ff62006-08-24 00:41:19 +0000247boolean = Boolean = bool
Fredrik Lundh3d9addd2002-06-27 21:36:21 +0000248
249##
250# Wrapper for XML-RPC DateTime values. This converts a time value to
251# the format used by XML-RPC.
252# <p>
Florent Xicluna2738a642011-11-01 00:06:58 +0100253# The value can be given as a datetime object, as a string in the
254# format "yyyymmddThh:mm:ss", as a 9-item time tuple (as returned by
Fredrik Lundh3d9addd2002-06-27 21:36:21 +0000255# time.localtime()), or an integer value (as returned by time.time()).
256# The wrapper uses time.localtime() to convert an integer to a time
257# tuple.
258#
Florent Xicluna2738a642011-11-01 00:06:58 +0100259# @param value The time, given as a datetime object, an ISO 8601 string,
260# a time tuple, or an integer time value.
261
262
263# Issue #13305: different format codes across platforms
264_day0 = datetime(1, 1, 1)
265if _day0.strftime('%Y') == '0001': # Mac OS X
266 def _iso8601_format(value):
267 return value.strftime("%Y%m%dT%H:%M:%S")
268elif _day0.strftime('%4Y') == '0001': # Linux
269 def _iso8601_format(value):
270 return value.strftime("%4Y%m%dT%H:%M:%S")
271else:
272 def _iso8601_format(value):
273 return value.strftime("%Y%m%dT%H:%M:%S").zfill(17)
274del _day0
275
Fredrik Lundhb9056332001-07-11 17:42:21 +0000276
Christian Heimesdae2a892008-04-19 00:55:37 +0000277def _strftime(value):
Florent Xiclunab6f019a2011-10-30 23:54:17 +0100278 if isinstance(value, datetime):
Florent Xicluna2738a642011-11-01 00:06:58 +0100279 return _iso8601_format(value)
Christian Heimesdae2a892008-04-19 00:55:37 +0000280
281 if not isinstance(value, (tuple, time.struct_time)):
282 if value == 0:
283 value = time.time()
284 value = time.localtime(value)
285
286 return "%04d%02d%02dT%02d:%02d:%02d" % value[:6]
287
Fredrik Lundhb9056332001-07-11 17:42:21 +0000288class DateTime:
Fredrik Lundhc4c062f2001-09-10 19:45:02 +0000289 """DateTime wrapper for an ISO 8601 string or time tuple or
290 localtime integer value to generate 'dateTime.iso8601' XML-RPC
291 value.
292 """
Fredrik Lundhb9056332001-07-11 17:42:21 +0000293
294 def __init__(self, value=0):
Christian Heimesdae2a892008-04-19 00:55:37 +0000295 if isinstance(value, str):
296 self.value = value
297 else:
298 self.value = _strftime(value)
Fredrik Lundhb9056332001-07-11 17:42:21 +0000299
Christian Heimes05e8be12008-02-23 18:30:17 +0000300 def make_comparable(self, other):
Fredrik Lundhb9056332001-07-11 17:42:21 +0000301 if isinstance(other, DateTime):
Christian Heimes05e8be12008-02-23 18:30:17 +0000302 s = self.value
303 o = other.value
Florent Xiclunab6f019a2011-10-30 23:54:17 +0100304 elif isinstance(other, datetime):
Christian Heimes05e8be12008-02-23 18:30:17 +0000305 s = self.value
Florent Xicluna2738a642011-11-01 00:06:58 +0100306 o = _iso8601_format(other)
Florent Xicluna3fa29f72011-10-30 20:18:50 +0100307 elif isinstance(other, str):
Christian Heimes05e8be12008-02-23 18:30:17 +0000308 s = self.value
309 o = other
310 elif hasattr(other, "timetuple"):
311 s = self.timetuple()
312 o = other.timetuple()
313 else:
314 otype = (hasattr(other, "__class__")
315 and other.__class__.__name__
316 or type(other))
317 raise TypeError("Can't compare %s and %s" %
318 (self.__class__.__name__, otype))
319 return s, o
320
321 def __lt__(self, other):
322 s, o = self.make_comparable(other)
323 return s < o
324
325 def __le__(self, other):
326 s, o = self.make_comparable(other)
327 return s <= o
328
329 def __gt__(self, other):
330 s, o = self.make_comparable(other)
331 return s > o
332
333 def __ge__(self, other):
334 s, o = self.make_comparable(other)
335 return s >= o
336
337 def __eq__(self, other):
338 s, o = self.make_comparable(other)
339 return s == o
Guido van Rossum47b9ff62006-08-24 00:41:19 +0000340
341 def __ne__(self, other):
Christian Heimes05e8be12008-02-23 18:30:17 +0000342 s, o = self.make_comparable(other)
343 return s != o
344
345 def timetuple(self):
346 return time.strptime(self.value, "%Y%m%dT%H:%M:%S")
347
Fredrik Lundh3d9addd2002-06-27 21:36:21 +0000348 ##
349 # Get date/time value.
350 #
351 # @return Date/time value, as an ISO 8601 string.
352
353 def __str__(self):
354 return self.value
355
Fredrik Lundhb9056332001-07-11 17:42:21 +0000356 def __repr__(self):
Florent Xiclunac4fec932011-10-30 20:19:32 +0100357 return "<DateTime %r at %x>" % (self.value, id(self))
Fredrik Lundhb9056332001-07-11 17:42:21 +0000358
359 def decode(self, data):
Neal Norwitzff113342007-04-17 08:42:15 +0000360 self.value = str(data).strip()
Fredrik Lundhb9056332001-07-11 17:42:21 +0000361
362 def encode(self, out):
363 out.write("<value><dateTime.iso8601>")
364 out.write(self.value)
365 out.write("</dateTime.iso8601></value>\n")
366
Fredrik Lundh3d9addd2002-06-27 21:36:21 +0000367def _datetime(data):
368 # decode xml element contents into a DateTime structure.
Fredrik Lundhb9056332001-07-11 17:42:21 +0000369 value = DateTime()
370 value.decode(data)
371 return value
372
Skip Montanaro174dd222005-05-14 20:54:16 +0000373def _datetime_type(data):
Florent Xiclunab6f019a2011-10-30 23:54:17 +0100374 return datetime.strptime(data, "%Y%m%dT%H:%M:%S")
Skip Montanaro174dd222005-05-14 20:54:16 +0000375
Fredrik Lundh3d9addd2002-06-27 21:36:21 +0000376##
377# Wrapper for binary data. This can be used to transport any kind
378# of binary data over XML-RPC, using BASE64 encoding.
379#
380# @param data An 8-bit string containing arbitrary data.
381
Fredrik Lundhb9056332001-07-11 17:42:21 +0000382class Binary:
Fred Drake1b410792001-09-04 18:55:03 +0000383 """Wrapper for binary data."""
Fredrik Lundhb9056332001-07-11 17:42:21 +0000384
385 def __init__(self, data=None):
Guido van Rossum54a40cb2007-08-27 22:27:41 +0000386 if data is None:
387 data = b""
388 else:
Florent Xicluna61665192011-11-15 20:53:25 +0100389 if not isinstance(data, (bytes, bytearray)):
390 raise TypeError("expected bytes or bytearray, not %s" %
Guido van Rossum54a40cb2007-08-27 22:27:41 +0000391 data.__class__.__name__)
392 data = bytes(data) # Make a copy of the bytes!
Fredrik Lundhb9056332001-07-11 17:42:21 +0000393 self.data = data
394
Fredrik Lundh3d9addd2002-06-27 21:36:21 +0000395 ##
396 # Get buffer contents.
397 #
398 # @return Buffer contents, as an 8-bit string.
399
400 def __str__(self):
Guido van Rossum54a40cb2007-08-27 22:27:41 +0000401 return str(self.data, "latin-1") # XXX encoding?!
Fredrik Lundh3d9addd2002-06-27 21:36:21 +0000402
Guido van Rossum47b9ff62006-08-24 00:41:19 +0000403 def __eq__(self, other):
Fredrik Lundhb9056332001-07-11 17:42:21 +0000404 if isinstance(other, Binary):
405 other = other.data
Guido van Rossum47b9ff62006-08-24 00:41:19 +0000406 return self.data == other
407
408 def __ne__(self, other):
409 if isinstance(other, Binary):
410 other = other.data
411 return self.data != other
Fredrik Lundhb9056332001-07-11 17:42:21 +0000412
413 def decode(self, data):
Georg Brandlb54d8012009-06-04 09:11:51 +0000414 self.data = base64.decodebytes(data)
Fredrik Lundhb9056332001-07-11 17:42:21 +0000415
416 def encode(self, out):
Fredrik Lundhb9056332001-07-11 17:42:21 +0000417 out.write("<value><base64>\n")
Georg Brandlb54d8012009-06-04 09:11:51 +0000418 encoded = base64.encodebytes(self.data)
Brett Cannon2dbde5e2007-07-30 03:50:35 +0000419 out.write(encoded.decode('ascii'))
Fredrik Lundhb9056332001-07-11 17:42:21 +0000420 out.write("</base64></value>\n")
421
Fredrik Lundh3d9addd2002-06-27 21:36:21 +0000422def _binary(data):
423 # decode xml element contents into a Binary structure
Fredrik Lundhb9056332001-07-11 17:42:21 +0000424 value = Binary()
425 value.decode(data)
426 return value
427
Skip Montanaro9a7c96a2003-01-22 18:17:25 +0000428WRAPPERS = (DateTime, Binary)
Fredrik Lundhb9056332001-07-11 17:42:21 +0000429
430# --------------------------------------------------------------------
431# XML parsers
432
Georg Brandlcef803f2009-06-04 09:04:53 +0000433class ExpatParser:
434 # fast expat parser for Python 2.0 and later.
435 def __init__(self, target):
436 self._parser = parser = expat.ParserCreate(None, None)
437 self._target = target
438 parser.StartElementHandler = target.start
439 parser.EndElementHandler = target.end
440 parser.CharacterDataHandler = target.data
441 encoding = None
442 target.xml(encoding, None)
Fredrik Lundhb9056332001-07-11 17:42:21 +0000443
Georg Brandlcef803f2009-06-04 09:04:53 +0000444 def feed(self, data):
445 self._parser.Parse(data, 0)
Fredrik Lundhb9056332001-07-11 17:42:21 +0000446
Georg Brandlcef803f2009-06-04 09:04:53 +0000447 def close(self):
448 self._parser.Parse("", 1) # end of data
449 del self._target, self._parser # get rid of circular references
Fredrik Lundhb9056332001-07-11 17:42:21 +0000450
Fredrik Lundhb9056332001-07-11 17:42:21 +0000451# --------------------------------------------------------------------
452# XML-RPC marshalling and unmarshalling code
453
Fredrik Lundh3d9addd2002-06-27 21:36:21 +0000454##
455# XML-RPC marshaller.
456#
457# @param encoding Default encoding for 8-bit strings. The default
458# value is None (interpreted as UTF-8).
459# @see dumps
460
Fredrik Lundhb9056332001-07-11 17:42:21 +0000461class Marshaller:
Fred Drake1b410792001-09-04 18:55:03 +0000462 """Generate an XML-RPC params chunk from a Python data structure.
Fredrik Lundhb9056332001-07-11 17:42:21 +0000463
Fredrik Lundhc4c062f2001-09-10 19:45:02 +0000464 Create a Marshaller instance for each set of parameters, and use
465 the "dumps" method to convert your data (represented as a tuple)
466 to an XML-RPC params chunk. To write a fault response, pass a
467 Fault instance instead. You may prefer to use the "dumps" module
468 function for this purpose.
Fred Drake1b410792001-09-04 18:55:03 +0000469 """
Fredrik Lundhb9056332001-07-11 17:42:21 +0000470
471 # by the way, if you don't understand what's going on in here,
472 # that's perfectly ok.
473
Georg Brandlfe991052009-09-16 15:54:04 +0000474 def __init__(self, encoding=None, allow_none=False):
Fredrik Lundhb9056332001-07-11 17:42:21 +0000475 self.memo = {}
476 self.data = None
477 self.encoding = encoding
Andrew M. Kuchlinga4c2b742003-04-25 00:26:51 +0000478 self.allow_none = allow_none
Tim Petersc2659cf2003-05-12 20:19:37 +0000479
Fredrik Lundhb9056332001-07-11 17:42:21 +0000480 dispatch = {}
481
482 def dumps(self, values):
Fredrik Lundh3d9addd2002-06-27 21:36:21 +0000483 out = []
484 write = out.append
485 dump = self.__dump
Fredrik Lundhb9056332001-07-11 17:42:21 +0000486 if isinstance(values, Fault):
487 # fault instance
488 write("<fault>\n")
Martin v. Löwis541342f2003-07-12 07:53:04 +0000489 dump({'faultCode': values.faultCode,
490 'faultString': values.faultString},
491 write)
Fredrik Lundhb9056332001-07-11 17:42:21 +0000492 write("</fault>\n")
493 else:
494 # parameter block
Fredrik Lundhc266bb02001-08-23 20:13:08 +0000495 # FIXME: the xml-rpc specification allows us to leave out
496 # the entire <params> block if there are no parameters.
497 # however, changing this may break older code (including
498 # old versions of xmlrpclib.py), so this is better left as
499 # is for now. See @XMLRPC3 for more information. /F
Fredrik Lundhb9056332001-07-11 17:42:21 +0000500 write("<params>\n")
501 for v in values:
502 write("<param>\n")
Fredrik Lundh3d9addd2002-06-27 21:36:21 +0000503 dump(v, write)
Fredrik Lundhb9056332001-07-11 17:42:21 +0000504 write("</param>\n")
505 write("</params>\n")
Neal Norwitzff113342007-04-17 08:42:15 +0000506 result = "".join(out)
Fredrik Lundhb9056332001-07-11 17:42:21 +0000507 return result
508
Fredrik Lundh3d9addd2002-06-27 21:36:21 +0000509 def __dump(self, value, write):
Fredrik Lundhb9056332001-07-11 17:42:21 +0000510 try:
511 f = self.dispatch[type(value)]
512 except KeyError:
Thomas Wouters89f507f2006-12-13 04:49:30 +0000513 # check if this object can be marshalled as a structure
Florent Xicluna93dfee12011-10-30 20:22:25 +0100514 if not hasattr(value, '__dict__'):
Collin Winterce36ad82007-08-30 01:19:48 +0000515 raise TypeError("cannot marshal %s objects" % type(value))
Thomas Wouters89f507f2006-12-13 04:49:30 +0000516 # check if this class is a sub-class of a basic type,
517 # because we don't know how to marshal these types
518 # (e.g. a string sub-class)
519 for type_ in type(value).__mro__:
520 if type_ in self.dispatch.keys():
Collin Winterce36ad82007-08-30 01:19:48 +0000521 raise TypeError("cannot marshal %s objects" % type(value))
Thomas Wouters89f507f2006-12-13 04:49:30 +0000522 # XXX(twouters): using "_arbitrary_instance" as key as a quick-fix
523 # for the p3yk merge, this should probably be fixed more neatly.
524 f = self.dispatch["_arbitrary_instance"]
525 f(self, value, write)
Fredrik Lundhb9056332001-07-11 17:42:21 +0000526
Andrew M. Kuchlinga4c2b742003-04-25 00:26:51 +0000527 def dump_nil (self, value, write):
528 if not self.allow_none:
Collin Winterce36ad82007-08-30 01:19:48 +0000529 raise TypeError("cannot marshal None unless allow_none is enabled")
Andrew M. Kuchlinga4c2b742003-04-25 00:26:51 +0000530 write("<value><nil/></value>")
Guido van Rossum13257902007-06-07 23:15:56 +0000531 dispatch[type(None)] = dump_nil
Tim Petersc2659cf2003-05-12 20:19:37 +0000532
Georg Brandl38eceaa2008-05-26 11:14:17 +0000533 def dump_bool(self, value, write):
534 write("<value><boolean>")
535 write(value and "1" or "0")
536 write("</boolean></value>\n")
537 dispatch[bool] = dump_bool
Skip Montanaro9a7c96a2003-01-22 18:17:25 +0000538
Fredrik Lundh3d9addd2002-06-27 21:36:21 +0000539 def dump_long(self, value, write):
Skip Montanaro5449e082001-10-17 22:53:33 +0000540 if value > MAXINT or value < MININT:
Serhiy Storchaka95949422013-08-27 19:40:23 +0300541 raise OverflowError("int exceeds XML-RPC limits")
Fredrik Lundh3d9addd2002-06-27 21:36:21 +0000542 write("<value><int>")
543 write(str(int(value)))
544 write("</int></value>\n")
Guido van Rossum13257902007-06-07 23:15:56 +0000545 dispatch[int] = dump_long
Skip Montanaro5e9c71b2001-10-10 15:56:34 +0000546
Florent Xicluna2bb96f52011-10-23 22:11:00 +0200547 # backward compatible
548 dump_int = dump_long
549
Fredrik Lundh3d9addd2002-06-27 21:36:21 +0000550 def dump_double(self, value, write):
551 write("<value><double>")
552 write(repr(value))
553 write("</double></value>\n")
Guido van Rossum13257902007-06-07 23:15:56 +0000554 dispatch[float] = dump_double
Fredrik Lundhb9056332001-07-11 17:42:21 +0000555
Guido van Rossum54ad5232007-05-27 09:17:48 +0000556 def dump_unicode(self, value, write, escape=escape):
Guido van Rossum54ad5232007-05-27 09:17:48 +0000557 write("<value><string>")
558 write(escape(value))
559 write("</string></value>\n")
560 dispatch[str] = dump_unicode
Fredrik Lundhb9056332001-07-11 17:42:21 +0000561
Florent Xicluna61665192011-11-15 20:53:25 +0100562 def dump_bytes(self, value, write):
563 write("<value><base64>\n")
564 encoded = base64.encodebytes(value)
565 write(encoded.decode('ascii'))
566 write("</base64></value>\n")
567 dispatch[bytes] = dump_bytes
568 dispatch[bytearray] = dump_bytes
569
Fredrik Lundh3d9addd2002-06-27 21:36:21 +0000570 def dump_array(self, value, write):
571 i = id(value)
Guido van Rossume2b70bc2006-08-18 22:13:04 +0000572 if i in self.memo:
Collin Winterce36ad82007-08-30 01:19:48 +0000573 raise TypeError("cannot marshal recursive sequences")
Fredrik Lundh3d9addd2002-06-27 21:36:21 +0000574 self.memo[i] = None
Fredrik Lundh1538c232001-10-01 19:42:03 +0000575 dump = self.__dump
Fredrik Lundhb9056332001-07-11 17:42:21 +0000576 write("<value><array><data>\n")
577 for v in value:
Fredrik Lundh3d9addd2002-06-27 21:36:21 +0000578 dump(v, write)
Fredrik Lundhb9056332001-07-11 17:42:21 +0000579 write("</data></array></value>\n")
Fredrik Lundh3d9addd2002-06-27 21:36:21 +0000580 del self.memo[i]
Guido van Rossum13257902007-06-07 23:15:56 +0000581 dispatch[tuple] = dump_array
582 dispatch[list] = dump_array
Fredrik Lundhb9056332001-07-11 17:42:21 +0000583
Fredrik Lundh3d9addd2002-06-27 21:36:21 +0000584 def dump_struct(self, value, write, escape=escape):
585 i = id(value)
Guido van Rossume2b70bc2006-08-18 22:13:04 +0000586 if i in self.memo:
Collin Winterce36ad82007-08-30 01:19:48 +0000587 raise TypeError("cannot marshal recursive dictionaries")
Fredrik Lundh3d9addd2002-06-27 21:36:21 +0000588 self.memo[i] = None
Fredrik Lundh1538c232001-10-01 19:42:03 +0000589 dump = self.__dump
Fredrik Lundhb9056332001-07-11 17:42:21 +0000590 write("<value><struct>\n")
Andrew M. Kuchling5962f452004-06-05 12:35:58 +0000591 for k, v in value.items():
Fredrik Lundhb9056332001-07-11 17:42:21 +0000592 write("<member>\n")
Guido van Rossum3172c5d2007-10-16 18:12:55 +0000593 if not isinstance(k, str):
Collin Winterce36ad82007-08-30 01:19:48 +0000594 raise TypeError("dictionary key must be string")
Fredrik Lundh1538c232001-10-01 19:42:03 +0000595 write("<name>%s</name>\n" % escape(k))
Andrew M. Kuchling5962f452004-06-05 12:35:58 +0000596 dump(v, write)
Fredrik Lundhb9056332001-07-11 17:42:21 +0000597 write("</member>\n")
598 write("</struct></value>\n")
Fredrik Lundh3d9addd2002-06-27 21:36:21 +0000599 del self.memo[i]
Guido van Rossum13257902007-06-07 23:15:56 +0000600 dispatch[dict] = dump_struct
Fredrik Lundhb9056332001-07-11 17:42:21 +0000601
Florent Xiclunab6f019a2011-10-30 23:54:17 +0100602 def dump_datetime(self, value, write):
603 write("<value><dateTime.iso8601>")
604 write(_strftime(value))
605 write("</dateTime.iso8601></value>\n")
606 dispatch[datetime] = dump_datetime
Fred Drakeba613c32005-02-10 18:33:30 +0000607
Fredrik Lundh3d9addd2002-06-27 21:36:21 +0000608 def dump_instance(self, value, write):
Fredrik Lundhb9056332001-07-11 17:42:21 +0000609 # check for special wrappers
610 if value.__class__ in WRAPPERS:
Fredrik Lundh3d9addd2002-06-27 21:36:21 +0000611 self.write = write
Fredrik Lundhb9056332001-07-11 17:42:21 +0000612 value.encode(self)
Fredrik Lundh3d9addd2002-06-27 21:36:21 +0000613 del self.write
Fredrik Lundhb9056332001-07-11 17:42:21 +0000614 else:
615 # store instance attributes as a struct (really?)
Fredrik Lundh3d9addd2002-06-27 21:36:21 +0000616 self.dump_struct(value.__dict__, write)
Guido van Rossume4dea982006-04-21 13:45:00 +0000617 dispatch[DateTime] = dump_instance
618 dispatch[Binary] = dump_instance
Thomas Wouters89f507f2006-12-13 04:49:30 +0000619 # XXX(twouters): using "_arbitrary_instance" as key as a quick-fix
620 # for the p3yk merge, this should probably be fixed more neatly.
621 dispatch["_arbitrary_instance"] = dump_instance
Fredrik Lundhb9056332001-07-11 17:42:21 +0000622
Fredrik Lundh3d9addd2002-06-27 21:36:21 +0000623##
624# XML-RPC unmarshaller.
625#
626# @see loads
627
Fredrik Lundhb9056332001-07-11 17:42:21 +0000628class Unmarshaller:
Fred Drake1b410792001-09-04 18:55:03 +0000629 """Unmarshal an XML-RPC response, based on incoming XML event
Fredrik Lundhb0e8e9b2001-09-10 21:45:42 +0000630 messages (start, data, end). Call close() to get the resulting
Fred Drake1b410792001-09-04 18:55:03 +0000631 data structure.
Fredrik Lundhb9056332001-07-11 17:42:21 +0000632
Fredrik Lundhb0e8e9b2001-09-10 21:45:42 +0000633 Note that this reader is fairly tolerant, and gladly accepts bogus
634 XML-RPC data without complaining (but not bogus XML).
Fred Drake1b410792001-09-04 18:55:03 +0000635 """
Fredrik Lundhb9056332001-07-11 17:42:21 +0000636
637 # and again, if you don't understand what's going on in here,
638 # that's perfectly ok.
639
Florent Xicluna61665192011-11-15 20:53:25 +0100640 def __init__(self, use_datetime=False, use_builtin_types=False):
Fredrik Lundhb9056332001-07-11 17:42:21 +0000641 self._type = None
642 self._stack = []
643 self._marks = []
644 self._data = []
645 self._methodname = None
646 self._encoding = "utf-8"
647 self.append = self._stack.append
Florent Xicluna61665192011-11-15 20:53:25 +0100648 self._use_datetime = use_builtin_types or use_datetime
649 self._use_bytes = use_builtin_types
Fredrik Lundhb9056332001-07-11 17:42:21 +0000650
651 def close(self):
652 # return response tuple and target method
653 if self._type is None or self._marks:
654 raise ResponseError()
655 if self._type == "fault":
Guido van Rossum68468eb2003-02-27 20:14:51 +0000656 raise Fault(**self._stack[0])
Fredrik Lundhb9056332001-07-11 17:42:21 +0000657 return tuple(self._stack)
658
659 def getmethodname(self):
660 return self._methodname
661
662 #
663 # event handlers
664
665 def xml(self, encoding, standalone):
666 self._encoding = encoding
667 # FIXME: assert standalone == 1 ???
668
669 def start(self, tag, attrs):
670 # prepare to handle this element
671 if tag == "array" or tag == "struct":
672 self._marks.append(len(self._stack))
673 self._data = []
674 self._value = (tag == "value")
675
676 def data(self, text):
677 self._data.append(text)
678
Neal Norwitzff113342007-04-17 08:42:15 +0000679 def end(self, tag):
Fredrik Lundhb9056332001-07-11 17:42:21 +0000680 # call the appropriate end tag handler
681 try:
682 f = self.dispatch[tag]
683 except KeyError:
684 pass # unknown tag ?
685 else:
Neal Norwitzff113342007-04-17 08:42:15 +0000686 return f(self, "".join(self._data))
Fredrik Lundhb9056332001-07-11 17:42:21 +0000687
688 #
689 # accelerator support
690
691 def end_dispatch(self, tag, data):
692 # dispatch data
693 try:
694 f = self.dispatch[tag]
695 except KeyError:
696 pass # unknown tag ?
697 else:
698 return f(self, data)
699
700 #
701 # element decoders
702
703 dispatch = {}
704
Andrew M. Kuchlinga4c2b742003-04-25 00:26:51 +0000705 def end_nil (self, data):
706 self.append(None)
707 self._value = 0
708 dispatch["nil"] = end_nil
Tim Petersc2659cf2003-05-12 20:19:37 +0000709
Fredrik Lundh1538c232001-10-01 19:42:03 +0000710 def end_boolean(self, data):
Fredrik Lundhb9056332001-07-11 17:42:21 +0000711 if data == "0":
712 self.append(False)
713 elif data == "1":
714 self.append(True)
715 else:
Collin Winterce36ad82007-08-30 01:19:48 +0000716 raise TypeError("bad boolean value")
Fredrik Lundhb9056332001-07-11 17:42:21 +0000717 self._value = 0
718 dispatch["boolean"] = end_boolean
719
Fredrik Lundh1538c232001-10-01 19:42:03 +0000720 def end_int(self, data):
721 self.append(int(data))
Fredrik Lundhb9056332001-07-11 17:42:21 +0000722 self._value = 0
723 dispatch["i4"] = end_int
Georg Brandlf08a9dd2008-06-10 16:57:31 +0000724 dispatch["i8"] = end_int
Fredrik Lundhb9056332001-07-11 17:42:21 +0000725 dispatch["int"] = end_int
726
Fredrik Lundh1538c232001-10-01 19:42:03 +0000727 def end_double(self, data):
728 self.append(float(data))
Fredrik Lundhb9056332001-07-11 17:42:21 +0000729 self._value = 0
730 dispatch["double"] = end_double
731
Fredrik Lundh1538c232001-10-01 19:42:03 +0000732 def end_string(self, data):
Fredrik Lundhb9056332001-07-11 17:42:21 +0000733 if self._encoding:
Alexandre Vassalottie671fd22009-07-22 02:32:34 +0000734 data = data.decode(self._encoding)
735 self.append(data)
Fredrik Lundhb9056332001-07-11 17:42:21 +0000736 self._value = 0
737 dispatch["string"] = end_string
738 dispatch["name"] = end_string # struct keys are always strings
739
740 def end_array(self, data):
Raymond Hettinger46ac8eb2002-06-30 03:39:14 +0000741 mark = self._marks.pop()
Fredrik Lundhb9056332001-07-11 17:42:21 +0000742 # map arrays to Python lists
743 self._stack[mark:] = [self._stack[mark:]]
744 self._value = 0
745 dispatch["array"] = end_array
746
747 def end_struct(self, data):
Raymond Hettinger46ac8eb2002-06-30 03:39:14 +0000748 mark = self._marks.pop()
Fredrik Lundhb9056332001-07-11 17:42:21 +0000749 # map structs to Python dictionaries
750 dict = {}
751 items = self._stack[mark:]
752 for i in range(0, len(items), 2):
Alexandre Vassalottie671fd22009-07-22 02:32:34 +0000753 dict[items[i]] = items[i+1]
Fredrik Lundhb9056332001-07-11 17:42:21 +0000754 self._stack[mark:] = [dict]
755 self._value = 0
756 dispatch["struct"] = end_struct
757
Fredrik Lundh1538c232001-10-01 19:42:03 +0000758 def end_base64(self, data):
Fredrik Lundhb9056332001-07-11 17:42:21 +0000759 value = Binary()
Guido van Rossum54a40cb2007-08-27 22:27:41 +0000760 value.decode(data.encode("ascii"))
Florent Xicluna61665192011-11-15 20:53:25 +0100761 if self._use_bytes:
762 value = value.data
Fredrik Lundhb9056332001-07-11 17:42:21 +0000763 self.append(value)
764 self._value = 0
765 dispatch["base64"] = end_base64
766
Fredrik Lundh1538c232001-10-01 19:42:03 +0000767 def end_dateTime(self, data):
Fredrik Lundhb9056332001-07-11 17:42:21 +0000768 value = DateTime()
Fredrik Lundh1538c232001-10-01 19:42:03 +0000769 value.decode(data)
Skip Montanaro174dd222005-05-14 20:54:16 +0000770 if self._use_datetime:
771 value = _datetime_type(data)
Fredrik Lundhb9056332001-07-11 17:42:21 +0000772 self.append(value)
773 dispatch["dateTime.iso8601"] = end_dateTime
774
775 def end_value(self, data):
Fredrik Lundh3d9addd2002-06-27 21:36:21 +0000776 # if we stumble upon a value element with no internal
Fredrik Lundhb9056332001-07-11 17:42:21 +0000777 # elements, treat it as a string element
778 if self._value:
779 self.end_string(data)
780 dispatch["value"] = end_value
781
782 def end_params(self, data):
783 self._type = "params"
784 dispatch["params"] = end_params
785
786 def end_fault(self, data):
787 self._type = "fault"
788 dispatch["fault"] = end_fault
789
Fredrik Lundh1538c232001-10-01 19:42:03 +0000790 def end_methodName(self, data):
Fredrik Lundhb9056332001-07-11 17:42:21 +0000791 if self._encoding:
Alexandre Vassalottie671fd22009-07-22 02:32:34 +0000792 data = data.decode(self._encoding)
Fredrik Lundhb9056332001-07-11 17:42:21 +0000793 self._methodname = data
794 self._type = "methodName" # no params
795 dispatch["methodName"] = end_methodName
796
Martin v. Löwis45394c22003-10-31 13:49:36 +0000797## Multicall support
798#
Fredrik Lundhb9056332001-07-11 17:42:21 +0000799
Martin v. Löwis45394c22003-10-31 13:49:36 +0000800class _MultiCallMethod:
801 # some lesser magic to store calls made to a MultiCall object
802 # for batch execution
803 def __init__(self, call_list, name):
804 self.__call_list = call_list
805 self.__name = name
806 def __getattr__(self, name):
807 return _MultiCallMethod(self.__call_list, "%s.%s" % (self.__name, name))
808 def __call__(self, *args):
809 self.__call_list.append((self.__name, args))
810
Martin v. Löwis12237b32004-08-22 16:04:50 +0000811class MultiCallIterator:
Martin v. Löwis45394c22003-10-31 13:49:36 +0000812 """Iterates over the results of a multicall. Exceptions are
Andrew Svetlov737fb892012-12-18 21:14:22 +0200813 raised in response to xmlrpc faults."""
Raymond Hettingercc523fc2003-11-02 09:47:05 +0000814
Martin v. Löwis12237b32004-08-22 16:04:50 +0000815 def __init__(self, results):
816 self.results = results
817
818 def __getitem__(self, i):
819 item = self.results[i]
820 if type(item) == type({}):
821 raise Fault(item['faultCode'], item['faultString'])
822 elif type(item) == type([]):
823 return item[0]
Martin v. Löwis45394c22003-10-31 13:49:36 +0000824 else:
Collin Winterce36ad82007-08-30 01:19:48 +0000825 raise ValueError("unexpected type in multicall result")
Raymond Hettingercc523fc2003-11-02 09:47:05 +0000826
Martin v. Löwis45394c22003-10-31 13:49:36 +0000827class MultiCall:
828 """server -> a object used to boxcar method calls
829
830 server should be a ServerProxy object.
831
832 Methods can be added to the MultiCall using normal
833 method call syntax e.g.:
834
835 multicall = MultiCall(server_proxy)
836 multicall.add(2,3)
837 multicall.get_address("Guido")
838
839 To execute the multicall, call the MultiCall object e.g.:
840
841 add_result, address = multicall()
842 """
Raymond Hettingercc523fc2003-11-02 09:47:05 +0000843
Martin v. Löwis45394c22003-10-31 13:49:36 +0000844 def __init__(self, server):
845 self.__server = server
846 self.__call_list = []
Raymond Hettingercc523fc2003-11-02 09:47:05 +0000847
Martin v. Löwis45394c22003-10-31 13:49:36 +0000848 def __repr__(self):
849 return "<MultiCall at %x>" % id(self)
Raymond Hettingercc523fc2003-11-02 09:47:05 +0000850
Martin v. Löwis45394c22003-10-31 13:49:36 +0000851 __str__ = __repr__
852
853 def __getattr__(self, name):
854 return _MultiCallMethod(self.__call_list, name)
855
856 def __call__(self):
857 marshalled_list = []
858 for name, args in self.__call_list:
859 marshalled_list.append({'methodName' : name, 'params' : args})
860
861 return MultiCallIterator(self.__server.system.multicall(marshalled_list))
Raymond Hettingercc523fc2003-11-02 09:47:05 +0000862
Fredrik Lundhb9056332001-07-11 17:42:21 +0000863# --------------------------------------------------------------------
864# convenience functions
865
Georg Brandl38eceaa2008-05-26 11:14:17 +0000866FastMarshaller = FastParser = FastUnmarshaller = None
867
Fredrik Lundh3d9addd2002-06-27 21:36:21 +0000868##
869# Create a parser object, and connect it to an unmarshalling instance.
870# This function picks the fastest available XML parser.
871#
872# return A (parser, unmarshaller) tuple.
873
Florent Xicluna61665192011-11-15 20:53:25 +0100874def getparser(use_datetime=False, use_builtin_types=False):
Fredrik Lundhb9056332001-07-11 17:42:21 +0000875 """getparser() -> parser, unmarshaller
876
Fredrik Lundhb0e8e9b2001-09-10 21:45:42 +0000877 Create an instance of the fastest available parser, and attach it
878 to an unmarshalling object. Return both objects.
Fredrik Lundhb9056332001-07-11 17:42:21 +0000879 """
880 if FastParser and FastUnmarshaller:
Florent Xicluna61665192011-11-15 20:53:25 +0100881 if use_builtin_types:
Skip Montanaro174dd222005-05-14 20:54:16 +0000882 mkdatetime = _datetime_type
Florent Xicluna61665192011-11-15 20:53:25 +0100883 mkbytes = base64.decodebytes
884 elif use_datetime:
885 mkdatetime = _datetime_type
886 mkbytes = _binary
Skip Montanaro174dd222005-05-14 20:54:16 +0000887 else:
888 mkdatetime = _datetime
Florent Xicluna61665192011-11-15 20:53:25 +0100889 mkbytes = _binary
890 target = FastUnmarshaller(True, False, mkbytes, mkdatetime, Fault)
Fredrik Lundhb9056332001-07-11 17:42:21 +0000891 parser = FastParser(target)
892 else:
Florent Xicluna61665192011-11-15 20:53:25 +0100893 target = Unmarshaller(use_datetime=use_datetime, use_builtin_types=use_builtin_types)
Fredrik Lundhb9056332001-07-11 17:42:21 +0000894 if FastParser:
895 parser = FastParser(target)
Fredrik Lundhb9056332001-07-11 17:42:21 +0000896 else:
Georg Brandlcef803f2009-06-04 09:04:53 +0000897 parser = ExpatParser(target)
Fredrik Lundhb9056332001-07-11 17:42:21 +0000898 return parser, target
899
Fredrik Lundh3d9addd2002-06-27 21:36:21 +0000900##
901# Convert a Python tuple or a Fault instance to an XML-RPC packet.
902#
903# @def dumps(params, **options)
904# @param params A tuple or Fault instance.
905# @keyparam methodname If given, create a methodCall request for
906# this method name.
907# @keyparam methodresponse If given, create a methodResponse packet.
908# If used with a tuple, the tuple must be a singleton (that is,
909# it must contain exactly one element).
910# @keyparam encoding The packet encoding.
911# @return A string containing marshalled data.
912
Andrew M. Kuchlinga4c2b742003-04-25 00:26:51 +0000913def dumps(params, methodname=None, methodresponse=None, encoding=None,
Georg Brandlfe991052009-09-16 15:54:04 +0000914 allow_none=False):
Fredrik Lundhb9056332001-07-11 17:42:21 +0000915 """data [,options] -> marshalled data
916
917 Convert an argument tuple or a Fault instance to an XML-RPC
918 request (or response, if the methodresponse option is used).
919
Fredrik Lundhb0e8e9b2001-09-10 21:45:42 +0000920 In addition to the data object, the following options can be given
921 as keyword arguments:
Fredrik Lundhb9056332001-07-11 17:42:21 +0000922
923 methodname: the method name for a methodCall packet
924
925 methodresponse: true to create a methodResponse packet.
926 If this option is used with a tuple, the tuple must be
927 a singleton (i.e. it can contain only one element).
928
929 encoding: the packet encoding (default is UTF-8)
930
Florent Xicluna61665192011-11-15 20:53:25 +0100931 All byte strings in the data structure are assumed to use the
Fredrik Lundhb9056332001-07-11 17:42:21 +0000932 packet encoding. Unicode strings are automatically converted,
Fredrik Lundhb0e8e9b2001-09-10 21:45:42 +0000933 where necessary.
Fredrik Lundhb9056332001-07-11 17:42:21 +0000934 """
935
Guido van Rossum13257902007-06-07 23:15:56 +0000936 assert isinstance(params, (tuple, Fault)), "argument must be tuple or Fault instance"
Fredrik Lundhb9056332001-07-11 17:42:21 +0000937 if isinstance(params, Fault):
938 methodresponse = 1
Guido van Rossum13257902007-06-07 23:15:56 +0000939 elif methodresponse and isinstance(params, tuple):
Fredrik Lundhb9056332001-07-11 17:42:21 +0000940 assert len(params) == 1, "response tuple must be a singleton"
941
Fredrik Lundh3d9addd2002-06-27 21:36:21 +0000942 if not encoding:
Fredrik Lundhb9056332001-07-11 17:42:21 +0000943 encoding = "utf-8"
944
Fredrik Lundh3d9addd2002-06-27 21:36:21 +0000945 if FastMarshaller:
946 m = FastMarshaller(encoding)
947 else:
Andrew M. Kuchlinga4c2b742003-04-25 00:26:51 +0000948 m = Marshaller(encoding, allow_none)
Fredrik Lundh3d9addd2002-06-27 21:36:21 +0000949
Fredrik Lundhb9056332001-07-11 17:42:21 +0000950 data = m.dumps(params)
951
952 if encoding != "utf-8":
Fredrik Lundh3d9addd2002-06-27 21:36:21 +0000953 xmlheader = "<?xml version='1.0' encoding='%s'?>\n" % str(encoding)
Fredrik Lundhb9056332001-07-11 17:42:21 +0000954 else:
955 xmlheader = "<?xml version='1.0'?>\n" # utf-8 is default
956
957 # standard XML-RPC wrappings
Fredrik Lundh3d9addd2002-06-27 21:36:21 +0000958 if methodname:
Fredrik Lundhb9056332001-07-11 17:42:21 +0000959 # a method call
Guido van Rossum3172c5d2007-10-16 18:12:55 +0000960 if not isinstance(methodname, str):
Fredrik Lundhb9056332001-07-11 17:42:21 +0000961 methodname = methodname.encode(encoding)
962 data = (
963 xmlheader,
964 "<methodCall>\n"
965 "<methodName>", methodname, "</methodName>\n",
966 data,
967 "</methodCall>\n"
968 )
969 elif methodresponse:
970 # a method response, or a fault structure
971 data = (
972 xmlheader,
973 "<methodResponse>\n",
974 data,
975 "</methodResponse>\n"
976 )
977 else:
978 return data # return as is
Neal Norwitzff113342007-04-17 08:42:15 +0000979 return "".join(data)
Fredrik Lundhb9056332001-07-11 17:42:21 +0000980
Fredrik Lundh3d9addd2002-06-27 21:36:21 +0000981##
982# Convert an XML-RPC packet to a Python object. If the XML-RPC packet
983# represents a fault condition, this function raises a Fault exception.
984#
985# @param data An XML-RPC packet, given as an 8-bit string.
Walter Dörwaldf0dfc7a2003-10-20 14:01:56 +0000986# @return A tuple containing the unpacked data, and the method name
Fredrik Lundh3d9addd2002-06-27 21:36:21 +0000987# (None if not present).
988# @see Fault
989
Florent Xicluna61665192011-11-15 20:53:25 +0100990def loads(data, use_datetime=False, use_builtin_types=False):
Fredrik Lundhb9056332001-07-11 17:42:21 +0000991 """data -> unmarshalled data, method name
992
993 Convert an XML-RPC packet to unmarshalled data plus a method
994 name (None if not present).
995
996 If the XML-RPC packet represents a fault condition, this function
997 raises a Fault exception.
998 """
Florent Xicluna61665192011-11-15 20:53:25 +0100999 p, u = getparser(use_datetime=use_datetime, use_builtin_types=use_builtin_types)
Fredrik Lundhb9056332001-07-11 17:42:21 +00001000 p.feed(data)
1001 p.close()
1002 return u.close(), u.getmethodname()
1003
Kristján Valur Jónsson985fc6a2009-07-01 10:01:31 +00001004##
1005# Encode a string using the gzip content encoding such as specified by the
1006# Content-Encoding: gzip
1007# in the HTTP header, as described in RFC 1952
1008#
1009# @param data the unencoded data
1010# @return the encoded data
1011
1012def gzip_encode(data):
1013 """data -> gzip encoded data
1014
1015 Encode data using the gzip content encoding as described in RFC 1952
1016 """
Kristján Valur Jónssonaefde242009-07-19 22:29:24 +00001017 if not gzip:
1018 raise NotImplementedError
Kristján Valur Jónsson985fc6a2009-07-01 10:01:31 +00001019 f = BytesIO()
1020 gzf = gzip.GzipFile(mode="wb", fileobj=f, compresslevel=1)
1021 gzf.write(data)
1022 gzf.close()
1023 encoded = f.getvalue()
1024 f.close()
1025 return encoded
1026
1027##
1028# Decode a string using the gzip content encoding such as specified by the
1029# Content-Encoding: gzip
1030# in the HTTP header, as described in RFC 1952
1031#
1032# @param data The encoded data
1033# @return the unencoded data
1034# @raises ValueError if data is not correctly coded.
1035
1036def gzip_decode(data):
1037 """gzip encoded data -> unencoded data
1038
1039 Decode data using the gzip content encoding as described in RFC 1952
1040 """
Kristján Valur Jónssonaefde242009-07-19 22:29:24 +00001041 if not gzip:
1042 raise NotImplementedError
Kristján Valur Jónsson985fc6a2009-07-01 10:01:31 +00001043 f = BytesIO(data)
1044 gzf = gzip.GzipFile(mode="rb", fileobj=f)
1045 try:
1046 decoded = gzf.read()
Andrew Svetlovf7a17b42012-12-25 16:47:37 +02001047 except OSError:
Kristján Valur Jónsson985fc6a2009-07-01 10:01:31 +00001048 raise ValueError("invalid data")
1049 f.close()
1050 gzf.close()
1051 return decoded
1052
1053##
1054# Return a decoded file-like object for the gzip encoding
1055# as described in RFC 1952.
1056#
1057# @param response A stream supporting a read() method
1058# @return a file-like object that the decoded data can be read() from
1059
Kristján Valur Jónsson9ab07312009-07-19 22:38:38 +00001060class GzipDecodedResponse(gzip.GzipFile if gzip else object):
Kristján Valur Jónsson985fc6a2009-07-01 10:01:31 +00001061 """a file-like object to decode a response encoded with the gzip
1062 method, as described in RFC 1952.
1063 """
1064 def __init__(self, response):
1065 #response doesn't support tell() and read(), required by
1066 #GzipFile
Kristján Valur Jónssonaefde242009-07-19 22:29:24 +00001067 if not gzip:
1068 raise NotImplementedError
Kristján Valur Jónsson985fc6a2009-07-01 10:01:31 +00001069 self.io = BytesIO(response.read())
1070 gzip.GzipFile.__init__(self, mode="rb", fileobj=self.io)
1071
1072 def close(self):
1073 gzip.GzipFile.close(self)
1074 self.io.close()
1075
Fredrik Lundhb9056332001-07-11 17:42:21 +00001076
1077# --------------------------------------------------------------------
1078# request dispatcher
1079
1080class _Method:
1081 # some magic to bind an XML-RPC method to an RPC server.
1082 # supports "nested" methods (e.g. examples.getStateName)
1083 def __init__(self, send, name):
1084 self.__send = send
1085 self.__name = name
1086 def __getattr__(self, name):
1087 return _Method(self.__send, "%s.%s" % (self.__name, name))
1088 def __call__(self, *args):
1089 return self.__send(self.__name, args)
1090
Fredrik Lundh3d9addd2002-06-27 21:36:21 +00001091##
1092# Standard transport class for XML-RPC over HTTP.
1093# <p>
1094# You can create custom transports by subclassing this method, and
1095# overriding selected methods.
Fredrik Lundhb9056332001-07-11 17:42:21 +00001096
1097class Transport:
Fredrik Lundhc4c062f2001-09-10 19:45:02 +00001098 """Handles an HTTP transaction to an XML-RPC server."""
Fredrik Lundhb9056332001-07-11 17:42:21 +00001099
1100 # client identifier (may be overridden)
Florent Xicluna75861df2011-10-30 20:39:24 +01001101 user_agent = "Python-xmlrpc/%s" % __version__
Fredrik Lundhb9056332001-07-11 17:42:21 +00001102
Kristján Valur Jónsson985fc6a2009-07-01 10:01:31 +00001103 #if true, we'll request gzip encoding
1104 accept_gzip_encoding = True
1105
1106 # if positive, encode request using gzip if it exceeds this threshold
1107 # note that many server will get confused, so only use it if you know
1108 # that they can decode such a request
1109 encode_threshold = None #None = don't encode
1110
Florent Xicluna61665192011-11-15 20:53:25 +01001111 def __init__(self, use_datetime=False, use_builtin_types=False):
Skip Montanaro174dd222005-05-14 20:54:16 +00001112 self._use_datetime = use_datetime
Florent Xicluna61665192011-11-15 20:53:25 +01001113 self._use_builtin_types = use_builtin_types
Kristján Valur Jónsson985fc6a2009-07-01 10:01:31 +00001114 self._connection = (None, None)
1115 self._extra_headers = []
Skip Montanaro174dd222005-05-14 20:54:16 +00001116
Fredrik Lundh3d9addd2002-06-27 21:36:21 +00001117 ##
1118 # Send a complete request, and parse the response.
Kristján Valur Jónsson985fc6a2009-07-01 10:01:31 +00001119 # Retry request if a cached connection has disconnected.
Fredrik Lundh3d9addd2002-06-27 21:36:21 +00001120 #
1121 # @param host Target host.
1122 # @param handler Target PRC handler.
1123 # @param request_body XML-RPC request body.
1124 # @param verbose Debugging flag.
1125 # @return Parsed response.
1126
Georg Brandlfe991052009-09-16 15:54:04 +00001127 def request(self, host, handler, request_body, verbose=False):
Kristján Valur Jónsson985fc6a2009-07-01 10:01:31 +00001128 #retry request once if cached connection has gone cold
1129 for i in (0, 1):
1130 try:
1131 return self.single_request(host, handler, request_body, verbose)
Andrew Svetlov0832af62012-12-18 23:10:48 +02001132 except OSError as e:
1133 if i or e.errno not in (errno.ECONNRESET, errno.ECONNABORTED,
1134 errno.EPIPE):
Kristján Valur Jónsson43535d92009-07-03 23:23:50 +00001135 raise
1136 except http.client.BadStatusLine: #close after we sent request
1137 if i:
Kristján Valur Jónsson985fc6a2009-07-01 10:01:31 +00001138 raise
1139
Georg Brandlfe991052009-09-16 15:54:04 +00001140 def single_request(self, host, handler, request_body, verbose=False):
Fredrik Lundhb9056332001-07-11 17:42:21 +00001141 # issue XML-RPC request
Kristján Valur Jónsson985fc6a2009-07-01 10:01:31 +00001142 try:
1143 http_conn = self.send_request(host, handler, request_body, verbose)
1144 resp = http_conn.getresponse()
1145 if resp.status == 200:
1146 self.verbose = verbose
1147 return self.parse_response(resp)
Fredrik Lundhb9056332001-07-11 17:42:21 +00001148
Kristján Valur Jónsson985fc6a2009-07-01 10:01:31 +00001149 except Fault:
1150 raise
1151 except Exception:
1152 #All unexpected errors leave connection in
1153 # a strange state, so we clear it.
1154 self.close()
1155 raise
Fredrik Lundhb9056332001-07-11 17:42:21 +00001156
Kristján Valur Jónsson985fc6a2009-07-01 10:01:31 +00001157 #We got an error response.
1158 #Discard any response data and raise exception
1159 if resp.getheader("content-length", ""):
1160 resp.read()
1161 raise ProtocolError(
1162 host + handler,
1163 resp.status, resp.reason,
1164 dict(resp.getheaders())
1165 )
Fredrik Lundhb9056332001-07-11 17:42:21 +00001166
Fredrik Lundh3d9addd2002-06-27 21:36:21 +00001167
1168 ##
1169 # Create parser.
1170 #
1171 # @return A 2-tuple containing a parser and a unmarshaller.
Fredrik Lundhb9056332001-07-11 17:42:21 +00001172
Fredrik Lundhc4c062f2001-09-10 19:45:02 +00001173 def getparser(self):
1174 # get parser and unmarshaller
Florent Xicluna61665192011-11-15 20:53:25 +01001175 return getparser(use_datetime=self._use_datetime,
1176 use_builtin_types=self._use_builtin_types)
Fredrik Lundhc4c062f2001-09-10 19:45:02 +00001177
Fredrik Lundh3d9addd2002-06-27 21:36:21 +00001178 ##
Fredrik Lundh1303c7c2002-10-22 18:23:00 +00001179 # Get authorization info from host parameter
1180 # Host may be a string, or a (host, x509-dict) tuple; if a string,
1181 # it is checked for a "user:pw@host" format, and a "Basic
1182 # Authentication" header is added if appropriate.
1183 #
1184 # @param host Host descriptor (URL or (URL, x509 info) tuple).
1185 # @return A 3-tuple containing (actual host, extra headers,
1186 # x509 info). The header and x509 fields may be None.
1187
1188 def get_host_info(self, host):
1189
1190 x509 = {}
Guido van Rossum13257902007-06-07 23:15:56 +00001191 if isinstance(host, tuple):
Fredrik Lundh1303c7c2002-10-22 18:23:00 +00001192 host, x509 = host
1193
Jeremy Hylton1afc1692008-06-18 20:49:58 +00001194 auth, host = urllib.parse.splituser(host)
Fredrik Lundh1303c7c2002-10-22 18:23:00 +00001195
1196 if auth:
Georg Brandlc8dcfb62009-02-13 10:50:01 +00001197 auth = urllib.parse.unquote_to_bytes(auth)
Georg Brandlb54d8012009-06-04 09:11:51 +00001198 auth = base64.encodebytes(auth).decode("utf-8")
Neal Norwitzff113342007-04-17 08:42:15 +00001199 auth = "".join(auth.split()) # get rid of whitespace
Fredrik Lundh1303c7c2002-10-22 18:23:00 +00001200 extra_headers = [
1201 ("Authorization", "Basic " + auth)
1202 ]
1203 else:
Kristján Valur Jónsson985fc6a2009-07-01 10:01:31 +00001204 extra_headers = []
Fredrik Lundh1303c7c2002-10-22 18:23:00 +00001205
1206 return host, extra_headers, x509
1207
1208 ##
Fredrik Lundh3d9addd2002-06-27 21:36:21 +00001209 # Connect to server.
1210 #
1211 # @param host Target host.
Jeremy Hylton5d8a88a2007-08-14 16:47:39 +00001212 # @return An HTTPConnection object
Fredrik Lundh3d9addd2002-06-27 21:36:21 +00001213
Fredrik Lundhb9056332001-07-11 17:42:21 +00001214 def make_connection(self, host):
Kristján Valur Jónsson985fc6a2009-07-01 10:01:31 +00001215 #return an existing connection if possible. This allows
1216 #HTTP/1.1 keep-alive.
1217 if self._connection and host == self._connection[0]:
1218 return self._connection[1]
Fredrik Lundhb9056332001-07-11 17:42:21 +00001219 # create a HTTP connection object from a host descriptor
Kristján Valur Jónsson985fc6a2009-07-01 10:01:31 +00001220 chost, self._extra_headers, x509 = self.get_host_info(host)
1221 self._connection = host, http.client.HTTPConnection(chost)
1222 return self._connection[1]
Jeremy Hylton5d8a88a2007-08-14 16:47:39 +00001223
Kristján Valur Jónsson985fc6a2009-07-01 10:01:31 +00001224 ##
1225 # Clear any cached connection object.
1226 # Used in the event of socket errors.
1227 #
1228 def close(self):
1229 if self._connection[1]:
1230 self._connection[1].close()
1231 self._connection = (None, None)
Fredrik Lundhb9056332001-07-11 17:42:21 +00001232
Fredrik Lundh3d9addd2002-06-27 21:36:21 +00001233 ##
Jeremy Hylton5d8a88a2007-08-14 16:47:39 +00001234 # Send HTTP request.
Fredrik Lundh3d9addd2002-06-27 21:36:21 +00001235 #
Jeremy Hylton5d8a88a2007-08-14 16:47:39 +00001236 # @param host Host descriptor (URL or (URL, x509 info) tuple).
1237 # @param handler Targer RPC handler (a path relative to host)
1238 # @param request_body The XML-RPC request body
1239 # @param debug Enable debugging if debug is true.
1240 # @return An HTTPConnection.
Fredrik Lundh3d9addd2002-06-27 21:36:21 +00001241
Jeremy Hylton5d8a88a2007-08-14 16:47:39 +00001242 def send_request(self, host, handler, request_body, debug):
Kristján Valur Jónsson985fc6a2009-07-01 10:01:31 +00001243 connection = self.make_connection(host)
1244 headers = self._extra_headers[:]
Jeremy Hylton5d8a88a2007-08-14 16:47:39 +00001245 if debug:
1246 connection.set_debuglevel(1)
Kristján Valur Jónssonaefde242009-07-19 22:29:24 +00001247 if self.accept_gzip_encoding and gzip:
Kristján Valur Jónsson985fc6a2009-07-01 10:01:31 +00001248 connection.putrequest("POST", handler, skip_accept_encoding=True)
1249 headers.append(("Accept-Encoding", "gzip"))
1250 else:
1251 connection.putrequest("POST", handler)
1252 headers.append(("Content-Type", "text/xml"))
1253 headers.append(("User-Agent", self.user_agent))
1254 self.send_headers(connection, headers)
1255 self.send_content(connection, request_body)
Jeremy Hylton5d8a88a2007-08-14 16:47:39 +00001256 return connection
Fredrik Lundhb9056332001-07-11 17:42:21 +00001257
Fredrik Lundh3d9addd2002-06-27 21:36:21 +00001258 ##
Kristján Valur Jónsson985fc6a2009-07-01 10:01:31 +00001259 # Send request headers.
1260 # This function provides a useful hook for subclassing
1261 #
1262 # @param connection httpConnection.
1263 # @param headers list of key,value pairs for HTTP headers
1264
1265 def send_headers(self, connection, headers):
1266 for key, val in headers:
1267 connection.putheader(key, val)
1268
1269 ##
1270 # Send request body.
1271 # This function provides a useful hook for subclassing
1272 #
1273 # @param connection httpConnection.
1274 # @param request_body XML-RPC request body.
1275
1276 def send_content(self, connection, request_body):
1277 #optionally encode the request
1278 if (self.encode_threshold is not None and
Kristján Valur Jónssonaefde242009-07-19 22:29:24 +00001279 self.encode_threshold < len(request_body) and
1280 gzip):
Kristján Valur Jónsson985fc6a2009-07-01 10:01:31 +00001281 connection.putheader("Content-Encoding", "gzip")
1282 request_body = gzip_encode(request_body)
1283
1284 connection.putheader("Content-Length", str(len(request_body)))
1285 connection.endheaders(request_body)
1286
1287 ##
Fredrik Lundh3d9addd2002-06-27 21:36:21 +00001288 # Parse response.
1289 #
1290 # @param file Stream.
1291 # @return Response tuple and target method.
1292
Kristján Valur Jónsson985fc6a2009-07-01 10:01:31 +00001293 def parse_response(self, response):
1294 # read response data from httpresponse, and parse it
Senthil Kumaranf34445f2010-12-08 08:04:49 +00001295 # Check for new http response object, otherwise it is a file object.
1296 if hasattr(response, 'getheader'):
1297 if response.getheader("Content-Encoding", "") == "gzip":
1298 stream = GzipDecodedResponse(response)
1299 else:
1300 stream = response
Kristján Valur Jónsson985fc6a2009-07-01 10:01:31 +00001301 else:
1302 stream = response
Fredrik Lundhb9056332001-07-11 17:42:21 +00001303
Fredrik Lundhc4c062f2001-09-10 19:45:02 +00001304 p, u = self.getparser()
Fredrik Lundhb9056332001-07-11 17:42:21 +00001305
1306 while 1:
Kristján Valur Jónsson985fc6a2009-07-01 10:01:31 +00001307 data = stream.read(1024)
1308 if not data:
Fredrik Lundhb9056332001-07-11 17:42:21 +00001309 break
1310 if self.verbose:
Kristján Valur Jónsson985fc6a2009-07-01 10:01:31 +00001311 print("body:", repr(data))
1312 p.feed(data)
Fredrik Lundhb9056332001-07-11 17:42:21 +00001313
Kristján Valur Jónsson985fc6a2009-07-01 10:01:31 +00001314 if stream is not response:
1315 stream.close()
Fredrik Lundhb9056332001-07-11 17:42:21 +00001316 p.close()
1317
1318 return u.close()
1319
Fredrik Lundh3d9addd2002-06-27 21:36:21 +00001320##
1321# Standard transport class for XML-RPC over HTTPS.
1322
Fredrik Lundhb9056332001-07-11 17:42:21 +00001323class SafeTransport(Transport):
Fredrik Lundhc4c062f2001-09-10 19:45:02 +00001324 """Handles an HTTPS transaction to an XML-RPC server."""
Fredrik Lundhb9056332001-07-11 17:42:21 +00001325
Benjamin Petersonc1da3d12014-11-29 23:32:57 -05001326 def __init__(self, use_datetime=False, use_builtin_types=False, *,
1327 context=None):
1328 super().__init__(use_datetime=use_datetime, use_builtin_types=use_builtin_types)
1329 self.context = context
1330
Fredrik Lundh3d9addd2002-06-27 21:36:21 +00001331 # FIXME: mostly untested
1332
Kristján Valur Jónsson985fc6a2009-07-01 10:01:31 +00001333 def make_connection(self, host):
1334 if self._connection and host == self._connection[0]:
1335 return self._connection[1]
1336
Senthil Kumaran6a0b5c42010-11-18 17:08:48 +00001337 if not hasattr(http.client, "HTTPSConnection"):
Fredrik Lundh1303c7c2002-10-22 18:23:00 +00001338 raise NotImplementedError(
Kristján Valur Jónsson985fc6a2009-07-01 10:01:31 +00001339 "your version of http.client doesn't support HTTPS")
1340 # create a HTTPS connection object from a host descriptor
1341 # host may be a string, or a (host, x509-dict) tuple
1342 chost, self._extra_headers, x509 = self.get_host_info(host)
1343 self._connection = host, http.client.HTTPSConnection(chost,
Benjamin Petersonc1da3d12014-11-29 23:32:57 -05001344 None, context=self.context, **(x509 or {}))
Kristján Valur Jónsson985fc6a2009-07-01 10:01:31 +00001345 return self._connection[1]
Fredrik Lundhb9056332001-07-11 17:42:21 +00001346
Fredrik Lundh3d9addd2002-06-27 21:36:21 +00001347##
1348# Standard server proxy. This class establishes a virtual connection
1349# to an XML-RPC server.
1350# <p>
1351# This class is available as ServerProxy and Server. New code should
1352# use ServerProxy, to avoid confusion.
1353#
1354# @def ServerProxy(uri, **options)
1355# @param uri The connection point on the server.
1356# @keyparam transport A transport factory, compatible with the
1357# standard transport class.
1358# @keyparam encoding The default encoding used for 8-bit strings
1359# (default is UTF-8).
1360# @keyparam verbose Use a true value to enable debugging output.
1361# (printed to standard output).
1362# @see Transport
1363
Fredrik Lundhb9056332001-07-11 17:42:21 +00001364class ServerProxy:
1365 """uri [,options] -> a logical connection to an XML-RPC server
1366
1367 uri is the connection point on the server, given as
1368 scheme://host/target.
1369
1370 The standard implementation always supports the "http" scheme. If
1371 SSL socket support is available (Python 2.0), it also supports
1372 "https".
1373
1374 If the target part and the slash preceding it are both omitted,
1375 "/RPC2" is assumed.
1376
1377 The following options can be given as keyword arguments:
1378
1379 transport: a transport factory
1380 encoding: the request encoding (default is UTF-8)
1381
1382 All 8-bit strings passed to the server proxy are assumed to use
1383 the given encoding.
1384 """
1385
Georg Brandlfe991052009-09-16 15:54:04 +00001386 def __init__(self, uri, transport=None, encoding=None, verbose=False,
Benjamin Petersonc1da3d12014-11-29 23:32:57 -05001387 allow_none=False, use_datetime=False, use_builtin_types=False,
1388 *, context=None):
Fredrik Lundhb9056332001-07-11 17:42:21 +00001389 # establish a "logical" server connection
1390
1391 # get the url
Jeremy Hylton1afc1692008-06-18 20:49:58 +00001392 type, uri = urllib.parse.splittype(uri)
Fredrik Lundhb9056332001-07-11 17:42:21 +00001393 if type not in ("http", "https"):
Andrew Svetlovf7a17b42012-12-25 16:47:37 +02001394 raise OSError("unsupported XML-RPC protocol")
Jeremy Hylton1afc1692008-06-18 20:49:58 +00001395 self.__host, self.__handler = urllib.parse.splithost(uri)
Fredrik Lundhb9056332001-07-11 17:42:21 +00001396 if not self.__handler:
1397 self.__handler = "/RPC2"
1398
1399 if transport is None:
1400 if type == "https":
Florent Xicluna61665192011-11-15 20:53:25 +01001401 handler = SafeTransport
Benjamin Petersonc1da3d12014-11-29 23:32:57 -05001402 extra_kwargs = {"context": context}
Fredrik Lundhb9056332001-07-11 17:42:21 +00001403 else:
Florent Xicluna61665192011-11-15 20:53:25 +01001404 handler = Transport
Benjamin Petersonc1da3d12014-11-29 23:32:57 -05001405 extra_kwargs = {}
Florent Xicluna61665192011-11-15 20:53:25 +01001406 transport = handler(use_datetime=use_datetime,
Benjamin Petersonc1da3d12014-11-29 23:32:57 -05001407 use_builtin_types=use_builtin_types,
1408 **extra_kwargs)
Fredrik Lundhb9056332001-07-11 17:42:21 +00001409 self.__transport = transport
1410
Senthil Kumaranb3af08f2009-04-01 20:20:43 +00001411 self.__encoding = encoding or 'utf-8'
Fredrik Lundhb9056332001-07-11 17:42:21 +00001412 self.__verbose = verbose
Andrew M. Kuchlinga4c2b742003-04-25 00:26:51 +00001413 self.__allow_none = allow_none
Tim Petersc2659cf2003-05-12 20:19:37 +00001414
Kristján Valur Jónsson985fc6a2009-07-01 10:01:31 +00001415 def __close(self):
1416 self.__transport.close()
1417
Fredrik Lundhb9056332001-07-11 17:42:21 +00001418 def __request(self, methodname, params):
1419 # call a method on the remote server
1420
Andrew M. Kuchlinga4c2b742003-04-25 00:26:51 +00001421 request = dumps(params, methodname, encoding=self.__encoding,
Senthil Kumaranb3af08f2009-04-01 20:20:43 +00001422 allow_none=self.__allow_none).encode(self.__encoding)
Fredrik Lundhb9056332001-07-11 17:42:21 +00001423
1424 response = self.__transport.request(
1425 self.__host,
1426 self.__handler,
1427 request,
1428 verbose=self.__verbose
1429 )
1430
1431 if len(response) == 1:
1432 response = response[0]
1433
1434 return response
1435
1436 def __repr__(self):
1437 return (
Fredrik Lundh78eedce2001-08-23 20:04:33 +00001438 "<ServerProxy for %s%s>" %
Fredrik Lundhb9056332001-07-11 17:42:21 +00001439 (self.__host, self.__handler)
1440 )
1441
1442 __str__ = __repr__
Raymond Hettingercc523fc2003-11-02 09:47:05 +00001443
Fredrik Lundhb9056332001-07-11 17:42:21 +00001444 def __getattr__(self, name):
1445 # magic method dispatcher
1446 return _Method(self.__request, name)
1447
1448 # note: to call a remote object with an non-standard name, use
1449 # result getattr(server, "strange-python-name")(args)
1450
Kristján Valur Jónsson985fc6a2009-07-01 10:01:31 +00001451 def __call__(self, attr):
1452 """A workaround to get special attributes on the ServerProxy
1453 without interfering with the magic __getattr__
1454 """
1455 if attr == "close":
1456 return self.__close
1457 elif attr == "transport":
1458 return self.__transport
1459 raise AttributeError("Attribute %r not found" % (attr,))
1460
Fredrik Lundh78eedce2001-08-23 20:04:33 +00001461# compatibility
Fredrik Lundh3d9addd2002-06-27 21:36:21 +00001462
Fredrik Lundhb9056332001-07-11 17:42:21 +00001463Server = ServerProxy
1464
1465# --------------------------------------------------------------------
1466# test code
1467
1468if __name__ == "__main__":
1469
1470 # simple test program (from the XML-RPC specification)
1471
Senthil Kumaran939e2db2014-01-12 16:06:58 -08001472 # local server, available from Lib/xmlrpc/server.py
1473 server = ServerProxy("http://localhost:8000")
Fredrik Lundhb9056332001-07-11 17:42:21 +00001474
Fredrik Lundhb9056332001-07-11 17:42:21 +00001475 try:
Guido van Rossumbe19ed72007-02-09 05:37:30 +00001476 print(server.currentTime.getCurrentTime())
Guido van Rossumb940e112007-01-10 16:19:56 +00001477 except Error as v:
Guido van Rossumbe19ed72007-02-09 05:37:30 +00001478 print("ERROR", v)
Martin v. Löwis12237b32004-08-22 16:04:50 +00001479
1480 multi = MultiCall(server)
Senthil Kumaran939e2db2014-01-12 16:06:58 -08001481 multi.getData()
1482 multi.pow(2,9)
1483 multi.add(1,2)
Martin v. Löwis12237b32004-08-22 16:04:50 +00001484 try:
1485 for response in multi():
Guido van Rossumbe19ed72007-02-09 05:37:30 +00001486 print(response)
Guido van Rossumb940e112007-01-10 16:19:56 +00001487 except Error as v:
Guido van Rossumbe19ed72007-02-09 05:37:30 +00001488 print("ERROR", v)