Cách handle invalid arguments với argparse trong Python?

Chúng ta viết lại code cho trước như sau:

# foo.py
import argparse

class InvalidArgError(Exception):
    pass

parser = argparse.ArgumentParser()
parser.add_argument("echo")
args = parser.parse_args()

try:
    print(args.echo)
    raise InvalidArgError
except InvalidArgError as e:
    print(e)

Khi script này chạy ở terminal như sau:

$ python foo.py echo bar

Chúng ta nhận được output sau:

usage: foo.py [-h] echo
foo.py: error: unrecognized arguments: bar