src/git_stats/config.py

   1 #!/usr/bin/env python
   2
   3
   4 import os
   5
   6 from git import Repo
   7
   8 def readAll(convert_camel_case=True):
   9   """Reads all config files and returns the result
  10
  11   When a value is read in multiple times the one read in
  12   the last is used. The following files are tried in the
  13   listed order:
  14   ~/.gitconfig
  15   .git/config
  16   config
  17   """
  18
  19   paths = []
  20
  21   path = "~/.gitconfig"
  22   path = os.path.expanduser(path)
  23   paths.append(path)
  24
  25   repo = Repo(".")
  26   path = os.path.join(repo.wd, '.git', 'config')
  27   paths.append(path)
  28
  29   paths.append("config")
  30
  31   result = {}
  32
  33   for path in paths:
  34     if not os.path.isfile(path):
  35       continue
  36
  37     config = read(path, convert_camel_case)
  38
  39     for key, value in config.iteritems():
  40       result[key] = value
  41
  42   return result
  43
  44 def _readVerse(verse, config, convert_camel_case):
  45   """Reads in a verse from a config file and stores the result in config
  46   """
  47
  48   for line in verse:
  49     # Remove spaces and newlines
  50     stripline = line.lstrip().rstrip()
  51
  52     if stripline == line.rstrip() or not stripline:
  53       break
  54
  55     # Extract the key:value pair
  56     splitline = stripline.split(' = ')
  57     key = splitline[0].rstrip()
  58     value = splitline[1].lstrip()
  59
  60     # Try to convert to a number
  61     try:
  62       intval = int(value)
  63       value = intval
  64     except ValueError:
  65       pass
  66
  67     if value == "False":
  68       value = False
  69
  70     if value == "True":
  71       value = True
  72
  73     if value == "None":
  74       value = None
  75
  76     if convert_camel_case and key.lower() != key:
  77       res = ""
  78
  79       for c in key:
  80         if c.isupper():
  81           res += '_'
  82
  83         res += c.lower()
  84
  85       key = res
  86
  87     config[key] = value
  88
  89
  90 def read(path, convert_camel_case=True):
  91   """Reads in the specified file
  92
  93   Understands 'True', 'False', 'None' and integer values.
  94   All camelCase keys are converted to dashed_form.
  95   The file is expected to have the following format:
  96   .*
  97   [GitStats]\n
  98     [key = value\n]*
  99   <any unindented or empty line>
 100   .*
 101
 102   Args:
 103     path: The location of the file to read in
 104
 105   Returns: A dictionary with the key:value pairs specified in the file.
 106   """
 107
 108   config = {}
 109
 110   file = open(path)
 111   lines = file.readlines()
 112
 113   while True:
 114     # Find our verse
 115     try:
 116       pos = lines.index("[GitStats]\n")
 117     except ValueError:
 118       return config
 119
 120     # Kick off the header
 121     pos += 1
 122     lines = lines[pos:]
 123
 124     _readVerse(lines, config, convert_camel_case)
 125
 126   return config
 127
 128 if __name__ == '__main__':
 129   result = readAll()
 130
 131   print("convert_camel_case is ON:")
 132   for key, value in result.iteritems():
 133     print(str(key) + ": " + str(value))
 134
 135   result = readAll(False)
 136
 137   print("")
 138   print("convertCamelCase is OFF:")
 139   for key, value in result.iteritems():
 140     print(str(key) + ": " + str(value))
 141