view python/cli.py @ 375:9314c1008189

aslightly better; still trouble at da end
author Jeff Hammel <jhammel@mozilla.com>
date Wed, 24 Jul 2013 03:10:27 -0700
parents e7948549afa1
children
line wrap: on
line source

#!/usr/bin/env python

"""
program illustrating command line in the form of
``--option foo`` or ``--option=foo`` goes to a (key,value) paid
and ``-tag`` gets appended to a list.  Further options go to a further list::

    >>> main(['-foo', '--bar=fleem', 'baz'])
    (['foo'], {'bar': 'fleem'}, ['baz'])
"""

import sys

class ParserError(Exception):
  """error for exceptions while parsing the command line"""

def main(_args=sys.argv[1:]):

  # return values
  _dict = {}
  tags = []
  args = []

  # parse the arguments
  key = None
  for arg in _args:
    if arg.startswith('---'):
      raise ParserError("arguments should start with '-' or '--' only")
    elif arg.startswith('--'):
      if key:
        raise ParserError("Key %s still open" % key)
      key = arg[2:]
      if '=' in key:
        key, value = key.split('=', 1)
        _dict[key] = value
        key = None
        continue
    elif arg.startswith('-'):
      if key:
        raise ParserError("Key %s still open" % key)
      tags.append(arg[1:])
      continue
    else:
      if key:
        _dict[key] = arg
        continue
      args.append(arg)

  # return values
  return (_dict, tags, args)

if __name__ == '__main__':
  try:
    _dict, tags, args = main()
  except ParserError, e:
    import pdb; pdb.set_trace() # for debugging
  print _dict
  print tags
  print args