| Author: | Andrew Ellerton |
|---|---|
| Last upated: | July 2006 |
#!/bin/env python
import xmlrpclib
class ProxyTransport(xmlrpclib.Transport):
"""Provides an XMl-RPC transport routing via a http proxy.
This is done by using urllib2, which in turn uses the environment
varable http_proxy and whatever else it is built to use (e.g. the
windows registry).
NOTE: the environment variable http_proxy should be set correctly.
See checkProxySetting() below.
Written from scratch but inspired by xmlrpc_urllib_transport.py
file from http://starship.python.net/crew/jjkunce/ by jjk.
A. Ellerton 2006-07-06
"""
def request(self, host, handler, request_body, verbose):
import urllib2
self.verbose=verbose
url='http://'+host+handler
if self.verbose: "ProxyTransport URL: [%s]"%url
request = urllib2.Request(url)
request.add_data(request_body)
# Note: 'Host' and 'Content-Length' are added automatically
request.add_header("User-Agent", self.user_agent)
request.add_header("Content-Type", "text/xml") # Important
proxy_handler=urllib2.ProxyHandler()
opener=urllib2.build_opener(proxy_handler)
f=opener.open(request)
return(self.parse_response(f))
def checkProxySetting():
"""If the variable 'http_proxy' is set, it will most likely be in one
of these forms (not real host/ports):
proxyhost:8080
http://proxyhost:8080
urlllib2 seems to require it to have 'http;//" at the start.
This routine does that, and returns the transport for xmlrpc.
"""
import os, re
try:
http_proxy=os.environ['http_proxy']
except KeyError:
return
# ensure the proxy has the 'http://' at the start
#
match = re.match("(http://)?([-_\.A-Za-z]+):(\d+)", http_proxy)
if not match: raise Exception("Proxy format not recognised: [%s]" % http_proxy)
os.environ['http_proxy'] = "http://%s:%s" % (match.group(2), match.group(3))
#print "Determined proxy: %s" % os.environ['http_proxy']
return
def test():
import sys, os
def nextArg():
try: return sys.argv.pop(1)
except: return None
checkProxySetting()
url=nextArg() or "http://betty.userland.com"
api=nextArg() or "examples.getStateName(32)" # "examples.getStateList([1,2])"
try:
server=xmlrpclib.Server(url, transport=ProxyTransport())
print "Url: %s" % url
try: print "Proxy: %s" % os.environ['http_proxy']
except KeyError: print "Proxy: (Apparently none)"
print "API: %s" % api
r = eval("server.%s"%api)
print "Result: ", r
except xmlrpclib.ProtocolError, e:
print "Connection error: %s" % e
except xmlrpclib.Fault, e:
print "Error: %s" % e
if __name__=='__main__':
# run with no parameters for basic test case.
test()
Back to software.