diff --git a/imgrate.py b/imgrate.py index 070ba75..67ab8cf 100755 --- a/imgrate.py +++ b/imgrate.py @@ -49,13 +49,18 @@ class imgrate: if __name__ == '__main__': - ap = ArgumentParser() - ap.add_argument('images', type=str, nargs='+', help='') - args = ap.parse_args() + def parse_arguments(): + ap = ArgumentParser() + ap.add_argument('images', type=str, nargs='+', help='') + return ap.parse_args() - ratings = {image: imgrate(image).quality for image in args.images} - maximum = max(ratings, key=ratings.get) - for rating in ratings.keys(): - maxmark = '*' if rating == maximum and len(ratings) > 1 else ' ' - print('{}imgrate("{}") = {}'.format(maxmark, rating, ratings[rating])) + def run(args): + ratings = {image: imgrate(image).quality for image in args.images} + maximum = max(ratings, key=ratings.get) + + for rating in ratings.keys(): + maxmark = '*' if rating == maximum and len(ratings) > 1 else ' ' + print('{}imgrate("{}") = {}'.format(maxmark, rating, ratings[rating])) + + run(parse_arguments())