Added argparse --nof argument

serputko · Sep 15, 2017 · 2c64be7 · 2c64be7
1 parent 058b6a2
commit 2c64be7
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/bin/parser.py b/bin/parser.py
@@ -95,7 +95,9 @@ def show_top_n_popular_transactions(self, n):
     parser = argparse.ArgumentParser()
     parser.add_argument('--path', nargs='?', type=str, help="path to log file to parse",
                         default='./data/test_task_2_logs.log')
+    parser.add_argument('--nof', nargs='?', type=int, help="number of transactions to display",
+                        default=25)
     args = parser.parse_args()
     p = Parser(args.path)
     p.show_rpm_for_all_transactions()
-    p.show_top_n_popular_transactions(25)
+    p.show_top_n_popular_transactions(args.nof)