Source code for soco.soap

# pylint: disable=fixme

# Disable while we have Python 2.x compatability
# pylint: disable=useless-object-inheritance

"""Classes for handling SoCo's basic SOAP requirements.

This module does not handle anything like the full `SOAP Specification
<http://www.w3.org/TR/soap/>`_ , but is enough for SoCo's needs. Sonos uses
SOAP for UPnP communications, and for communication with third party music
services.
"""

# The state of Python's SOAP libraries is poor. In any event, the two main
# libraries, PySimpleSOAP and SUDS (or the more up-to-date SUDS-Jurko),
# are too complex for our needs. SUDS requires a WSDL file to be parsed,
# and although SONOS provides one in relation to music services (at
# http://musicpartners.sonos.com/sites/default/files/Sonos.wsdl) the various
# music services themselves provide buggy, incomplete or old
# implementations which cause SUDS to break. PySimpleSOAP can work without a
# WSDL file, but contains various bugs which mean that we would have to use
# a patched version (upstream releases are infrequent).  Since SONOS only
# appears to use basic SOAP features, and after experimenting with other
# libraries, it seems best to write our own.

# Some is the same as that in services.py.
# TODO: refactor services.py to depend on this code


import logging
from xml.sax.saxutils import escape

import requests

from .exceptions import SoCoException
from .utils import prettify
from .xml import XML

_LOG = logging.getLogger(__name__)


[docs]class SoapFault(SoCoException): """An exception encapsulating a SOAP Fault.""" def __init__(self, faultcode, faultstring, detail=None): """ Args: faultcode (str): The SOAP faultcode. faultstring (str): The SOAP faultstring. detail (:obj:`~xml.etree.ElementTree.Element`): The SOAP fault detail, as an ElementTree :obj:`~xml.etree.ElementTree.Element`. Defaults to `None`. """ self.faultcode = faultcode self.faultstring = faultstring self.detail = detail self.detail_string = XML.tostring(detail) if detail is not None else "" super().__init__(faultcode, faultstring) def __str__(self): return "{}: {}".format(self.faultcode, self.faultstring) def __repr__(self): return "SoapFault(faultcode={}, faultstring={}, detail={})".format( repr(self.faultcode), repr(self.faultstring), repr(self.detail) )
# Sonos uses SOAP to send commands in the RPC form. A complete RPC SOAP # message should look something like this. See generally # http://www.w3.org/TR/2000/NOTE-SOAP-20000508/ # POST Endpoint URL HTTP/1.1 # HOST: Host of Endpoint URL:port # CONTENT-LENGTH: bytes in body # CONTENT-TYPE: text/xml; charset="utf-8" # SOAPACTION: URI # # <?xml version="1.0"?> # <s:Envelope # xmlns:s="http://schemas.xmlsoap.org/soap/envelope/" # s:encodingStyle="http://schemas.xmlsoap.org/soap/encoding/"> # <s:Header> # </Header elements go here> # </s:Header> # <s:Body> # <ns:MethodName xmlns:ns="MethodNamespace>" # <param1>value</param1> # ... # <param_n>value</param_n> # </ns:MethodName> # </s:Body> # </s:Envelope> # pylint: disable=too-many-instance-attributes, too-many-arguments
[docs]class SoapMessage: """A SOAP Message representing a remote procedure call. Uses the `Requests <http://www.python-requests.org/en/latest/>`_ library for communication with a SOAP server. """ def __init__( self, endpoint, method, parameters=None, http_headers=None, soap_action=None, soap_header=None, namespace=None, **request_args ): """ Args: endpoint (str): The SOAP endpoint URL for this client. method (str): The name of the method to call. parameters (list): A list of (name, value) tuples containing the parameters to pass to the method. Default `None`. http_headers (dict): A dict in the form ``{'Header': 'Value,..}`` containing http headers to use for the http request. ``Content-type`` and ``SOAPACTION`` headers will be created automatically, so do not include them here. Use this, for example, to set a user-agent. soap_action (str): The value of the ``SOAPACTION`` header. Default 'None`. soap_header (str): A string representation of the XML to be used for the SOAP Header. Default `None`. namespace (str): The namespace URI to use for the method and parameters. `None`, by default. **request_args: Other keyword parameters will be passed to the Requests request which is used to handle the http communication. For example, a timeout value can be set. """ self.endpoint = endpoint self.method = method self.parameters = [] if parameters is None else parameters self.http_headers = http_headers self.soap_action = soap_action self.soap_header = soap_header self.namespace = namespace self.request_args = request_args # pylint:disable=no-self-use
[docs] def prepare_headers(self, http_headers, soap_action): """Prepare the http headers for sending. Add the ``SOAPACTION`` header to the others. Args: http_headers (dict): A dict in the form ``{'Header': 'Value,..}`` containing http headers to use for the http request. soap_action (str): The value of the ``SOAPACTION`` header. Returns: dict: headers including the ``SOAPACTION`` header. """ headers = {"Content-Type": 'text/xml; charset="utf-8"'} if soap_action is not None: headers.update({"SOAPACTION": '"{}"'.format(soap_action)}) if http_headers is not None: headers.update(http_headers) return headers
[docs] def prepare_soap_header(self, soap_header): """Prepare the SOAP header for sending. Wraps the soap header in appropriate tags. Args: soap_header (str): A string representation of the XML to be used for the SOAP Header Returns: str: The soap header wrapped in appropriate tags. """ if soap_header is not None: return "<s:Header>{}</s:Header>".format(soap_header) else: return ""
[docs] def prepare_soap_body(self, method, parameters, namespace): """Prepare the SOAP message body for sending. Args: method (str): The name of the method to call. parameters (list): A list of (name, value) tuples containing the parameters to pass to the method. namespace (str): The XML namespace to use for the method. Returns: str: A properly formatted SOAP Body. """ tags = [] for name, value in parameters: tag = "<{name}>{value}</{name}>".format( name=name, value=escape("%s" % value, {'"': "&quot;"}) ) # % converts to unicode because we are using unicode literals. # Avoids use of 'unicode' function which does not exist in python 3 tags.append(tag) wrapped_params = "".join(tags) # Prepare the SOAP Body if namespace is not None: soap_body = ( '<{method} xmlns="{namespace}">' "{params}" "</{method}>".format( method=method, params=wrapped_params, namespace=namespace ) ) else: soap_body = ( "<{method}>" "{params}" "</{method}>".format(method=method, params=wrapped_params) ) return soap_body
[docs] def prepare_soap_envelope(self, prepared_soap_header, prepared_soap_body): """Prepare the SOAP Envelope for sending. Args: prepared_soap_header (str): A SOAP Header prepared by `prepare_soap_header` prepared_soap_body (str): A SOAP Body prepared by `prepare_soap_body` Returns: str: A prepared SOAP Envelope """ # pylint: disable=bad-continuation soap_env_template = ( '<?xml version="1.0"?>' '<s:Envelope xmlns:s="http://schemas.xmlsoap.org/soap/envelope/"' ' s:encodingStyle="http://schemas.xmlsoap.org/soap/encoding/">' "{soap_header}" "<s:Body>" "{soap_body}" "</s:Body>" "</s:Envelope>" ) # noqa PEP8 return soap_env_template.format( soap_header=prepared_soap_header, soap_body=prepared_soap_body )
[docs] def prepare(self): """Prepare the SOAP message for sending to the server.""" headers = self.prepare_headers(self.http_headers, self.soap_action) soap_header = self.prepare_soap_header(self.soap_header) soap_body = self.prepare_soap_body(self.method, self.parameters, self.namespace) data = self.prepare_soap_envelope(soap_header, soap_body) return (headers, data)
[docs] def call(self): """Call the SOAP method on the server. Returns: str: the decapusulated SOAP response from the server, still encoded as utf-8. Raises: SoapFault: if a SOAP error occurs. ~requests.exceptions.HTTPError: if an http error occurs. """ headers, data = self.prepare() # Check log level before logging XML, since prettifying it is # expensive if _LOG.isEnabledFor(logging.DEBUG): _LOG.debug("Sending %s, %s", headers, prettify(data)) response = requests.post( self.endpoint, headers=headers, data=data.encode("utf-8"), **self.request_args ) _LOG.debug("Received %s, %s", response.headers, response.text) status = response.status_code if status == 200: # The response is good. Extract the Body tree = XML.fromstring(response.content) # Get the first child of the <Body> tag. NB There should only be # one if the RPC standard is followed. body = tree.find("{http://schemas.xmlsoap.org/soap/envelope/}Body")[0] return body elif status == 500: # We probably have a SOAP Fault tree = XML.fromstring(response.content) fault = tree.find(".//{http://schemas.xmlsoap.org/soap/envelope/}Fault") if fault is None: # Not a SOAP fault. Must be something else. response.raise_for_status() faultcode = fault.findtext("faultcode") faultstring = fault.findtext("faultstring") faultdetail = fault.find("detail") raise SoapFault(faultcode, faultstring, faultdetail) else: # Something else has gone wrong. Probably a network error. Let # Requests handle it response.raise_for_status() return None