From a81c85eb56cdd2d4b930cfa6655692005c596b61 Mon Sep 17 00:00:00 2001 From: Klaus-Uwe Mitterer Date: Mon, 22 Jun 2015 19:44:42 +0200 Subject: [PATCH] Make verifier use argparse. Not better, but prettier. --- verifier.py | 17 +++++++++-------- 1 file changed, 9 insertions(+), 8 deletions(-) diff --git a/verifier.py b/verifier.py index 2a336a1..f5de9c8 100755 --- a/verifier.py +++ b/verifier.py @@ -1,6 +1,6 @@ #!/usr/bin/env python3 -import sys +import argparse, sys def verifier(i1, i2): try: @@ -11,10 +11,11 @@ def verifier(i1, i2): raise ValueError("Arguments must be integers.") if __name__ == "__main__": - try: - if verifier(*sys.argv[1:]): - print("Have fun.") - else: - print("Nope.") - except TypeError: - raise ValueError(sys.argv[0] + " takes exactly two arguments.") + parser = argparse.ArgumentParser() + parser.add_argument("age1", type=int) + parser.add_argument("age2", type=int) + args = parser.parse_args() + if verifier(args.age1, args.age2): + print("Have fun.") + else: + print("Nope.")