specify a lighthouse server

This commit is contained in:
Jack 2016-08-03 17:44:32 -04:00
parent 9fedf14e55
commit e8d04b1921

View file

@ -8,8 +8,8 @@ log = logging.getLogger(__name__)
class LighthouseClient(object): class LighthouseClient(object):
def __init__(self): def __init__(self, servers=None):
self.servers = SEARCH_SERVERS self.servers = servers or SEARCH_SERVERS
def _get_random_server(self): def _get_random_server(self):
return Proxy(random.choice(self.servers)) return Proxy(random.choice(self.servers))