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 bug
  24 from git_stats import branch
  25 from git_stats import commit
  26 from git_stats import diff
  27 from git_stats import index
  28 from git_stats import matcher
  29 from git_stats import tests
  30
  31 def version(*args):
  32   """Shows a silly version message
  33   """
  34
  35   print("Current version is 0.1-alpha_pre_dinner_build")
  36   return 0
  37
  38 class DispatchException(Exception):
  39   """This exception is raised when something went wrong during dispatching
  40   """
  41
  42   pass
  43
  44 class Dispatcher():
  45   """This class provides basic dispatching functionality
  46   """
  47
  48
  49   def __init__(self, commands):
  50     self.commands = commands
  51
  52   def showUsageMessage(self):
  53     """Shows a usage message, listing all available commands
  54     """
  55
  56     print("Available commands are:")
  57
  58     for key in sorted(self.commands):
  59       print(key)
  60
  61   def dispatch(self, argv):
  62     """Dispatches a command with the specified arguments
  63
  64     Args:
  65       argv: The arguments to parse and dispatch.
  66     """
  67
  68     if not len(argv) > 0:
  69       self.showUsageMessage()
  70       return 1
  71
  72     command = argv[0]
  73
  74     for key, value in self.commands.iteritems():
  75       if key.startswith(command):
  76         func = value
  77         break
  78     else:
  79       raise DispatchException("Unknown command '" + command + "'.")
  80
  81     # When not specifying a command, throw in the --help switch
  82     if len(argv) == 1:
  83       argv.append("--help")
  84
  85     return func(*argv[1:])
  86
  87 commands = {
  88   "author" : author.dispatch,
  89   "bug"    :    bug.dispatch,
  90   "branch" : branch.dispatch,
  91   "commit" : commit.dispatch,
  92   "diff"   :   diff.dispatch,
  93   "index"  :  index.dispatch,
  94   "matcher":matcher.dispatch,
  95   "tests"  :  tests.dispatch,
  96   "-v"     :         version,
  97 }
  98
  99 def main(args):
 100   """Main routine, dispatches the command specified in args
 101
 102   If called with sys.argv, cut off the first element!
 103   """
 104
 105   result = Dispatcher(commands).dispatch(args)
 106   return result
 107
 108 if __name__ == '__main__':
 109   import sys
 110   result = main(sys.argv[1:])
 111   sys.exit(result)
 112