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.")