src/git_stats/index.py

   1 #!/usr/bin/env python
   2
   3 import os
   4 import sys
   5
   6 from optparse import OptionParser
   7 from git_python import Git
   8
   9 from git_stats import commit
  10
  11 def stagedChanges(ignoreAddedFiles=False):
  12   """Returns the paths to all files that have changes staged.
  13   When calling this function the current directory should be the root
  14   of the git directory that should be acted on.
  15
  16   Params:
  17     ignoreAddedFiles: When True files that added newly are ignored.
  18
  19   Returns:
  20     A list of paths that were changed.
  21   """
  22
  23   git = Git(".")
  24   result = git.diff_index("--cached", "--name-status", "HEAD")
  25
  26   log = result.split('\n')
  27
  28   changed = []
  29
  30   for line in log:
  31     # Skip the last empty line
  32     if len(line.lstrip()) == 0:
  33       continue
  34
  35     splitline = line.split('\t')
  36
  37     # Skip files that were freshly added
  38     if splitline[0] == 'A' and ignoreAddedFiles:
  39       continue
  40
  41     changed.append(splitline[1])
  42
  43   return changed
  44
  45 def touched(showAll):
  46   """Returns which commits touched the same files as the staged changes.
  47
  48   Args:
  49     showAll: When true, all files, even new ones, are searched.
  50
  51   Returns: If there were no staged changes, None is returned.
  52     Otherwise the list of matching commits is returned.
  53   """
  54
  55   staged = stagedChanges(not showAll)
  56
  57   if not staged:
  58     return None
  59
  60   touched = commit.commitsThatTouched(staged)
  61   return touched
  62
  63 def dispatch(*args):
  64   """Dispatches index related commands
  65   """
  66
  67   progname = os.path.basename(sys.argv[0]) + " index"
  68
  69   parser = OptionParser(prog=progname)
  70
  71   parser.add_option(
  72       "-a", "--all",
  73       action="store_true",
  74       help="search even new files, which usually have no matching commits")
  75
  76   (options, args) = parser.parse_args(list(args))
  77
  78   result = touched(options.all)
  79
  80   if not result:
  81     print("No changes staged (or only new files added)")
  82     return
  83
  84   commit.prettyPrint(result)
  85