openmedialibrary_platform/Darwin/lib/python3.5/xmlrpc/client.py

1498 lines
47 KiB
Python
Raw Normal View History

2013-10-11 17:28:32 +00:00
#
# XML-RPC CLIENT LIBRARY
# $Id$
#
# an XML-RPC client interface for Python.
#
# the marshalling and response parser code can also be used to
# implement XML-RPC servers.
#
# Notes:
# this version is designed to work with Python 2.1 or newer.
#
# History:
# 1999-01-14 fl Created
# 1999-01-15 fl Changed dateTime to use localtime
# 1999-01-16 fl Added Binary/base64 element, default to RPC2 service
# 1999-01-19 fl Fixed array data element (from Skip Montanaro)
# 1999-01-21 fl Fixed dateTime constructor, etc.
# 1999-02-02 fl Added fault handling, handle empty sequences, etc.
# 1999-02-10 fl Fixed problem with empty responses (from Skip Montanaro)
# 1999-06-20 fl Speed improvements, pluggable parsers/transports (0.9.8)
# 2000-11-28 fl Changed boolean to check the truth value of its argument
# 2001-02-24 fl Added encoding/Unicode/SafeTransport patches
# 2001-02-26 fl Added compare support to wrappers (0.9.9/1.0b1)
# 2001-03-28 fl Make sure response tuple is a singleton
# 2001-03-29 fl Don't require empty params element (from Nicholas Riley)
# 2001-06-10 fl Folded in _xmlrpclib accelerator support (1.0b2)
# 2001-08-20 fl Base xmlrpclib.Error on built-in Exception (from Paul Prescod)
# 2001-09-03 fl Allow Transport subclass to override getparser
# 2001-09-10 fl Lazy import of urllib, cgi, xmllib (20x import speedup)
# 2001-10-01 fl Remove containers from memo cache when done with them
# 2001-10-01 fl Use faster escape method (80% dumps speedup)
# 2001-10-02 fl More dumps microtuning
# 2001-10-04 fl Make sure import expat gets a parser (from Guido van Rossum)
# 2001-10-10 sm Allow long ints to be passed as ints if they don't overflow
# 2001-10-17 sm Test for int and long overflow (allows use on 64-bit systems)
# 2001-11-12 fl Use repr() to marshal doubles (from Paul Felix)
# 2002-03-17 fl Avoid buffered read when possible (from James Rucker)
# 2002-04-07 fl Added pythondoc comments
# 2002-04-16 fl Added __str__ methods to datetime/binary wrappers
# 2002-05-15 fl Added error constants (from Andrew Kuchling)
# 2002-06-27 fl Merged with Python CVS version
# 2002-10-22 fl Added basic authentication (based on code from Phillip Eby)
# 2003-01-22 sm Add support for the bool type
# 2003-02-27 gvr Remove apply calls
# 2003-04-24 sm Use cStringIO if available
# 2003-04-25 ak Add support for nil
# 2003-06-15 gn Add support for time.struct_time
# 2003-07-12 gp Correct marshalling of Faults
# 2003-10-31 mvl Add multicall support
# 2004-08-20 mvl Bump minimum supported Python version to 2.1
2015-11-25 01:25:01 +00:00
# 2014-12-02 ch/doko Add workaround for gzip bomb vulnerability
2013-10-11 17:28:32 +00:00
#
# Copyright (c) 1999-2002 by Secret Labs AB.
# Copyright (c) 1999-2002 by Fredrik Lundh.
#
# info@pythonware.com
# http://www.pythonware.com
#
# --------------------------------------------------------------------
# The XML-RPC client interface is
#
# Copyright (c) 1999-2002 by Secret Labs AB
# Copyright (c) 1999-2002 by Fredrik Lundh
#
# By obtaining, using, and/or copying this software and/or its
# associated documentation, you agree that you have read, understood,
# and will comply with the following terms and conditions:
#
# Permission to use, copy, modify, and distribute this software and
# its associated documentation for any purpose and without fee is
# hereby granted, provided that the above copyright notice appears in
# all copies, and that both that copyright notice and this permission
# notice appear in supporting documentation, and that the name of
# Secret Labs AB or the author not be used in advertising or publicity
# pertaining to distribution of the software without specific, written
# prior permission.
#
# SECRET LABS AB AND THE AUTHOR DISCLAIMS ALL WARRANTIES WITH REGARD
# TO THIS SOFTWARE, INCLUDING ALL IMPLIED WARRANTIES OF MERCHANT-
# ABILITY AND FITNESS. IN NO EVENT SHALL SECRET LABS AB OR THE AUTHOR
# BE LIABLE FOR ANY SPECIAL, INDIRECT OR CONSEQUENTIAL DAMAGES OR ANY
# DAMAGES WHATSOEVER RESULTING FROM LOSS OF USE, DATA OR PROFITS,
# WHETHER IN AN ACTION OF CONTRACT, NEGLIGENCE OR OTHER TORTIOUS
# ACTION, ARISING OUT OF OR IN CONNECTION WITH THE USE OR PERFORMANCE
# OF THIS SOFTWARE.
# --------------------------------------------------------------------
"""
An XML-RPC client interface for Python.
The marshalling and response parser code can also be used to
implement XML-RPC servers.
Exported exceptions:
Error Base class for client errors
ProtocolError Indicates an HTTP protocol error
ResponseError Indicates a broken response package
Fault Indicates an XML-RPC fault package
Exported classes:
ServerProxy Represents a logical connection to an XML-RPC server
MultiCall Executor of boxcared xmlrpc requests
DateTime dateTime wrapper for an ISO 8601 string or time tuple or
localtime integer value to generate a "dateTime.iso8601"
XML-RPC value
Binary binary data wrapper
Marshaller Generate an XML-RPC params chunk from a Python data structure
Unmarshaller Unmarshal an XML-RPC response from incoming XML event message
Transport Handles an HTTP transaction to an XML-RPC server
SafeTransport Handles an HTTPS transaction to an XML-RPC server
Exported constants:
2014-09-30 16:15:32 +00:00
(none)
2013-10-11 17:28:32 +00:00
Exported functions:
getparser Create instance of the fastest available parser & attach
to an unmarshalling object
dumps Convert an argument tuple or a Fault instance to an XML-RPC
request (or response, if the methodresponse option is used).
loads Convert an XML-RPC packet to unmarshalled data plus a method
name (None if not present).
"""
2014-09-30 16:15:32 +00:00
import base64
import sys
import time
from datetime import datetime
import http.client
import urllib.parse
from xml.parsers import expat
2013-10-11 17:28:32 +00:00
import errno
2014-09-30 16:15:32 +00:00
from io import BytesIO
2013-10-11 17:28:32 +00:00
try:
import gzip
except ImportError:
gzip = None #python can be built without zlib/gzip support
# --------------------------------------------------------------------
# Internal stuff
2014-09-30 16:15:32 +00:00
def escape(s):
s = s.replace("&", "&")
s = s.replace("<", "&lt;")
return s.replace(">", "&gt;",)
2013-10-11 17:28:32 +00:00
2014-09-30 16:15:32 +00:00
# used in User-Agent header sent
__version__ = sys.version[:3]
2013-10-11 17:28:32 +00:00
# xmlrpc integer limits
2014-09-30 16:15:32 +00:00
MAXINT = 2**31-1
MININT = -2**31
2013-10-11 17:28:32 +00:00
# --------------------------------------------------------------------
# Error constants (from Dan Libby's specification at
# http://xmlrpc-epi.sourceforge.net/specs/rfc.fault_codes.php)
# Ranges of errors
PARSE_ERROR = -32700
SERVER_ERROR = -32600
APPLICATION_ERROR = -32500
SYSTEM_ERROR = -32400
TRANSPORT_ERROR = -32300
# Specific errors
NOT_WELLFORMED_ERROR = -32700
UNSUPPORTED_ENCODING = -32701
INVALID_ENCODING_CHAR = -32702
INVALID_XMLRPC = -32600
METHOD_NOT_FOUND = -32601
INVALID_METHOD_PARAMS = -32602
INTERNAL_ERROR = -32603
# --------------------------------------------------------------------
# Exceptions
##
# Base class for all kinds of client-side errors.
class Error(Exception):
"""Base class for client errors."""
def __str__(self):
return repr(self)
##
# Indicates an HTTP-level protocol error. This is raised by the HTTP
# transport layer, if the server returns an error code other than 200
# (OK).
#
# @param url The target URL.
# @param errcode The HTTP error code.
# @param errmsg The HTTP error message.
# @param headers The HTTP header dictionary.
class ProtocolError(Error):
"""Indicates an HTTP protocol error."""
def __init__(self, url, errcode, errmsg, headers):
Error.__init__(self)
self.url = url
self.errcode = errcode
self.errmsg = errmsg
self.headers = headers
def __repr__(self):
return (
2015-11-25 01:25:01 +00:00
"<%s for %s: %s %s>" %
(self.__class__.__name__, self.url, self.errcode, self.errmsg)
2013-10-11 17:28:32 +00:00
)
##
# Indicates a broken XML-RPC response package. This exception is
# raised by the unmarshalling layer, if the XML-RPC response is
# malformed.
class ResponseError(Error):
"""Indicates a broken response package."""
pass
##
# Indicates an XML-RPC fault response package. This exception is
# raised by the unmarshalling layer, if the XML-RPC response contains
2014-09-30 16:15:32 +00:00
# a fault string. This exception can also be used as a class, to
2013-10-11 17:28:32 +00:00
# generate a fault XML-RPC message.
#
# @param faultCode The XML-RPC fault code.
# @param faultString The XML-RPC fault string.
class Fault(Error):
"""Indicates an XML-RPC fault package."""
def __init__(self, faultCode, faultString, **extra):
Error.__init__(self)
self.faultCode = faultCode
self.faultString = faultString
def __repr__(self):
2015-11-25 01:25:01 +00:00
return "<%s %s: %r>" % (self.__class__.__name__,
self.faultCode, self.faultString)
2013-10-11 17:28:32 +00:00
# --------------------------------------------------------------------
# Special values
##
2014-09-30 16:15:32 +00:00
# Backwards compatibility
2013-10-11 17:28:32 +00:00
2014-09-30 16:15:32 +00:00
boolean = Boolean = bool
2013-10-11 17:28:32 +00:00
##
# Wrapper for XML-RPC DateTime values. This converts a time value to
# the format used by XML-RPC.
# <p>
2014-09-30 16:15:32 +00:00
# The value can be given as a datetime object, as a string in the
# format "yyyymmddThh:mm:ss", as a 9-item time tuple (as returned by
2013-10-11 17:28:32 +00:00
# time.localtime()), or an integer value (as returned by time.time()).
# The wrapper uses time.localtime() to convert an integer to a time
# tuple.
#
2014-09-30 16:15:32 +00:00
# @param value The time, given as a datetime object, an ISO 8601 string,
# a time tuple, or an integer time value.
# Issue #13305: different format codes across platforms
_day0 = datetime(1, 1, 1)
if _day0.strftime('%Y') == '0001': # Mac OS X
def _iso8601_format(value):
return value.strftime("%Y%m%dT%H:%M:%S")
elif _day0.strftime('%4Y') == '0001': # Linux
def _iso8601_format(value):
return value.strftime("%4Y%m%dT%H:%M:%S")
else:
def _iso8601_format(value):
return value.strftime("%Y%m%dT%H:%M:%S").zfill(17)
del _day0
2013-10-11 17:28:32 +00:00
def _strftime(value):
2014-09-30 16:15:32 +00:00
if isinstance(value, datetime):
return _iso8601_format(value)
2013-10-11 17:28:32 +00:00
2014-09-30 16:15:32 +00:00
if not isinstance(value, (tuple, time.struct_time)):
2013-10-11 17:28:32 +00:00
if value == 0:
value = time.time()
value = time.localtime(value)
return "%04d%02d%02dT%02d:%02d:%02d" % value[:6]
class DateTime:
"""DateTime wrapper for an ISO 8601 string or time tuple or
localtime integer value to generate 'dateTime.iso8601' XML-RPC
value.
"""
def __init__(self, value=0):
2014-09-30 16:15:32 +00:00
if isinstance(value, str):
2013-10-11 17:28:32 +00:00
self.value = value
else:
self.value = _strftime(value)
def make_comparable(self, other):
if isinstance(other, DateTime):
s = self.value
o = other.value
2014-09-30 16:15:32 +00:00
elif isinstance(other, datetime):
2013-10-11 17:28:32 +00:00
s = self.value
2014-09-30 16:15:32 +00:00
o = _iso8601_format(other)
elif isinstance(other, str):
2013-10-11 17:28:32 +00:00
s = self.value
o = other
elif hasattr(other, "timetuple"):
s = self.timetuple()
o = other.timetuple()
else:
otype = (hasattr(other, "__class__")
and other.__class__.__name__
or type(other))
raise TypeError("Can't compare %s and %s" %
(self.__class__.__name__, otype))
return s, o
def __lt__(self, other):
s, o = self.make_comparable(other)
return s < o
def __le__(self, other):
s, o = self.make_comparable(other)
return s <= o
def __gt__(self, other):
s, o = self.make_comparable(other)
return s > o
def __ge__(self, other):
s, o = self.make_comparable(other)
return s >= o
def __eq__(self, other):
s, o = self.make_comparable(other)
return s == o
def timetuple(self):
return time.strptime(self.value, "%Y%m%dT%H:%M:%S")
##
# Get date/time value.
#
# @return Date/time value, as an ISO 8601 string.
def __str__(self):
return self.value
def __repr__(self):
2015-11-25 01:25:01 +00:00
return "<%s %r at %#x>" % (self.__class__.__name__, self.value, id(self))
2013-10-11 17:28:32 +00:00
def decode(self, data):
2014-09-30 16:15:32 +00:00
self.value = str(data).strip()
2013-10-11 17:28:32 +00:00
def encode(self, out):
out.write("<value><dateTime.iso8601>")
out.write(self.value)
out.write("</dateTime.iso8601></value>\n")
def _datetime(data):
# decode xml element contents into a DateTime structure.
value = DateTime()
value.decode(data)
return value
def _datetime_type(data):
2014-09-30 16:15:32 +00:00
return datetime.strptime(data, "%Y%m%dT%H:%M:%S")
2013-10-11 17:28:32 +00:00
##
# Wrapper for binary data. This can be used to transport any kind
# of binary data over XML-RPC, using BASE64 encoding.
#
# @param data An 8-bit string containing arbitrary data.
class Binary:
"""Wrapper for binary data."""
def __init__(self, data=None):
2014-09-30 16:15:32 +00:00
if data is None:
data = b""
else:
if not isinstance(data, (bytes, bytearray)):
raise TypeError("expected bytes or bytearray, not %s" %
data.__class__.__name__)
data = bytes(data) # Make a copy of the bytes!
2013-10-11 17:28:32 +00:00
self.data = data
##
# Get buffer contents.
#
# @return Buffer contents, as an 8-bit string.
def __str__(self):
2014-09-30 16:15:32 +00:00
return str(self.data, "latin-1") # XXX encoding?!
def __eq__(self, other):
if isinstance(other, Binary):
other = other.data
return self.data == other
2013-10-11 17:28:32 +00:00
def decode(self, data):
2014-09-30 16:15:32 +00:00
self.data = base64.decodebytes(data)
2013-10-11 17:28:32 +00:00
def encode(self, out):
out.write("<value><base64>\n")
2014-09-30 16:15:32 +00:00
encoded = base64.encodebytes(self.data)
out.write(encoded.decode('ascii'))
2013-10-11 17:28:32 +00:00
out.write("</base64></value>\n")
def _binary(data):
# decode xml element contents into a Binary structure
value = Binary()
value.decode(data)
return value
WRAPPERS = (DateTime, Binary)
# --------------------------------------------------------------------
# XML parsers
2014-09-30 16:15:32 +00:00
class ExpatParser:
# fast expat parser for Python 2.0 and later.
def __init__(self, target):
self._parser = parser = expat.ParserCreate(None, None)
self._target = target
parser.StartElementHandler = target.start
parser.EndElementHandler = target.end
parser.CharacterDataHandler = target.data
encoding = None
target.xml(encoding, None)
2013-10-11 17:28:32 +00:00
2014-09-30 16:15:32 +00:00
def feed(self, data):
self._parser.Parse(data, 0)
2013-10-11 17:28:32 +00:00
2014-09-30 16:15:32 +00:00
def close(self):
2015-11-25 01:25:01 +00:00
try:
parser = self._parser
except AttributeError:
pass
else:
del self._target, self._parser # get rid of circular references
parser.Parse(b"", True) # end of data
2013-10-11 17:28:32 +00:00
# --------------------------------------------------------------------
# XML-RPC marshalling and unmarshalling code
##
# XML-RPC marshaller.
#
# @param encoding Default encoding for 8-bit strings. The default
# value is None (interpreted as UTF-8).
# @see dumps
class Marshaller:
"""Generate an XML-RPC params chunk from a Python data structure.
Create a Marshaller instance for each set of parameters, and use
the "dumps" method to convert your data (represented as a tuple)
to an XML-RPC params chunk. To write a fault response, pass a
Fault instance instead. You may prefer to use the "dumps" module
function for this purpose.
"""
# by the way, if you don't understand what's going on in here,
# that's perfectly ok.
2014-09-30 16:15:32 +00:00
def __init__(self, encoding=None, allow_none=False):
2013-10-11 17:28:32 +00:00
self.memo = {}
self.data = None
self.encoding = encoding
self.allow_none = allow_none
dispatch = {}
def dumps(self, values):
out = []
write = out.append
dump = self.__dump
if isinstance(values, Fault):
# fault instance
write("<fault>\n")
dump({'faultCode': values.faultCode,
'faultString': values.faultString},
write)
write("</fault>\n")
else:
# parameter block
# FIXME: the xml-rpc specification allows us to leave out
# the entire <params> block if there are no parameters.
# however, changing this may break older code (including
# old versions of xmlrpclib.py), so this is better left as
# is for now. See @XMLRPC3 for more information. /F
write("<params>\n")
for v in values:
write("<param>\n")
dump(v, write)
write("</param>\n")
write("</params>\n")
2014-09-30 16:15:32 +00:00
result = "".join(out)
2013-10-11 17:28:32 +00:00
return result
def __dump(self, value, write):
try:
f = self.dispatch[type(value)]
except KeyError:
# check if this object can be marshalled as a structure
2014-09-30 16:15:32 +00:00
if not hasattr(value, '__dict__'):
raise TypeError("cannot marshal %s objects" % type(value))
2013-10-11 17:28:32 +00:00
# check if this class is a sub-class of a basic type,
# because we don't know how to marshal these types
# (e.g. a string sub-class)
for type_ in type(value).__mro__:
if type_ in self.dispatch.keys():
2014-09-30 16:15:32 +00:00
raise TypeError("cannot marshal %s objects" % type(value))
# XXX(twouters): using "_arbitrary_instance" as key as a quick-fix
# for the p3yk merge, this should probably be fixed more neatly.
f = self.dispatch["_arbitrary_instance"]
2013-10-11 17:28:32 +00:00
f(self, value, write)
def dump_nil (self, value, write):
if not self.allow_none:
2014-09-30 16:15:32 +00:00
raise TypeError("cannot marshal None unless allow_none is enabled")
2013-10-11 17:28:32 +00:00
write("<value><nil/></value>")
2014-09-30 16:15:32 +00:00
dispatch[type(None)] = dump_nil
2013-10-11 17:28:32 +00:00
2014-09-30 16:15:32 +00:00
def dump_bool(self, value, write):
write("<value><boolean>")
write(value and "1" or "0")
write("</boolean></value>\n")
dispatch[bool] = dump_bool
2013-10-11 17:28:32 +00:00
def dump_long(self, value, write):
if value > MAXINT or value < MININT:
2014-09-30 16:15:32 +00:00
raise OverflowError("int exceeds XML-RPC limits")
2013-10-11 17:28:32 +00:00
write("<value><int>")
write(str(int(value)))
write("</int></value>\n")
2014-09-30 16:15:32 +00:00
dispatch[int] = dump_long
# backward compatible
dump_int = dump_long
2013-10-11 17:28:32 +00:00
def dump_double(self, value, write):
write("<value><double>")
write(repr(value))
write("</double></value>\n")
2014-09-30 16:15:32 +00:00
dispatch[float] = dump_double
2013-10-11 17:28:32 +00:00
2014-09-30 16:15:32 +00:00
def dump_unicode(self, value, write, escape=escape):
2013-10-11 17:28:32 +00:00
write("<value><string>")
write(escape(value))
write("</string></value>\n")
2014-09-30 16:15:32 +00:00
dispatch[str] = dump_unicode
2013-10-11 17:28:32 +00:00
2014-09-30 16:15:32 +00:00
def dump_bytes(self, value, write):
write("<value><base64>\n")
encoded = base64.encodebytes(value)
write(encoded.decode('ascii'))
write("</base64></value>\n")
dispatch[bytes] = dump_bytes
dispatch[bytearray] = dump_bytes
2013-10-11 17:28:32 +00:00
def dump_array(self, value, write):
i = id(value)
if i in self.memo:
2014-09-30 16:15:32 +00:00
raise TypeError("cannot marshal recursive sequences")
2013-10-11 17:28:32 +00:00
self.memo[i] = None
dump = self.__dump
write("<value><array><data>\n")
for v in value:
dump(v, write)
write("</data></array></value>\n")
del self.memo[i]
2014-09-30 16:15:32 +00:00
dispatch[tuple] = dump_array
dispatch[list] = dump_array
2013-10-11 17:28:32 +00:00
def dump_struct(self, value, write, escape=escape):
i = id(value)
if i in self.memo:
2014-09-30 16:15:32 +00:00
raise TypeError("cannot marshal recursive dictionaries")
2013-10-11 17:28:32 +00:00
self.memo[i] = None
dump = self.__dump
write("<value><struct>\n")
for k, v in value.items():
write("<member>\n")
2014-09-30 16:15:32 +00:00
if not isinstance(k, str):
raise TypeError("dictionary key must be string")
2013-10-11 17:28:32 +00:00
write("<name>%s</name>\n" % escape(k))
dump(v, write)
write("</member>\n")
write("</struct></value>\n")
del self.memo[i]
2014-09-30 16:15:32 +00:00
dispatch[dict] = dump_struct
2013-10-11 17:28:32 +00:00
2014-09-30 16:15:32 +00:00
def dump_datetime(self, value, write):
write("<value><dateTime.iso8601>")
write(_strftime(value))
write("</dateTime.iso8601></value>\n")
dispatch[datetime] = dump_datetime
2013-10-11 17:28:32 +00:00
def dump_instance(self, value, write):
# check for special wrappers
if value.__class__ in WRAPPERS:
self.write = write
value.encode(self)
del self.write
else:
# store instance attributes as a struct (really?)
self.dump_struct(value.__dict__, write)
2014-09-30 16:15:32 +00:00
dispatch[DateTime] = dump_instance
dispatch[Binary] = dump_instance
# XXX(twouters): using "_arbitrary_instance" as key as a quick-fix
# for the p3yk merge, this should probably be fixed more neatly.
dispatch["_arbitrary_instance"] = dump_instance
2013-10-11 17:28:32 +00:00
##
# XML-RPC unmarshaller.
#
# @see loads
class Unmarshaller:
"""Unmarshal an XML-RPC response, based on incoming XML event
messages (start, data, end). Call close() to get the resulting
data structure.
Note that this reader is fairly tolerant, and gladly accepts bogus
XML-RPC data without complaining (but not bogus XML).
"""
# and again, if you don't understand what's going on in here,
# that's perfectly ok.
2014-09-30 16:15:32 +00:00
def __init__(self, use_datetime=False, use_builtin_types=False):
2013-10-11 17:28:32 +00:00
self._type = None
self._stack = []
self._marks = []
self._data = []
self._methodname = None
self._encoding = "utf-8"
self.append = self._stack.append
2014-09-30 16:15:32 +00:00
self._use_datetime = use_builtin_types or use_datetime
self._use_bytes = use_builtin_types
2013-10-11 17:28:32 +00:00
def close(self):
# return response tuple and target method
if self._type is None or self._marks:
raise ResponseError()
if self._type == "fault":
raise Fault(**self._stack[0])
return tuple(self._stack)
def getmethodname(self):
return self._methodname
#
# event handlers
def xml(self, encoding, standalone):
self._encoding = encoding
# FIXME: assert standalone == 1 ???
def start(self, tag, attrs):
# prepare to handle this element
if tag == "array" or tag == "struct":
self._marks.append(len(self._stack))
self._data = []
self._value = (tag == "value")
def data(self, text):
self._data.append(text)
2014-09-30 16:15:32 +00:00
def end(self, tag):
2013-10-11 17:28:32 +00:00
# call the appropriate end tag handler
try:
f = self.dispatch[tag]
except KeyError:
pass # unknown tag ?
else:
2014-09-30 16:15:32 +00:00
return f(self, "".join(self._data))
2013-10-11 17:28:32 +00:00
#
# accelerator support
def end_dispatch(self, tag, data):
# dispatch data
try:
f = self.dispatch[tag]
except KeyError:
pass # unknown tag ?
else:
return f(self, data)
#
# element decoders
dispatch = {}
def end_nil (self, data):
self.append(None)
self._value = 0
dispatch["nil"] = end_nil
def end_boolean(self, data):
if data == "0":
self.append(False)
elif data == "1":
self.append(True)
else:
2014-09-30 16:15:32 +00:00
raise TypeError("bad boolean value")
2013-10-11 17:28:32 +00:00
self._value = 0
dispatch["boolean"] = end_boolean
def end_int(self, data):
self.append(int(data))
self._value = 0
dispatch["i4"] = end_int
dispatch["i8"] = end_int
dispatch["int"] = end_int
def end_double(self, data):
self.append(float(data))
self._value = 0
dispatch["double"] = end_double
def end_string(self, data):
if self._encoding:
2014-09-30 16:15:32 +00:00
data = data.decode(self._encoding)
self.append(data)
2013-10-11 17:28:32 +00:00
self._value = 0
dispatch["string"] = end_string
dispatch["name"] = end_string # struct keys are always strings
def end_array(self, data):
mark = self._marks.pop()
# map arrays to Python lists
self._stack[mark:] = [self._stack[mark:]]
self._value = 0
dispatch["array"] = end_array
def end_struct(self, data):
mark = self._marks.pop()
# map structs to Python dictionaries
dict = {}
items = self._stack[mark:]
for i in range(0, len(items), 2):
2014-09-30 16:15:32 +00:00
dict[items[i]] = items[i+1]
2013-10-11 17:28:32 +00:00
self._stack[mark:] = [dict]
self._value = 0
dispatch["struct"] = end_struct
def end_base64(self, data):
value = Binary()
2014-09-30 16:15:32 +00:00
value.decode(data.encode("ascii"))
if self._use_bytes:
value = value.data
2013-10-11 17:28:32 +00:00
self.append(value)
self._value = 0
dispatch["base64"] = end_base64
def end_dateTime(self, data):
value = DateTime()
value.decode(data)
if self._use_datetime:
value = _datetime_type(data)
self.append(value)
dispatch["dateTime.iso8601"] = end_dateTime
def end_value(self, data):
# if we stumble upon a value element with no internal
# elements, treat it as a string element
if self._value:
self.end_string(data)
dispatch["value"] = end_value
def end_params(self, data):
self._type = "params"
dispatch["params"] = end_params
def end_fault(self, data):
self._type = "fault"
dispatch["fault"] = end_fault
def end_methodName(self, data):
if self._encoding:
2014-09-30 16:15:32 +00:00
data = data.decode(self._encoding)
2013-10-11 17:28:32 +00:00
self._methodname = data
self._type = "methodName" # no params
dispatch["methodName"] = end_methodName
## Multicall support
#
class _MultiCallMethod:
# some lesser magic to store calls made to a MultiCall object
# for batch execution
def __init__(self, call_list, name):
self.__call_list = call_list
self.__name = name
def __getattr__(self, name):
return _MultiCallMethod(self.__call_list, "%s.%s" % (self.__name, name))
def __call__(self, *args):
self.__call_list.append((self.__name, args))
class MultiCallIterator:
"""Iterates over the results of a multicall. Exceptions are
raised in response to xmlrpc faults."""
def __init__(self, results):
self.results = results
def __getitem__(self, i):
item = self.results[i]
if type(item) == type({}):
raise Fault(item['faultCode'], item['faultString'])
elif type(item) == type([]):
return item[0]
else:
2014-09-30 16:15:32 +00:00
raise ValueError("unexpected type in multicall result")
2013-10-11 17:28:32 +00:00
class MultiCall:
"""server -> a object used to boxcar method calls
server should be a ServerProxy object.
Methods can be added to the MultiCall using normal
method call syntax e.g.:
multicall = MultiCall(server_proxy)
multicall.add(2,3)
multicall.get_address("Guido")
To execute the multicall, call the MultiCall object e.g.:
add_result, address = multicall()
"""
def __init__(self, server):
self.__server = server
self.__call_list = []
def __repr__(self):
2015-11-25 01:25:01 +00:00
return "<%s at %#x>" % (self.__class__.__name__, id(self))
2013-10-11 17:28:32 +00:00
__str__ = __repr__
def __getattr__(self, name):
return _MultiCallMethod(self.__call_list, name)
def __call__(self):
marshalled_list = []
for name, args in self.__call_list:
marshalled_list.append({'methodName' : name, 'params' : args})
return MultiCallIterator(self.__server.system.multicall(marshalled_list))
# --------------------------------------------------------------------
# convenience functions
2014-09-30 16:15:32 +00:00
FastMarshaller = FastParser = FastUnmarshaller = None
2013-10-11 17:28:32 +00:00
##
# Create a parser object, and connect it to an unmarshalling instance.
# This function picks the fastest available XML parser.
#
# return A (parser, unmarshaller) tuple.
2014-09-30 16:15:32 +00:00
def getparser(use_datetime=False, use_builtin_types=False):
2013-10-11 17:28:32 +00:00
"""getparser() -> parser, unmarshaller
Create an instance of the fastest available parser, and attach it
to an unmarshalling object. Return both objects.
"""
if FastParser and FastUnmarshaller:
2014-09-30 16:15:32 +00:00
if use_builtin_types:
mkdatetime = _datetime_type
mkbytes = base64.decodebytes
elif use_datetime:
2013-10-11 17:28:32 +00:00
mkdatetime = _datetime_type
2014-09-30 16:15:32 +00:00
mkbytes = _binary
2013-10-11 17:28:32 +00:00
else:
mkdatetime = _datetime
2014-09-30 16:15:32 +00:00
mkbytes = _binary
target = FastUnmarshaller(True, False, mkbytes, mkdatetime, Fault)
2013-10-11 17:28:32 +00:00
parser = FastParser(target)
else:
2014-09-30 16:15:32 +00:00
target = Unmarshaller(use_datetime=use_datetime, use_builtin_types=use_builtin_types)
2013-10-11 17:28:32 +00:00
if FastParser:
parser = FastParser(target)
else:
2014-09-30 16:15:32 +00:00
parser = ExpatParser(target)
2013-10-11 17:28:32 +00:00
return parser, target
##
# Convert a Python tuple or a Fault instance to an XML-RPC packet.
#
# @def dumps(params, **options)
# @param params A tuple or Fault instance.
# @keyparam methodname If given, create a methodCall request for
# this method name.
# @keyparam methodresponse If given, create a methodResponse packet.
# If used with a tuple, the tuple must be a singleton (that is,
# it must contain exactly one element).
# @keyparam encoding The packet encoding.
# @return A string containing marshalled data.
def dumps(params, methodname=None, methodresponse=None, encoding=None,
2014-09-30 16:15:32 +00:00
allow_none=False):
2013-10-11 17:28:32 +00:00
"""data [,options] -> marshalled data
Convert an argument tuple or a Fault instance to an XML-RPC
request (or response, if the methodresponse option is used).
In addition to the data object, the following options can be given
as keyword arguments:
methodname: the method name for a methodCall packet
methodresponse: true to create a methodResponse packet.
If this option is used with a tuple, the tuple must be
a singleton (i.e. it can contain only one element).
encoding: the packet encoding (default is UTF-8)
2014-09-30 16:15:32 +00:00
All byte strings in the data structure are assumed to use the
2013-10-11 17:28:32 +00:00
packet encoding. Unicode strings are automatically converted,
where necessary.
"""
2014-09-30 16:15:32 +00:00
assert isinstance(params, (tuple, Fault)), "argument must be tuple or Fault instance"
2013-10-11 17:28:32 +00:00
if isinstance(params, Fault):
methodresponse = 1
2014-09-30 16:15:32 +00:00
elif methodresponse and isinstance(params, tuple):
2013-10-11 17:28:32 +00:00
assert len(params) == 1, "response tuple must be a singleton"
if not encoding:
encoding = "utf-8"
if FastMarshaller:
m = FastMarshaller(encoding)
else:
m = Marshaller(encoding, allow_none)
data = m.dumps(params)
if encoding != "utf-8":
xmlheader = "<?xml version='1.0' encoding='%s'?>\n" % str(encoding)
else:
xmlheader = "<?xml version='1.0'?>\n" # utf-8 is default
# standard XML-RPC wrappings
if methodname:
# a method call
2014-09-30 16:15:32 +00:00
if not isinstance(methodname, str):
2013-10-11 17:28:32 +00:00
methodname = methodname.encode(encoding)
data = (
xmlheader,
"<methodCall>\n"
"<methodName>", methodname, "</methodName>\n",
data,
"</methodCall>\n"
)
elif methodresponse:
# a method response, or a fault structure
data = (
xmlheader,
"<methodResponse>\n",
data,
"</methodResponse>\n"
)
else:
return data # return as is
2014-09-30 16:15:32 +00:00
return "".join(data)
2013-10-11 17:28:32 +00:00
##
# Convert an XML-RPC packet to a Python object. If the XML-RPC packet
# represents a fault condition, this function raises a Fault exception.
#
# @param data An XML-RPC packet, given as an 8-bit string.
# @return A tuple containing the unpacked data, and the method name
# (None if not present).
# @see Fault
2014-09-30 16:15:32 +00:00
def loads(data, use_datetime=False, use_builtin_types=False):
2013-10-11 17:28:32 +00:00
"""data -> unmarshalled data, method name
Convert an XML-RPC packet to unmarshalled data plus a method
name (None if not present).
If the XML-RPC packet represents a fault condition, this function
raises a Fault exception.
"""
2014-09-30 16:15:32 +00:00
p, u = getparser(use_datetime=use_datetime, use_builtin_types=use_builtin_types)
2013-10-11 17:28:32 +00:00
p.feed(data)
p.close()
return u.close(), u.getmethodname()
##
# Encode a string using the gzip content encoding such as specified by the
# Content-Encoding: gzip
# in the HTTP header, as described in RFC 1952
#
# @param data the unencoded data
# @return the encoded data
def gzip_encode(data):
"""data -> gzip encoded data
Encode data using the gzip content encoding as described in RFC 1952
"""
if not gzip:
raise NotImplementedError
2014-09-30 16:15:32 +00:00
f = BytesIO()
2015-11-25 01:25:01 +00:00
with gzip.GzipFile(mode="wb", fileobj=f, compresslevel=1) as gzf:
gzf.write(data)
return f.getvalue()
2013-10-11 17:28:32 +00:00
##
# Decode a string using the gzip content encoding such as specified by the
# Content-Encoding: gzip
# in the HTTP header, as described in RFC 1952
#
# @param data The encoded data
2015-11-25 01:25:01 +00:00
# @keyparam max_decode Maximum bytes to decode (20MB default), use negative
# values for unlimited decoding
2013-10-11 17:28:32 +00:00
# @return the unencoded data
# @raises ValueError if data is not correctly coded.
2015-11-25 01:25:01 +00:00
# @raises ValueError if max gzipped payload length exceeded
2013-10-11 17:28:32 +00:00
2015-11-25 01:25:01 +00:00
def gzip_decode(data, max_decode=20971520):
2013-10-11 17:28:32 +00:00
"""gzip encoded data -> unencoded data
Decode data using the gzip content encoding as described in RFC 1952
"""
if not gzip:
raise NotImplementedError
2015-11-25 01:25:01 +00:00
with gzip.GzipFile(mode="rb", fileobj=BytesIO(data)) as gzf:
try:
if max_decode < 0: # no limit
decoded = gzf.read()
else:
decoded = gzf.read(max_decode + 1)
except OSError:
raise ValueError("invalid data")
if max_decode >= 0 and len(decoded) > max_decode:
raise ValueError("max gzipped payload length exceeded")
2013-10-11 17:28:32 +00:00
return decoded
##
# Return a decoded file-like object for the gzip encoding
# as described in RFC 1952.
#
# @param response A stream supporting a read() method
# @return a file-like object that the decoded data can be read() from
class GzipDecodedResponse(gzip.GzipFile if gzip else object):
"""a file-like object to decode a response encoded with the gzip
method, as described in RFC 1952.
"""
def __init__(self, response):
#response doesn't support tell() and read(), required by
#GzipFile
if not gzip:
raise NotImplementedError
2014-09-30 16:15:32 +00:00
self.io = BytesIO(response.read())
gzip.GzipFile.__init__(self, mode="rb", fileobj=self.io)
2013-10-11 17:28:32 +00:00
def close(self):
2015-11-25 01:25:01 +00:00
try:
gzip.GzipFile.close(self)
finally:
self.io.close()
2013-10-11 17:28:32 +00:00
# --------------------------------------------------------------------
# request dispatcher
class _Method:
# some magic to bind an XML-RPC method to an RPC server.
# supports "nested" methods (e.g. examples.getStateName)
def __init__(self, send, name):
self.__send = send
self.__name = name
def __getattr__(self, name):
return _Method(self.__send, "%s.%s" % (self.__name, name))
def __call__(self, *args):
return self.__send(self.__name, args)
##
# Standard transport class for XML-RPC over HTTP.
# <p>
# You can create custom transports by subclassing this method, and
# overriding selected methods.
class Transport:
"""Handles an HTTP transaction to an XML-RPC server."""
# client identifier (may be overridden)
2014-09-30 16:15:32 +00:00
user_agent = "Python-xmlrpc/%s" % __version__
2013-10-11 17:28:32 +00:00
#if true, we'll request gzip encoding
accept_gzip_encoding = True
# if positive, encode request using gzip if it exceeds this threshold
# note that many server will get confused, so only use it if you know
# that they can decode such a request
encode_threshold = None #None = don't encode
2014-09-30 16:15:32 +00:00
def __init__(self, use_datetime=False, use_builtin_types=False):
2013-10-11 17:28:32 +00:00
self._use_datetime = use_datetime
2014-09-30 16:15:32 +00:00
self._use_builtin_types = use_builtin_types
2013-10-11 17:28:32 +00:00
self._connection = (None, None)
self._extra_headers = []
2014-09-30 16:15:32 +00:00
2013-10-11 17:28:32 +00:00
##
# Send a complete request, and parse the response.
# Retry request if a cached connection has disconnected.
#
# @param host Target host.
# @param handler Target PRC handler.
# @param request_body XML-RPC request body.
# @param verbose Debugging flag.
# @return Parsed response.
2014-09-30 16:15:32 +00:00
def request(self, host, handler, request_body, verbose=False):
2013-10-11 17:28:32 +00:00
#retry request once if cached connection has gone cold
for i in (0, 1):
try:
return self.single_request(host, handler, request_body, verbose)
2014-09-30 16:15:32 +00:00
except OSError as e:
if i or e.errno not in (errno.ECONNRESET, errno.ECONNABORTED,
errno.EPIPE):
2013-10-11 17:28:32 +00:00
raise
2015-11-25 01:25:01 +00:00
except http.client.RemoteDisconnected:
2013-10-11 17:28:32 +00:00
if i:
raise
2014-09-30 16:15:32 +00:00
def single_request(self, host, handler, request_body, verbose=False):
2013-10-11 17:28:32 +00:00
# issue XML-RPC request
try:
2014-09-30 16:15:32 +00:00
http_conn = self.send_request(host, handler, request_body, verbose)
resp = http_conn.getresponse()
if resp.status == 200:
2013-10-11 17:28:32 +00:00
self.verbose = verbose
2014-09-30 16:15:32 +00:00
return self.parse_response(resp)
2013-10-11 17:28:32 +00:00
except Fault:
raise
except Exception:
2014-09-30 16:15:32 +00:00
#All unexpected errors leave connection in
2013-10-11 17:28:32 +00:00
# a strange state, so we clear it.
self.close()
raise
2014-09-30 16:15:32 +00:00
#We got an error response.
#Discard any response data and raise exception
if resp.getheader("content-length", ""):
resp.read()
2013-10-11 17:28:32 +00:00
raise ProtocolError(
host + handler,
2014-09-30 16:15:32 +00:00
resp.status, resp.reason,
dict(resp.getheaders())
2013-10-11 17:28:32 +00:00
)
2014-09-30 16:15:32 +00:00
2013-10-11 17:28:32 +00:00
##
# Create parser.
#
# @return A 2-tuple containing a parser and a unmarshaller.
def getparser(self):
# get parser and unmarshaller
2014-09-30 16:15:32 +00:00
return getparser(use_datetime=self._use_datetime,
use_builtin_types=self._use_builtin_types)
2013-10-11 17:28:32 +00:00
##
# Get authorization info from host parameter
# Host may be a string, or a (host, x509-dict) tuple; if a string,
# it is checked for a "user:pw@host" format, and a "Basic
# Authentication" header is added if appropriate.
#
# @param host Host descriptor (URL or (URL, x509 info) tuple).
# @return A 3-tuple containing (actual host, extra headers,
# x509 info). The header and x509 fields may be None.
def get_host_info(self, host):
x509 = {}
2014-09-30 16:15:32 +00:00
if isinstance(host, tuple):
2013-10-11 17:28:32 +00:00
host, x509 = host
2014-09-30 16:15:32 +00:00
auth, host = urllib.parse.splituser(host)
2013-10-11 17:28:32 +00:00
if auth:
2014-09-30 16:15:32 +00:00
auth = urllib.parse.unquote_to_bytes(auth)
auth = base64.encodebytes(auth).decode("utf-8")
auth = "".join(auth.split()) # get rid of whitespace
2013-10-11 17:28:32 +00:00
extra_headers = [
("Authorization", "Basic " + auth)
]
else:
2014-09-30 16:15:32 +00:00
extra_headers = []
2013-10-11 17:28:32 +00:00
return host, extra_headers, x509
##
# Connect to server.
#
# @param host Target host.
2014-09-30 16:15:32 +00:00
# @return An HTTPConnection object
2013-10-11 17:28:32 +00:00
def make_connection(self, host):
#return an existing connection if possible. This allows
#HTTP/1.1 keep-alive.
if self._connection and host == self._connection[0]:
return self._connection[1]
# create a HTTP connection object from a host descriptor
chost, self._extra_headers, x509 = self.get_host_info(host)
2014-09-30 16:15:32 +00:00
self._connection = host, http.client.HTTPConnection(chost)
2013-10-11 17:28:32 +00:00
return self._connection[1]
##
# Clear any cached connection object.
# Used in the event of socket errors.
#
def close(self):
2015-11-25 01:25:01 +00:00
host, connection = self._connection
if connection:
2013-10-11 17:28:32 +00:00
self._connection = (None, None)
2015-11-25 01:25:01 +00:00
connection.close()
2013-10-11 17:28:32 +00:00
##
2014-09-30 16:15:32 +00:00
# Send HTTP request.
2013-10-11 17:28:32 +00:00
#
2014-09-30 16:15:32 +00:00
# @param host Host descriptor (URL or (URL, x509 info) tuple).
# @param handler Targer RPC handler (a path relative to host)
# @param request_body The XML-RPC request body
# @param debug Enable debugging if debug is true.
# @return An HTTPConnection.
def send_request(self, host, handler, request_body, debug):
connection = self.make_connection(host)
headers = self._extra_headers[:]
if debug:
connection.set_debuglevel(1)
if self.accept_gzip_encoding and gzip:
2013-10-11 17:28:32 +00:00
connection.putrequest("POST", handler, skip_accept_encoding=True)
2014-09-30 16:15:32 +00:00
headers.append(("Accept-Encoding", "gzip"))
2013-10-11 17:28:32 +00:00
else:
connection.putrequest("POST", handler)
2014-09-30 16:15:32 +00:00
headers.append(("Content-Type", "text/xml"))
headers.append(("User-Agent", self.user_agent))
self.send_headers(connection, headers)
self.send_content(connection, request_body)
return connection
2013-10-11 17:28:32 +00:00
##
2014-09-30 16:15:32 +00:00
# Send request headers.
# This function provides a useful hook for subclassing
2013-10-11 17:28:32 +00:00
#
2014-09-30 16:15:32 +00:00
# @param connection httpConnection.
# @param headers list of key,value pairs for HTTP headers
2013-10-11 17:28:32 +00:00
2014-09-30 16:15:32 +00:00
def send_headers(self, connection, headers):
for key, val in headers:
connection.putheader(key, val)
2013-10-11 17:28:32 +00:00
##
# Send request body.
2014-09-30 16:15:32 +00:00
# This function provides a useful hook for subclassing
2013-10-11 17:28:32 +00:00
#
2014-09-30 16:15:32 +00:00
# @param connection httpConnection.
2013-10-11 17:28:32 +00:00
# @param request_body XML-RPC request body.
def send_content(self, connection, request_body):
#optionally encode the request
if (self.encode_threshold is not None and
self.encode_threshold < len(request_body) and
gzip):
connection.putheader("Content-Encoding", "gzip")
request_body = gzip_encode(request_body)
connection.putheader("Content-Length", str(len(request_body)))
connection.endheaders(request_body)
##
# Parse response.
#
# @param file Stream.
# @return Response tuple and target method.
def parse_response(self, response):
# read response data from httpresponse, and parse it
2014-09-30 16:15:32 +00:00
# Check for new http response object, otherwise it is a file object.
if hasattr(response, 'getheader'):
2013-10-11 17:28:32 +00:00
if response.getheader("Content-Encoding", "") == "gzip":
stream = GzipDecodedResponse(response)
else:
stream = response
else:
stream = response
p, u = self.getparser()
while 1:
data = stream.read(1024)
if not data:
break
if self.verbose:
2014-09-30 16:15:32 +00:00
print("body:", repr(data))
2013-10-11 17:28:32 +00:00
p.feed(data)
if stream is not response:
stream.close()
p.close()
return u.close()
##
# Standard transport class for XML-RPC over HTTPS.
class SafeTransport(Transport):
"""Handles an HTTPS transaction to an XML-RPC server."""
2015-11-25 01:25:01 +00:00
def __init__(self, use_datetime=False, use_builtin_types=False, *,
context=None):
super().__init__(use_datetime=use_datetime, use_builtin_types=use_builtin_types)
self.context = context
2013-10-11 17:28:32 +00:00
# FIXME: mostly untested
def make_connection(self, host):
if self._connection and host == self._connection[0]:
return self._connection[1]
2014-09-30 16:15:32 +00:00
if not hasattr(http.client, "HTTPSConnection"):
raise NotImplementedError(
"your version of http.client doesn't support HTTPS")
2013-10-11 17:28:32 +00:00
# create a HTTPS connection object from a host descriptor
# host may be a string, or a (host, x509-dict) tuple
2014-09-30 16:15:32 +00:00
chost, self._extra_headers, x509 = self.get_host_info(host)
self._connection = host, http.client.HTTPSConnection(chost,
2015-11-25 01:25:01 +00:00
None, context=self.context, **(x509 or {}))
2014-09-30 16:15:32 +00:00
return self._connection[1]
2013-10-11 17:28:32 +00:00
##
# Standard server proxy. This class establishes a virtual connection
# to an XML-RPC server.
# <p>
# This class is available as ServerProxy and Server. New code should
# use ServerProxy, to avoid confusion.
#
# @def ServerProxy(uri, **options)
# @param uri The connection point on the server.
# @keyparam transport A transport factory, compatible with the
# standard transport class.
# @keyparam encoding The default encoding used for 8-bit strings
# (default is UTF-8).
# @keyparam verbose Use a true value to enable debugging output.
# (printed to standard output).
# @see Transport
class ServerProxy:
"""uri [,options] -> a logical connection to an XML-RPC server
uri is the connection point on the server, given as
scheme://host/target.
The standard implementation always supports the "http" scheme. If
SSL socket support is available (Python 2.0), it also supports
"https".
If the target part and the slash preceding it are both omitted,
"/RPC2" is assumed.
The following options can be given as keyword arguments:
transport: a transport factory
encoding: the request encoding (default is UTF-8)
All 8-bit strings passed to the server proxy are assumed to use
the given encoding.
"""
2014-09-30 16:15:32 +00:00
def __init__(self, uri, transport=None, encoding=None, verbose=False,
2015-11-25 01:25:01 +00:00
allow_none=False, use_datetime=False, use_builtin_types=False,
*, context=None):
2013-10-11 17:28:32 +00:00
# establish a "logical" server connection
# get the url
2014-09-30 16:15:32 +00:00
type, uri = urllib.parse.splittype(uri)
2013-10-11 17:28:32 +00:00
if type not in ("http", "https"):
2014-09-30 16:15:32 +00:00
raise OSError("unsupported XML-RPC protocol")
self.__host, self.__handler = urllib.parse.splithost(uri)
2013-10-11 17:28:32 +00:00
if not self.__handler:
self.__handler = "/RPC2"
if transport is None:
if type == "https":
2014-09-30 16:15:32 +00:00
handler = SafeTransport
2015-11-25 01:25:01 +00:00
extra_kwargs = {"context": context}
2013-10-11 17:28:32 +00:00
else:
2014-09-30 16:15:32 +00:00
handler = Transport
2015-11-25 01:25:01 +00:00
extra_kwargs = {}
2014-09-30 16:15:32 +00:00
transport = handler(use_datetime=use_datetime,
2015-11-25 01:25:01 +00:00
use_builtin_types=use_builtin_types,
**extra_kwargs)
2013-10-11 17:28:32 +00:00
self.__transport = transport
2014-09-30 16:15:32 +00:00
self.__encoding = encoding or 'utf-8'
2013-10-11 17:28:32 +00:00
self.__verbose = verbose
self.__allow_none = allow_none
def __close(self):
self.__transport.close()
def __request(self, methodname, params):
# call a method on the remote server
request = dumps(params, methodname, encoding=self.__encoding,
2014-09-30 16:15:32 +00:00
allow_none=self.__allow_none).encode(self.__encoding)
2013-10-11 17:28:32 +00:00
response = self.__transport.request(
self.__host,
self.__handler,
request,
verbose=self.__verbose
)
if len(response) == 1:
response = response[0]
return response
def __repr__(self):
return (
2015-11-25 01:25:01 +00:00
"<%s for %s%s>" %
(self.__class__.__name__, self.__host, self.__handler)
2013-10-11 17:28:32 +00:00
)
__str__ = __repr__
def __getattr__(self, name):
# magic method dispatcher
return _Method(self.__request, name)
# note: to call a remote object with an non-standard name, use
# result getattr(server, "strange-python-name")(args)
def __call__(self, attr):
"""A workaround to get special attributes on the ServerProxy
without interfering with the magic __getattr__
"""
if attr == "close":
return self.__close
elif attr == "transport":
return self.__transport
raise AttributeError("Attribute %r not found" % (attr,))
2015-11-25 01:25:01 +00:00
def __enter__(self):
return self
def __exit__(self, *args):
self.__close()
2013-10-11 17:28:32 +00:00
# compatibility
Server = ServerProxy
# --------------------------------------------------------------------
# test code
if __name__ == "__main__":
# simple test program (from the XML-RPC specification)
2014-09-30 16:15:32 +00:00
# local server, available from Lib/xmlrpc/server.py
server = ServerProxy("http://localhost:8000")
2013-10-11 17:28:32 +00:00
try:
2014-09-30 16:15:32 +00:00
print(server.currentTime.getCurrentTime())
except Error as v:
print("ERROR", v)
2013-10-11 17:28:32 +00:00
multi = MultiCall(server)
2014-09-30 16:15:32 +00:00
multi.getData()
multi.pow(2,9)
multi.add(1,2)
2013-10-11 17:28:32 +00:00
try:
for response in multi():
2014-09-30 16:15:32 +00:00
print(response)
except Error as v:
print("ERROR", v)