tools/metrics/histograms/update_net_error_codes.py

   1 #!/usr/bin/env python
   2 # Copyright 2014 The Chromium Authors. All rights reserved.
   3 # Use of this source code is governed by a BSD-style license that can be
   4 # found in the LICENSE file.
   5
   6 """Updates NetErrorCodes enum in histograms.xml file with values read
   7  from net_error_list.h.
   8
   9 If the file was pretty-printed, the updated version is pretty-printed too.
  10 """
  11
  12 import os.path
  13 import re
  14 import sys
  15
  16 sys.path.append(os.path.join(os.path.dirname(__file__), '..', 'common'))
  17 import path_util
  18
  19 import update_histogram_enum
  20
  21 NET_ERROR_LIST_PATH = 'net/base/net_error_list.h'
  22
  23 POSITIVE_ERROR_REGEX = re.compile(r'^NET_ERROR\(([\w]+), -([0-9]+)\)')
  24 NEGATIVE_ERROR_REGEX = re.compile(r'^NET_ERROR\(([\w]+), (-[0-9]+)\)')
  25
  26 def ReadNetErrorCodes(filename, error_regex):
  27   """Reads in values from net_error_list.h, returning a dictionary mapping
  28   error code to error name.
  29   """
  30   # Read the file as a list of lines
  31   with open(path_util.GetInputFile(filename)) as f:
  32     content = f.readlines()
  33
  34   # Parse out lines that are net errors.
  35   errors = {}
  36   for line in content:
  37     m = error_regex.match(line)
  38     if m:
  39       errors[int(m.group(2))] = m.group(1)
  40   return errors
  41
  42
  43 def main():
  44   if len(sys.argv) > 1:
  45     print >>sys.stderr, 'No arguments expected!'
  46     sys.stderr.write(__doc__)
  47     sys.exit(1)
  48
  49   update_histogram_enum.UpdateHistogramFromDict(
  50       'NetErrorCodes',
  51       ReadNetErrorCodes(NET_ERROR_LIST_PATH, POSITIVE_ERROR_REGEX),
  52       NET_ERROR_LIST_PATH)
  53
  54   update_histogram_enum.UpdateHistogramFromDict(
  55       'CombinedHttpResponseAndNetErrorCode',
  56       ReadNetErrorCodes(NET_ERROR_LIST_PATH, NEGATIVE_ERROR_REGEX),
  57       NET_ERROR_LIST_PATH)
  58
  59 if __name__ == '__main__':
  60   main()