Inhibit python subprocess from generating .pyc files anymore.

presubmit_support.py

 #!/usr/bin/env python
 # Copyright (c) 2011 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.
 
 """Enables directory-specific presubmit checks to run at upload and/or commit.
 """
 
 __version__ = '1.6.1'
 
@@ skipping 1017 matching lines @@
     rietveld: rietveld object.
 
   Warning:
     If may_prompt is true, output_stream SHOULD be sys.stdout and input_stream
     SHOULD be sys.stdin.
 
   Return:
     A PresubmitOutput object. Use output.should_continue() to figure out
     if there were errors or warnings and the caller should abort.
   """
-  output = PresubmitOutput(input_stream, output_stream)
-  if committing:
-    output.write("Running presubmit commit checks ...\n")
-  else:
-    output.write("Running presubmit upload checks ...\n")
-  start_time = time.time()
-  presubmit_files = ListRelevantPresubmitFiles(change.AbsoluteLocalPaths(True),
-                                               change.RepositoryRoot())
-  if not presubmit_files and verbose:
-    output.write("Warning, no presubmit.py found.\n")
-  results = []
-  executer = PresubmitExecuter(change, committing, tbr, rietveld, verbose)
-  if default_presubmit:
-    if verbose:
-      output.write("Running default presubmit script.\n")
-    fake_path = os.path.join(change.RepositoryRoot(), 'PRESUBMIT.py')
-    results += executer.ExecPresubmitScript(default_presubmit, fake_path)
-  for filename in presubmit_files:
-    filename = os.path.abspath(filename)
-    if verbose:
-      output.write("Running %s\n" % filename)
-    # Accept CRLF presubmit script.
-    presubmit_script = gclient_utils.FileRead(filename, 'rU')
-    results += executer.ExecPresubmitScript(presubmit_script, filename)
+  old_environ = os.environ
+  try:
+    # Make sure python subprocesses won't generate .pyc files.
+    os.environ = os.environ.copy()
+    os.environ['PYTHONDONTWRITEBYTECODE'] = '1'
 
-  errors = []
-  notifications = []
-  warnings = []
-  for result in results:
-    if result.fatal:
-      errors.append(result)
-    elif result.should_prompt:
-      warnings.append(result)
+    output = PresubmitOutput(input_stream, output_stream)
+    if committing:
+      output.write("Running presubmit commit checks ...\n")
     else:
-      notifications.append(result)
+      output.write("Running presubmit upload checks ...\n")
+    start_time = time.time()
+    presubmit_files = ListRelevantPresubmitFiles(
+        change.AbsoluteLocalPaths(True), change.RepositoryRoot())
+    if not presubmit_files and verbose:
+      output.write("Warning, no presubmit.py found.\n")
+    results = []
+    executer = PresubmitExecuter(change, committing, tbr, rietveld, verbose)
+    if default_presubmit:
+      if verbose:
+        output.write("Running default presubmit script.\n")
+      fake_path = os.path.join(change.RepositoryRoot(), 'PRESUBMIT.py')
+      results += executer.ExecPresubmitScript(default_presubmit, fake_path)
+    for filename in presubmit_files:
+      filename = os.path.abspath(filename)
+      if verbose:
+        output.write("Running %s\n" % filename)
+      # Accept CRLF presubmit script.
+      presubmit_script = gclient_utils.FileRead(filename, 'rU')
+      results += executer.ExecPresubmitScript(presubmit_script, filename)
 
-  output.write('\n')
-  for name, items in (('Messages', notifications),
-                      ('Warnings', warnings),
-                      ('ERRORS', errors)):
-    if items:
-      output.write('** Presubmit %s **\n' % name)
-      for item in items:
-        item.handle(output)
-        output.write('\n')
+    errors = []
+    notifications = []
+    warnings = []
+    for result in results:
+      if result.fatal:
+        errors.append(result)
+      elif result.should_prompt:
+        warnings.append(result)
+      else:
+        notifications.append(result)
 
-  total_time = time.time() - start_time
-  if total_time > 1.0:
-    output.write("Presubmit checks took %.1fs to calculate.\n\n" % total_time)
+    output.write('\n')
+    for name, items in (('Messages', notifications),
+                        ('Warnings', warnings),
+                        ('ERRORS', errors)):
+      if items:
+        output.write('** Presubmit %s **\n' % name)
+        for item in items:
+          item.handle(output)
+          output.write('\n')
 
-  if not errors:
-    if not warnings:
-      output.write('Presubmit checks passed.\n')
-    elif may_prompt:
-      output.prompt_yes_no('There were presubmit warnings. '
-                          'Are you sure you wish to continue? (y/N): ')
-    else:
-      output.fail()
+    total_time = time.time() - start_time
+    if total_time > 1.0:
+      output.write("Presubmit checks took %.1fs to calculate.\n\n" % total_time)
 
-  global _ASKED_FOR_FEEDBACK
-  # Ask for feedback one time out of 5.
-  if (len(results) and random.randint(0, 4) == 0 and not _ASKED_FOR_FEEDBACK):
-    output.write("Was the presubmit check useful? Please send feedback "
-                 "& hate mail to maruel@chromium.org!\n")
-    _ASKED_FOR_FEEDBACK = True
-  return output
+    if not errors:
+      if not warnings:
+        output.write('Presubmit checks passed.\n')
+      elif may_prompt:
+        output.prompt_yes_no('There were presubmit warnings. '
+                            'Are you sure you wish to continue? (y/N): ')
+      else:
+        output.fail()
+
+    global _ASKED_FOR_FEEDBACK
+    # Ask for feedback one time out of 5.
+    if (len(results) and random.randint(0, 4) == 0 and not _ASKED_FOR_FEEDBACK):
+      output.write("Was the presubmit check useful? Please send feedback "
+                  "& hate mail to maruel@chromium.org!\n")
+      _ASKED_FOR_FEEDBACK = True
+    return output
+  finally:
+    os.environ = old_environ
 
 
 def ScanSubDirs(mask, recursive):
   if not recursive:
     return [x for x in glob.glob(mask) if '.svn' not in x and '.git' not in x]
   else:
     results = []
     for root, dirs, files in os.walk('.'):
       if '.svn' in dirs:
         dirs.remove('.svn')
@@ skipping 91 matching lines @@
   except PresubmitFailure, e:
     print >> sys.stderr, e
     print >> sys.stderr, 'Maybe your depot_tools is out of date?'
     print >> sys.stderr, 'If all fails, contact maruel@'
     return 2
 
 
 if __name__ == '__main__':
   fix_encoding.fix_encoding()
   sys.exit(Main(None))