xmlrpcclient.py

   1 from twisted.internet.protocol import ClientFactory
   2 from twisted.protocols.http import HTTPClient
   3 from twisted.internet.defer import Deferred
   4
   5 from xmlrpclib import loads, dumps
   6
   7 USER_AGENT = 'Python/Twisted XMLRPC 0.1'
   8 class XMLRPCClient(HTTPClient):
   9     def connectionMade(self):
  10         payload = dumps(self.args, self.method)
  11         self.sendCommand('POST', '/RPC2')
  12         self.sendHeader('User-Agent', USER_AGENT)
  13         self.sendHeader('Content-Type', 'text/xml')
  14         self.sendHeader('Content-Length', len(payload))
  15         self.endHeaders()
  16         self.transport.write(payload)
  17         self.transport.write('\r\n')
  18
  19     def handleResponse(self, buf):
  20         try:
  21             args, name = loads(buf)
  22         except Exception, e:
  23             print "response decode error: " + `e`
  24             self.d.errback()
  25         else:
  26             apply(self.d.callback, args)
  27
  28 class XMLRPCClientFactory(ClientFactory):
  29     def __init__(self, method, args, callback=None, errback=None):
  30         self.method = method
  31         self.args = args
  32         self.d = Deferred()
  33         if callback:
  34             self.d.addCallback(callback)
  35         if errback:
  36             self.d.addErrback(errback)
  37
  38     def buildProtocol(self, addr):
  39         prot =  XMLRPCClient()
  40         prot.method = self.method
  41         prot.args = self.args
  42         prot.d = self.d
  43         return prot