Gooey/gooey/mockapplications/example_argparse_souce_in_try.py

#!/usr/local/bin/python2.7
# encoding: utf-8
'''
bin.example_argparse_souce -- shortdesc

bin.example_argparse_souce is a description

It defines classes_and_methods

@author:     user_name

@copyright:  2013 organization_name. All rights reserved.

@license:    license

@contact:    user_email
@deffield    updated: Updated
'''

import sys
import os

from argparse import ArgumentParser
from argparse import RawDescriptionHelpFormatter

from gooey.gooey_decorator import Gooey

__all__ = []
__version__ = 0.1
__date__ = '2013-12-13'
__updated__ = '2013-12-13'

DEBUG = 0
TESTRUN = 0
PROFILE = 0


class CLIError(Exception):
  '''Generic exception to raise and log different fatal errors.'''

  def __init__(self, msg):
    super(CLIError).__init__(type(self))
    self.msg = "E: %s" % msg

  def __str__(self):
    return self.msg

  def __unicode__(self):
    return self.msg


@Gooey
def main(argv=None):  # IGNORE:C0111
  '''Command line options.'''

  if argv is None:
    argv = sys.argv
  else:
    sys.argv.extend(argv)

  program_name = os.path.basename(sys.argv[0])
  program_version = "v%s" % __version__
  program_build_date = str(__updated__)
  program_version_message = '%%(prog)s %s (%s)' % (program_version, program_build_date)
  program_shortdesc = __import__('__main__').__doc__.split("\n")[1]
  program_license = '''%s

  Created by user_name on %s.
  Copyright 2013 organization_name. All rights reserved.

  Licensed under the Apache License 2.0
  http://www.apache.org/licenses/LICENSE-2.0

  Distributed on an "AS IS" basis without warranties
  or conditions of any kind, either express or implied.

USAGE
''' % (program_shortdesc, str(__date__))

  try:
    # Setup argument parser
    parser = ArgumentParser(description='Example Argparse Program', formatter_class=RawDescriptionHelpFormatter)

    parser.add_argument("filename", help="filename")

    parser.add_argument("-r", "--recursive", dest="recurse", action="store_true",
                        help="recurse into subfolders [default: %(default)s]")

    parser.add_argument("-v", "--verbose", dest="verbose", action="count",
                        help="set verbosity level [default: %(default)s]")

    parser.add_argument("-i", "--include", action="append", nargs='+',
                        help="only include paths matching this regex pattern. Note: exclude is given preference over include. ",
                        metavar="RE")

    parser.add_argument("-m", "--mycoolargument", help="mycoolargument")

    parser.add_argument("-e", "--exclude", dest="exclude",
                        help="exclude paths matching this regex pattern. [default: %(default)s]", metavar="RE")

    parser.add_argument('-V', '--version', action='version')

    parser.add_argument('-T', '--tester', choices=['yes', 'no'])

    parser.add_argument(dest="paths", help="paths to folder(s) with source file(s) [default: %(default)s]",
                        metavar="path", nargs='+')

    # 				for i in parser._actions:
    # 					print i
    # Process arguments
    args = parser.parse_args()

    paths = args.paths
    verbose = args.verbose
    recurse = args.recurse
    inpat = args.include
    expat = args.exclude

    if verbose > 0:
      print("Verbose mode on")
      if recurse:
        print("Recursive mode on")
      else:
        print("Recursive mode off")

    if inpat and expat and inpat == expat:
      raise CLIError("include and exclude pattern are equal! Nothing will be processed.")

    for inpath in paths:
      ### do something with inpath ###
      print(inpath)
    return 0
  except KeyboardInterrupt:
    ### handle keyboard interrupt ###
    return 0
  except Exception, e:
    if DEBUG or TESTRUN:
      raise (e)
    indent = len(program_name) * " "
    sys.stderr.write(program_name + ": " + repr(e) + "\n")
    sys.stderr.write(indent + "  for help use --help")
    return 2


if __name__ == "__main__":
  if DEBUG:
    sys.argv.append("-h")
    # 				sys.argv.append("-v")
    sys.argv.append("-r")
    main()
    sys.exit()
  if TESTRUN:
    import doctest

    doctest.testmod()
  if PROFILE:
    import cProfile
    import pstats

    profile_filename = 'bin.example_argparse_souce_profile.txt'
    cProfile.run('main()', profile_filename)
    statsfile = open("profile_stats.txt", "wb")
    p = pstats.Stats(profile_filename, stream=statsfile)
    stats = p.strip_dirs().sort_stats('cumulative')
    stats.print_stats()
    statsfile.close()
    sys.exit(0)
  sys.exit(main())