Chcę Python argparse rzucić wyjątek, a nie użycie

Question

Nie sądzę, że jest to możliwe, ale chcę obsługiwać wyjątki od argparse siebie.

Na przykład:

import argparse 
parser = argparse.ArgumentParser() 
parser.add_argument('--foo', help='foo help', required=True) 
try: 
    args = parser.parse_args() 
except: 
    do_something() 

Kiedy uruchamiam go:

$ myapp.py 
usage: myapp --foo foo 
myapp: error: argument --foo is required 

Ale chcę wpaść wyjątkiem zamiast.

Answer 1

Można podklasy ArgumentParser i zastąpić metodę error zrobić coś innego, gdy wystąpi błąd:

class ArgumentParserError(Exception): pass 

class ThrowingArgumentParser(argparse.ArgumentParser): 
    def error(self, message): 
     raise ArgumentParserError(message) 

parser = ThrowingArgumentParser() 
parser.add_argument(...) 
... 

Answer 2

w moim przypadku, argparse drukuje „zbyt mało argumentów”, a następnie zamknąć. po przeczytaniu kodu argparse, znalazłem, że po prostu wywołuje sys.exit() po wydrukowaniu jakiegoś komunikatu. ponieważ sys.exit() nie robi nic poza zgłoszeniem wyjątku SystemExit, możesz po prostu uchwycić ten wyjątek.

spróbuj więc sprawdzić, czy to działa.

try: 
     args = parser.parse_args(args) 
    except SystemExit: 
     .... your handler here ... 
     return