def __init__(self, entity_class): entity_name_lower = naming.convert(entity_class.__name__, 'camelcase', 'lowercase_underscore') self._indent = '' self._f = open(os.path.join(TEST_DIR, '{0}.py'.format(entity_name_lower)), 'a')
entity_domains_parser = entity_subparsers.add_parser('domains') entity_domains_parser.set_defaults(func=entity_domains) entity_search_parser = entity_subparsers.add_parser('search') entity_search_parser.add_argument('query') entity_search_parser.add_argument('--fields', default=None) entity_search_parser.set_defaults(func=entity_search) entity_list_parser = entity_subparsers.add_parser('list') entity_list_parser.add_argument('domain') entity_list_parser.add_argument('--fields', default=None) entity_list_parser.set_defaults(func=entity_list) if name == 'user': add_extra_user_parsers(entity_subparsers) for entity_class in get_entity_classes(): add_entity_parser(entity_class, naming.convert(entity_class.__name__, 'camelcase', 'lowercase_underscore')) def now(args): print session.now() now_parser = subparsers.add_parser('now') now_parser.set_defaults(func=now) args = parser.parse_args() args.func(args)