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 68a00a692206fe52a1bc31230adc36af61a8e743..70f67f86fa0cf867171b7f6c9e7c80c604d9a663 100644 |
--- a/tools/telemetry/telemetry/core/backends/chrome/extension_backend.py |
+++ b/tools/telemetry/telemetry/core/backends/chrome/extension_backend.py |
@@ -25,16 +25,12 @@ |
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 i, context_id in enumerate(self._extension_backend_list): |
+ for context_id in self._extension_backend_list: |
if self.ContextIdToExtensionId(context_id) == extension_id: |
- extensions.append(self._extension_backend_list[i]) |
+ extensions.append( |
+ self._extension_backend_list.GetBackendFromContextId(context_id)) |
if not extensions: |
raise KeyError('Cannot find an extension with id=%s' % extension_id) |
return extensions |