Better solution for grapic check in argparse