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 # 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 os.sys.path.insert(0, dir)
  19
  20 from git import Git
  21
  22 from git_stats import author
  23 from git_stats import branch
  24 from git_stats import commit
  25 from git_stats import diff
  26 from git_stats import index
  27
  28 class DispatchException(Exception):
  29   """This exception is raised when something went wrong during dispatching
  30   """
  31
  32   pass
  33
  34 class Dispatcher():
  35   """This class provides basic dispatching functionality
  36   """
  37
  38   def __init__(self, commands):
  39     self.commands = commands
  40
  41   def showUsageMessage(self):
  42     print("Available commands are:")
  43     for key in self.commands.iterkeys():
  44       print(key)
  45
  46   def dispatch(self, argv):
  47     if not len(argv) > 1:
  48       self.showUsageMessage()
  49       return 1
  50
  51     command = argv[1]
  52
  53     for key, value in self.commands.iteritems():
  54       if key.startswith(command):
  55         func = value
  56         break
  57     else:
  58       raise DispatchException("Unknown command '" + command + "'.")
  59
  60     # When not specifying a command, throw in the --help switch
  61     if len(argv) == 2:
  62       argv.append("--help")
  63
  64     return func(*argv[2:])
  65
  66 commands = {
  67   "author" : author.dispatch,
  68   "branch" : branch.dispatch,
  69   "commit" : commit.dispatch,
  70   "diff"   :   diff.dispatch,
  71   "index"  :  index.dispatch,
  72 }
  73
  74
  75 if __name__ == '__main__':
  76   import sys
  77   result = Dispatcher(commands).dispatch(sys.argv)
  78   sys.exit(result)
  79