class MyHandler(BaseHTTPServer.BaseHTTPRequestHandler):
def do_POST(s):
"""
Respond to a POST request.
"""
request_body = None
try:
length = int(s.headers['Content-Length'])
request_body = s.rfile.read(length) # get the request body
except Exception, e:
print "WARN - content-length header does not exist and payload is empty... "
resp, code = mock_responses.response(s.path, request_body)
if (resp.find(mock_responses.WARN) >= 0 or resp.find(
mock_responses.ERROR) >= 0) and code == mock_responses.EMPTY:
print resp
else:
s.send_response(code)
if mock_responses.isHadoop(s.path):
headers = mock_responses.headersHADOOP.items()
else:
headers = mock_responses.headersCKAN.items()
for h, v in headers:
s.send_header(h, v)
if code == mock_responses.REDIRECT:
s.send_header(mock_responses.LOCATION,
mock_responses.HADOOP_LOCATION)
s.end_headers()
s.wfile.write(resp)
def do_PUT(s):
"""
Respond to a PUT request.
"""
resp, code = mock_responses.response(s.path)
if (resp.find (mock_responses.WARN)>=0 or resp.find (mock_responses.ERROR)>=0) and code == mock_responses.EMPTY:
print resp
else:
s.send_response(code)
for h, v in mock_responses.headersHADOOP.items():
s.send_header(h, v)
if code == mock_responses.REDIRECT:
s.send_header(mock_responses.LOCATION,mock_responses.HADOOP_LOCATION)
s.end_headers()
s.wfile.write(resp)
def do_GET(s):
"""
Respond to a GET request.
"""
resp, code = mock_responses.response(s.path)
if (resp.find (mock_responses.WARN)>=0 or resp.find (mock_responses.ERROR)>=0) and code == mock_responses.EMPTY:
print resp
else:
s.send_response(code)
if mock_responses.isHadoop(s.path):headers = mock_responses.headersHADOOP.items()
else:headers=mock_responses.headersCKAN.items()
for h, v in headers:
s.send_header(h, v)
s.end_headers()
s.wfile.write(resp)
def do_PUT(s):
"""
Respond to a PUT request.
"""
resp, code = mock_responses.response(s.path)
if (resp.find(mock_responses.WARN) >= 0 or resp.find(
mock_responses.ERROR) >= 0) and code == mock_responses.EMPTY:
print resp
else:
s.send_response(code)
for h, v in mock_responses.headersHADOOP.items():
s.send_header(h, v)
if code == mock_responses.REDIRECT:
s.send_header(mock_responses.LOCATION,
mock_responses.HADOOP_LOCATION)
s.end_headers()
s.wfile.write(resp)
def do_GET(s):
"""
Respond to a GET request.
"""
resp, code = mock_responses.response(s.path)
if (resp.find(mock_responses.WARN) >= 0 or resp.find(
mock_responses.ERROR) >= 0) and code == mock_responses.EMPTY:
print resp
else:
s.send_response(code)
if mock_responses.isHadoop(s.path):
headers = mock_responses.headersHADOOP.items()
else:
headers = mock_responses.headersCKAN.items()
for h, v in headers:
s.send_header(h, v)
s.end_headers()
s.wfile.write(resp)
