Chromium Code Reviews
chromiumcodereview-hr@appspot.gserviceaccount.com (chromiumcodereview-hr) | Please choose your nickname with Settings | Help | Chromium Project | Gerrit Changes | Sign out
(314)

Side by Side Diff: testing/legion/common_lib.py

Issue 1124763003: Update from https://crrev.com/327068 (Closed) Base URL: git@github.com:domokit/mojo.git@master
Patch Set: update nacl, buildtools, fix display_change_notifier_unittest Created 5 years, 7 months ago
Use n/p to move between diff chunks; N/P to move between comments. Draft comments are only viewable by you.
Jump to:
View unified diff | Download patch
OLDNEW
1 # Copyright 2015 The Chromium Authors. All rights reserved. 1 # Copyright 2015 The Chromium Authors. All rights reserved.
2 # Use of this source code is governed by a BSD-style license that can be 2 # Use of this source code is governed by a BSD-style license that can be
3 # found in the LICENSE file. 3 # found in the LICENSE file.
4 4
5 """Common library methods used by both coordinator and task machines.""" 5 """Common library methods used by both coordinator and task machines."""
6 6
7 import argparse 7 import argparse
8 import logging 8 import logging
9 import os 9 import os
10 import socket 10 import socket
11 import xmlrpclib 11 import xmlrpclib
12 12
13 LOGGING_LEVELS = ['DEBUG', 'INFO', 'WARNING', 'WARN', 'ERROR'] 13 LOGGING_LEVELS = ['DEBUG', 'INFO', 'WARNING', 'WARN', 'ERROR']
14 MY_IP = socket.gethostbyname(socket.gethostname()) 14 MY_IP = socket.gethostbyname(socket.gethostname())
15 SERVER_ADDRESS = '' 15 SERVER_ADDRESS = ''
16 SERVER_PORT = 31710 16 SERVER_PORT = 31710
17 DEFAULT_TIMEOUT_SECS = 20 * 60 # 30 minutes 17 DEFAULT_TIMEOUT_SECS = 20 * 60 # 30 minutes
18 THIS_DIR = os.path.dirname(os.path.abspath(__file__)) 18 THIS_DIR = os.path.dirname(os.path.abspath(__file__))
19 SWARMING_DIR = os.path.join(THIS_DIR, '..', '..', 'tools', 'swarming_client') 19 SWARMING_DIR = os.path.join(THIS_DIR, '..', '..', 'tools', 'swarming_client')
20 20
21 21
22 def InitLogging(): 22 def InitLogging():
23 """Initialize the logging module. 23 """Initialize the logging module.
24 24
25 Raises: 25 Raises:
26 argparse.ArgumentError if the --verbosity arg is incorrect. 26 argparse.ArgumentError if the --verbosity arg is incorrect.
27 """ 27 """
28 parser = argparse.ArgumentParser() 28 parser = argparse.ArgumentParser()
29 logging_action = parser.add_argument('--verbosity', default='ERROR') 29 logging_action = parser.add_argument('--verbosity', default='INFO')
30 args, _ = parser.parse_known_args() 30 args, _ = parser.parse_known_args()
31 if args.verbosity not in LOGGING_LEVELS: 31 if args.verbosity not in LOGGING_LEVELS:
32 raise argparse.ArgumentError( 32 raise argparse.ArgumentError(
33 logging_action, 'Only levels %s supported' % str(LOGGING_LEVELS)) 33 logging_action, 'Only levels %s supported' % str(LOGGING_LEVELS))
34 logging.basicConfig( 34 logging.basicConfig(
35 format='%(asctime)s %(filename)s:%(lineno)s %(levelname)s] %(message)s', 35 format='%(asctime)s %(filename)s:%(lineno)s %(levelname)s] %(message)s',
36 datefmt='%H:%M:%S', level=args.verbosity) 36 datefmt='%H:%M:%S', level=args.verbosity)
37 37
38 38
39 def ConnectToServer(server): 39 def ConnectToServer(server):
40 """Connect to an RPC server.""" 40 """Connect to an RPC server."""
41 addr = 'http://%s:%d' % (server, SERVER_PORT) 41 addr = 'http://%s:%d' % (server, SERVER_PORT)
42 logging.debug('Connecting to RPC server at %s', addr) 42 logging.debug('Connecting to RPC server at %s', addr)
43 return xmlrpclib.Server(addr) 43 return xmlrpclib.Server(addr, allow_none=True)
OLDNEW

Powered by Google App Engine
This is Rietveld 408576698