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

Unified Diff: tools/telemetry/telemetry/internal/util/binary_manager.py

Issue 1280903003: Add dependency_manager initialization to binary_manager (Closed) Base URL: https://chromium.googlesource.com/chromium/src.git@master
Patch Set: Fix and refactor the unittest. Created 5 years, 4 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
Index: tools/telemetry/telemetry/internal/util/binary_manager.py
diff --git a/tools/telemetry/telemetry/internal/util/binary_manager.py b/tools/telemetry/telemetry/internal/util/binary_manager.py
index bfa2c2c014916d81503a7d6fde30ff064543222f..c6921ab31776f43a1a13c020127bfc169da431f5 100644
--- a/tools/telemetry/telemetry/internal/util/binary_manager.py
+++ b/tools/telemetry/telemetry/internal/util/binary_manager.py
@@ -2,18 +2,63 @@
# Use of this source code is governed by a BSD-style license that can be
# found in the LICENSE file.
+#TODO(aiolos): Remove the debug logging once the entire Dependency Manager
+#system has landed.
+import logging
+import os
+
from catapult_base import support_binaries
+from catapult_base import dependency_manager
+from telemetry.core import exceptions
+from telemetry.core import util
+
+
+TELEMETRY_PROJECT_CONFIG = os.path.join(
+ util.GetTelemetryDir(), 'telemetry', 'internal', 'binary_dependencies.json')
+
+
+_dependency_manager = None
+
+
+def NeedsInit():
+ return not _dependency_manager
+
+
+def InitDependencyManager(environment_config):
+ global _dependency_manager
+ if _dependency_manager:
+ raise exceptions.InitializationError(
+ 'Trying to re-initialize the binary manager with config %s'
+ % environment_config)
+ config_files = [TELEMETRY_PROJECT_CONFIG]
+ if environment_config:
+ config_files.append(environment_config)
+ logging.debug('Initializing the dependency manager with config files: %s.'
+ % config_files)
+ _dependency_manager = dependency_manager.DependencyManager(config_files)
+ logging.debug('Successfully initialized the dependency manager.')
def FetchPath(binary_name, platform, arch):
""" Return a path to the appropriate executable for <binary_name>, downloading
from cloud storage if needed, or None if it cannot be found.
"""
+ logging.debug('Called FetchPath for binary: %s on platform: %s and arch: %s'
+ % (binary_name, platform, arch))
+ if _dependency_manager is None:
+ raise exceptions.InitializationError(
+ 'Called FetchPath with uninitialized binary manager.')
return support_binaries.FindPath(binary_name, platform, arch)
+
def LocalPath(binary_name, platform, arch):
""" Return a local path to the given binary name, or None if an executable
cannot be found. Will not download the executable.
"""
+ logging.debug('Called LocalPath for binary: %s on platform: %s and arch: '
+ '%s' % (binary_name, platform, arch))
+ if _dependency_manager is None:
+ raise exceptions.InitializationError(
+ 'Called LocalPath with uninitialized binary manager.')
del platform, arch
return support_binaries.FindLocallyBuiltPath(binary_name)
« no previous file with comments | « tools/telemetry/telemetry/core/exceptions.py ('k') | tools/telemetry/telemetry/internal/util/binary_manager_unittest.py » ('j') | no next file with comments »

Powered by Google App Engine
This is Rietveld 408576698