xref: /external/autotest/cli/rpc.py

#
# Copyright 2008 Google Inc. All Rights Reserved.
#

import os, getpass
from autotest_lib.frontend.afe import rpc_client_lib
from autotest_lib.frontend.afe.json_rpc import proxy


def get_autotest_server(web_server=None):
    if not web_server:
        if 'AUTOTEST_WEB' in os.environ:
            web_server = os.environ['AUTOTEST_WEB']
        else:
            web_server = 'http://autotest'

    # if the name doesn't start with http://,
    # nonexistant hosts get an obscure error
    if not web_server.startswith('http://'):
        web_server = 'http://' + web_server

    return web_server


class afe_comm(object):
    """Handles the AFE setup and communication through RPC"""
    def __init__(self, web_server=None):
        self.web_server = get_autotest_server(web_server)
        self.proxy = self._connect()

    def _connect(self):
        # This does not fail even if the address is wrong.
        # We need to wait for an actual RPC to fail
        headers = {'AUTHORIZATION' : getpass.getuser()}
        rpc_server = self.web_server + "/afe/server/noauth/rpc/"
        return rpc_client_lib.get_proxy(rpc_server, headers=headers)


    def run(self, op, **data):
        function = getattr(self.proxy, op)
        return function(**data)