cola/dag.py

   1 import argparse
   2 import sys
   3
   4 from cola import app
   5 from cola.widgets.dag import git_dag
   6
   7
   8 def main(argv=None):
   9     """Run git-dag"""
  10     app.initialize()
  11     args = parse_args(argv=argv)
  12     return args.func(args)
  13
  14
  15 def shortcut_launch():
  16     """Run git-dag from a Windows shortcut"""
  17     return main(argv=['--prompt'])
  18
  19
  20 def parse_args(argv=None):
  21     """Parse command-line arguments"""
  22     if argv is None:
  23         argv = sys.argv[1:]
  24     parser = argparse.ArgumentParser()
  25     parser.set_defaults(func=cmd_dag)
  26
  27     app.add_common_arguments(parser)
  28     parser.add_argument(
  29         '-c',
  30         '--count',
  31         '--max-count',
  32         metavar='<count>',
  33         type=int,
  34         default=1000,
  35         help='number of commits to display',
  36     )
  37     parser.add_argument('args', nargs='*', metavar='<args>', help='git log arguments')
  38     args, rest = parser.parse_known_args(args=argv)
  39     if rest:
  40         # splice unknown arguments to the beginning ~
  41         # these are forwarded to git-log(1).
  42         args.args[:0] = rest
  43     return args
  44
  45
  46 def cmd_dag(args):
  47     """Run git-dag via the `git cola dag` sub-command"""
  48     context = app.application_init(args)
  49     view = git_dag(context, args=args, show=False)
  50     return app.application_start(context, view)