# HG changeset patch # User Charly COSTE # Date 1258818862 -3600 # Node ID 60cc60f0058d8b9e1590c81e80851ba054f1abd2 # Parent 23416c27b59298b52d4a9f292be14273e7cb4849 Oops, forgot to add this file in the new command system commit Signed-off-by: Charly COSTE diff --git a/argparse_modified.py b/argparse_modified.py new file mode 100644 --- /dev/null +++ b/argparse_modified.py @@ -0,0 +1,32 @@ +#!/usr/bin/env python +# -*- coding: utf-8 -*- + +# This program is free software: you can redistribute it and/or modify +# it under the terms of the GNU General Public License as published by +# the Free Software Foundation, either version 3 of the License, or +# (at your option) any later version. +# +# This program is distributed in the hope that it will be useful, +# but WITHOUT ANY WARRANTY; without even the implied warranty of +# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the +# GNU General Public License for more details. +# +# You should have received a copy of the GNU General Public License +# along with this program. If not, see . + +from argparse import ArgumentParser as OriginalArgumentParser + +class ParseException(Exception): pass + +class ArgumentParser(OriginalArgumentParser): + + def _print_message(self, message, file=None): + if not hasattr(self, 'ret'): + self.ret = '' + self.ret += message + + def exit(self, status=0, message=''): + if hasattr(self, 'ret') and isinstance(self.ret, basestring): + raise ParseException(status, self.ret+message) + else: + raise ParseException(status, message)