123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130 |
- import logging
- import BaseHTTPServer
- import optparse
- import xml.etree.ElementTree as ET
- import xml.parsers.expat
- logging.basicConfig(level = logging.DEBUG,
- format='%(asctime)s %(levelname)s %(message)s')
- options = None
- class ConciergeHandler(BaseHTTPServer.BaseHTTPRequestHandler):
- def logRequest(self):
- logging.info('[ConciergeHandler] %(command)s request: %(host)s:%(port)d --- %(path)s',
- dict(command = self.command,
- host = self.client_address[0],
- port = self.client_address[1],
- path = self.path))
- def logResponse(self, status):
- logging.info('[ConciergeHandler] %(command)s returned %(status)d',
- dict(command = self.command,
- status = status))
-
- def do_HEAD(self):
- self.logRequest()
-
- self.send_response(200)
- self.send_header('Content-type', 'text/html')
- self.end_headers()
- self.logResponse(200)
- def dumpXml(self, xml):
- logging.debug('[ConciergeHandler] %s', xml.tag)
- for attr in xml.attrib:
- logging.debug('[ConciergeHandler] %s [%s] %s', xml.tag, attr, xml.attrib[attr])
- for kid in xml.getchildren():
- self.dumpXml(kid)
- def do_POST(self):
- self.logRequest()
- hdrs = {}
- for hdr in self.headers.headers:
- logging.debug('[ConciergeHandler] POST: header: %s', hdr.rstrip())
- length = int(self.headers.getheader('Content-Length'))
- content = self.rfile.read(length)
- self.rfile.close()
-
- logging.debug('[ConciergeHandler] POST: content: %s', content)
- try:
- postXml = ET.fromstring(content)
- self.dumpXml(postXml)
- except xml.parsers.expat.ExpatError, xmlError:
- logging.error('[ConciergeHandler] POST illformed:%s', xmlError)
- self.send_response(500)
- return
- if not options.fail:
- self.send_response(200)
- self.send_header('Content-Type', 'text/html')
- self.send_header('Content-Length', len('<success/>'))
- self.end_headers()
- self.logResponse(200)
- self.wfile.write('<success/>')
- self.wfile.close()
- else:
- self.send_response(500)
- self.send_header('Content-Type', 'text/html')
- self.send_header('Content-Length', len('<error>gotcha!</error>'))
- self.end_headers()
- self.wfile.write('<error>gotcha!</error>')
- self.wfile.close()
- self.logResponse(500)
- def log_request(code, size):
- pass
- if __name__ == '__main__':
- logging.info('[ConciergeServer] Concierge Broker Test Server starting')
- parser = optparse.OptionParser()
- parser.add_option('-p', '--port', dest = 'port', help = 'port to listen on', metavar = 'PORT')
- parser.add_option('-f', '--fail', dest = 'fail', action = 'store_true', help = 'always fail POST requests')
- (options, args) = parser.parse_args()
- httpServer = BaseHTTPServer.HTTPServer(('', 8080), ConciergeHandler)
- try:
- httpServer.serve_forever()
- except KeyboardInterrupt:
- logging.info('[ConciergeServer] terminating')
- httpServer.server_close()
|