src/git_stats/stats.py

   1 #!/usr/bin/env python
   2
   3 """This module contains commands related to statistics on git repositories.
   4
   5 Each command here is merely a wrapper around other, more complicated, commands.
   6 """
   7
   8 import os
   9
  10 # Figure out what we are running from
  11 # Then get the directory we are in and use that instead
  12 # Ofcourse, this only works if git_python is in the dir we are run from
  13
  14 file = __file__
  15 path = os.path.abspath(file)
  16 dir = os.path.dirname(path)
  17
  18 split = os.path.split(dir)
  19 dir = split[0]
  20
  21 if not split[1] == "git_stats":
  22   raise Exception("Please run this script inside the git_stats directory")
  23
  24 os.sys.path.insert(0, dir)
  25
  26 from git_python import Git
  27
  28 from git_stats import author
  29 from git_stats import branch
  30 from git_stats import commit
  31 from git_stats import diff
  32 from git_stats import index
  33
  34 class DispatchException(Exception):
  35   """This exception is raised when something went wrong during dispatching
  36   """
  37
  38   pass
  39
  40 class Dispatcher():
  41   """This class provides basic dispatching functionality
  42   """
  43
  44   def __init__(self, commands):
  45     self.commands = commands
  46
  47   def showUsageMessage(self):
  48     print("Available commands are:")
  49     for key in self.commands.iterkeys():
  50       print(key)
  51
  52   def dispatch(self, argv):
  53     if not len(argv) > 1:
  54       self.showUsageMessage()
  55       return 1
  56
  57     command = argv[1]
  58
  59     for key, value in self.commands.iteritems():
  60       if key.startswith(command):
  61         func = value
  62         break
  63     else:
  64       raise DispatchException("Unknown command '" + command + "'.")
  65
  66     return func(*argv[2:])
  67
  68 commands = {
  69   "author" : author.dispatch,
  70   "branch" : branch.dispatch,
  71   "commit" : commit.dispatch,
  72   "diff"   :   diff.dispatch,
  73   "index"  :  index.dispatch,
  74 }
  75
  76
  77 if __name__ == '__main__':
  78   import sys
  79   result = Dispatcher(commands).dispatch(sys.argv)
  80   sys.exit(result)
  81