blob: 18378670976a15084582b603a6e01999e4c3ce15 [file] [log] [blame] [edit]
#!/usr/bin/python
import cgi, os, sys, urllib2
import common
from autotest_lib.client.common_lib import global_config
from autotest_lib.client.bin import utils
from autotest_lib.frontend.afe.json_rpc import serviceHandler
_PAGE = """\
Status: 302 Found
Content-Type: text/plain
Location: %s\r\n\r
"""
# Define function for retrieving logs
def _retrieve_logs_dummy(job_path):
pass
site_retrieve_logs = utils.import_site_function(__file__,
"autotest_lib.tko.site_retrieve_logs", "site_retrieve_logs",
_retrieve_logs_dummy)
site_find_repository_host = utils.import_site_function(__file__,
"autotest_lib.tko.site_retrieve_logs", "site_find_repository_host",
_retrieve_logs_dummy)
form = cgi.FieldStorage(keep_blank_values=True)
# determine if this is a JSON-RPC request. we support both so that the new TKO
# client can use its RPC client code, but the old TKO can still use simple GET
# params.
_is_json_request = form.has_key('callback')
def _get_requested_path():
if _is_json_request:
request_data = form['request'].value
request = serviceHandler.ServiceHandler.translateRequest(request_data)
parameters = request['params'][0]
return parameters['path']
return form['job'].value
def find_repository_host(job_path):
"""Find the machine holding the given logs and return a URL to the logs"""
site_repo_info = site_find_repository_host(job_path)
if site_repo_info is not None:
return site_repo_info
config = global_config.global_config
drones = config.get_config_value('SCHEDULER', 'drones')
results_host = config.get_config_value('SCHEDULER', 'results_host')
archive_host = config.get_config_value('SCHEDULER', 'archive_host',
default='')
results_repos = [results_host]
for drone in drones.split(','):
drone = drone.strip()
if drone not in results_repos:
results_repos.append(drone)
if archive_host and archive_host not in results_repos:
results_repos.append(archive_host)
for drone in results_repos:
if drone == 'localhost':
continue
http_path = 'http://%s%s' % (drone, job_path)
try:
utils.urlopen(http_path)
return 'http', utils.normalize_hostname(drone), job_path
except urllib2.URLError:
pass
def get_full_url(info, log_path):
if info is not None:
protocol, host, path = info
prefix = '%s://%s' % (protocol, host)
else:
prefix = ''
path = log_path
if _is_json_request:
return '%s/tko/jsonp_fetcher.cgi?%s' % (prefix,
os.environ['QUERY_STRING'])
else:
return prefix + path
log_path = _get_requested_path()
info = find_repository_host(log_path)
site_retrieve_logs(log_path)
print _PAGE % get_full_url(info, log_path)