Fix python scripts in src/tools/

tools/heapcheck/chrome_tests.py

Index: tools/heapcheck/chrome_tests.py
diff --git a/tools/heapcheck/chrome_tests.py b/tools/heapcheck/chrome_tests.py
index e4cadd2ccff599f041b8ac2c1b15472881082a52..f01231d3175b6ab4217be4ed02847a75c8a8621a 100755
--- a/tools/heapcheck/chrome_tests.py
+++ b/tools/heapcheck/chrome_tests.py
@@ -1,5 +1,4 @@
-#!/usr/bin/python
-
+#!/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.
@@ -417,7 +416,11 @@ class ChromeTests(object):
     # summary list for long, but will be useful for someone reviewing this bot.
     return ret
 
-def _main(_):
+
+def main():
+  if not sys.platform.startswith('linux'):
+    logging.error("Heap checking works only on Linux at the moment.")
+    return 1
   parser = optparse.OptionParser("usage: %prog -b <dir> -t <test> "
                                  "[-t <test> ...]")
   parser.disable_interspersed_args()
@@ -457,9 +460,4 @@ def _main(_):
 
 
 if __name__ == "__main__":
-  if sys.platform.startswith('linux'):
-    ret = _main(sys.argv)
-  else:
-    logging.error("Heap checking works only on Linux at the moment.")
-    ret = 1
-  sys.exit(ret)
+  sys.exit(main())