blob: bd59f326fbe2c22d2d92c34f848f7f1c2f22eeeb [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
131import time
Georg Brandl24420152008-05-26 16:32:26 +0000132import http.client
Georg Brandlcef803f2009-06-04 09:04:53 +0000133from xml.parsers import expat
Kristján Valur Jónsson985fc6a2009-07-01 10:01:31 +0000134import socket
135import errno
136from io import BytesIO
Kristján Valur Jónssonaefde242009-07-19 22:29:24 +0000137try:
138 import gzip
139except ImportError:
140 gzip = None #python can be built without zlib/gzip support
Fredrik Lundh1538c232001-10-01 19:42:03 +0000141
Fredrik Lundh3d9addd2002-06-27 21:36:21 +0000142# --------------------------------------------------------------------
143# Internal stuff
144
Fredrik Lundhb9056332001-07-11 17:42:21 +0000145try:
Fred Drakeba613c32005-02-10 18:33:30 +0000146 import datetime
147except ImportError:
148 datetime = None
149
Neal Norwitzff113342007-04-17 08:42:15 +0000150def escape(s):
151 s = s.replace("&", "&")
152 s = s.replace("<", "&lt;")
153 return s.replace(">", "&gt;",)
Fredrik Lundh1538c232001-10-01 19:42:03 +0000154
Fredrik Lundh3d9addd2002-06-27 21:36:21 +0000155__version__ = "1.0.1"
156
157# xmlrpc integer limits
Guido van Rossume2a383d2007-01-15 16:59:06 +0000158MAXINT = 2**31-1
159MININT = -2**31
Fredrik Lundh3d9addd2002-06-27 21:36:21 +0000160
161# --------------------------------------------------------------------
162# Error constants (from Dan Libby's specification at
163# http://xmlrpc-epi.sourceforge.net/specs/rfc.fault_codes.php)
164
165# Ranges of errors
166PARSE_ERROR = -32700
167SERVER_ERROR = -32600
168APPLICATION_ERROR = -32500
169SYSTEM_ERROR = -32400
170TRANSPORT_ERROR = -32300
171
172# Specific errors
173NOT_WELLFORMED_ERROR = -32700
174UNSUPPORTED_ENCODING = -32701
175INVALID_ENCODING_CHAR = -32702
176INVALID_XMLRPC = -32600
177METHOD_NOT_FOUND = -32601
178INVALID_METHOD_PARAMS = -32602
179INTERNAL_ERROR = -32603
Fredrik Lundhb9056332001-07-11 17:42:21 +0000180
181# --------------------------------------------------------------------
182# Exceptions
183
Fredrik Lundh3d9addd2002-06-27 21:36:21 +0000184##
185# Base class for all kinds of client-side errors.
186
Fredrik Lundh78eedce2001-08-23 20:04:33 +0000187class Error(Exception):
Fred Drake1b410792001-09-04 18:55:03 +0000188 """Base class for client errors."""
Fredrik Lundhb0e8e9b2001-09-10 21:45:42 +0000189 def __str__(self):
190 return repr(self)
Fredrik Lundhb9056332001-07-11 17:42:21 +0000191
Fredrik Lundh3d9addd2002-06-27 21:36:21 +0000192##
193# Indicates an HTTP-level protocol error. This is raised by the HTTP
194# transport layer, if the server returns an error code other than 200
195# (OK).
196#
197# @param url The target URL.
198# @param errcode The HTTP error code.
199# @param errmsg The HTTP error message.
200# @param headers The HTTP header dictionary.
201
Fredrik Lundhb9056332001-07-11 17:42:21 +0000202class ProtocolError(Error):
Fred Drake1b410792001-09-04 18:55:03 +0000203 """Indicates an HTTP protocol error."""
Fredrik Lundhb9056332001-07-11 17:42:21 +0000204 def __init__(self, url, errcode, errmsg, headers):
Fredrik Lundhb0e8e9b2001-09-10 21:45:42 +0000205 Error.__init__(self)
Fredrik Lundhb9056332001-07-11 17:42:21 +0000206 self.url = url
207 self.errcode = errcode
208 self.errmsg = errmsg
209 self.headers = headers
210 def __repr__(self):
211 return (
212 "<ProtocolError for %s: %s %s>" %
213 (self.url, self.errcode, self.errmsg)
214 )
215
Fredrik Lundh3d9addd2002-06-27 21:36:21 +0000216##
217# Indicates a broken XML-RPC response package. This exception is
218# raised by the unmarshalling layer, if the XML-RPC response is
219# malformed.
220
Fredrik Lundhb9056332001-07-11 17:42:21 +0000221class ResponseError(Error):
Fredrik Lundhc4c062f2001-09-10 19:45:02 +0000222 """Indicates a broken response package."""
Fredrik Lundhb9056332001-07-11 17:42:21 +0000223 pass
224
Fredrik Lundh3d9addd2002-06-27 21:36:21 +0000225##
226# Indicates an XML-RPC fault response package. This exception is
227# raised by the unmarshalling layer, if the XML-RPC response contains
Florent Xiclunac4fec932011-10-30 20:19:32 +0100228# a fault string. This exception can also be used as a class, to
Fredrik Lundh3d9addd2002-06-27 21:36:21 +0000229# generate a fault XML-RPC message.
230#
231# @param faultCode The XML-RPC fault code.
232# @param faultString The XML-RPC fault string.
233
Fredrik Lundhb9056332001-07-11 17:42:21 +0000234class Fault(Error):
Fredrik Lundhc4c062f2001-09-10 19:45:02 +0000235 """Indicates an XML-RPC fault package."""
Fredrik Lundhb9056332001-07-11 17:42:21 +0000236 def __init__(self, faultCode, faultString, **extra):
Fredrik Lundhb0e8e9b2001-09-10 21:45:42 +0000237 Error.__init__(self)
Fredrik Lundhb9056332001-07-11 17:42:21 +0000238 self.faultCode = faultCode
239 self.faultString = faultString
240 def __repr__(self):
Florent Xiclunac4fec932011-10-30 20:19:32 +0100241 return "<Fault %s: %r>" % (self.faultCode, self.faultString)
Fredrik Lundhb9056332001-07-11 17:42:21 +0000242
243# --------------------------------------------------------------------
244# Special values
245
Fredrik Lundh3d9addd2002-06-27 21:36:21 +0000246##
Georg Brandl38eceaa2008-05-26 11:14:17 +0000247# Backwards compatibility
Fredrik Lundh3d9addd2002-06-27 21:36:21 +0000248
Guido van Rossum47b9ff62006-08-24 00:41:19 +0000249boolean = Boolean = bool
Fredrik Lundh3d9addd2002-06-27 21:36:21 +0000250
251##
252# Wrapper for XML-RPC DateTime values. This converts a time value to
253# the format used by XML-RPC.
254# <p>
255# The value can be given as a string in the format
256# "yyyymmddThh:mm:ss", as a 9-item time tuple (as returned by
257# time.localtime()), or an integer value (as returned by time.time()).
258# The wrapper uses time.localtime() to convert an integer to a time
259# tuple.
260#
261# @param value The time, given as an ISO 8601 string, a time
262# tuple, or a integer time value.
Fredrik Lundhb9056332001-07-11 17:42:21 +0000263
Christian Heimesdae2a892008-04-19 00:55:37 +0000264def _strftime(value):
265 if datetime:
266 if isinstance(value, datetime.datetime):
267 return "%04d%02d%02dT%02d:%02d:%02d" % (
268 value.year, value.month, value.day,
269 value.hour, value.minute, value.second)
270
271 if not isinstance(value, (tuple, time.struct_time)):
272 if value == 0:
273 value = time.time()
274 value = time.localtime(value)
275
276 return "%04d%02d%02dT%02d:%02d:%02d" % value[:6]
277
Fredrik Lundhb9056332001-07-11 17:42:21 +0000278class DateTime:
Fredrik Lundhc4c062f2001-09-10 19:45:02 +0000279 """DateTime wrapper for an ISO 8601 string or time tuple or
280 localtime integer value to generate 'dateTime.iso8601' XML-RPC
281 value.
282 """
Fredrik Lundhb9056332001-07-11 17:42:21 +0000283
284 def __init__(self, value=0):
Christian Heimesdae2a892008-04-19 00:55:37 +0000285 if isinstance(value, str):
286 self.value = value
287 else:
288 self.value = _strftime(value)
Fredrik Lundhb9056332001-07-11 17:42:21 +0000289
Christian Heimes05e8be12008-02-23 18:30:17 +0000290 def make_comparable(self, other):
Fredrik Lundhb9056332001-07-11 17:42:21 +0000291 if isinstance(other, DateTime):
Christian Heimes05e8be12008-02-23 18:30:17 +0000292 s = self.value
293 o = other.value
294 elif datetime and isinstance(other, datetime.datetime):
295 s = self.value
296 o = other.strftime("%Y%m%dT%H:%M:%S")
Florent Xicluna3fa29f72011-10-30 20:18:50 +0100297 elif isinstance(other, str):
Christian Heimes05e8be12008-02-23 18:30:17 +0000298 s = self.value
299 o = other
300 elif hasattr(other, "timetuple"):
301 s = self.timetuple()
302 o = other.timetuple()
303 else:
304 otype = (hasattr(other, "__class__")
305 and other.__class__.__name__
306 or type(other))
307 raise TypeError("Can't compare %s and %s" %
308 (self.__class__.__name__, otype))
309 return s, o
310
311 def __lt__(self, other):
312 s, o = self.make_comparable(other)
313 return s < o
314
315 def __le__(self, other):
316 s, o = self.make_comparable(other)
317 return s <= o
318
319 def __gt__(self, other):
320 s, o = self.make_comparable(other)
321 return s > o
322
323 def __ge__(self, other):
324 s, o = self.make_comparable(other)
325 return s >= o
326
327 def __eq__(self, other):
328 s, o = self.make_comparable(other)
329 return s == o
Guido van Rossum47b9ff62006-08-24 00:41:19 +0000330
331 def __ne__(self, other):
Christian Heimes05e8be12008-02-23 18:30:17 +0000332 s, o = self.make_comparable(other)
333 return s != o
334
335 def timetuple(self):
336 return time.strptime(self.value, "%Y%m%dT%H:%M:%S")
337
Fredrik Lundh3d9addd2002-06-27 21:36:21 +0000338 ##
339 # Get date/time value.
340 #
341 # @return Date/time value, as an ISO 8601 string.
342
343 def __str__(self):
344 return self.value
345
Fredrik Lundhb9056332001-07-11 17:42:21 +0000346 def __repr__(self):
Florent Xiclunac4fec932011-10-30 20:19:32 +0100347 return "<DateTime %r at %x>" % (self.value, id(self))
Fredrik Lundhb9056332001-07-11 17:42:21 +0000348
349 def decode(self, data):
Neal Norwitzff113342007-04-17 08:42:15 +0000350 self.value = str(data).strip()
Fredrik Lundhb9056332001-07-11 17:42:21 +0000351
352 def encode(self, out):
353 out.write("<value><dateTime.iso8601>")
354 out.write(self.value)
355 out.write("</dateTime.iso8601></value>\n")
356
Fredrik Lundh3d9addd2002-06-27 21:36:21 +0000357def _datetime(data):
358 # decode xml element contents into a DateTime structure.
Fredrik Lundhb9056332001-07-11 17:42:21 +0000359 value = DateTime()
360 value.decode(data)
361 return value
362
Skip Montanaro174dd222005-05-14 20:54:16 +0000363def _datetime_type(data):
364 t = time.strptime(data, "%Y%m%dT%H:%M:%S")
365 return datetime.datetime(*tuple(t)[:6])
366
Fredrik Lundh3d9addd2002-06-27 21:36:21 +0000367##
368# Wrapper for binary data. This can be used to transport any kind
369# of binary data over XML-RPC, using BASE64 encoding.
370#
371# @param data An 8-bit string containing arbitrary data.
372
Fredrik Lundhb9056332001-07-11 17:42:21 +0000373class Binary:
Fred Drake1b410792001-09-04 18:55:03 +0000374 """Wrapper for binary data."""
Fredrik Lundhb9056332001-07-11 17:42:21 +0000375
376 def __init__(self, data=None):
Guido van Rossum54a40cb2007-08-27 22:27:41 +0000377 if data is None:
378 data = b""
379 else:
380 if not isinstance(data, bytes):
381 raise TypeError("expected bytes, not %s" %
382 data.__class__.__name__)
383 data = bytes(data) # Make a copy of the bytes!
Fredrik Lundhb9056332001-07-11 17:42:21 +0000384 self.data = data
385
Fredrik Lundh3d9addd2002-06-27 21:36:21 +0000386 ##
387 # Get buffer contents.
388 #
389 # @return Buffer contents, as an 8-bit string.
390
391 def __str__(self):
Guido van Rossum54a40cb2007-08-27 22:27:41 +0000392 return str(self.data, "latin-1") # XXX encoding?!
Fredrik Lundh3d9addd2002-06-27 21:36:21 +0000393
Guido van Rossum47b9ff62006-08-24 00:41:19 +0000394 def __eq__(self, other):
Fredrik Lundhb9056332001-07-11 17:42:21 +0000395 if isinstance(other, Binary):
396 other = other.data
Guido van Rossum47b9ff62006-08-24 00:41:19 +0000397 return self.data == other
398
399 def __ne__(self, other):
400 if isinstance(other, Binary):
401 other = other.data
402 return self.data != other
Fredrik Lundhb9056332001-07-11 17:42:21 +0000403
404 def decode(self, data):
Georg Brandlb54d8012009-06-04 09:11:51 +0000405 self.data = base64.decodebytes(data)
Fredrik Lundhb9056332001-07-11 17:42:21 +0000406
407 def encode(self, out):
Fredrik Lundhb9056332001-07-11 17:42:21 +0000408 out.write("<value><base64>\n")
Georg Brandlb54d8012009-06-04 09:11:51 +0000409 encoded = base64.encodebytes(self.data)
Brett Cannon2dbde5e2007-07-30 03:50:35 +0000410 out.write(encoded.decode('ascii'))
411 out.write('\n')
Fredrik Lundhb9056332001-07-11 17:42:21 +0000412 out.write("</base64></value>\n")
413
Fredrik Lundh3d9addd2002-06-27 21:36:21 +0000414def _binary(data):
415 # decode xml element contents into a Binary structure
Fredrik Lundhb9056332001-07-11 17:42:21 +0000416 value = Binary()
417 value.decode(data)
418 return value
419
Skip Montanaro9a7c96a2003-01-22 18:17:25 +0000420WRAPPERS = (DateTime, Binary)
Fredrik Lundhb9056332001-07-11 17:42:21 +0000421
422# --------------------------------------------------------------------
423# XML parsers
424
Georg Brandlcef803f2009-06-04 09:04:53 +0000425class ExpatParser:
426 # fast expat parser for Python 2.0 and later.
427 def __init__(self, target):
428 self._parser = parser = expat.ParserCreate(None, None)
429 self._target = target
430 parser.StartElementHandler = target.start
431 parser.EndElementHandler = target.end
432 parser.CharacterDataHandler = target.data
433 encoding = None
434 target.xml(encoding, None)
Fredrik Lundhb9056332001-07-11 17:42:21 +0000435
Georg Brandlcef803f2009-06-04 09:04:53 +0000436 def feed(self, data):
437 self._parser.Parse(data, 0)
Fredrik Lundhb9056332001-07-11 17:42:21 +0000438
Georg Brandlcef803f2009-06-04 09:04:53 +0000439 def close(self):
440 self._parser.Parse("", 1) # end of data
441 del self._target, self._parser # get rid of circular references
Fredrik Lundhb9056332001-07-11 17:42:21 +0000442
Fredrik Lundhb9056332001-07-11 17:42:21 +0000443# --------------------------------------------------------------------
444# XML-RPC marshalling and unmarshalling code
445
Fredrik Lundh3d9addd2002-06-27 21:36:21 +0000446##
447# XML-RPC marshaller.
448#
449# @param encoding Default encoding for 8-bit strings. The default
450# value is None (interpreted as UTF-8).
451# @see dumps
452
Fredrik Lundhb9056332001-07-11 17:42:21 +0000453class Marshaller:
Fred Drake1b410792001-09-04 18:55:03 +0000454 """Generate an XML-RPC params chunk from a Python data structure.
Fredrik Lundhb9056332001-07-11 17:42:21 +0000455
Fredrik Lundhc4c062f2001-09-10 19:45:02 +0000456 Create a Marshaller instance for each set of parameters, and use
457 the "dumps" method to convert your data (represented as a tuple)
458 to an XML-RPC params chunk. To write a fault response, pass a
459 Fault instance instead. You may prefer to use the "dumps" module
460 function for this purpose.
Fred Drake1b410792001-09-04 18:55:03 +0000461 """
Fredrik Lundhb9056332001-07-11 17:42:21 +0000462
463 # by the way, if you don't understand what's going on in here,
464 # that's perfectly ok.
465
Georg Brandlfe991052009-09-16 15:54:04 +0000466 def __init__(self, encoding=None, allow_none=False):
Fredrik Lundhb9056332001-07-11 17:42:21 +0000467 self.memo = {}
468 self.data = None
469 self.encoding = encoding
Andrew M. Kuchlinga4c2b742003-04-25 00:26:51 +0000470 self.allow_none = allow_none
Tim Petersc2659cf2003-05-12 20:19:37 +0000471
Fredrik Lundhb9056332001-07-11 17:42:21 +0000472 dispatch = {}
473
474 def dumps(self, values):
Fredrik Lundh3d9addd2002-06-27 21:36:21 +0000475 out = []
476 write = out.append
477 dump = self.__dump
Fredrik Lundhb9056332001-07-11 17:42:21 +0000478 if isinstance(values, Fault):
479 # fault instance
480 write("<fault>\n")
Martin v. Löwis541342f2003-07-12 07:53:04 +0000481 dump({'faultCode': values.faultCode,
482 'faultString': values.faultString},
483 write)
Fredrik Lundhb9056332001-07-11 17:42:21 +0000484 write("</fault>\n")
485 else:
486 # parameter block
Fredrik Lundhc266bb02001-08-23 20:13:08 +0000487 # FIXME: the xml-rpc specification allows us to leave out
488 # the entire <params> block if there are no parameters.
489 # however, changing this may break older code (including
490 # old versions of xmlrpclib.py), so this is better left as
491 # is for now. See @XMLRPC3 for more information. /F
Fredrik Lundhb9056332001-07-11 17:42:21 +0000492 write("<params>\n")
493 for v in values:
494 write("<param>\n")
Fredrik Lundh3d9addd2002-06-27 21:36:21 +0000495 dump(v, write)
Fredrik Lundhb9056332001-07-11 17:42:21 +0000496 write("</param>\n")
497 write("</params>\n")
Neal Norwitzff113342007-04-17 08:42:15 +0000498 result = "".join(out)
Fredrik Lundhb9056332001-07-11 17:42:21 +0000499 return result
500
Fredrik Lundh3d9addd2002-06-27 21:36:21 +0000501 def __dump(self, value, write):
Fredrik Lundhb9056332001-07-11 17:42:21 +0000502 try:
503 f = self.dispatch[type(value)]
504 except KeyError:
Thomas Wouters89f507f2006-12-13 04:49:30 +0000505 # check if this object can be marshalled as a structure
506 try:
507 value.__dict__
508 except:
Collin Winterce36ad82007-08-30 01:19:48 +0000509 raise TypeError("cannot marshal %s objects" % type(value))
Thomas Wouters89f507f2006-12-13 04:49:30 +0000510 # check if this class is a sub-class of a basic type,
511 # because we don't know how to marshal these types
512 # (e.g. a string sub-class)
513 for type_ in type(value).__mro__:
514 if type_ in self.dispatch.keys():
Collin Winterce36ad82007-08-30 01:19:48 +0000515 raise TypeError("cannot marshal %s objects" % type(value))
Thomas Wouters89f507f2006-12-13 04:49:30 +0000516 # XXX(twouters): using "_arbitrary_instance" as key as a quick-fix
517 # for the p3yk merge, this should probably be fixed more neatly.
518 f = self.dispatch["_arbitrary_instance"]
519 f(self, value, write)
Fredrik Lundhb9056332001-07-11 17:42:21 +0000520
Andrew M. Kuchlinga4c2b742003-04-25 00:26:51 +0000521 def dump_nil (self, value, write):
522 if not self.allow_none:
Collin Winterce36ad82007-08-30 01:19:48 +0000523 raise TypeError("cannot marshal None unless allow_none is enabled")
Andrew M. Kuchlinga4c2b742003-04-25 00:26:51 +0000524 write("<value><nil/></value>")
Guido van Rossum13257902007-06-07 23:15:56 +0000525 dispatch[type(None)] = dump_nil
Tim Petersc2659cf2003-05-12 20:19:37 +0000526
Fredrik Lundh3d9addd2002-06-27 21:36:21 +0000527 def dump_int(self, value, write):
Skip Montanaro5449e082001-10-17 22:53:33 +0000528 # in case ints are > 32 bits
529 if value > MAXINT or value < MININT:
Collin Winterce36ad82007-08-30 01:19:48 +0000530 raise OverflowError("int exceeds XML-RPC limits")
Fredrik Lundh3d9addd2002-06-27 21:36:21 +0000531 write("<value><int>")
532 write(str(value))
533 write("</int></value>\n")
Guido van Rossum13257902007-06-07 23:15:56 +0000534 #dispatch[int] = dump_int
Fredrik Lundhb9056332001-07-11 17:42:21 +0000535
Georg Brandl38eceaa2008-05-26 11:14:17 +0000536 def dump_bool(self, value, write):
537 write("<value><boolean>")
538 write(value and "1" or "0")
539 write("</boolean></value>\n")
540 dispatch[bool] = dump_bool
Skip Montanaro9a7c96a2003-01-22 18:17:25 +0000541
Fredrik Lundh3d9addd2002-06-27 21:36:21 +0000542 def dump_long(self, value, write):
Skip Montanaro5449e082001-10-17 22:53:33 +0000543 if value > MAXINT or value < MININT:
Collin Winterce36ad82007-08-30 01:19:48 +0000544 raise OverflowError("long int exceeds XML-RPC limits")
Fredrik Lundh3d9addd2002-06-27 21:36:21 +0000545 write("<value><int>")
546 write(str(int(value)))
547 write("</int></value>\n")
Guido van Rossum13257902007-06-07 23:15:56 +0000548 dispatch[int] = dump_long
Skip Montanaro5e9c71b2001-10-10 15:56:34 +0000549
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
Fredrik Lundh3d9addd2002-06-27 21:36:21 +0000556 def dump_string(self, value, write, escape=escape):
557 write("<value><string>")
558 write(escape(value))
559 write("</string></value>\n")
Guido van Rossum98297ee2007-11-06 21:34:58 +0000560 dispatch[bytes] = dump_string
Fredrik Lundhb9056332001-07-11 17:42:21 +0000561
Guido van Rossum54ad5232007-05-27 09:17:48 +0000562 def dump_unicode(self, value, write, escape=escape):
Guido van Rossum54ad5232007-05-27 09:17:48 +0000563 write("<value><string>")
564 write(escape(value))
565 write("</string></value>\n")
566 dispatch[str] = dump_unicode
Fredrik Lundhb9056332001-07-11 17:42:21 +0000567
Fredrik Lundh3d9addd2002-06-27 21:36:21 +0000568 def dump_array(self, value, write):
569 i = id(value)
Guido van Rossume2b70bc2006-08-18 22:13:04 +0000570 if i in self.memo:
Collin Winterce36ad82007-08-30 01:19:48 +0000571 raise TypeError("cannot marshal recursive sequences")
Fredrik Lundh3d9addd2002-06-27 21:36:21 +0000572 self.memo[i] = None
Fredrik Lundh1538c232001-10-01 19:42:03 +0000573 dump = self.__dump
Fredrik Lundhb9056332001-07-11 17:42:21 +0000574 write("<value><array><data>\n")
575 for v in value:
Fredrik Lundh3d9addd2002-06-27 21:36:21 +0000576 dump(v, write)
Fredrik Lundhb9056332001-07-11 17:42:21 +0000577 write("</data></array></value>\n")
Fredrik Lundh3d9addd2002-06-27 21:36:21 +0000578 del self.memo[i]
Guido van Rossum13257902007-06-07 23:15:56 +0000579 dispatch[tuple] = dump_array
580 dispatch[list] = dump_array
Fredrik Lundhb9056332001-07-11 17:42:21 +0000581
Fredrik Lundh3d9addd2002-06-27 21:36:21 +0000582 def dump_struct(self, value, write, escape=escape):
583 i = id(value)
Guido van Rossume2b70bc2006-08-18 22:13:04 +0000584 if i in self.memo:
Collin Winterce36ad82007-08-30 01:19:48 +0000585 raise TypeError("cannot marshal recursive dictionaries")
Fredrik Lundh3d9addd2002-06-27 21:36:21 +0000586 self.memo[i] = None
Fredrik Lundh1538c232001-10-01 19:42:03 +0000587 dump = self.__dump
Fredrik Lundhb9056332001-07-11 17:42:21 +0000588 write("<value><struct>\n")
Andrew M. Kuchling5962f452004-06-05 12:35:58 +0000589 for k, v in value.items():
Fredrik Lundhb9056332001-07-11 17:42:21 +0000590 write("<member>\n")
Guido van Rossum3172c5d2007-10-16 18:12:55 +0000591 if not isinstance(k, str):
Collin Winterce36ad82007-08-30 01:19:48 +0000592 raise TypeError("dictionary key must be string")
Fredrik Lundh1538c232001-10-01 19:42:03 +0000593 write("<name>%s</name>\n" % escape(k))
Andrew M. Kuchling5962f452004-06-05 12:35:58 +0000594 dump(v, write)
Fredrik Lundhb9056332001-07-11 17:42:21 +0000595 write("</member>\n")
596 write("</struct></value>\n")
Fredrik Lundh3d9addd2002-06-27 21:36:21 +0000597 del self.memo[i]
Guido van Rossum13257902007-06-07 23:15:56 +0000598 dispatch[dict] = dump_struct
Fredrik Lundhb9056332001-07-11 17:42:21 +0000599
Fred Drakeba613c32005-02-10 18:33:30 +0000600 if datetime:
601 def dump_datetime(self, value, write):
602 write("<value><dateTime.iso8601>")
Christian Heimesdae2a892008-04-19 00:55:37 +0000603 write(_strftime(value))
Fred Drakeba613c32005-02-10 18:33:30 +0000604 write("</dateTime.iso8601></value>\n")
605 dispatch[datetime.datetime] = dump_datetime
606
Fredrik Lundh3d9addd2002-06-27 21:36:21 +0000607 def dump_instance(self, value, write):
Fredrik Lundhb9056332001-07-11 17:42:21 +0000608 # check for special wrappers
609 if value.__class__ in WRAPPERS:
Fredrik Lundh3d9addd2002-06-27 21:36:21 +0000610 self.write = write
Fredrik Lundhb9056332001-07-11 17:42:21 +0000611 value.encode(self)
Fredrik Lundh3d9addd2002-06-27 21:36:21 +0000612 del self.write
Fredrik Lundhb9056332001-07-11 17:42:21 +0000613 else:
614 # store instance attributes as a struct (really?)
Fredrik Lundh3d9addd2002-06-27 21:36:21 +0000615 self.dump_struct(value.__dict__, write)
Guido van Rossume4dea982006-04-21 13:45:00 +0000616 dispatch[DateTime] = dump_instance
617 dispatch[Binary] = dump_instance
Thomas Wouters89f507f2006-12-13 04:49:30 +0000618 # XXX(twouters): using "_arbitrary_instance" as key as a quick-fix
619 # for the p3yk merge, this should probably be fixed more neatly.
620 dispatch["_arbitrary_instance"] = dump_instance
Fredrik Lundhb9056332001-07-11 17:42:21 +0000621
Fredrik Lundh3d9addd2002-06-27 21:36:21 +0000622##
623# XML-RPC unmarshaller.
624#
625# @see loads
626
Fredrik Lundhb9056332001-07-11 17:42:21 +0000627class Unmarshaller:
Fred Drake1b410792001-09-04 18:55:03 +0000628 """Unmarshal an XML-RPC response, based on incoming XML event
Fredrik Lundhb0e8e9b2001-09-10 21:45:42 +0000629 messages (start, data, end). Call close() to get the resulting
Fred Drake1b410792001-09-04 18:55:03 +0000630 data structure.
Fredrik Lundhb9056332001-07-11 17:42:21 +0000631
Fredrik Lundhb0e8e9b2001-09-10 21:45:42 +0000632 Note that this reader is fairly tolerant, and gladly accepts bogus
633 XML-RPC data without complaining (but not bogus XML).
Fred Drake1b410792001-09-04 18:55:03 +0000634 """
Fredrik Lundhb9056332001-07-11 17:42:21 +0000635
636 # and again, if you don't understand what's going on in here,
637 # that's perfectly ok.
638
Georg Brandlfe991052009-09-16 15:54:04 +0000639 def __init__(self, use_datetime=False):
Fredrik Lundhb9056332001-07-11 17:42:21 +0000640 self._type = None
641 self._stack = []
642 self._marks = []
643 self._data = []
644 self._methodname = None
645 self._encoding = "utf-8"
646 self.append = self._stack.append
Skip Montanaro174dd222005-05-14 20:54:16 +0000647 self._use_datetime = use_datetime
648 if use_datetime and not datetime:
Collin Winterce36ad82007-08-30 01:19:48 +0000649 raise ValueError("the datetime module is not available")
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"))
Fredrik Lundhb9056332001-07-11 17:42:21 +0000761 self.append(value)
762 self._value = 0
763 dispatch["base64"] = end_base64
764
Fredrik Lundh1538c232001-10-01 19:42:03 +0000765 def end_dateTime(self, data):
Fredrik Lundhb9056332001-07-11 17:42:21 +0000766 value = DateTime()
Fredrik Lundh1538c232001-10-01 19:42:03 +0000767 value.decode(data)
Skip Montanaro174dd222005-05-14 20:54:16 +0000768 if self._use_datetime:
769 value = _datetime_type(data)
Fredrik Lundhb9056332001-07-11 17:42:21 +0000770 self.append(value)
771 dispatch["dateTime.iso8601"] = end_dateTime
772
773 def end_value(self, data):
Fredrik Lundh3d9addd2002-06-27 21:36:21 +0000774 # if we stumble upon a value element with no internal
Fredrik Lundhb9056332001-07-11 17:42:21 +0000775 # elements, treat it as a string element
776 if self._value:
777 self.end_string(data)
778 dispatch["value"] = end_value
779
780 def end_params(self, data):
781 self._type = "params"
782 dispatch["params"] = end_params
783
784 def end_fault(self, data):
785 self._type = "fault"
786 dispatch["fault"] = end_fault
787
Fredrik Lundh1538c232001-10-01 19:42:03 +0000788 def end_methodName(self, data):
Fredrik Lundhb9056332001-07-11 17:42:21 +0000789 if self._encoding:
Alexandre Vassalottie671fd22009-07-22 02:32:34 +0000790 data = data.decode(self._encoding)
Fredrik Lundhb9056332001-07-11 17:42:21 +0000791 self._methodname = data
792 self._type = "methodName" # no params
793 dispatch["methodName"] = end_methodName
794
Martin v. Löwis45394c22003-10-31 13:49:36 +0000795## Multicall support
796#
Fredrik Lundhb9056332001-07-11 17:42:21 +0000797
Martin v. Löwis45394c22003-10-31 13:49:36 +0000798class _MultiCallMethod:
799 # some lesser magic to store calls made to a MultiCall object
800 # for batch execution
801 def __init__(self, call_list, name):
802 self.__call_list = call_list
803 self.__name = name
804 def __getattr__(self, name):
805 return _MultiCallMethod(self.__call_list, "%s.%s" % (self.__name, name))
806 def __call__(self, *args):
807 self.__call_list.append((self.__name, args))
808
Martin v. Löwis12237b32004-08-22 16:04:50 +0000809class MultiCallIterator:
Martin v. Löwis45394c22003-10-31 13:49:36 +0000810 """Iterates over the results of a multicall. Exceptions are
811 thrown in response to xmlrpc faults."""
Raymond Hettingercc523fc2003-11-02 09:47:05 +0000812
Martin v. Löwis12237b32004-08-22 16:04:50 +0000813 def __init__(self, results):
814 self.results = results
815
816 def __getitem__(self, i):
817 item = self.results[i]
818 if type(item) == type({}):
819 raise Fault(item['faultCode'], item['faultString'])
820 elif type(item) == type([]):
821 return item[0]
Martin v. Löwis45394c22003-10-31 13:49:36 +0000822 else:
Collin Winterce36ad82007-08-30 01:19:48 +0000823 raise ValueError("unexpected type in multicall result")
Raymond Hettingercc523fc2003-11-02 09:47:05 +0000824
Martin v. Löwis45394c22003-10-31 13:49:36 +0000825class MultiCall:
826 """server -> a object used to boxcar method calls
827
828 server should be a ServerProxy object.
829
830 Methods can be added to the MultiCall using normal
831 method call syntax e.g.:
832
833 multicall = MultiCall(server_proxy)
834 multicall.add(2,3)
835 multicall.get_address("Guido")
836
837 To execute the multicall, call the MultiCall object e.g.:
838
839 add_result, address = multicall()
840 """
Raymond Hettingercc523fc2003-11-02 09:47:05 +0000841
Martin v. Löwis45394c22003-10-31 13:49:36 +0000842 def __init__(self, server):
843 self.__server = server
844 self.__call_list = []
Raymond Hettingercc523fc2003-11-02 09:47:05 +0000845
Martin v. Löwis45394c22003-10-31 13:49:36 +0000846 def __repr__(self):
847 return "<MultiCall at %x>" % id(self)
Raymond Hettingercc523fc2003-11-02 09:47:05 +0000848
Martin v. Löwis45394c22003-10-31 13:49:36 +0000849 __str__ = __repr__
850
851 def __getattr__(self, name):
852 return _MultiCallMethod(self.__call_list, name)
853
854 def __call__(self):
855 marshalled_list = []
856 for name, args in self.__call_list:
857 marshalled_list.append({'methodName' : name, 'params' : args})
858
859 return MultiCallIterator(self.__server.system.multicall(marshalled_list))
Raymond Hettingercc523fc2003-11-02 09:47:05 +0000860
Fredrik Lundhb9056332001-07-11 17:42:21 +0000861# --------------------------------------------------------------------
862# convenience functions
863
Georg Brandl38eceaa2008-05-26 11:14:17 +0000864FastMarshaller = FastParser = FastUnmarshaller = None
865
Fredrik Lundh3d9addd2002-06-27 21:36:21 +0000866##
867# Create a parser object, and connect it to an unmarshalling instance.
868# This function picks the fastest available XML parser.
869#
870# return A (parser, unmarshaller) tuple.
871
Georg Brandlfe991052009-09-16 15:54:04 +0000872def getparser(use_datetime=False):
Fredrik Lundhb9056332001-07-11 17:42:21 +0000873 """getparser() -> parser, unmarshaller
874
Fredrik Lundhb0e8e9b2001-09-10 21:45:42 +0000875 Create an instance of the fastest available parser, and attach it
876 to an unmarshalling object. Return both objects.
Fredrik Lundhb9056332001-07-11 17:42:21 +0000877 """
Skip Montanaro174dd222005-05-14 20:54:16 +0000878 if use_datetime and not datetime:
Collin Winterce36ad82007-08-30 01:19:48 +0000879 raise ValueError("the datetime module is not available")
Fredrik Lundhb9056332001-07-11 17:42:21 +0000880 if FastParser and FastUnmarshaller:
Skip Montanaro174dd222005-05-14 20:54:16 +0000881 if use_datetime:
882 mkdatetime = _datetime_type
883 else:
884 mkdatetime = _datetime
885 target = FastUnmarshaller(True, False, _binary, mkdatetime, Fault)
Fredrik Lundhb9056332001-07-11 17:42:21 +0000886 parser = FastParser(target)
887 else:
Skip Montanaro174dd222005-05-14 20:54:16 +0000888 target = Unmarshaller(use_datetime=use_datetime)
Fredrik Lundhb9056332001-07-11 17:42:21 +0000889 if FastParser:
890 parser = FastParser(target)
Fredrik Lundhb9056332001-07-11 17:42:21 +0000891 else:
Georg Brandlcef803f2009-06-04 09:04:53 +0000892 parser = ExpatParser(target)
Fredrik Lundhb9056332001-07-11 17:42:21 +0000893 return parser, target
894
Fredrik Lundh3d9addd2002-06-27 21:36:21 +0000895##
896# Convert a Python tuple or a Fault instance to an XML-RPC packet.
897#
898# @def dumps(params, **options)
899# @param params A tuple or Fault instance.
900# @keyparam methodname If given, create a methodCall request for
901# this method name.
902# @keyparam methodresponse If given, create a methodResponse packet.
903# If used with a tuple, the tuple must be a singleton (that is,
904# it must contain exactly one element).
905# @keyparam encoding The packet encoding.
906# @return A string containing marshalled data.
907
Andrew M. Kuchlinga4c2b742003-04-25 00:26:51 +0000908def dumps(params, methodname=None, methodresponse=None, encoding=None,
Georg Brandlfe991052009-09-16 15:54:04 +0000909 allow_none=False):
Fredrik Lundhb9056332001-07-11 17:42:21 +0000910 """data [,options] -> marshalled data
911
912 Convert an argument tuple or a Fault instance to an XML-RPC
913 request (or response, if the methodresponse option is used).
914
Fredrik Lundhb0e8e9b2001-09-10 21:45:42 +0000915 In addition to the data object, the following options can be given
916 as keyword arguments:
Fredrik Lundhb9056332001-07-11 17:42:21 +0000917
918 methodname: the method name for a methodCall packet
919
920 methodresponse: true to create a methodResponse packet.
921 If this option is used with a tuple, the tuple must be
922 a singleton (i.e. it can contain only one element).
923
924 encoding: the packet encoding (default is UTF-8)
925
926 All 8-bit strings in the data structure are assumed to use the
927 packet encoding. Unicode strings are automatically converted,
Fredrik Lundhb0e8e9b2001-09-10 21:45:42 +0000928 where necessary.
Fredrik Lundhb9056332001-07-11 17:42:21 +0000929 """
930
Guido van Rossum13257902007-06-07 23:15:56 +0000931 assert isinstance(params, (tuple, Fault)), "argument must be tuple or Fault instance"
Fredrik Lundhb9056332001-07-11 17:42:21 +0000932 if isinstance(params, Fault):
933 methodresponse = 1
Guido van Rossum13257902007-06-07 23:15:56 +0000934 elif methodresponse and isinstance(params, tuple):
Fredrik Lundhb9056332001-07-11 17:42:21 +0000935 assert len(params) == 1, "response tuple must be a singleton"
936
Fredrik Lundh3d9addd2002-06-27 21:36:21 +0000937 if not encoding:
Fredrik Lundhb9056332001-07-11 17:42:21 +0000938 encoding = "utf-8"
939
Fredrik Lundh3d9addd2002-06-27 21:36:21 +0000940 if FastMarshaller:
941 m = FastMarshaller(encoding)
942 else:
Andrew M. Kuchlinga4c2b742003-04-25 00:26:51 +0000943 m = Marshaller(encoding, allow_none)
Fredrik Lundh3d9addd2002-06-27 21:36:21 +0000944
Fredrik Lundhb9056332001-07-11 17:42:21 +0000945 data = m.dumps(params)
946
947 if encoding != "utf-8":
Fredrik Lundh3d9addd2002-06-27 21:36:21 +0000948 xmlheader = "<?xml version='1.0' encoding='%s'?>\n" % str(encoding)
Fredrik Lundhb9056332001-07-11 17:42:21 +0000949 else:
950 xmlheader = "<?xml version='1.0'?>\n" # utf-8 is default
951
952 # standard XML-RPC wrappings
Fredrik Lundh3d9addd2002-06-27 21:36:21 +0000953 if methodname:
Fredrik Lundhb9056332001-07-11 17:42:21 +0000954 # a method call
Guido van Rossum3172c5d2007-10-16 18:12:55 +0000955 if not isinstance(methodname, str):
Fredrik Lundhb9056332001-07-11 17:42:21 +0000956 methodname = methodname.encode(encoding)
957 data = (
958 xmlheader,
959 "<methodCall>\n"
960 "<methodName>", methodname, "</methodName>\n",
961 data,
962 "</methodCall>\n"
963 )
964 elif methodresponse:
965 # a method response, or a fault structure
966 data = (
967 xmlheader,
968 "<methodResponse>\n",
969 data,
970 "</methodResponse>\n"
971 )
972 else:
973 return data # return as is
Neal Norwitzff113342007-04-17 08:42:15 +0000974 return "".join(data)
Fredrik Lundhb9056332001-07-11 17:42:21 +0000975
Fredrik Lundh3d9addd2002-06-27 21:36:21 +0000976##
977# Convert an XML-RPC packet to a Python object. If the XML-RPC packet
978# represents a fault condition, this function raises a Fault exception.
979#
980# @param data An XML-RPC packet, given as an 8-bit string.
Walter Dörwaldf0dfc7a2003-10-20 14:01:56 +0000981# @return A tuple containing the unpacked data, and the method name
Fredrik Lundh3d9addd2002-06-27 21:36:21 +0000982# (None if not present).
983# @see Fault
984
Georg Brandlfe991052009-09-16 15:54:04 +0000985def loads(data, use_datetime=False):
Fredrik Lundhb9056332001-07-11 17:42:21 +0000986 """data -> unmarshalled data, method name
987
988 Convert an XML-RPC packet to unmarshalled data plus a method
989 name (None if not present).
990
991 If the XML-RPC packet represents a fault condition, this function
992 raises a Fault exception.
993 """
Skip Montanaro174dd222005-05-14 20:54:16 +0000994 p, u = getparser(use_datetime=use_datetime)
Fredrik Lundhb9056332001-07-11 17:42:21 +0000995 p.feed(data)
996 p.close()
997 return u.close(), u.getmethodname()
998
Kristján Valur Jónsson985fc6a2009-07-01 10:01:31 +0000999##
1000# Encode a string using the gzip content encoding such as specified by the
1001# Content-Encoding: gzip
1002# in the HTTP header, as described in RFC 1952
1003#
1004# @param data the unencoded data
1005# @return the encoded data
1006
1007def gzip_encode(data):
1008 """data -> gzip encoded data
1009
1010 Encode data using the gzip content encoding as described in RFC 1952
1011 """
Kristján Valur Jónssonaefde242009-07-19 22:29:24 +00001012 if not gzip:
1013 raise NotImplementedError
Kristján Valur Jónsson985fc6a2009-07-01 10:01:31 +00001014 f = BytesIO()
1015 gzf = gzip.GzipFile(mode="wb", fileobj=f, compresslevel=1)
1016 gzf.write(data)
1017 gzf.close()
1018 encoded = f.getvalue()
1019 f.close()
1020 return encoded
1021
1022##
1023# Decode a string using the gzip content encoding such as specified by the
1024# Content-Encoding: gzip
1025# in the HTTP header, as described in RFC 1952
1026#
1027# @param data The encoded data
1028# @return the unencoded data
1029# @raises ValueError if data is not correctly coded.
1030
1031def gzip_decode(data):
1032 """gzip encoded data -> unencoded data
1033
1034 Decode data using the gzip content encoding as described in RFC 1952
1035 """
Kristján Valur Jónssonaefde242009-07-19 22:29:24 +00001036 if not gzip:
1037 raise NotImplementedError
Kristján Valur Jónsson985fc6a2009-07-01 10:01:31 +00001038 f = BytesIO(data)
1039 gzf = gzip.GzipFile(mode="rb", fileobj=f)
1040 try:
1041 decoded = gzf.read()
1042 except IOError:
1043 raise ValueError("invalid data")
1044 f.close()
1045 gzf.close()
1046 return decoded
1047
1048##
1049# Return a decoded file-like object for the gzip encoding
1050# as described in RFC 1952.
1051#
1052# @param response A stream supporting a read() method
1053# @return a file-like object that the decoded data can be read() from
1054
Kristján Valur Jónsson9ab07312009-07-19 22:38:38 +00001055class GzipDecodedResponse(gzip.GzipFile if gzip else object):
Kristján Valur Jónsson985fc6a2009-07-01 10:01:31 +00001056 """a file-like object to decode a response encoded with the gzip
1057 method, as described in RFC 1952.
1058 """
1059 def __init__(self, response):
1060 #response doesn't support tell() and read(), required by
1061 #GzipFile
Kristján Valur Jónssonaefde242009-07-19 22:29:24 +00001062 if not gzip:
1063 raise NotImplementedError
Kristján Valur Jónsson985fc6a2009-07-01 10:01:31 +00001064 self.io = BytesIO(response.read())
1065 gzip.GzipFile.__init__(self, mode="rb", fileobj=self.io)
1066
1067 def close(self):
1068 gzip.GzipFile.close(self)
1069 self.io.close()
1070
Fredrik Lundhb9056332001-07-11 17:42:21 +00001071
1072# --------------------------------------------------------------------
1073# request dispatcher
1074
1075class _Method:
1076 # some magic to bind an XML-RPC method to an RPC server.
1077 # supports "nested" methods (e.g. examples.getStateName)
1078 def __init__(self, send, name):
1079 self.__send = send
1080 self.__name = name
1081 def __getattr__(self, name):
1082 return _Method(self.__send, "%s.%s" % (self.__name, name))
1083 def __call__(self, *args):
1084 return self.__send(self.__name, args)
1085
Fredrik Lundh3d9addd2002-06-27 21:36:21 +00001086##
1087# Standard transport class for XML-RPC over HTTP.
1088# <p>
1089# You can create custom transports by subclassing this method, and
1090# overriding selected methods.
Fredrik Lundhb9056332001-07-11 17:42:21 +00001091
1092class Transport:
Fredrik Lundhc4c062f2001-09-10 19:45:02 +00001093 """Handles an HTTP transaction to an XML-RPC server."""
Fredrik Lundhb9056332001-07-11 17:42:21 +00001094
1095 # client identifier (may be overridden)
1096 user_agent = "xmlrpclib.py/%s (by www.pythonware.com)" % __version__
1097
Kristján Valur Jónsson985fc6a2009-07-01 10:01:31 +00001098 #if true, we'll request gzip encoding
1099 accept_gzip_encoding = True
1100
1101 # if positive, encode request using gzip if it exceeds this threshold
1102 # note that many server will get confused, so only use it if you know
1103 # that they can decode such a request
1104 encode_threshold = None #None = don't encode
1105
Georg Brandlfe991052009-09-16 15:54:04 +00001106 def __init__(self, use_datetime=False):
Skip Montanaro174dd222005-05-14 20:54:16 +00001107 self._use_datetime = use_datetime
Kristján Valur Jónsson985fc6a2009-07-01 10:01:31 +00001108 self._connection = (None, None)
1109 self._extra_headers = []
Skip Montanaro174dd222005-05-14 20:54:16 +00001110
Fredrik Lundh3d9addd2002-06-27 21:36:21 +00001111 ##
1112 # Send a complete request, and parse the response.
Kristján Valur Jónsson985fc6a2009-07-01 10:01:31 +00001113 # Retry request if a cached connection has disconnected.
Fredrik Lundh3d9addd2002-06-27 21:36:21 +00001114 #
1115 # @param host Target host.
1116 # @param handler Target PRC handler.
1117 # @param request_body XML-RPC request body.
1118 # @param verbose Debugging flag.
1119 # @return Parsed response.
1120
Georg Brandlfe991052009-09-16 15:54:04 +00001121 def request(self, host, handler, request_body, verbose=False):
Kristján Valur Jónsson985fc6a2009-07-01 10:01:31 +00001122 #retry request once if cached connection has gone cold
1123 for i in (0, 1):
1124 try:
1125 return self.single_request(host, handler, request_body, verbose)
Kristján Valur Jónsson43535d92009-07-03 23:23:50 +00001126 except socket.error as e:
Victor Stinner756f5472010-07-24 02:24:55 +00001127 if i or e.errno not in (errno.ECONNRESET, errno.ECONNABORTED, errno.EPIPE):
Kristján Valur Jónsson43535d92009-07-03 23:23:50 +00001128 raise
1129 except http.client.BadStatusLine: #close after we sent request
1130 if i:
Kristján Valur Jónsson985fc6a2009-07-01 10:01:31 +00001131 raise
1132
Georg Brandlfe991052009-09-16 15:54:04 +00001133 def single_request(self, host, handler, request_body, verbose=False):
Fredrik Lundhb9056332001-07-11 17:42:21 +00001134 # issue XML-RPC request
Kristján Valur Jónsson985fc6a2009-07-01 10:01:31 +00001135 try:
1136 http_conn = self.send_request(host, handler, request_body, verbose)
1137 resp = http_conn.getresponse()
1138 if resp.status == 200:
1139 self.verbose = verbose
1140 return self.parse_response(resp)
Fredrik Lundhb9056332001-07-11 17:42:21 +00001141
Kristján Valur Jónsson985fc6a2009-07-01 10:01:31 +00001142 except Fault:
1143 raise
1144 except Exception:
1145 #All unexpected errors leave connection in
1146 # a strange state, so we clear it.
1147 self.close()
1148 raise
Fredrik Lundhb9056332001-07-11 17:42:21 +00001149
Kristján Valur Jónsson985fc6a2009-07-01 10:01:31 +00001150 #We got an error response.
1151 #Discard any response data and raise exception
1152 if resp.getheader("content-length", ""):
1153 resp.read()
1154 raise ProtocolError(
1155 host + handler,
1156 resp.status, resp.reason,
1157 dict(resp.getheaders())
1158 )
Fredrik Lundhb9056332001-07-11 17:42:21 +00001159
Fredrik Lundh3d9addd2002-06-27 21:36:21 +00001160
1161 ##
1162 # Create parser.
1163 #
1164 # @return A 2-tuple containing a parser and a unmarshaller.
Fredrik Lundhb9056332001-07-11 17:42:21 +00001165
Fredrik Lundhc4c062f2001-09-10 19:45:02 +00001166 def getparser(self):
1167 # get parser and unmarshaller
Skip Montanaro174dd222005-05-14 20:54:16 +00001168 return getparser(use_datetime=self._use_datetime)
Fredrik Lundhc4c062f2001-09-10 19:45:02 +00001169
Fredrik Lundh3d9addd2002-06-27 21:36:21 +00001170 ##
Fredrik Lundh1303c7c2002-10-22 18:23:00 +00001171 # Get authorization info from host parameter
1172 # Host may be a string, or a (host, x509-dict) tuple; if a string,
1173 # it is checked for a "user:pw@host" format, and a "Basic
1174 # Authentication" header is added if appropriate.
1175 #
1176 # @param host Host descriptor (URL or (URL, x509 info) tuple).
1177 # @return A 3-tuple containing (actual host, extra headers,
1178 # x509 info). The header and x509 fields may be None.
1179
1180 def get_host_info(self, host):
1181
1182 x509 = {}
Guido van Rossum13257902007-06-07 23:15:56 +00001183 if isinstance(host, tuple):
Fredrik Lundh1303c7c2002-10-22 18:23:00 +00001184 host, x509 = host
1185
Jeremy Hylton1afc1692008-06-18 20:49:58 +00001186 import urllib.parse
1187 auth, host = urllib.parse.splituser(host)
Fredrik Lundh1303c7c2002-10-22 18:23:00 +00001188
1189 if auth:
Georg Brandlc8dcfb62009-02-13 10:50:01 +00001190 auth = urllib.parse.unquote_to_bytes(auth)
Georg Brandlb54d8012009-06-04 09:11:51 +00001191 auth = base64.encodebytes(auth).decode("utf-8")
Neal Norwitzff113342007-04-17 08:42:15 +00001192 auth = "".join(auth.split()) # get rid of whitespace
Fredrik Lundh1303c7c2002-10-22 18:23:00 +00001193 extra_headers = [
1194 ("Authorization", "Basic " + auth)
1195 ]
1196 else:
Kristján Valur Jónsson985fc6a2009-07-01 10:01:31 +00001197 extra_headers = []
Fredrik Lundh1303c7c2002-10-22 18:23:00 +00001198
1199 return host, extra_headers, x509
1200
1201 ##
Fredrik Lundh3d9addd2002-06-27 21:36:21 +00001202 # Connect to server.
1203 #
1204 # @param host Target host.
Jeremy Hylton5d8a88a2007-08-14 16:47:39 +00001205 # @return An HTTPConnection object
Fredrik Lundh3d9addd2002-06-27 21:36:21 +00001206
Fredrik Lundhb9056332001-07-11 17:42:21 +00001207 def make_connection(self, host):
Kristján Valur Jónsson985fc6a2009-07-01 10:01:31 +00001208 #return an existing connection if possible. This allows
1209 #HTTP/1.1 keep-alive.
1210 if self._connection and host == self._connection[0]:
1211 return self._connection[1]
Fredrik Lundhb9056332001-07-11 17:42:21 +00001212 # create a HTTP connection object from a host descriptor
Kristján Valur Jónsson985fc6a2009-07-01 10:01:31 +00001213 chost, self._extra_headers, x509 = self.get_host_info(host)
1214 self._connection = host, http.client.HTTPConnection(chost)
1215 return self._connection[1]
Jeremy Hylton5d8a88a2007-08-14 16:47:39 +00001216
Kristján Valur Jónsson985fc6a2009-07-01 10:01:31 +00001217 ##
1218 # Clear any cached connection object.
1219 # Used in the event of socket errors.
1220 #
1221 def close(self):
1222 if self._connection[1]:
1223 self._connection[1].close()
1224 self._connection = (None, None)
Fredrik Lundhb9056332001-07-11 17:42:21 +00001225
Fredrik Lundh3d9addd2002-06-27 21:36:21 +00001226 ##
Jeremy Hylton5d8a88a2007-08-14 16:47:39 +00001227 # Send HTTP request.
Fredrik Lundh3d9addd2002-06-27 21:36:21 +00001228 #
Jeremy Hylton5d8a88a2007-08-14 16:47:39 +00001229 # @param host Host descriptor (URL or (URL, x509 info) tuple).
1230 # @param handler Targer RPC handler (a path relative to host)
1231 # @param request_body The XML-RPC request body
1232 # @param debug Enable debugging if debug is true.
1233 # @return An HTTPConnection.
Fredrik Lundh3d9addd2002-06-27 21:36:21 +00001234
Jeremy Hylton5d8a88a2007-08-14 16:47:39 +00001235 def send_request(self, host, handler, request_body, debug):
Kristján Valur Jónsson985fc6a2009-07-01 10:01:31 +00001236 connection = self.make_connection(host)
1237 headers = self._extra_headers[:]
Jeremy Hylton5d8a88a2007-08-14 16:47:39 +00001238 if debug:
1239 connection.set_debuglevel(1)
Kristján Valur Jónssonaefde242009-07-19 22:29:24 +00001240 if self.accept_gzip_encoding and gzip:
Kristján Valur Jónsson985fc6a2009-07-01 10:01:31 +00001241 connection.putrequest("POST", handler, skip_accept_encoding=True)
1242 headers.append(("Accept-Encoding", "gzip"))
1243 else:
1244 connection.putrequest("POST", handler)
1245 headers.append(("Content-Type", "text/xml"))
1246 headers.append(("User-Agent", self.user_agent))
1247 self.send_headers(connection, headers)
1248 self.send_content(connection, request_body)
Jeremy Hylton5d8a88a2007-08-14 16:47:39 +00001249 return connection
Fredrik Lundhb9056332001-07-11 17:42:21 +00001250
Fredrik Lundh3d9addd2002-06-27 21:36:21 +00001251 ##
Kristján Valur Jónsson985fc6a2009-07-01 10:01:31 +00001252 # Send request headers.
1253 # This function provides a useful hook for subclassing
1254 #
1255 # @param connection httpConnection.
1256 # @param headers list of key,value pairs for HTTP headers
1257
1258 def send_headers(self, connection, headers):
1259 for key, val in headers:
1260 connection.putheader(key, val)
1261
1262 ##
1263 # Send request body.
1264 # This function provides a useful hook for subclassing
1265 #
1266 # @param connection httpConnection.
1267 # @param request_body XML-RPC request body.
1268
1269 def send_content(self, connection, request_body):
1270 #optionally encode the request
1271 if (self.encode_threshold is not None and
Kristján Valur Jónssonaefde242009-07-19 22:29:24 +00001272 self.encode_threshold < len(request_body) and
1273 gzip):
Kristján Valur Jónsson985fc6a2009-07-01 10:01:31 +00001274 connection.putheader("Content-Encoding", "gzip")
1275 request_body = gzip_encode(request_body)
1276
1277 connection.putheader("Content-Length", str(len(request_body)))
1278 connection.endheaders(request_body)
1279
1280 ##
Fredrik Lundh3d9addd2002-06-27 21:36:21 +00001281 # Parse response.
1282 #
1283 # @param file Stream.
1284 # @return Response tuple and target method.
1285
Kristján Valur Jónsson985fc6a2009-07-01 10:01:31 +00001286 def parse_response(self, response):
1287 # read response data from httpresponse, and parse it
Senthil Kumaranf34445f2010-12-08 08:04:49 +00001288 # Check for new http response object, otherwise it is a file object.
1289 if hasattr(response, 'getheader'):
1290 if response.getheader("Content-Encoding", "") == "gzip":
1291 stream = GzipDecodedResponse(response)
1292 else:
1293 stream = response
Kristján Valur Jónsson985fc6a2009-07-01 10:01:31 +00001294 else:
1295 stream = response
Fredrik Lundhb9056332001-07-11 17:42:21 +00001296
Fredrik Lundhc4c062f2001-09-10 19:45:02 +00001297 p, u = self.getparser()
Fredrik Lundhb9056332001-07-11 17:42:21 +00001298
1299 while 1:
Kristján Valur Jónsson985fc6a2009-07-01 10:01:31 +00001300 data = stream.read(1024)
1301 if not data:
Fredrik Lundhb9056332001-07-11 17:42:21 +00001302 break
1303 if self.verbose:
Kristján Valur Jónsson985fc6a2009-07-01 10:01:31 +00001304 print("body:", repr(data))
1305 p.feed(data)
Fredrik Lundhb9056332001-07-11 17:42:21 +00001306
Kristján Valur Jónsson985fc6a2009-07-01 10:01:31 +00001307 if stream is not response:
1308 stream.close()
Fredrik Lundhb9056332001-07-11 17:42:21 +00001309 p.close()
1310
1311 return u.close()
1312
Fredrik Lundh3d9addd2002-06-27 21:36:21 +00001313##
1314# Standard transport class for XML-RPC over HTTPS.
1315
Fredrik Lundhb9056332001-07-11 17:42:21 +00001316class SafeTransport(Transport):
Fredrik Lundhc4c062f2001-09-10 19:45:02 +00001317 """Handles an HTTPS transaction to an XML-RPC server."""
Fredrik Lundhb9056332001-07-11 17:42:21 +00001318
Fredrik Lundh3d9addd2002-06-27 21:36:21 +00001319 # FIXME: mostly untested
1320
Kristján Valur Jónsson985fc6a2009-07-01 10:01:31 +00001321 def make_connection(self, host):
1322 if self._connection and host == self._connection[0]:
1323 return self._connection[1]
1324
Senthil Kumaran6a0b5c42010-11-18 17:08:48 +00001325 if not hasattr(http.client, "HTTPSConnection"):
Fredrik Lundh1303c7c2002-10-22 18:23:00 +00001326 raise NotImplementedError(
Kristján Valur Jónsson985fc6a2009-07-01 10:01:31 +00001327 "your version of http.client doesn't support HTTPS")
1328 # create a HTTPS connection object from a host descriptor
1329 # host may be a string, or a (host, x509-dict) tuple
1330 chost, self._extra_headers, x509 = self.get_host_info(host)
1331 self._connection = host, http.client.HTTPSConnection(chost,
1332 None, **(x509 or {}))
1333 return self._connection[1]
Fredrik Lundhb9056332001-07-11 17:42:21 +00001334
Fredrik Lundh3d9addd2002-06-27 21:36:21 +00001335##
1336# Standard server proxy. This class establishes a virtual connection
1337# to an XML-RPC server.
1338# <p>
1339# This class is available as ServerProxy and Server. New code should
1340# use ServerProxy, to avoid confusion.
1341#
1342# @def ServerProxy(uri, **options)
1343# @param uri The connection point on the server.
1344# @keyparam transport A transport factory, compatible with the
1345# standard transport class.
1346# @keyparam encoding The default encoding used for 8-bit strings
1347# (default is UTF-8).
1348# @keyparam verbose Use a true value to enable debugging output.
1349# (printed to standard output).
1350# @see Transport
1351
Fredrik Lundhb9056332001-07-11 17:42:21 +00001352class ServerProxy:
1353 """uri [,options] -> a logical connection to an XML-RPC server
1354
1355 uri is the connection point on the server, given as
1356 scheme://host/target.
1357
1358 The standard implementation always supports the "http" scheme. If
1359 SSL socket support is available (Python 2.0), it also supports
1360 "https".
1361
1362 If the target part and the slash preceding it are both omitted,
1363 "/RPC2" is assumed.
1364
1365 The following options can be given as keyword arguments:
1366
1367 transport: a transport factory
1368 encoding: the request encoding (default is UTF-8)
1369
1370 All 8-bit strings passed to the server proxy are assumed to use
1371 the given encoding.
1372 """
1373
Georg Brandlfe991052009-09-16 15:54:04 +00001374 def __init__(self, uri, transport=None, encoding=None, verbose=False,
1375 allow_none=False, use_datetime=False):
Fredrik Lundhb9056332001-07-11 17:42:21 +00001376 # establish a "logical" server connection
1377
1378 # get the url
Jeremy Hylton1afc1692008-06-18 20:49:58 +00001379 import urllib.parse
1380 type, uri = urllib.parse.splittype(uri)
Fredrik Lundhb9056332001-07-11 17:42:21 +00001381 if type not in ("http", "https"):
Collin Winterce36ad82007-08-30 01:19:48 +00001382 raise IOError("unsupported XML-RPC protocol")
Jeremy Hylton1afc1692008-06-18 20:49:58 +00001383 self.__host, self.__handler = urllib.parse.splithost(uri)
Fredrik Lundhb9056332001-07-11 17:42:21 +00001384 if not self.__handler:
1385 self.__handler = "/RPC2"
1386
1387 if transport is None:
1388 if type == "https":
Skip Montanaro174dd222005-05-14 20:54:16 +00001389 transport = SafeTransport(use_datetime=use_datetime)
Fredrik Lundhb9056332001-07-11 17:42:21 +00001390 else:
Skip Montanaro174dd222005-05-14 20:54:16 +00001391 transport = Transport(use_datetime=use_datetime)
Fredrik Lundhb9056332001-07-11 17:42:21 +00001392 self.__transport = transport
1393
Senthil Kumaranb3af08f2009-04-01 20:20:43 +00001394 self.__encoding = encoding or 'utf-8'
Fredrik Lundhb9056332001-07-11 17:42:21 +00001395 self.__verbose = verbose
Andrew M. Kuchlinga4c2b742003-04-25 00:26:51 +00001396 self.__allow_none = allow_none
Tim Petersc2659cf2003-05-12 20:19:37 +00001397
Kristján Valur Jónsson985fc6a2009-07-01 10:01:31 +00001398 def __close(self):
1399 self.__transport.close()
1400
Fredrik Lundhb9056332001-07-11 17:42:21 +00001401 def __request(self, methodname, params):
1402 # call a method on the remote server
1403
Andrew M. Kuchlinga4c2b742003-04-25 00:26:51 +00001404 request = dumps(params, methodname, encoding=self.__encoding,
Senthil Kumaranb3af08f2009-04-01 20:20:43 +00001405 allow_none=self.__allow_none).encode(self.__encoding)
Fredrik Lundhb9056332001-07-11 17:42:21 +00001406
1407 response = self.__transport.request(
1408 self.__host,
1409 self.__handler,
1410 request,
1411 verbose=self.__verbose
1412 )
1413
1414 if len(response) == 1:
1415 response = response[0]
1416
1417 return response
1418
1419 def __repr__(self):
1420 return (
Fredrik Lundh78eedce2001-08-23 20:04:33 +00001421 "<ServerProxy for %s%s>" %
Fredrik Lundhb9056332001-07-11 17:42:21 +00001422 (self.__host, self.__handler)
1423 )
1424
1425 __str__ = __repr__
Raymond Hettingercc523fc2003-11-02 09:47:05 +00001426
Fredrik Lundhb9056332001-07-11 17:42:21 +00001427 def __getattr__(self, name):
1428 # magic method dispatcher
1429 return _Method(self.__request, name)
1430
1431 # note: to call a remote object with an non-standard name, use
1432 # result getattr(server, "strange-python-name")(args)
1433
Kristján Valur Jónsson985fc6a2009-07-01 10:01:31 +00001434 def __call__(self, attr):
1435 """A workaround to get special attributes on the ServerProxy
1436 without interfering with the magic __getattr__
1437 """
1438 if attr == "close":
1439 return self.__close
1440 elif attr == "transport":
1441 return self.__transport
1442 raise AttributeError("Attribute %r not found" % (attr,))
1443
Fredrik Lundh78eedce2001-08-23 20:04:33 +00001444# compatibility
Fredrik Lundh3d9addd2002-06-27 21:36:21 +00001445
Fredrik Lundhb9056332001-07-11 17:42:21 +00001446Server = ServerProxy
1447
1448# --------------------------------------------------------------------
1449# test code
1450
1451if __name__ == "__main__":
1452
1453 # simple test program (from the XML-RPC specification)
1454
Fredrik Lundh78eedce2001-08-23 20:04:33 +00001455 # server = ServerProxy("http://localhost:8000") # local server
Martin v. Löwis12237b32004-08-22 16:04:50 +00001456 server = ServerProxy("http://time.xmlrpc.com/RPC2")
Fredrik Lundhb9056332001-07-11 17:42:21 +00001457
Fredrik Lundhb9056332001-07-11 17:42:21 +00001458 try:
Guido van Rossumbe19ed72007-02-09 05:37:30 +00001459 print(server.currentTime.getCurrentTime())
Guido van Rossumb940e112007-01-10 16:19:56 +00001460 except Error as v:
Guido van Rossumbe19ed72007-02-09 05:37:30 +00001461 print("ERROR", v)
Martin v. Löwis12237b32004-08-22 16:04:50 +00001462
Jeremy Hylton5d8a88a2007-08-14 16:47:39 +00001463 # The server at xmlrpc.com doesn't seem to support multicall anymore.
Martin v. Löwis12237b32004-08-22 16:04:50 +00001464 multi = MultiCall(server)
1465 multi.currentTime.getCurrentTime()
1466 multi.currentTime.getCurrentTime()
1467 try:
1468 for response in multi():
Guido van Rossumbe19ed72007-02-09 05:37:30 +00001469 print(response)
Guido van Rossumb940e112007-01-10 16:19:56 +00001470 except Error as v:
Guido van Rossumbe19ed72007-02-09 05:37:30 +00001471 print("ERROR", v)