tools/telemetry/catapult_base/refactor/__init__.py

   1 # Copyright 2015 The Chromium Authors. All rights reserved.
   2 # Use of this source code is governed by a BSD-style license that can be
   3 # found in the LICENSE file.
   4
   5 """Style-preserving Python code transforms.
   6
   7 This module provides components for modifying and querying Python code. They can
   8 be used to build custom refactorings and linters.
   9 """
  10
  11 import functools
  12 import multiprocessing
  13
  14 # pylint: disable=wildcard-import
  15 from catapult_base.refactor.annotated_symbol import *
  16 from catapult_base.refactor.module import Module
  17
  18
  19 def _TransformFile(transform, file_path):
  20   module = Module(file_path)
  21   result = transform(module)
  22   module.Write()
  23   return result
  24
  25
  26 def Transform(transform, file_paths):
  27   transform = functools.partial(_TransformFile, transform)
  28   return multiprocessing.Pool().map(transform, file_paths)