src/git_stats/index.py

   1 #!/usr/bin/env python
   2
   3 from git_python import Git
   4
   5 def stagedChanges(ignoreAddedFiles):
   6   """Returns the paths to all files that have changes staged.
   7   When calling this function the current directory should be the root
   8   of the git directory that should be acted on.
   9
  10   Params:
  11     ignoreAddedFiles: When True files that added newly are ignored.
  12
  13   Returns:
  14     A list of paths that were changed.
  15   """
  16
  17   git = Git(".")
  18   result = git.diff_index("--cached", "--name-status", "HEAD")
  19
  20   log = result.split('\n')
  21
  22   changed = []
  23
  24   for line in log:
  25     # Skip the last empty line
  26     if len(line.lstrip()) == 0:
  27       continue
  28
  29     splitline = line.split('\t')
  30
  31     # Skip files that were freshly added
  32     if splitline[0] == 'A' and ignoreAddedFiles:
  33       continue
  34
  35     changed.append(splitline[1])
  36
  37   return changed
  38
  39 if __name__ == '__main__':
  40   import os
  41   os.chdir("../..")
  42
  43   staged = stagedChanges(True)
  44
  45   if not staged:
  46     print("No changes staged (or only new files added)")
  47   else:
  48     for path in staged:
  49       print(path)
  50