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