Index: tools/telemetry/telemetry/core/backends/chrome/extension_backend.py |
diff --git a/tools/telemetry/telemetry/core/backends/chrome/extension_backend.py b/tools/telemetry/telemetry/core/backends/chrome/extension_backend.py |
index 70f67f86fa0cf867171b7f6c9e7c80c604d9a663..68a00a692206fe52a1bc31230adc36af61a8e743 100644 |
--- a/tools/telemetry/telemetry/core/backends/chrome/extension_backend.py |
+++ b/tools/telemetry/telemetry/core/backends/chrome/extension_backend.py |
@@ -25,12 +25,16 @@ |
def __init__(self, browser_backend): |
self._extension_backend_list = ExtensionBackendList(browser_backend) |
+ def __contains__(self, extension_id): |
+ return (extension_id in |
+ (self.ContextIdToExtensionId(context_id) |
+ for context_id in self._extension_backend_list)) |
+ |
def __getitem__(self, extension_id): |
extensions = [] |
- for context_id in self._extension_backend_list: |
+ for i, context_id in enumerate(self._extension_backend_list): |
if self.ContextIdToExtensionId(context_id) == extension_id: |
- extensions.append( |
- self._extension_backend_list.GetBackendFromContextId(context_id)) |
+ extensions.append(self._extension_backend_list[i]) |
if not extensions: |
raise KeyError('Cannot find an extension with id=%s' % extension_id) |
return extensions |