src/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 environname = "GIT_STATS_PATH"
  11
  12 # Try to retreive the path from the environment
  13 # This is so that we can then import from git_python
  14 if os.environ.has_key(environname):
  15   dir = os.environ[environname]
  16   dir = os.path.expanduser(dir)
  17   dir = os.path.expandvars(dir)
  18 else:
  19   # Figure out what we are running from
  20   # Then get the directory we are in and use that instead
  21   # Ofcourse, this only works if git_python is in the dir we are run from
  22
  23   file = __file__
  24   path = os.path.abspath(file)
  25   dir = os.path.dirname(path)
  26
  27   if not os.path.isdir(os.path.join(dir, "git_stats")):
  28     raise Exception("Cannot find git_stats directory, please set $GIT_STATS_PATH")
  29
  30 os.sys.path.insert(0, dir)
  31
  32 from git_python import Git
  33
  34 from git_stats import author
  35 from git_stats import branch
  36 from git_stats import commit
  37 from git_stats import diff
  38 from git_stats import index
  39
  40 class DispatchException(Exception):
  41   """This exception is raised when something went wrong during dispatching
  42   """
  43
  44   pass
  45
  46 class Dispatcher():
  47   """This class provides basic dispatching functionality
  48   """
  49
  50   def __init__(self, commands):
  51     self.commands = commands
  52
  53   def showUsageMessage(self):
  54     print("Available commands are:")
  55     for key in self.commands.iterkeys():
  56       print(key)
  57
  58   def dispatch(self, argv):
  59     if not len(argv) > 1:
  60       self.showUsageMessage()
  61       return 1
  62
  63     command = argv[1]
  64
  65     for key, value in self.commands.iteritems():
  66       if key.startswith(command):
  67         func = value
  68         break
  69     else:
  70       raise DispatchException("Unknown command '" + command + "'.")
  71
  72     return func(*argv[2:])
  73
  74 commands = {
  75   "author" : author.dispatch,
  76   "branch" : branch.dispatch,
  77   "commit" : commit.dispatch,
  78   "diff"   :   diff.dispatch,
  79   "index"  :  index.dispatch,
  80 }
  81
  82
  83 if __name__ == '__main__':
  84   import sys
  85   result = Dispatcher(commands).dispatch(sys.argv)
  86   sys.exit(result)
  87