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

Unified Diff: testing/legion/common_lib.py

Issue 890773003: Adding the initial code for Omnibot multi-machine support (Closed) Base URL: https://chromium.googlesource.com/chromium/src.git@master
Patch Set: Created 5 years, 10 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 side-by-side diff with in-line comments
Download patch
« no previous file with comments | « testing/legion/client_rpc_server.py ('k') | testing/legion/discovery_server.py » ('j') | no next file with comments »
Expand Comments ('e') | Collapse Comments ('c') | Show Comments Hide Comments ('s')
Index: testing/legion/common_lib.py
diff --git a/testing/legion/common_lib.py b/testing/legion/common_lib.py
new file mode 100644
index 0000000000000000000000000000000000000000..b66481e07638e5950185e6b055b210dd1f3515ed
--- /dev/null
+++ b/testing/legion/common_lib.py
@@ -0,0 +1,40 @@
+# Copyright 2015 The Chromium Authors. All rights reserved.
+# Use of this source code is governed by a BSD-style license that can be
+# found in the LICENSE file.
+
+"""Common library methods used by both host and client controllers."""
+
+import argparse
+import logging
+import socket
+import xmlrpclib
+
+LOGGING_LEVELS = ['DEBUG', 'INFO', 'WARNING', 'WARN', 'ERROR']
+MY_IP = socket.gethostbyname(socket.gethostname())
+SERVER_ADDRESS = ''
+SERVER_PORT = 31710
+DEFAULT_TIMEOUT_SECS = 20 * 60 # 30 minutes
+
+
+def InitLogging():
+ """Initialize the logging module.
+
+ Raises:
+ argparse.ArgumentError if the --verbosity arg is incorrect.
+ """
+ parser = argparse.ArgumentParser()
+ logging_action = parser.add_argument('--verbosity', default='ERROR')
+ args, _ = parser.parse_known_args()
+ if args.verbosity not in LOGGING_LEVELS:
+ raise argparse.ArgumentError(
+ logging_action, 'Only levels %s supported' % str(LOGGING_LEVELS))
+ logging.basicConfig(
+ format='%(asctime)s %(filename)s:%(lineno)s %(levelname)s] %(message)s',
+ datefmt='%H:%M:%S', level=args.verbosity)
+
+
+def ConnectToServer(server):
+ """Connect to an RPC server."""
+ addr = 'http://%s:%d' % (server, SERVER_PORT)
+ logging.debug('Connecting to RPC server at %s', addr)
+ return xmlrpclib.Server(addr)
« no previous file with comments | « testing/legion/client_rpc_server.py ('k') | testing/legion/discovery_server.py » ('j') | no next file with comments »

Powered by Google App Engine
This is Rietveld 408576698