blob: 2dd3e1763a928b8ea133d7055410eb4325a38af0 [file] [log] [blame]
This module provides utility functions useful when writing clients for the RPC
__author__ = ' (Steve Howard)'
import getpass, os
from json_rpc import proxy
from autotest_lib.client.common_lib import utils
class AuthError(Exception):
def get_proxy(*args, **kwargs):
"""Use this to access the AFE or TKO RPC interfaces."""
return proxy.ServiceProxy(*args, **kwargs)
def _base_authorization_headers(username, server):
Don't call this directly, call authorization_headers().
This implementation may be overridden by site code.
@returns A dictionary of authorization headers to pass in to get_proxy().
if not username:
if 'AUTOTEST_USER' in os.environ:
username = os.environ['AUTOTEST_USER']
username = getpass.getuser()
return {'AUTHORIZATION' : username}
authorization_headers = utils.import_site_function(
__file__, 'autotest_lib.frontend.afe.site_rpc_client_lib',
'authorization_headers', _base_authorization_headers)