blob: ff422652e69eb3fe2ee00276582b31bb836b86ae [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 socket
138import errno
139from io import BytesIO
Kristján Valur Jónssonaefde242009-07-19 22:29:24 +0000140try:
141 import gzip
142except ImportError:
143 gzip = None #python can be built without zlib/gzip support
Fredrik Lundh1538c232001-10-01 19:42:03 +0000144
Fredrik Lundh3d9addd2002-06-27 21:36:21 +0000145# --------------------------------------------------------------------
146# Internal stuff
147
Neal Norwitzff113342007-04-17 08:42:15 +0000148def escape(s):
149 s = s.replace("&", "&")
150 s = s.replace("<", "&lt;")
151 return s.replace(">", "&gt;",)
Fredrik Lundh1538c232001-10-01 19:42:03 +0000152
Florent Xicluna75861df2011-10-30 20:39:24 +0100153# used in User-Agent header sent
154__version__ = sys.version[:3]
Fredrik Lundh3d9addd2002-06-27 21:36:21 +0000155
156# xmlrpc integer limits
Guido van Rossume2a383d2007-01-15 16:59:06 +0000157MAXINT = 2**31-1
158MININT = -2**31
Fredrik Lundh3d9addd2002-06-27 21:36:21 +0000159
160# --------------------------------------------------------------------
161# Error constants (from Dan Libby's specification at
162# http://xmlrpc-epi.sourceforge.net/specs/rfc.fault_codes.php)
163
164# Ranges of errors
165PARSE_ERROR = -32700
166SERVER_ERROR = -32600
167APPLICATION_ERROR = -32500
168SYSTEM_ERROR = -32400
169TRANSPORT_ERROR = -32300
170
171# Specific errors
172NOT_WELLFORMED_ERROR = -32700
173UNSUPPORTED_ENCODING = -32701
174INVALID_ENCODING_CHAR = -32702
175INVALID_XMLRPC = -32600
176METHOD_NOT_FOUND = -32601
177INVALID_METHOD_PARAMS = -32602
178INTERNAL_ERROR = -32603
Fredrik Lundhb9056332001-07-11 17:42:21 +0000179
180# --------------------------------------------------------------------
181# Exceptions
182
Fredrik Lundh3d9addd2002-06-27 21:36:21 +0000183##
184# Base class for all kinds of client-side errors.
185
Fredrik Lundh78eedce2001-08-23 20:04:33 +0000186class Error(Exception):
Fred Drake1b410792001-09-04 18:55:03 +0000187 """Base class for client errors."""
Fredrik Lundhb0e8e9b2001-09-10 21:45:42 +0000188 def __str__(self):
189 return repr(self)
Fredrik Lundhb9056332001-07-11 17:42:21 +0000190
Fredrik Lundh3d9addd2002-06-27 21:36:21 +0000191##
192# Indicates an HTTP-level protocol error. This is raised by the HTTP
193# transport layer, if the server returns an error code other than 200
194# (OK).
195#
196# @param url The target URL.
197# @param errcode The HTTP error code.
198# @param errmsg The HTTP error message.
199# @param headers The HTTP header dictionary.
200
Fredrik Lundhb9056332001-07-11 17:42:21 +0000201class ProtocolError(Error):
Fred Drake1b410792001-09-04 18:55:03 +0000202 """Indicates an HTTP protocol error."""
Fredrik Lundhb9056332001-07-11 17:42:21 +0000203 def __init__(self, url, errcode, errmsg, headers):
Fredrik Lundhb0e8e9b2001-09-10 21:45:42 +0000204 Error.__init__(self)
Fredrik Lundhb9056332001-07-11 17:42:21 +0000205 self.url = url
206 self.errcode = errcode
207 self.errmsg = errmsg
208 self.headers = headers
209 def __repr__(self):
210 return (
211 "<ProtocolError for %s: %s %s>" %
212 (self.url, self.errcode, self.errmsg)
213 )
214
Fredrik Lundh3d9addd2002-06-27 21:36:21 +0000215##
216# Indicates a broken XML-RPC response package. This exception is
217# raised by the unmarshalling layer, if the XML-RPC response is
218# malformed.
219
Fredrik Lundhb9056332001-07-11 17:42:21 +0000220class ResponseError(Error):
Fredrik Lundhc4c062f2001-09-10 19:45:02 +0000221 """Indicates a broken response package."""
Fredrik Lundhb9056332001-07-11 17:42:21 +0000222 pass
223
Fredrik Lundh3d9addd2002-06-27 21:36:21 +0000224##
225# Indicates an XML-RPC fault response package. This exception is
226# raised by the unmarshalling layer, if the XML-RPC response contains
Florent Xiclunac4fec932011-10-30 20:19:32 +0100227# a fault string. This exception can also be used as a class, to
Fredrik Lundh3d9addd2002-06-27 21:36:21 +0000228# generate a fault XML-RPC message.
229#
230# @param faultCode The XML-RPC fault code.
231# @param faultString The XML-RPC fault string.
232
Fredrik Lundhb9056332001-07-11 17:42:21 +0000233class Fault(Error):
Fredrik Lundhc4c062f2001-09-10 19:45:02 +0000234 """Indicates an XML-RPC fault package."""
Fredrik Lundhb9056332001-07-11 17:42:21 +0000235 def __init__(self, faultCode, faultString, **extra):
Fredrik Lundhb0e8e9b2001-09-10 21:45:42 +0000236 Error.__init__(self)
Fredrik Lundhb9056332001-07-11 17:42:21 +0000237 self.faultCode = faultCode
238 self.faultString = faultString
239 def __repr__(self):
Florent Xiclunac4fec932011-10-30 20:19:32 +0100240 return "<Fault %s: %r>" % (self.faultCode, self.faultString)
Fredrik Lundhb9056332001-07-11 17:42:21 +0000241
242# --------------------------------------------------------------------
243# Special values
244
Fredrik Lundh3d9addd2002-06-27 21:36:21 +0000245##
Georg Brandl38eceaa2008-05-26 11:14:17 +0000246# Backwards compatibility
Fredrik Lundh3d9addd2002-06-27 21:36:21 +0000247
Guido van Rossum47b9ff62006-08-24 00:41:19 +0000248boolean = Boolean = bool
Fredrik Lundh3d9addd2002-06-27 21:36:21 +0000249
250##
251# Wrapper for XML-RPC DateTime values. This converts a time value to
252# the format used by XML-RPC.
253# <p>
Florent Xicluna2738a642011-11-01 00:06:58 +0100254# The value can be given as a datetime object, as a string in the
255# format "yyyymmddThh:mm:ss", as a 9-item time tuple (as returned by
Fredrik Lundh3d9addd2002-06-27 21:36:21 +0000256# time.localtime()), or an integer value (as returned by time.time()).
257# The wrapper uses time.localtime() to convert an integer to a time
258# tuple.
259#
Florent Xicluna2738a642011-11-01 00:06:58 +0100260# @param value The time, given as a datetime object, an ISO 8601 string,
261# a time tuple, or an integer time value.
262
263
264# Issue #13305: different format codes across platforms
265_day0 = datetime(1, 1, 1)
266if _day0.strftime('%Y') == '0001': # Mac OS X
267 def _iso8601_format(value):
268 return value.strftime("%Y%m%dT%H:%M:%S")
269elif _day0.strftime('%4Y') == '0001': # Linux
270 def _iso8601_format(value):
271 return value.strftime("%4Y%m%dT%H:%M:%S")
272else:
273 def _iso8601_format(value):
274 return value.strftime("%Y%m%dT%H:%M:%S").zfill(17)
275del _day0
276
Fredrik Lundhb9056332001-07-11 17:42:21 +0000277
Christian Heimesdae2a892008-04-19 00:55:37 +0000278def _strftime(value):
Florent Xiclunab6f019a2011-10-30 23:54:17 +0100279 if isinstance(value, datetime):
Florent Xicluna2738a642011-11-01 00:06:58 +0100280 return _iso8601_format(value)
Christian Heimesdae2a892008-04-19 00:55:37 +0000281
282 if not isinstance(value, (tuple, time.struct_time)):
283 if value == 0:
284 value = time.time()
285 value = time.localtime(value)
286
287 return "%04d%02d%02dT%02d:%02d:%02d" % value[:6]
288
Fredrik Lundhb9056332001-07-11 17:42:21 +0000289class DateTime:
Fredrik Lundhc4c062f2001-09-10 19:45:02 +0000290 """DateTime wrapper for an ISO 8601 string or time tuple or
291 localtime integer value to generate 'dateTime.iso8601' XML-RPC
292 value.
293 """
Fredrik Lundhb9056332001-07-11 17:42:21 +0000294
295 def __init__(self, value=0):
Christian Heimesdae2a892008-04-19 00:55:37 +0000296 if isinstance(value, str):
297 self.value = value
298 else:
299 self.value = _strftime(value)
Fredrik Lundhb9056332001-07-11 17:42:21 +0000300
Christian Heimes05e8be12008-02-23 18:30:17 +0000301 def make_comparable(self, other):
Fredrik Lundhb9056332001-07-11 17:42:21 +0000302 if isinstance(other, DateTime):
Christian Heimes05e8be12008-02-23 18:30:17 +0000303 s = self.value
304 o = other.value
Florent Xiclunab6f019a2011-10-30 23:54:17 +0100305 elif isinstance(other, datetime):
Christian Heimes05e8be12008-02-23 18:30:17 +0000306 s = self.value
Florent Xicluna2738a642011-11-01 00:06:58 +0100307 o = _iso8601_format(other)
Florent Xicluna3fa29f72011-10-30 20:18:50 +0100308 elif isinstance(other, str):
Christian Heimes05e8be12008-02-23 18:30:17 +0000309 s = self.value
310 o = other
311 elif hasattr(other, "timetuple"):
312 s = self.timetuple()
313 o = other.timetuple()
314 else:
315 otype = (hasattr(other, "__class__")
316 and other.__class__.__name__
317 or type(other))
318 raise TypeError("Can't compare %s and %s" %
319 (self.__class__.__name__, otype))
320 return s, o
321
322 def __lt__(self, other):
323 s, o = self.make_comparable(other)
324 return s < o
325
326 def __le__(self, other):
327 s, o = self.make_comparable(other)
328 return s <= o
329
330 def __gt__(self, other):
331 s, o = self.make_comparable(other)
332 return s > o
333
334 def __ge__(self, other):
335 s, o = self.make_comparable(other)
336 return s >= o
337
338 def __eq__(self, other):
339 s, o = self.make_comparable(other)
340 return s == o
Guido van Rossum47b9ff62006-08-24 00:41:19 +0000341
342 def __ne__(self, other):
Christian Heimes05e8be12008-02-23 18:30:17 +0000343 s, o = self.make_comparable(other)
344 return s != o
345
346 def timetuple(self):
347 return time.strptime(self.value, "%Y%m%dT%H:%M:%S")
348
Fredrik Lundh3d9addd2002-06-27 21:36:21 +0000349 ##
350 # Get date/time value.
351 #
352 # @return Date/time value, as an ISO 8601 string.
353
354 def __str__(self):
355 return self.value
356
Fredrik Lundhb9056332001-07-11 17:42:21 +0000357 def __repr__(self):
Florent Xiclunac4fec932011-10-30 20:19:32 +0100358 return "<DateTime %r at %x>" % (self.value, id(self))
Fredrik Lundhb9056332001-07-11 17:42:21 +0000359
360 def decode(self, data):
Neal Norwitzff113342007-04-17 08:42:15 +0000361 self.value = str(data).strip()
Fredrik Lundhb9056332001-07-11 17:42:21 +0000362
363 def encode(self, out):
364 out.write("<value><dateTime.iso8601>")
365 out.write(self.value)
366 out.write("</dateTime.iso8601></value>\n")
367
Fredrik Lundh3d9addd2002-06-27 21:36:21 +0000368def _datetime(data):
369 # decode xml element contents into a DateTime structure.
Fredrik Lundhb9056332001-07-11 17:42:21 +0000370 value = DateTime()
371 value.decode(data)
372 return value
373
Skip Montanaro174dd222005-05-14 20:54:16 +0000374def _datetime_type(data):
Florent Xiclunab6f019a2011-10-30 23:54:17 +0100375 return datetime.strptime(data, "%Y%m%dT%H:%M:%S")
Skip Montanaro174dd222005-05-14 20:54:16 +0000376
Fredrik Lundh3d9addd2002-06-27 21:36:21 +0000377##
378# Wrapper for binary data. This can be used to transport any kind
379# of binary data over XML-RPC, using BASE64 encoding.
380#
381# @param data An 8-bit string containing arbitrary data.
382
Fredrik Lundhb9056332001-07-11 17:42:21 +0000383class Binary:
Fred Drake1b410792001-09-04 18:55:03 +0000384 """Wrapper for binary data."""
Fredrik Lundhb9056332001-07-11 17:42:21 +0000385
386 def __init__(self, data=None):
Guido van Rossum54a40cb2007-08-27 22:27:41 +0000387 if data is None:
388 data = b""
389 else:
Florent Xicluna61665192011-11-15 20:53:25 +0100390 if not isinstance(data, (bytes, bytearray)):
391 raise TypeError("expected bytes or bytearray, not %s" %
Guido van Rossum54a40cb2007-08-27 22:27:41 +0000392 data.__class__.__name__)
393 data = bytes(data) # Make a copy of the bytes!
Fredrik Lundhb9056332001-07-11 17:42:21 +0000394 self.data = data
395
Fredrik Lundh3d9addd2002-06-27 21:36:21 +0000396 ##
397 # Get buffer contents.
398 #
399 # @return Buffer contents, as an 8-bit string.
400
401 def __str__(self):
Guido van Rossum54a40cb2007-08-27 22:27:41 +0000402 return str(self.data, "latin-1") # XXX encoding?!
Fredrik Lundh3d9addd2002-06-27 21:36:21 +0000403
Guido van Rossum47b9ff62006-08-24 00:41:19 +0000404 def __eq__(self, other):
Fredrik Lundhb9056332001-07-11 17:42:21 +0000405 if isinstance(other, Binary):
406 other = other.data
Guido van Rossum47b9ff62006-08-24 00:41:19 +0000407 return self.data == other
408
409 def __ne__(self, other):
410 if isinstance(other, Binary):
411 other = other.data
412 return self.data != other
Fredrik Lundhb9056332001-07-11 17:42:21 +0000413
414 def decode(self, data):
Georg Brandlb54d8012009-06-04 09:11:51 +0000415 self.data = base64.decodebytes(data)
Fredrik Lundhb9056332001-07-11 17:42:21 +0000416
417 def encode(self, out):
Fredrik Lundhb9056332001-07-11 17:42:21 +0000418 out.write("<value><base64>\n")
Georg Brandlb54d8012009-06-04 09:11:51 +0000419 encoded = base64.encodebytes(self.data)
Brett Cannon2dbde5e2007-07-30 03:50:35 +0000420 out.write(encoded.decode('ascii'))
Fredrik Lundhb9056332001-07-11 17:42:21 +0000421 out.write("</base64></value>\n")
422
Fredrik Lundh3d9addd2002-06-27 21:36:21 +0000423def _binary(data):
424 # decode xml element contents into a Binary structure
Fredrik Lundhb9056332001-07-11 17:42:21 +0000425 value = Binary()
426 value.decode(data)
427 return value
428
Skip Montanaro9a7c96a2003-01-22 18:17:25 +0000429WRAPPERS = (DateTime, Binary)
Fredrik Lundhb9056332001-07-11 17:42:21 +0000430
431# --------------------------------------------------------------------
432# XML parsers
433
Georg Brandlcef803f2009-06-04 09:04:53 +0000434class ExpatParser:
435 # fast expat parser for Python 2.0 and later.
436 def __init__(self, target):
437 self._parser = parser = expat.ParserCreate(None, None)
438 self._target = target
439 parser.StartElementHandler = target.start
440 parser.EndElementHandler = target.end
441 parser.CharacterDataHandler = target.data
442 encoding = None
443 target.xml(encoding, None)
Fredrik Lundhb9056332001-07-11 17:42:21 +0000444
Georg Brandlcef803f2009-06-04 09:04:53 +0000445 def feed(self, data):
446 self._parser.Parse(data, 0)
Fredrik Lundhb9056332001-07-11 17:42:21 +0000447
Georg Brandlcef803f2009-06-04 09:04:53 +0000448 def close(self):
449 self._parser.Parse("", 1) # end of data
450 del self._target, self._parser # get rid of circular references
Fredrik Lundhb9056332001-07-11 17:42:21 +0000451
Fredrik Lundhb9056332001-07-11 17:42:21 +0000452# --------------------------------------------------------------------
453# XML-RPC marshalling and unmarshalling code
454
Fredrik Lundh3d9addd2002-06-27 21:36:21 +0000455##
456# XML-RPC marshaller.
457#
458# @param encoding Default encoding for 8-bit strings. The default
459# value is None (interpreted as UTF-8).
460# @see dumps
461
Fredrik Lundhb9056332001-07-11 17:42:21 +0000462class Marshaller:
Fred Drake1b410792001-09-04 18:55:03 +0000463 """Generate an XML-RPC params chunk from a Python data structure.
Fredrik Lundhb9056332001-07-11 17:42:21 +0000464
Fredrik Lundhc4c062f2001-09-10 19:45:02 +0000465 Create a Marshaller instance for each set of parameters, and use
466 the "dumps" method to convert your data (represented as a tuple)
467 to an XML-RPC params chunk. To write a fault response, pass a
468 Fault instance instead. You may prefer to use the "dumps" module
469 function for this purpose.
Fred Drake1b410792001-09-04 18:55:03 +0000470 """
Fredrik Lundhb9056332001-07-11 17:42:21 +0000471
472 # by the way, if you don't understand what's going on in here,
473 # that's perfectly ok.
474
Georg Brandlfe991052009-09-16 15:54:04 +0000475 def __init__(self, encoding=None, allow_none=False):
Fredrik Lundhb9056332001-07-11 17:42:21 +0000476 self.memo = {}
477 self.data = None
478 self.encoding = encoding
Andrew M. Kuchlinga4c2b742003-04-25 00:26:51 +0000479 self.allow_none = allow_none
Tim Petersc2659cf2003-05-12 20:19:37 +0000480
Fredrik Lundhb9056332001-07-11 17:42:21 +0000481 dispatch = {}
482
483 def dumps(self, values):
Fredrik Lundh3d9addd2002-06-27 21:36:21 +0000484 out = []
485 write = out.append
486 dump = self.__dump
Fredrik Lundhb9056332001-07-11 17:42:21 +0000487 if isinstance(values, Fault):
488 # fault instance
489 write("<fault>\n")
Martin v. Löwis541342f2003-07-12 07:53:04 +0000490 dump({'faultCode': values.faultCode,
491 'faultString': values.faultString},
492 write)
Fredrik Lundhb9056332001-07-11 17:42:21 +0000493 write("</fault>\n")
494 else:
495 # parameter block
Fredrik Lundhc266bb02001-08-23 20:13:08 +0000496 # FIXME: the xml-rpc specification allows us to leave out
497 # the entire <params> block if there are no parameters.
498 # however, changing this may break older code (including
499 # old versions of xmlrpclib.py), so this is better left as
500 # is for now. See @XMLRPC3 for more information. /F
Fredrik Lundhb9056332001-07-11 17:42:21 +0000501 write("<params>\n")
502 for v in values:
503 write("<param>\n")
Fredrik Lundh3d9addd2002-06-27 21:36:21 +0000504 dump(v, write)
Fredrik Lundhb9056332001-07-11 17:42:21 +0000505 write("</param>\n")
506 write("</params>\n")
Neal Norwitzff113342007-04-17 08:42:15 +0000507 result = "".join(out)
Fredrik Lundhb9056332001-07-11 17:42:21 +0000508 return result
509
Fredrik Lundh3d9addd2002-06-27 21:36:21 +0000510 def __dump(self, value, write):
Fredrik Lundhb9056332001-07-11 17:42:21 +0000511 try:
512 f = self.dispatch[type(value)]
513 except KeyError:
Thomas Wouters89f507f2006-12-13 04:49:30 +0000514 # check if this object can be marshalled as a structure
Florent Xicluna93dfee12011-10-30 20:22:25 +0100515 if not hasattr(value, '__dict__'):
Collin Winterce36ad82007-08-30 01:19:48 +0000516 raise TypeError("cannot marshal %s objects" % type(value))
Thomas Wouters89f507f2006-12-13 04:49:30 +0000517 # check if this class is a sub-class of a basic type,
518 # because we don't know how to marshal these types
519 # (e.g. a string sub-class)
520 for type_ in type(value).__mro__:
521 if type_ in self.dispatch.keys():
Collin Winterce36ad82007-08-30 01:19:48 +0000522 raise TypeError("cannot marshal %s objects" % type(value))
Thomas Wouters89f507f2006-12-13 04:49:30 +0000523 # XXX(twouters): using "_arbitrary_instance" as key as a quick-fix
524 # for the p3yk merge, this should probably be fixed more neatly.
525 f = self.dispatch["_arbitrary_instance"]
526 f(self, value, write)
Fredrik Lundhb9056332001-07-11 17:42:21 +0000527
Andrew M. Kuchlinga4c2b742003-04-25 00:26:51 +0000528 def dump_nil (self, value, write):
529 if not self.allow_none:
Collin Winterce36ad82007-08-30 01:19:48 +0000530 raise TypeError("cannot marshal None unless allow_none is enabled")
Andrew M. Kuchlinga4c2b742003-04-25 00:26:51 +0000531 write("<value><nil/></value>")
Guido van Rossum13257902007-06-07 23:15:56 +0000532 dispatch[type(None)] = dump_nil
Tim Petersc2659cf2003-05-12 20:19:37 +0000533
Georg Brandl38eceaa2008-05-26 11:14:17 +0000534 def dump_bool(self, value, write):
535 write("<value><boolean>")
536 write(value and "1" or "0")
537 write("</boolean></value>\n")
538 dispatch[bool] = dump_bool
Skip Montanaro9a7c96a2003-01-22 18:17:25 +0000539
Fredrik Lundh3d9addd2002-06-27 21:36:21 +0000540 def dump_long(self, value, write):
Skip Montanaro5449e082001-10-17 22:53:33 +0000541 if value > MAXINT or value < MININT:
Collin Winterce36ad82007-08-30 01:19:48 +0000542 raise OverflowError("long int exceeds XML-RPC limits")
Fredrik Lundh3d9addd2002-06-27 21:36:21 +0000543 write("<value><int>")
544 write(str(int(value)))
545 write("</int></value>\n")
Guido van Rossum13257902007-06-07 23:15:56 +0000546 dispatch[int] = dump_long
Skip Montanaro5e9c71b2001-10-10 15:56:34 +0000547
Florent Xicluna2bb96f52011-10-23 22:11:00 +0200548 # backward compatible
549 dump_int = dump_long
550
Fredrik Lundh3d9addd2002-06-27 21:36:21 +0000551 def dump_double(self, value, write):
552 write("<value><double>")
553 write(repr(value))
554 write("</double></value>\n")
Guido van Rossum13257902007-06-07 23:15:56 +0000555 dispatch[float] = dump_double
Fredrik Lundhb9056332001-07-11 17:42:21 +0000556
Guido van Rossum54ad5232007-05-27 09:17:48 +0000557 def dump_unicode(self, value, write, escape=escape):
Guido van Rossum54ad5232007-05-27 09:17:48 +0000558 write("<value><string>")
559 write(escape(value))
560 write("</string></value>\n")
561 dispatch[str] = dump_unicode
Fredrik Lundhb9056332001-07-11 17:42:21 +0000562
Florent Xicluna61665192011-11-15 20:53:25 +0100563 def dump_bytes(self, value, write):
564 write("<value><base64>\n")
565 encoded = base64.encodebytes(value)
566 write(encoded.decode('ascii'))
567 write("</base64></value>\n")
568 dispatch[bytes] = dump_bytes
569 dispatch[bytearray] = dump_bytes
570
Fredrik Lundh3d9addd2002-06-27 21:36:21 +0000571 def dump_array(self, value, write):
572 i = id(value)
Guido van Rossume2b70bc2006-08-18 22:13:04 +0000573 if i in self.memo:
Collin Winterce36ad82007-08-30 01:19:48 +0000574 raise TypeError("cannot marshal recursive sequences")
Fredrik Lundh3d9addd2002-06-27 21:36:21 +0000575 self.memo[i] = None
Fredrik Lundh1538c232001-10-01 19:42:03 +0000576 dump = self.__dump
Fredrik Lundhb9056332001-07-11 17:42:21 +0000577 write("<value><array><data>\n")
578 for v in value:
Fredrik Lundh3d9addd2002-06-27 21:36:21 +0000579 dump(v, write)
Fredrik Lundhb9056332001-07-11 17:42:21 +0000580 write("</data></array></value>\n")
Fredrik Lundh3d9addd2002-06-27 21:36:21 +0000581 del self.memo[i]
Guido van Rossum13257902007-06-07 23:15:56 +0000582 dispatch[tuple] = dump_array
583 dispatch[list] = dump_array
Fredrik Lundhb9056332001-07-11 17:42:21 +0000584
Fredrik Lundh3d9addd2002-06-27 21:36:21 +0000585 def dump_struct(self, value, write, escape=escape):
586 i = id(value)
Guido van Rossume2b70bc2006-08-18 22:13:04 +0000587 if i in self.memo:
Collin Winterce36ad82007-08-30 01:19:48 +0000588 raise TypeError("cannot marshal recursive dictionaries")
Fredrik Lundh3d9addd2002-06-27 21:36:21 +0000589 self.memo[i] = None
Fredrik Lundh1538c232001-10-01 19:42:03 +0000590 dump = self.__dump
Fredrik Lundhb9056332001-07-11 17:42:21 +0000591 write("<value><struct>\n")
Andrew M. Kuchling5962f452004-06-05 12:35:58 +0000592 for k, v in value.items():
Fredrik Lundhb9056332001-07-11 17:42:21 +0000593 write("<member>\n")
Guido van Rossum3172c5d2007-10-16 18:12:55 +0000594 if not isinstance(k, str):
Collin Winterce36ad82007-08-30 01:19:48 +0000595 raise TypeError("dictionary key must be string")
Fredrik Lundh1538c232001-10-01 19:42:03 +0000596 write("<name>%s</name>\n" % escape(k))
Andrew M. Kuchling5962f452004-06-05 12:35:58 +0000597 dump(v, write)
Fredrik Lundhb9056332001-07-11 17:42:21 +0000598 write("</member>\n")
599 write("</struct></value>\n")
Fredrik Lundh3d9addd2002-06-27 21:36:21 +0000600 del self.memo[i]
Guido van Rossum13257902007-06-07 23:15:56 +0000601 dispatch[dict] = dump_struct
Fredrik Lundhb9056332001-07-11 17:42:21 +0000602
Florent Xiclunab6f019a2011-10-30 23:54:17 +0100603 def dump_datetime(self, value, write):
604 write("<value><dateTime.iso8601>")
605 write(_strftime(value))
606 write("</dateTime.iso8601></value>\n")
607 dispatch[datetime] = dump_datetime
Fred Drakeba613c32005-02-10 18:33:30 +0000608
Fredrik Lundh3d9addd2002-06-27 21:36:21 +0000609 def dump_instance(self, value, write):
Fredrik Lundhb9056332001-07-11 17:42:21 +0000610 # check for special wrappers
611 if value.__class__ in WRAPPERS:
Fredrik Lundh3d9addd2002-06-27 21:36:21 +0000612 self.write = write
Fredrik Lundhb9056332001-07-11 17:42:21 +0000613 value.encode(self)
Fredrik Lundh3d9addd2002-06-27 21:36:21 +0000614 del self.write
Fredrik Lundhb9056332001-07-11 17:42:21 +0000615 else:
616 # store instance attributes as a struct (really?)
Fredrik Lundh3d9addd2002-06-27 21:36:21 +0000617 self.dump_struct(value.__dict__, write)
Guido van Rossume4dea982006-04-21 13:45:00 +0000618 dispatch[DateTime] = dump_instance
619 dispatch[Binary] = dump_instance
Thomas Wouters89f507f2006-12-13 04:49:30 +0000620 # XXX(twouters): using "_arbitrary_instance" as key as a quick-fix
621 # for the p3yk merge, this should probably be fixed more neatly.
622 dispatch["_arbitrary_instance"] = dump_instance
Fredrik Lundhb9056332001-07-11 17:42:21 +0000623
Fredrik Lundh3d9addd2002-06-27 21:36:21 +0000624##
625# XML-RPC unmarshaller.
626#
627# @see loads
628
Fredrik Lundhb9056332001-07-11 17:42:21 +0000629class Unmarshaller:
Fred Drake1b410792001-09-04 18:55:03 +0000630 """Unmarshal an XML-RPC response, based on incoming XML event
Fredrik Lundhb0e8e9b2001-09-10 21:45:42 +0000631 messages (start, data, end). Call close() to get the resulting
Fred Drake1b410792001-09-04 18:55:03 +0000632 data structure.
Fredrik Lundhb9056332001-07-11 17:42:21 +0000633
Fredrik Lundhb0e8e9b2001-09-10 21:45:42 +0000634 Note that this reader is fairly tolerant, and gladly accepts bogus
635 XML-RPC data without complaining (but not bogus XML).
Fred Drake1b410792001-09-04 18:55:03 +0000636 """
Fredrik Lundhb9056332001-07-11 17:42:21 +0000637
638 # and again, if you don't understand what's going on in here,
639 # that's perfectly ok.
640
Florent Xicluna61665192011-11-15 20:53:25 +0100641 def __init__(self, use_datetime=False, use_builtin_types=False):
Fredrik Lundhb9056332001-07-11 17:42:21 +0000642 self._type = None
643 self._stack = []
644 self._marks = []
645 self._data = []
646 self._methodname = None
647 self._encoding = "utf-8"
648 self.append = self._stack.append
Florent Xicluna61665192011-11-15 20:53:25 +0100649 self._use_datetime = use_builtin_types or use_datetime
650 self._use_bytes = use_builtin_types
Fredrik Lundhb9056332001-07-11 17:42:21 +0000651
652 def close(self):
653 # return response tuple and target method
654 if self._type is None or self._marks:
655 raise ResponseError()
656 if self._type == "fault":
Guido van Rossum68468eb2003-02-27 20:14:51 +0000657 raise Fault(**self._stack[0])
Fredrik Lundhb9056332001-07-11 17:42:21 +0000658 return tuple(self._stack)
659
660 def getmethodname(self):
661 return self._methodname
662
663 #
664 # event handlers
665
666 def xml(self, encoding, standalone):
667 self._encoding = encoding
668 # FIXME: assert standalone == 1 ???
669
670 def start(self, tag, attrs):
671 # prepare to handle this element
672 if tag == "array" or tag == "struct":
673 self._marks.append(len(self._stack))
674 self._data = []
675 self._value = (tag == "value")
676
677 def data(self, text):
678 self._data.append(text)
679
Neal Norwitzff113342007-04-17 08:42:15 +0000680 def end(self, tag):
Fredrik Lundhb9056332001-07-11 17:42:21 +0000681 # call the appropriate end tag handler
682 try:
683 f = self.dispatch[tag]
684 except KeyError:
685 pass # unknown tag ?
686 else:
Neal Norwitzff113342007-04-17 08:42:15 +0000687 return f(self, "".join(self._data))
Fredrik Lundhb9056332001-07-11 17:42:21 +0000688
689 #
690 # accelerator support
691
692 def end_dispatch(self, tag, data):
693 # dispatch data
694 try:
695 f = self.dispatch[tag]
696 except KeyError:
697 pass # unknown tag ?
698 else:
699 return f(self, data)
700
701 #
702 # element decoders
703
704 dispatch = {}
705
Andrew M. Kuchlinga4c2b742003-04-25 00:26:51 +0000706 def end_nil (self, data):
707 self.append(None)
708 self._value = 0
709 dispatch["nil"] = end_nil
Tim Petersc2659cf2003-05-12 20:19:37 +0000710
Fredrik Lundh1538c232001-10-01 19:42:03 +0000711 def end_boolean(self, data):
Fredrik Lundhb9056332001-07-11 17:42:21 +0000712 if data == "0":
713 self.append(False)
714 elif data == "1":
715 self.append(True)
716 else:
Collin Winterce36ad82007-08-30 01:19:48 +0000717 raise TypeError("bad boolean value")
Fredrik Lundhb9056332001-07-11 17:42:21 +0000718 self._value = 0
719 dispatch["boolean"] = end_boolean
720
Fredrik Lundh1538c232001-10-01 19:42:03 +0000721 def end_int(self, data):
722 self.append(int(data))
Fredrik Lundhb9056332001-07-11 17:42:21 +0000723 self._value = 0
724 dispatch["i4"] = end_int
Georg Brandlf08a9dd2008-06-10 16:57:31 +0000725 dispatch["i8"] = end_int
Fredrik Lundhb9056332001-07-11 17:42:21 +0000726 dispatch["int"] = end_int
727
Fredrik Lundh1538c232001-10-01 19:42:03 +0000728 def end_double(self, data):
729 self.append(float(data))
Fredrik Lundhb9056332001-07-11 17:42:21 +0000730 self._value = 0
731 dispatch["double"] = end_double
732
Fredrik Lundh1538c232001-10-01 19:42:03 +0000733 def end_string(self, data):
Fredrik Lundhb9056332001-07-11 17:42:21 +0000734 if self._encoding:
Alexandre Vassalottie671fd22009-07-22 02:32:34 +0000735 data = data.decode(self._encoding)
736 self.append(data)
Fredrik Lundhb9056332001-07-11 17:42:21 +0000737 self._value = 0
738 dispatch["string"] = end_string
739 dispatch["name"] = end_string # struct keys are always strings
740
741 def end_array(self, data):
Raymond Hettinger46ac8eb2002-06-30 03:39:14 +0000742 mark = self._marks.pop()
Fredrik Lundhb9056332001-07-11 17:42:21 +0000743 # map arrays to Python lists
744 self._stack[mark:] = [self._stack[mark:]]
745 self._value = 0
746 dispatch["array"] = end_array
747
748 def end_struct(self, data):
Raymond Hettinger46ac8eb2002-06-30 03:39:14 +0000749 mark = self._marks.pop()
Fredrik Lundhb9056332001-07-11 17:42:21 +0000750 # map structs to Python dictionaries
751 dict = {}
752 items = self._stack[mark:]
753 for i in range(0, len(items), 2):
Alexandre Vassalottie671fd22009-07-22 02:32:34 +0000754 dict[items[i]] = items[i+1]
Fredrik Lundhb9056332001-07-11 17:42:21 +0000755 self._stack[mark:] = [dict]
756 self._value = 0
757 dispatch["struct"] = end_struct
758
Fredrik Lundh1538c232001-10-01 19:42:03 +0000759 def end_base64(self, data):
Fredrik Lundhb9056332001-07-11 17:42:21 +0000760 value = Binary()
Guido van Rossum54a40cb2007-08-27 22:27:41 +0000761 value.decode(data.encode("ascii"))
Florent Xicluna61665192011-11-15 20:53:25 +0100762 if self._use_bytes:
763 value = value.data
Fredrik Lundhb9056332001-07-11 17:42:21 +0000764 self.append(value)
765 self._value = 0
766 dispatch["base64"] = end_base64
767
Fredrik Lundh1538c232001-10-01 19:42:03 +0000768 def end_dateTime(self, data):
Fredrik Lundhb9056332001-07-11 17:42:21 +0000769 value = DateTime()
Fredrik Lundh1538c232001-10-01 19:42:03 +0000770 value.decode(data)
Skip Montanaro174dd222005-05-14 20:54:16 +0000771 if self._use_datetime:
772 value = _datetime_type(data)
Fredrik Lundhb9056332001-07-11 17:42:21 +0000773 self.append(value)
774 dispatch["dateTime.iso8601"] = end_dateTime
775
776 def end_value(self, data):
Fredrik Lundh3d9addd2002-06-27 21:36:21 +0000777 # if we stumble upon a value element with no internal
Fredrik Lundhb9056332001-07-11 17:42:21 +0000778 # elements, treat it as a string element
779 if self._value:
780 self.end_string(data)
781 dispatch["value"] = end_value
782
783 def end_params(self, data):
784 self._type = "params"
785 dispatch["params"] = end_params
786
787 def end_fault(self, data):
788 self._type = "fault"
789 dispatch["fault"] = end_fault
790
Fredrik Lundh1538c232001-10-01 19:42:03 +0000791 def end_methodName(self, data):
Fredrik Lundhb9056332001-07-11 17:42:21 +0000792 if self._encoding:
Alexandre Vassalottie671fd22009-07-22 02:32:34 +0000793 data = data.decode(self._encoding)
Fredrik Lundhb9056332001-07-11 17:42:21 +0000794 self._methodname = data
795 self._type = "methodName" # no params
796 dispatch["methodName"] = end_methodName
797
Martin v. Löwis45394c22003-10-31 13:49:36 +0000798## Multicall support
799#
Fredrik Lundhb9056332001-07-11 17:42:21 +0000800
Martin v. Löwis45394c22003-10-31 13:49:36 +0000801class _MultiCallMethod:
802 # some lesser magic to store calls made to a MultiCall object
803 # for batch execution
804 def __init__(self, call_list, name):
805 self.__call_list = call_list
806 self.__name = name
807 def __getattr__(self, name):
808 return _MultiCallMethod(self.__call_list, "%s.%s" % (self.__name, name))
809 def __call__(self, *args):
810 self.__call_list.append((self.__name, args))
811
Martin v. Löwis12237b32004-08-22 16:04:50 +0000812class MultiCallIterator:
Martin v. Löwis45394c22003-10-31 13:49:36 +0000813 """Iterates over the results of a multicall. Exceptions are
Andrew Svetlov737fb892012-12-18 21:14:22 +0200814 raised in response to xmlrpc faults."""
Raymond Hettingercc523fc2003-11-02 09:47:05 +0000815
Martin v. Löwis12237b32004-08-22 16:04:50 +0000816 def __init__(self, results):
817 self.results = results
818
819 def __getitem__(self, i):
820 item = self.results[i]
821 if type(item) == type({}):
822 raise Fault(item['faultCode'], item['faultString'])
823 elif type(item) == type([]):
824 return item[0]
Martin v. Löwis45394c22003-10-31 13:49:36 +0000825 else:
Collin Winterce36ad82007-08-30 01:19:48 +0000826 raise ValueError("unexpected type in multicall result")
Raymond Hettingercc523fc2003-11-02 09:47:05 +0000827
Martin v. Löwis45394c22003-10-31 13:49:36 +0000828class MultiCall:
829 """server -> a object used to boxcar method calls
830
831 server should be a ServerProxy object.
832
833 Methods can be added to the MultiCall using normal
834 method call syntax e.g.:
835
836 multicall = MultiCall(server_proxy)
837 multicall.add(2,3)
838 multicall.get_address("Guido")
839
840 To execute the multicall, call the MultiCall object e.g.:
841
842 add_result, address = multicall()
843 """
Raymond Hettingercc523fc2003-11-02 09:47:05 +0000844
Martin v. Löwis45394c22003-10-31 13:49:36 +0000845 def __init__(self, server):
846 self.__server = server
847 self.__call_list = []
Raymond Hettingercc523fc2003-11-02 09:47:05 +0000848
Martin v. Löwis45394c22003-10-31 13:49:36 +0000849 def __repr__(self):
850 return "<MultiCall at %x>" % id(self)
Raymond Hettingercc523fc2003-11-02 09:47:05 +0000851
Martin v. Löwis45394c22003-10-31 13:49:36 +0000852 __str__ = __repr__
853
854 def __getattr__(self, name):
855 return _MultiCallMethod(self.__call_list, name)
856
857 def __call__(self):
858 marshalled_list = []
859 for name, args in self.__call_list:
860 marshalled_list.append({'methodName' : name, 'params' : args})
861
862 return MultiCallIterator(self.__server.system.multicall(marshalled_list))
Raymond Hettingercc523fc2003-11-02 09:47:05 +0000863
Fredrik Lundhb9056332001-07-11 17:42:21 +0000864# --------------------------------------------------------------------
865# convenience functions
866
Georg Brandl38eceaa2008-05-26 11:14:17 +0000867FastMarshaller = FastParser = FastUnmarshaller = None
868
Fredrik Lundh3d9addd2002-06-27 21:36:21 +0000869##
870# Create a parser object, and connect it to an unmarshalling instance.
871# This function picks the fastest available XML parser.
872#
873# return A (parser, unmarshaller) tuple.
874
Florent Xicluna61665192011-11-15 20:53:25 +0100875def getparser(use_datetime=False, use_builtin_types=False):
Fredrik Lundhb9056332001-07-11 17:42:21 +0000876 """getparser() -> parser, unmarshaller
877
Fredrik Lundhb0e8e9b2001-09-10 21:45:42 +0000878 Create an instance of the fastest available parser, and attach it
879 to an unmarshalling object. Return both objects.
Fredrik Lundhb9056332001-07-11 17:42:21 +0000880 """
881 if FastParser and FastUnmarshaller:
Florent Xicluna61665192011-11-15 20:53:25 +0100882 if use_builtin_types:
Skip Montanaro174dd222005-05-14 20:54:16 +0000883 mkdatetime = _datetime_type
Florent Xicluna61665192011-11-15 20:53:25 +0100884 mkbytes = base64.decodebytes
885 elif use_datetime:
886 mkdatetime = _datetime_type
887 mkbytes = _binary
Skip Montanaro174dd222005-05-14 20:54:16 +0000888 else:
889 mkdatetime = _datetime
Florent Xicluna61665192011-11-15 20:53:25 +0100890 mkbytes = _binary
891 target = FastUnmarshaller(True, False, mkbytes, mkdatetime, Fault)
Fredrik Lundhb9056332001-07-11 17:42:21 +0000892 parser = FastParser(target)
893 else:
Florent Xicluna61665192011-11-15 20:53:25 +0100894 target = Unmarshaller(use_datetime=use_datetime, use_builtin_types=use_builtin_types)
Fredrik Lundhb9056332001-07-11 17:42:21 +0000895 if FastParser:
896 parser = FastParser(target)
Fredrik Lundhb9056332001-07-11 17:42:21 +0000897 else:
Georg Brandlcef803f2009-06-04 09:04:53 +0000898 parser = ExpatParser(target)
Fredrik Lundhb9056332001-07-11 17:42:21 +0000899 return parser, target
900
Fredrik Lundh3d9addd2002-06-27 21:36:21 +0000901##
902# Convert a Python tuple or a Fault instance to an XML-RPC packet.
903#
904# @def dumps(params, **options)
905# @param params A tuple or Fault instance.
906# @keyparam methodname If given, create a methodCall request for
907# this method name.
908# @keyparam methodresponse If given, create a methodResponse packet.
909# If used with a tuple, the tuple must be a singleton (that is,
910# it must contain exactly one element).
911# @keyparam encoding The packet encoding.
912# @return A string containing marshalled data.
913
Andrew M. Kuchlinga4c2b742003-04-25 00:26:51 +0000914def dumps(params, methodname=None, methodresponse=None, encoding=None,
Georg Brandlfe991052009-09-16 15:54:04 +0000915 allow_none=False):
Fredrik Lundhb9056332001-07-11 17:42:21 +0000916 """data [,options] -> marshalled data
917
918 Convert an argument tuple or a Fault instance to an XML-RPC
919 request (or response, if the methodresponse option is used).
920
Fredrik Lundhb0e8e9b2001-09-10 21:45:42 +0000921 In addition to the data object, the following options can be given
922 as keyword arguments:
Fredrik Lundhb9056332001-07-11 17:42:21 +0000923
924 methodname: the method name for a methodCall packet
925
926 methodresponse: true to create a methodResponse packet.
927 If this option is used with a tuple, the tuple must be
928 a singleton (i.e. it can contain only one element).
929
930 encoding: the packet encoding (default is UTF-8)
931
Florent Xicluna61665192011-11-15 20:53:25 +0100932 All byte strings in the data structure are assumed to use the
Fredrik Lundhb9056332001-07-11 17:42:21 +0000933 packet encoding. Unicode strings are automatically converted,
Fredrik Lundhb0e8e9b2001-09-10 21:45:42 +0000934 where necessary.
Fredrik Lundhb9056332001-07-11 17:42:21 +0000935 """
936
Guido van Rossum13257902007-06-07 23:15:56 +0000937 assert isinstance(params, (tuple, Fault)), "argument must be tuple or Fault instance"
Fredrik Lundhb9056332001-07-11 17:42:21 +0000938 if isinstance(params, Fault):
939 methodresponse = 1
Guido van Rossum13257902007-06-07 23:15:56 +0000940 elif methodresponse and isinstance(params, tuple):
Fredrik Lundhb9056332001-07-11 17:42:21 +0000941 assert len(params) == 1, "response tuple must be a singleton"
942
Fredrik Lundh3d9addd2002-06-27 21:36:21 +0000943 if not encoding:
Fredrik Lundhb9056332001-07-11 17:42:21 +0000944 encoding = "utf-8"
945
Fredrik Lundh3d9addd2002-06-27 21:36:21 +0000946 if FastMarshaller:
947 m = FastMarshaller(encoding)
948 else:
Andrew M. Kuchlinga4c2b742003-04-25 00:26:51 +0000949 m = Marshaller(encoding, allow_none)
Fredrik Lundh3d9addd2002-06-27 21:36:21 +0000950
Fredrik Lundhb9056332001-07-11 17:42:21 +0000951 data = m.dumps(params)
952
953 if encoding != "utf-8":
Fredrik Lundh3d9addd2002-06-27 21:36:21 +0000954 xmlheader = "<?xml version='1.0' encoding='%s'?>\n" % str(encoding)
Fredrik Lundhb9056332001-07-11 17:42:21 +0000955 else:
956 xmlheader = "<?xml version='1.0'?>\n" # utf-8 is default
957
958 # standard XML-RPC wrappings
Fredrik Lundh3d9addd2002-06-27 21:36:21 +0000959 if methodname:
Fredrik Lundhb9056332001-07-11 17:42:21 +0000960 # a method call
Guido van Rossum3172c5d2007-10-16 18:12:55 +0000961 if not isinstance(methodname, str):
Fredrik Lundhb9056332001-07-11 17:42:21 +0000962 methodname = methodname.encode(encoding)
963 data = (
964 xmlheader,
965 "<methodCall>\n"
966 "<methodName>", methodname, "</methodName>\n",
967 data,
968 "</methodCall>\n"
969 )
970 elif methodresponse:
971 # a method response, or a fault structure
972 data = (
973 xmlheader,
974 "<methodResponse>\n",
975 data,
976 "</methodResponse>\n"
977 )
978 else:
979 return data # return as is
Neal Norwitzff113342007-04-17 08:42:15 +0000980 return "".join(data)
Fredrik Lundhb9056332001-07-11 17:42:21 +0000981
Fredrik Lundh3d9addd2002-06-27 21:36:21 +0000982##
983# Convert an XML-RPC packet to a Python object. If the XML-RPC packet
984# represents a fault condition, this function raises a Fault exception.
985#
986# @param data An XML-RPC packet, given as an 8-bit string.
Walter Dörwaldf0dfc7a2003-10-20 14:01:56 +0000987# @return A tuple containing the unpacked data, and the method name
Fredrik Lundh3d9addd2002-06-27 21:36:21 +0000988# (None if not present).
989# @see Fault
990
Florent Xicluna61665192011-11-15 20:53:25 +0100991def loads(data, use_datetime=False, use_builtin_types=False):
Fredrik Lundhb9056332001-07-11 17:42:21 +0000992 """data -> unmarshalled data, method name
993
994 Convert an XML-RPC packet to unmarshalled data plus a method
995 name (None if not present).
996
997 If the XML-RPC packet represents a fault condition, this function
998 raises a Fault exception.
999 """
Florent Xicluna61665192011-11-15 20:53:25 +01001000 p, u = getparser(use_datetime=use_datetime, use_builtin_types=use_builtin_types)
Fredrik Lundhb9056332001-07-11 17:42:21 +00001001 p.feed(data)
1002 p.close()
1003 return u.close(), u.getmethodname()
1004
Kristján Valur Jónsson985fc6a2009-07-01 10:01:31 +00001005##
1006# Encode a string using the gzip content encoding such as specified by the
1007# Content-Encoding: gzip
1008# in the HTTP header, as described in RFC 1952
1009#
1010# @param data the unencoded data
1011# @return the encoded data
1012
1013def gzip_encode(data):
1014 """data -> gzip encoded data
1015
1016 Encode data using the gzip content encoding as described in RFC 1952
1017 """
Kristján Valur Jónssonaefde242009-07-19 22:29:24 +00001018 if not gzip:
1019 raise NotImplementedError
Kristján Valur Jónsson985fc6a2009-07-01 10:01:31 +00001020 f = BytesIO()
1021 gzf = gzip.GzipFile(mode="wb", fileobj=f, compresslevel=1)
1022 gzf.write(data)
1023 gzf.close()
1024 encoded = f.getvalue()
1025 f.close()
1026 return encoded
1027
1028##
1029# Decode a string using the gzip content encoding such as specified by the
1030# Content-Encoding: gzip
1031# in the HTTP header, as described in RFC 1952
1032#
1033# @param data The encoded data
1034# @return the unencoded data
1035# @raises ValueError if data is not correctly coded.
1036
1037def gzip_decode(data):
1038 """gzip encoded data -> unencoded data
1039
1040 Decode data using the gzip content encoding as described in RFC 1952
1041 """
Kristján Valur Jónssonaefde242009-07-19 22:29:24 +00001042 if not gzip:
1043 raise NotImplementedError
Kristján Valur Jónsson985fc6a2009-07-01 10:01:31 +00001044 f = BytesIO(data)
1045 gzf = gzip.GzipFile(mode="rb", fileobj=f)
1046 try:
1047 decoded = gzf.read()
Andrew Svetlovf7a17b42012-12-25 16:47:37 +02001048 except OSError:
Kristján Valur Jónsson985fc6a2009-07-01 10:01:31 +00001049 raise ValueError("invalid data")
1050 f.close()
1051 gzf.close()
1052 return decoded
1053
1054##
1055# Return a decoded file-like object for the gzip encoding
1056# as described in RFC 1952.
1057#
1058# @param response A stream supporting a read() method
1059# @return a file-like object that the decoded data can be read() from
1060
Kristján Valur Jónsson9ab07312009-07-19 22:38:38 +00001061class GzipDecodedResponse(gzip.GzipFile if gzip else object):
Kristján Valur Jónsson985fc6a2009-07-01 10:01:31 +00001062 """a file-like object to decode a response encoded with the gzip
1063 method, as described in RFC 1952.
1064 """
1065 def __init__(self, response):
1066 #response doesn't support tell() and read(), required by
1067 #GzipFile
Kristján Valur Jónssonaefde242009-07-19 22:29:24 +00001068 if not gzip:
1069 raise NotImplementedError
Kristján Valur Jónsson985fc6a2009-07-01 10:01:31 +00001070 self.io = BytesIO(response.read())
1071 gzip.GzipFile.__init__(self, mode="rb", fileobj=self.io)
1072
1073 def close(self):
1074 gzip.GzipFile.close(self)
1075 self.io.close()
1076
Fredrik Lundhb9056332001-07-11 17:42:21 +00001077
1078# --------------------------------------------------------------------
1079# request dispatcher
1080
1081class _Method:
1082 # some magic to bind an XML-RPC method to an RPC server.
1083 # supports "nested" methods (e.g. examples.getStateName)
1084 def __init__(self, send, name):
1085 self.__send = send
1086 self.__name = name
1087 def __getattr__(self, name):
1088 return _Method(self.__send, "%s.%s" % (self.__name, name))
1089 def __call__(self, *args):
1090 return self.__send(self.__name, args)
1091
Fredrik Lundh3d9addd2002-06-27 21:36:21 +00001092##
1093# Standard transport class for XML-RPC over HTTP.
1094# <p>
1095# You can create custom transports by subclassing this method, and
1096# overriding selected methods.
Fredrik Lundhb9056332001-07-11 17:42:21 +00001097
1098class Transport:
Fredrik Lundhc4c062f2001-09-10 19:45:02 +00001099 """Handles an HTTP transaction to an XML-RPC server."""
Fredrik Lundhb9056332001-07-11 17:42:21 +00001100
1101 # client identifier (may be overridden)
Florent Xicluna75861df2011-10-30 20:39:24 +01001102 user_agent = "Python-xmlrpc/%s" % __version__
Fredrik Lundhb9056332001-07-11 17:42:21 +00001103
Kristján Valur Jónsson985fc6a2009-07-01 10:01:31 +00001104 #if true, we'll request gzip encoding
1105 accept_gzip_encoding = True
1106
1107 # if positive, encode request using gzip if it exceeds this threshold
1108 # note that many server will get confused, so only use it if you know
1109 # that they can decode such a request
1110 encode_threshold = None #None = don't encode
1111
Florent Xicluna61665192011-11-15 20:53:25 +01001112 def __init__(self, use_datetime=False, use_builtin_types=False):
Skip Montanaro174dd222005-05-14 20:54:16 +00001113 self._use_datetime = use_datetime
Florent Xicluna61665192011-11-15 20:53:25 +01001114 self._use_builtin_types = use_builtin_types
Kristján Valur Jónsson985fc6a2009-07-01 10:01:31 +00001115 self._connection = (None, None)
1116 self._extra_headers = []
Skip Montanaro174dd222005-05-14 20:54:16 +00001117
Fredrik Lundh3d9addd2002-06-27 21:36:21 +00001118 ##
1119 # Send a complete request, and parse the response.
Kristján Valur Jónsson985fc6a2009-07-01 10:01:31 +00001120 # Retry request if a cached connection has disconnected.
Fredrik Lundh3d9addd2002-06-27 21:36:21 +00001121 #
1122 # @param host Target host.
1123 # @param handler Target PRC handler.
1124 # @param request_body XML-RPC request body.
1125 # @param verbose Debugging flag.
1126 # @return Parsed response.
1127
Georg Brandlfe991052009-09-16 15:54:04 +00001128 def request(self, host, handler, request_body, verbose=False):
Kristján Valur Jónsson985fc6a2009-07-01 10:01:31 +00001129 #retry request once if cached connection has gone cold
1130 for i in (0, 1):
1131 try:
1132 return self.single_request(host, handler, request_body, verbose)
Andrew Svetlov0832af62012-12-18 23:10:48 +02001133 except OSError as e:
1134 if i or e.errno not in (errno.ECONNRESET, errno.ECONNABORTED,
1135 errno.EPIPE):
Kristján Valur Jónsson43535d92009-07-03 23:23:50 +00001136 raise
1137 except http.client.BadStatusLine: #close after we sent request
1138 if i:
Kristján Valur Jónsson985fc6a2009-07-01 10:01:31 +00001139 raise
1140
Georg Brandlfe991052009-09-16 15:54:04 +00001141 def single_request(self, host, handler, request_body, verbose=False):
Fredrik Lundhb9056332001-07-11 17:42:21 +00001142 # issue XML-RPC request
Kristján Valur Jónsson985fc6a2009-07-01 10:01:31 +00001143 try:
1144 http_conn = self.send_request(host, handler, request_body, verbose)
1145 resp = http_conn.getresponse()
1146 if resp.status == 200:
1147 self.verbose = verbose
1148 return self.parse_response(resp)
Fredrik Lundhb9056332001-07-11 17:42:21 +00001149
Kristján Valur Jónsson985fc6a2009-07-01 10:01:31 +00001150 except Fault:
1151 raise
1152 except Exception:
1153 #All unexpected errors leave connection in
1154 # a strange state, so we clear it.
1155 self.close()
1156 raise
Fredrik Lundhb9056332001-07-11 17:42:21 +00001157
Kristján Valur Jónsson985fc6a2009-07-01 10:01:31 +00001158 #We got an error response.
1159 #Discard any response data and raise exception
1160 if resp.getheader("content-length", ""):
1161 resp.read()
1162 raise ProtocolError(
1163 host + handler,
1164 resp.status, resp.reason,
1165 dict(resp.getheaders())
1166 )
Fredrik Lundhb9056332001-07-11 17:42:21 +00001167
Fredrik Lundh3d9addd2002-06-27 21:36:21 +00001168
1169 ##
1170 # Create parser.
1171 #
1172 # @return A 2-tuple containing a parser and a unmarshaller.
Fredrik Lundhb9056332001-07-11 17:42:21 +00001173
Fredrik Lundhc4c062f2001-09-10 19:45:02 +00001174 def getparser(self):
1175 # get parser and unmarshaller
Florent Xicluna61665192011-11-15 20:53:25 +01001176 return getparser(use_datetime=self._use_datetime,
1177 use_builtin_types=self._use_builtin_types)
Fredrik Lundhc4c062f2001-09-10 19:45:02 +00001178
Fredrik Lundh3d9addd2002-06-27 21:36:21 +00001179 ##
Fredrik Lundh1303c7c2002-10-22 18:23:00 +00001180 # Get authorization info from host parameter
1181 # Host may be a string, or a (host, x509-dict) tuple; if a string,
1182 # it is checked for a "user:pw@host" format, and a "Basic
1183 # Authentication" header is added if appropriate.
1184 #
1185 # @param host Host descriptor (URL or (URL, x509 info) tuple).
1186 # @return A 3-tuple containing (actual host, extra headers,
1187 # x509 info). The header and x509 fields may be None.
1188
1189 def get_host_info(self, host):
1190
1191 x509 = {}
Guido van Rossum13257902007-06-07 23:15:56 +00001192 if isinstance(host, tuple):
Fredrik Lundh1303c7c2002-10-22 18:23:00 +00001193 host, x509 = host
1194
Jeremy Hylton1afc1692008-06-18 20:49:58 +00001195 auth, host = urllib.parse.splituser(host)
Fredrik Lundh1303c7c2002-10-22 18:23:00 +00001196
1197 if auth:
Georg Brandlc8dcfb62009-02-13 10:50:01 +00001198 auth = urllib.parse.unquote_to_bytes(auth)
Georg Brandlb54d8012009-06-04 09:11:51 +00001199 auth = base64.encodebytes(auth).decode("utf-8")
Neal Norwitzff113342007-04-17 08:42:15 +00001200 auth = "".join(auth.split()) # get rid of whitespace
Fredrik Lundh1303c7c2002-10-22 18:23:00 +00001201 extra_headers = [
1202 ("Authorization", "Basic " + auth)
1203 ]
1204 else:
Kristján Valur Jónsson985fc6a2009-07-01 10:01:31 +00001205 extra_headers = []
Fredrik Lundh1303c7c2002-10-22 18:23:00 +00001206
1207 return host, extra_headers, x509
1208
1209 ##
Fredrik Lundh3d9addd2002-06-27 21:36:21 +00001210 # Connect to server.
1211 #
1212 # @param host Target host.
Jeremy Hylton5d8a88a2007-08-14 16:47:39 +00001213 # @return An HTTPConnection object
Fredrik Lundh3d9addd2002-06-27 21:36:21 +00001214
Fredrik Lundhb9056332001-07-11 17:42:21 +00001215 def make_connection(self, host):
Kristján Valur Jónsson985fc6a2009-07-01 10:01:31 +00001216 #return an existing connection if possible. This allows
1217 #HTTP/1.1 keep-alive.
1218 if self._connection and host == self._connection[0]:
1219 return self._connection[1]
Fredrik Lundhb9056332001-07-11 17:42:21 +00001220 # create a HTTP connection object from a host descriptor
Kristján Valur Jónsson985fc6a2009-07-01 10:01:31 +00001221 chost, self._extra_headers, x509 = self.get_host_info(host)
1222 self._connection = host, http.client.HTTPConnection(chost)
1223 return self._connection[1]
Jeremy Hylton5d8a88a2007-08-14 16:47:39 +00001224
Kristján Valur Jónsson985fc6a2009-07-01 10:01:31 +00001225 ##
1226 # Clear any cached connection object.
1227 # Used in the event of socket errors.
1228 #
1229 def close(self):
1230 if self._connection[1]:
1231 self._connection[1].close()
1232 self._connection = (None, None)
Fredrik Lundhb9056332001-07-11 17:42:21 +00001233
Fredrik Lundh3d9addd2002-06-27 21:36:21 +00001234 ##
Jeremy Hylton5d8a88a2007-08-14 16:47:39 +00001235 # Send HTTP request.
Fredrik Lundh3d9addd2002-06-27 21:36:21 +00001236 #
Jeremy Hylton5d8a88a2007-08-14 16:47:39 +00001237 # @param host Host descriptor (URL or (URL, x509 info) tuple).
1238 # @param handler Targer RPC handler (a path relative to host)
1239 # @param request_body The XML-RPC request body
1240 # @param debug Enable debugging if debug is true.
1241 # @return An HTTPConnection.
Fredrik Lundh3d9addd2002-06-27 21:36:21 +00001242
Jeremy Hylton5d8a88a2007-08-14 16:47:39 +00001243 def send_request(self, host, handler, request_body, debug):
Kristján Valur Jónsson985fc6a2009-07-01 10:01:31 +00001244 connection = self.make_connection(host)
1245 headers = self._extra_headers[:]
Jeremy Hylton5d8a88a2007-08-14 16:47:39 +00001246 if debug:
1247 connection.set_debuglevel(1)
Kristján Valur Jónssonaefde242009-07-19 22:29:24 +00001248 if self.accept_gzip_encoding and gzip:
Kristján Valur Jónsson985fc6a2009-07-01 10:01:31 +00001249 connection.putrequest("POST", handler, skip_accept_encoding=True)
1250 headers.append(("Accept-Encoding", "gzip"))
1251 else:
1252 connection.putrequest("POST", handler)
1253 headers.append(("Content-Type", "text/xml"))
1254 headers.append(("User-Agent", self.user_agent))
1255 self.send_headers(connection, headers)
1256 self.send_content(connection, request_body)
Jeremy Hylton5d8a88a2007-08-14 16:47:39 +00001257 return connection
Fredrik Lundhb9056332001-07-11 17:42:21 +00001258
Fredrik Lundh3d9addd2002-06-27 21:36:21 +00001259 ##
Kristján Valur Jónsson985fc6a2009-07-01 10:01:31 +00001260 # Send request headers.
1261 # This function provides a useful hook for subclassing
1262 #
1263 # @param connection httpConnection.
1264 # @param headers list of key,value pairs for HTTP headers
1265
1266 def send_headers(self, connection, headers):
1267 for key, val in headers:
1268 connection.putheader(key, val)
1269
1270 ##
1271 # Send request body.
1272 # This function provides a useful hook for subclassing
1273 #
1274 # @param connection httpConnection.
1275 # @param request_body XML-RPC request body.
1276
1277 def send_content(self, connection, request_body):
1278 #optionally encode the request
1279 if (self.encode_threshold is not None and
Kristján Valur Jónssonaefde242009-07-19 22:29:24 +00001280 self.encode_threshold < len(request_body) and
1281 gzip):
Kristján Valur Jónsson985fc6a2009-07-01 10:01:31 +00001282 connection.putheader("Content-Encoding", "gzip")
1283 request_body = gzip_encode(request_body)
1284
1285 connection.putheader("Content-Length", str(len(request_body)))
1286 connection.endheaders(request_body)
1287
1288 ##
Fredrik Lundh3d9addd2002-06-27 21:36:21 +00001289 # Parse response.
1290 #
1291 # @param file Stream.
1292 # @return Response tuple and target method.
1293
Kristján Valur Jónsson985fc6a2009-07-01 10:01:31 +00001294 def parse_response(self, response):
1295 # read response data from httpresponse, and parse it
Senthil Kumaranf34445f2010-12-08 08:04:49 +00001296 # Check for new http response object, otherwise it is a file object.
1297 if hasattr(response, 'getheader'):
1298 if response.getheader("Content-Encoding", "") == "gzip":
1299 stream = GzipDecodedResponse(response)
1300 else:
1301 stream = response
Kristján Valur Jónsson985fc6a2009-07-01 10:01:31 +00001302 else:
1303 stream = response
Fredrik Lundhb9056332001-07-11 17:42:21 +00001304
Fredrik Lundhc4c062f2001-09-10 19:45:02 +00001305 p, u = self.getparser()
Fredrik Lundhb9056332001-07-11 17:42:21 +00001306
1307 while 1:
Kristján Valur Jónsson985fc6a2009-07-01 10:01:31 +00001308 data = stream.read(1024)
1309 if not data:
Fredrik Lundhb9056332001-07-11 17:42:21 +00001310 break
1311 if self.verbose:
Kristján Valur Jónsson985fc6a2009-07-01 10:01:31 +00001312 print("body:", repr(data))
1313 p.feed(data)
Fredrik Lundhb9056332001-07-11 17:42:21 +00001314
Kristján Valur Jónsson985fc6a2009-07-01 10:01:31 +00001315 if stream is not response:
1316 stream.close()
Fredrik Lundhb9056332001-07-11 17:42:21 +00001317 p.close()
1318
1319 return u.close()
1320
Fredrik Lundh3d9addd2002-06-27 21:36:21 +00001321##
1322# Standard transport class for XML-RPC over HTTPS.
1323
Fredrik Lundhb9056332001-07-11 17:42:21 +00001324class SafeTransport(Transport):
Fredrik Lundhc4c062f2001-09-10 19:45:02 +00001325 """Handles an HTTPS transaction to an XML-RPC server."""
Fredrik Lundhb9056332001-07-11 17:42:21 +00001326
Fredrik Lundh3d9addd2002-06-27 21:36:21 +00001327 # FIXME: mostly untested
1328
Kristján Valur Jónsson985fc6a2009-07-01 10:01:31 +00001329 def make_connection(self, host):
1330 if self._connection and host == self._connection[0]:
1331 return self._connection[1]
1332
Senthil Kumaran6a0b5c42010-11-18 17:08:48 +00001333 if not hasattr(http.client, "HTTPSConnection"):
Fredrik Lundh1303c7c2002-10-22 18:23:00 +00001334 raise NotImplementedError(
Kristján Valur Jónsson985fc6a2009-07-01 10:01:31 +00001335 "your version of http.client doesn't support HTTPS")
1336 # create a HTTPS connection object from a host descriptor
1337 # host may be a string, or a (host, x509-dict) tuple
1338 chost, self._extra_headers, x509 = self.get_host_info(host)
1339 self._connection = host, http.client.HTTPSConnection(chost,
1340 None, **(x509 or {}))
1341 return self._connection[1]
Fredrik Lundhb9056332001-07-11 17:42:21 +00001342
Fredrik Lundh3d9addd2002-06-27 21:36:21 +00001343##
1344# Standard server proxy. This class establishes a virtual connection
1345# to an XML-RPC server.
1346# <p>
1347# This class is available as ServerProxy and Server. New code should
1348# use ServerProxy, to avoid confusion.
1349#
1350# @def ServerProxy(uri, **options)
1351# @param uri The connection point on the server.
1352# @keyparam transport A transport factory, compatible with the
1353# standard transport class.
1354# @keyparam encoding The default encoding used for 8-bit strings
1355# (default is UTF-8).
1356# @keyparam verbose Use a true value to enable debugging output.
1357# (printed to standard output).
1358# @see Transport
1359
Fredrik Lundhb9056332001-07-11 17:42:21 +00001360class ServerProxy:
1361 """uri [,options] -> a logical connection to an XML-RPC server
1362
1363 uri is the connection point on the server, given as
1364 scheme://host/target.
1365
1366 The standard implementation always supports the "http" scheme. If
1367 SSL socket support is available (Python 2.0), it also supports
1368 "https".
1369
1370 If the target part and the slash preceding it are both omitted,
1371 "/RPC2" is assumed.
1372
1373 The following options can be given as keyword arguments:
1374
1375 transport: a transport factory
1376 encoding: the request encoding (default is UTF-8)
1377
1378 All 8-bit strings passed to the server proxy are assumed to use
1379 the given encoding.
1380 """
1381
Georg Brandlfe991052009-09-16 15:54:04 +00001382 def __init__(self, uri, transport=None, encoding=None, verbose=False,
Florent Xicluna61665192011-11-15 20:53:25 +01001383 allow_none=False, use_datetime=False, use_builtin_types=False):
Fredrik Lundhb9056332001-07-11 17:42:21 +00001384 # establish a "logical" server connection
1385
1386 # get the url
Jeremy Hylton1afc1692008-06-18 20:49:58 +00001387 type, uri = urllib.parse.splittype(uri)
Fredrik Lundhb9056332001-07-11 17:42:21 +00001388 if type not in ("http", "https"):
Andrew Svetlovf7a17b42012-12-25 16:47:37 +02001389 raise OSError("unsupported XML-RPC protocol")
Jeremy Hylton1afc1692008-06-18 20:49:58 +00001390 self.__host, self.__handler = urllib.parse.splithost(uri)
Fredrik Lundhb9056332001-07-11 17:42:21 +00001391 if not self.__handler:
1392 self.__handler = "/RPC2"
1393
1394 if transport is None:
1395 if type == "https":
Florent Xicluna61665192011-11-15 20:53:25 +01001396 handler = SafeTransport
Fredrik Lundhb9056332001-07-11 17:42:21 +00001397 else:
Florent Xicluna61665192011-11-15 20:53:25 +01001398 handler = Transport
1399 transport = handler(use_datetime=use_datetime,
1400 use_builtin_types=use_builtin_types)
Fredrik Lundhb9056332001-07-11 17:42:21 +00001401 self.__transport = transport
1402
Senthil Kumaranb3af08f2009-04-01 20:20:43 +00001403 self.__encoding = encoding or 'utf-8'
Fredrik Lundhb9056332001-07-11 17:42:21 +00001404 self.__verbose = verbose
Andrew M. Kuchlinga4c2b742003-04-25 00:26:51 +00001405 self.__allow_none = allow_none
Tim Petersc2659cf2003-05-12 20:19:37 +00001406
Kristján Valur Jónsson985fc6a2009-07-01 10:01:31 +00001407 def __close(self):
1408 self.__transport.close()
1409
Fredrik Lundhb9056332001-07-11 17:42:21 +00001410 def __request(self, methodname, params):
1411 # call a method on the remote server
1412
Andrew M. Kuchlinga4c2b742003-04-25 00:26:51 +00001413 request = dumps(params, methodname, encoding=self.__encoding,
Senthil Kumaranb3af08f2009-04-01 20:20:43 +00001414 allow_none=self.__allow_none).encode(self.__encoding)
Fredrik Lundhb9056332001-07-11 17:42:21 +00001415
1416 response = self.__transport.request(
1417 self.__host,
1418 self.__handler,
1419 request,
1420 verbose=self.__verbose
1421 )
1422
1423 if len(response) == 1:
1424 response = response[0]
1425
1426 return response
1427
1428 def __repr__(self):
1429 return (
Fredrik Lundh78eedce2001-08-23 20:04:33 +00001430 "<ServerProxy for %s%s>" %
Fredrik Lundhb9056332001-07-11 17:42:21 +00001431 (self.__host, self.__handler)
1432 )
1433
1434 __str__ = __repr__
Raymond Hettingercc523fc2003-11-02 09:47:05 +00001435
Fredrik Lundhb9056332001-07-11 17:42:21 +00001436 def __getattr__(self, name):
1437 # magic method dispatcher
1438 return _Method(self.__request, name)
1439
1440 # note: to call a remote object with an non-standard name, use
1441 # result getattr(server, "strange-python-name")(args)
1442
Kristján Valur Jónsson985fc6a2009-07-01 10:01:31 +00001443 def __call__(self, attr):
1444 """A workaround to get special attributes on the ServerProxy
1445 without interfering with the magic __getattr__
1446 """
1447 if attr == "close":
1448 return self.__close
1449 elif attr == "transport":
1450 return self.__transport
1451 raise AttributeError("Attribute %r not found" % (attr,))
1452
Fredrik Lundh78eedce2001-08-23 20:04:33 +00001453# compatibility
Fredrik Lundh3d9addd2002-06-27 21:36:21 +00001454
Fredrik Lundhb9056332001-07-11 17:42:21 +00001455Server = ServerProxy
1456
1457# --------------------------------------------------------------------
1458# test code
1459
1460if __name__ == "__main__":
1461
1462 # simple test program (from the XML-RPC specification)
1463
Fredrik Lundh78eedce2001-08-23 20:04:33 +00001464 # server = ServerProxy("http://localhost:8000") # local server
Martin v. Löwis12237b32004-08-22 16:04:50 +00001465 server = ServerProxy("http://time.xmlrpc.com/RPC2")
Fredrik Lundhb9056332001-07-11 17:42:21 +00001466
Fredrik Lundhb9056332001-07-11 17:42:21 +00001467 try:
Guido van Rossumbe19ed72007-02-09 05:37:30 +00001468 print(server.currentTime.getCurrentTime())
Guido van Rossumb940e112007-01-10 16:19:56 +00001469 except Error as v:
Guido van Rossumbe19ed72007-02-09 05:37:30 +00001470 print("ERROR", v)
Martin v. Löwis12237b32004-08-22 16:04:50 +00001471
Jeremy Hylton5d8a88a2007-08-14 16:47:39 +00001472 # The server at xmlrpc.com doesn't seem to support multicall anymore.
Martin v. Löwis12237b32004-08-22 16:04:50 +00001473 multi = MultiCall(server)
1474 multi.currentTime.getCurrentTime()
1475 multi.currentTime.getCurrentTime()
1476 try:
1477 for response in multi():
Guido van Rossumbe19ed72007-02-09 05:37:30 +00001478 print(response)
Guido van Rossumb940e112007-01-10 16:19:56 +00001479 except Error as v:
Guido van Rossumbe19ed72007-02-09 05:37:30 +00001480 print("ERROR", v)