Add module dir to sys.path in FindPythonDependencies

tools/telemetry/telemetry/internal/util/find_dependencies.py

 # Copyright 2014 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.
 
 import fnmatch
 import imp
 import logging
 import optparse
 import os
 import sys
@@ skipping 18 matching lines @@
     return []
   deps_paths = bootstrap.ListAllDepsPaths(deps_file)
   return set(os.path.realpath(os.path.join(
       path.GetChromiumSrcDir(), os.pardir, deps_path))
       for deps_path in deps_paths)
 
 
 def FindPythonDependencies(module_path):
   logging.info('Finding Python dependencies of %s' % module_path)
 
-  # Load the module to inherit its sys.path modifications.
-  imp.load_source(
-      os.path.splitext(os.path.basename(module_path))[0], module_path)
+  sys_path = sys.path
+  sys.path = list(sys_path)
+  try:
+    # Load the module to inherit its sys.path modifications.
+    sys.path.insert(0, os.path.abspath(os.path.dirname(module_path)))
+    imp.load_source(
+        os.path.splitext(os.path.basename(module_path))[0], module_path)
 
-  # Analyze the module for its imports.
-  graph = modulegraph.ModuleGraph()
-  graph.run_script(module_path)
+    # Analyze the module for its imports.
+    graph = modulegraph.ModuleGraph()
+    graph.run_script(module_path)
 
-  # Filter for only imports in Chromium.
-  for node in graph.nodes():
-    if not node.filename:
-      continue
-    module_path = os.path.realpath(node.filename)
+    # Filter for only imports in Chromium.
+    for node in graph.nodes():
+      if not node.filename:
+        continue
+      module_path = os.path.realpath(node.filename)
 
-    _, incoming_edges = graph.get_edges(node)
-    message = 'Discovered %s (Imported by: %s)' % (
-        node.filename, ', '.join(
-            d.filename for d in incoming_edges
-            if d is not None and d.filename is not None))
-    logging.info(message)
+      _, incoming_edges = graph.get_edges(node)
+      message = 'Discovered %s (Imported by: %s)' % (
+          node.filename, ', '.join(
+              d.filename for d in incoming_edges
+              if d is not None and d.filename is not None))
+      logging.info(message)
 
-    # This check is done after the logging/printing above to make sure that we
-    # also print out the dependency edges that include python packages that are
-    # not in chromium.
-    if not path.IsSubpath(module_path, path.GetChromiumSrcDir()):
-      continue
+      # This check is done after the logging/printing above to make sure that
+      # we also print out the dependency edges that include python packages
+      # that are not in chromium.
+      if not path.IsSubpath(module_path, path.GetChromiumSrcDir()):
+        continue
 
-    yield module_path
-    if node.packagepath is not None:
-      for p in node.packagepath:
-        yield p
+      yield module_path
+      if node.packagepath is not None:
+        for p in node.packagepath:
+          yield p
+
+  finally:
+    sys.path = sys_path
 
 
 def FindPageSetDependencies(base_dir):
   logging.info('Finding page sets in %s' % base_dir)
 
   # Add base_dir to path so our imports relative to base_dir will work.
   sys.path.append(base_dir)
   tests = discover.DiscoverClasses(base_dir, base_dir, benchmark.Benchmark,
                                    index_by_class_name=True)
 
@@ skipping 146 matching lines @@
 
   def Run(self, args):
     target_paths = args.positional_args
     dependencies = FindDependencies(target_paths, args)
     if args.zip:
       ZipDependencies(target_paths, dependencies, args)
       print 'Zip archive written to %s.' % args.zip
     else:
       print '\n'.join(sorted(dependencies))
     return 0