Make .pyc cleaning optional in clean_local_files.

scripts/slave/recipe_modules/chromium_android/api.py

 # Copyright 2013 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 contextlib
 import datetime
 import json
 import os
 import pipes
 import re
 import sys
+import textwrap
 import urllib
 
 from recipe_engine.types import freeze
 from recipe_engine import recipe_api
 
 def _TimestampToIsoFormat(timestamp):
   return datetime.datetime.utcfromtimestamp(timestamp).strftime('%Y%m%dT%H%M%S')
 
 
 class AndroidApi(recipe_api.RecipeApi):
@@ skipping 108 matching lines @@
 
     # TODO(sivachandra): Manufacture gclient spec such that it contains "src"
     # solution + repo_name solution. Then checkout will be automatically
     # correctly set by gclient.checkout
     self.m.path['checkout'] = self.m.path['slave_build'].join('src')
 
     self.clean_local_files()
 
     return result
 
-  def clean_local_files(self):
+  def clean_local_files(self, clean_pyc_files=True):
     target = self.c.BUILD_CONFIG
     debug_info_dumps = self.m.path['checkout'].join('out',
                                                     target,
                                                     'debug_info_dumps')
     test_logs = self.m.path['checkout'].join('out', target, 'test_logs')
     build_product = self.m.path['checkout'].join('out', 'build_product.zip')
-    self.m.python.inline(
-        'clean local files',
-        """
-          import shutil, sys, os
-          shutil.rmtree(sys.argv[1], True)
-          shutil.rmtree(sys.argv[2], True)
-          try:
-            os.remove(sys.argv[3])
-          except OSError:
-            pass
+    python_inline_script = textwrap.dedent("""
+        import shutil, sys, os
+        shutil.rmtree(sys.argv[1], True)
+        shutil.rmtree(sys.argv[2], True)
+        try:
+          os.remove(sys.argv[3])
+        except OSError:
+          pass
+        """)
+    if clean_pyc_files:
+      python_inline_script += textwrap.dedent("""\
           for base, _dirs, files in os.walk(sys.argv[4]):
             for f in files:
               if f.endswith('.pyc'):
                 os.remove(os.path.join(base, f))
-        """,
-        args=[debug_info_dumps, test_logs, build_product,
-              self.m.path['checkout']],
-        infra_step=True,
+      """)
+
+    self.m.python.inline(
+      'clean local files',
+      python_inline_script,
+      args=[debug_info_dumps, test_logs, build_product,
+            self.m.path['checkout']],
+      infra_step=True,
     )
 
   def run_tree_truth(self, additional_repos=None):
     # TODO(sivachandra): The downstream ToT builder will require
     # 'Show Revisions' step.
     repos = ['src']
     if additional_repos:
       repos.extend(additional_repos)
     if self.c.REPO_NAME not in repos and self.c.REPO_NAME:
       repos.append(self.c.REPO_NAME)
@@ skipping 1391 matching lines @@
       script = self.c.test_runner
       if wrapper_script_suite_name:
         script = self.m.chromium.output_dir.join('bin', 'run_%s' %
                                                  wrapper_script_suite_name)
       else:
         env = kwargs.get('env', {})
         env['CHROMIUM_OUTPUT_DIR'] = env.get('CHROMIUM_OUTPUT_DIR',
                                              self.m.chromium.output_dir)
         kwargs['env'] = env
       return self.m.python(step_name, script, args, **kwargs)