safexmlrpc.py

   1 # $Id$
   2 # $URL$
   3
   4 """Leverage curl to make XMLRPC requests that check the server's credentials."""
   5
   6 import xmlrpclib
   7
   8 import curlwrapper
   9
  10
  11 class CertificateCheckingSafeTransport (xmlrpclib.Transport):
  12
  13     def __init__(self, cacert, timeout):
  14         self.cacert = cacert
  15         self.timeout = timeout
  16
  17     def request(self, host, handler, request_body, verbose=0):
  18         self.verbose = verbose
  19         url='https://%s%s' % (host, handler)
  20         # this might raise an xmlrpclib.Protocolerror exception
  21         contents = curlwrapper.retrieve(url,
  22                                         cacert = self.cacert,
  23                                         postdata = request_body,
  24                                         timeout = self.timeout)
  25         return xmlrpclib.loads(contents)[0]
  26
  27 class ServerProxy(xmlrpclib.ServerProxy):
  28
  29     def __init__(self, uri, cacert, timeout = 300, **kwds):
  30         xmlrpclib.ServerProxy.__init__(self, uri,
  31                                        CertificateCheckingSafeTransport(cacert, timeout),
  32                                        **kwds)