Fix python scripts in src/tools/

tools/python/google/gethash_timer.py

-#!/usr/bin/python
-# Copyright (c) 2006-2008 The Chromium Authors. All rights reserved.
+#!/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.
 
-# Issue a series of GetHash requests to the SafeBrowsing servers and measure the
-# response times.
-#
-# Usage:
-#
-#   $ ./gethash_timer.py --period=600 --samples=20 --output=resp.csv
-#
-#   --period (or -p):  The amount of time (in seconds) to wait between GetHash
-#                      requests. Using a value of more than 300 (5 minutes) to
-#                      include the effect of DNS.
-#
-#   --samples (or -s): The number of requests to issue. If this parameter is not
-#                      specified, the test will run indefinitely.
-#
-#   --output (or -o):  The path to a file where the output will be written in
-#                      CSV format: sample_number,response_code,elapsed_time_ms
+"""Issue a series of GetHash requests to the SafeBrowsing servers and measure
+the response times.
+
+Usage:
+
+  $ ./gethash_timer.py --period=600 --samples=20 --output=resp.csv
+
+  --period (or -p):  The amount of time (in seconds) to wait between GetHash
+                     requests. Using a value of more than 300 (5 minutes) to
+                     include the effect of DNS.
+
+  --samples (or -s): The number of requests to issue. If this parameter is not
+                     specified, the test will run indefinitely.
+
+  --output (or -o):  The path to a file where the output will be written in
+                     CSV format: sample_number,response_code,elapsed_time_ms
+"""
 
 import getopt
 import httplib
 import sys
 import time
 
 _GETHASH_HOST = 'safebrowsing.clients.google.com'
-_GETHASH_REQUEST = '/safebrowsing/gethash?client=googleclient&appver=1.0&pver=2.1'
+_GETHASH_REQUEST = (
+    '/safebrowsing/gethash?client=googleclient&appver=1.0&pver=2.1')
 
 # Global logging file handle.
 g_file_handle = None
 
 
 def IssueGetHash(prefix):
   '''Issue one GetHash request to the safebrowsing servers.
   Args:
     prefix: A 4 byte value to look up on the server.
   Returns:
@@ skipping 68 matching lines @@
   '''Open a file for logging results.
   Args:
     file_name: A path to a file to store the output.
   Returns:
     None.
   '''
   global g_file_handle
   g_file_handle = open(file_name, 'w')
 
 
-if __name__ == '__main__':
+def main():
   period = 10
   samples = None
 
   options, args = getopt.getopt(sys.argv[1:],
                                 's:p:o:',
                                 ['samples=', 'period=', 'output='])
   for option, value in options:
     if option == '-s' or option == '--samples':
       samples = int(value)
     elif option == '-p' or option == '--period':
       period = float(value)
     elif option == '-o' or option == '--output':
       file_name = value
     else:
       print 'Bad option: %s' % option
-      sys.exit(1)
+      return 1
   try:
     print 'Starting Timed GetHash ----------'
     SetupOutputFile(file_name)
     RunTimedGetHash(period, samples)
   except KeyboardInterrupt:
     pass
 
   print 'Timed GetHash complete ----------'
   g_file_handle.close()
 
+
+if __name__ == '__main__':
+  sys.exit(main())