# minisoap - minimal SOAP client

#

# Copyright (c) 2011, 2013  Radek Brich <radek.brich@devl.cz>

#

# Permission is hereby granted, free of charge, to any person obtaining a copy

# of this software and associated documentation files (the "Software"), to deal

# in the Software without restriction, including without limitation the rights

# to use, copy, modify, merge, publish, distribute, sublicense, and/or sell

# copies of the Software, and to permit persons to whom the Software is

# furnished to do so, subject to the following conditions:

#

# The above copyright notice and this permission notice shall be included in

# all copies or substantial portions of the Software.

#

# THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR

# IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,

# FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE

# AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER

# LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,

# OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN

# THE SOFTWARE.

from http.client import HTTPConnection

from urllib.parse import urlparse

from lxml import etree

from collections import OrderedDict

debug = 0

HTTPConnection.debuglevel = debug

class SoapError(Exception):

    pass

class SoapResponse(OrderedDict):

    def __init__(self, name):

        OrderedDict.__init__(self)

        self.name = name

    def _printout(self, out, res, level):

        indent = level * 2 * ' '

        for k in res.keys():

            if type(res[k]) is list:

                for row in res[k]:

                    out.extend([indent, k, '\n'])

                    self._printout(out, row, level+1)

            else:

                out.extend([indent, k, ': ', str(res[k]), '\n'])

    def __str__(self):

        out = [self.name, '\n']

        self._printout(out, self, 1)

        return ''.join(out).rstrip()

class SoapClient:

    def __init__(self, url, namespace):

        self.url = url

        self.parsedurl = urlparse(url)

        self.namespace = namespace

        self.conn = HTTPConnection(self.parsedurl.netloc)

    def _build_soap_message(self, name, args):

        SOAP = 'http://schemas.xmlsoap.org/soap/envelope/'

        root = etree.Element('{%s}Envelope' % SOAP, nsmap={'soap':SOAP})

        body = etree.SubElement(root, '{%s}Body' % SOAP)

        func = etree.SubElement(body, '{%s}%s' % (self.namespace, name), nsmap={None:self.namespace})

        for key in args.keys():

            param = etree.SubElement(func, key)

            param.text = str(args[key])

        if debug:

            print(etree.tostring(root, pretty_print=True).decode('utf8'))

        return root

    def _break_params(self, func, d):

        for param in func.iterchildren():

            if len(param) == 0:

                d[param.tag] = param.text

            else:

                if not param.tag in d:

                    d[param.tag] = []

                newd = OrderedDict()

                self._break_params(param, newd)

                d[param.tag].append(newd)

    def _break_soap_message(self, root):

        if not root.tag.endswith('}Envelope'):

            raise SoapError('Reply has no Envelope.')

        SOAP = root.tag.split('}')[0][1:]

        body_list = list(root.iterchildren('{%s}Body' % SOAP))

        if len(body_list) != 1:

            raise SoapError('Bad SOAP format (%d Body elements)' % len(body_list))

        body = body_list[0]

        func_list =list(body.iterchildren())

        if len(func_list) != 1:

            raise SoapError('Bad SOAP format (Body has %d children, should have one)' % len(body_list))

        func = func_list[0]

        if debug:

            print(etree.tostring(func, pretty_print=True).decode('utf8'))

        res = SoapResponse(func.tag)

        self._break_params(func, res)

        return res

    def _serialize_xml(self, root):

        return etree.tostring(root,

            pretty_print=True,

            encoding='utf-8',

            xml_declaration=True)

    def _parse_xml(self, file):

        tree = etree.parse(file)

        return tree.getroot()

    def _send_soap(self, body):

        headers = {

            'Content-type': "text/xml;charset='utf-8'",

            'Soapaction': "''",

            'Accept': 'text/xml'}

        self.conn.request('POST', self.parsedurl.path, body, headers)

        resp = self.conn.getresponse()

        if resp.status != 200:

            raise SoapError('Server replied %d %s' % (resp.status, resp.reason))

        return resp

    def __getattr__(self, name):

        def fnc(**kwargs):

            root = self._build_soap_message(name, kwargs)

            body = self._serialize_xml(root)

            resp = self._send_soap(body)

            root = self._parse_xml(resp)

            return self._break_soap_message(root)

        return fnc

if __name__ == '__main__':

    service = SoapClient('http://10.0.0.1:34567/', namespace='http://devl.cz/')

    res = service.my_test_request()

    print(res)