From 235f5ae8cba218fff5ee9f49841177971f73e7c2 Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?Ricardo=20Montan=CC=83ana?= <rmontanana@gmail.com>
Date: Tue, 15 Feb 2022 18:02:31 +0100
Subject: [PATCH] Add -n to list

---
 src/Results.py |  9 ++++++++-
 src/list.py    | 14 +++++++++++---
 2 files changed, 19 insertions(+), 4 deletions(-)

diff --git a/src/Results.py b/src/Results.py
index e85812c..fc2ebd9 100644
--- a/src/Results.py
+++ b/src/Results.py
@@ -701,7 +701,12 @@ class Summary:
             self.data.append(entry)
 
     def list_results(
-        self, score=None, model=None, input_data=None, sort_key="date"
+        self,
+        score=None,
+        model=None,
+        input_data=None,
+        sort_key="date",
+        number=0,
     ) -> None:
         """Print the list of results"""
         data = self.data.copy() if input_data is None else input_data
@@ -716,6 +721,8 @@ class Summary:
             else x[sort_key],
             reverse=True,
         )
+        if number > 0:
+            data = data[:number]
         max_file = max(len(x["file"]) for x in data)
         max_title = max(len(x["title"]) for x in data)
         print(TextColor.LINE1, end="")
diff --git a/src/list.py b/src/list.py
index 9f118ea..770b77b 100755
--- a/src/list.py
+++ b/src/list.py
@@ -37,13 +37,21 @@ def parse_arguments():
         default="date",
         help="key to sort results",
     )
+    ap.add_argument(
+        "-n",
+        "--number",
+        type=int,
+        required=False,
+        default=0,
+        help="number of results to show, 0 to any",
+    )
     args = ap.parse_args()
 
-    return (args.excel, args.score, args.model, args.key)
+    return (args.excel, args.score, args.model, args.key, args.number)
 
 
-(excel, score, model, key) = parse_arguments()
+(excel, score, model, key, number) = parse_arguments()
 
 data = Summary()
 data.acquire()
-data.list_results(score=score, model=model, sort_key=key)
+data.list_results(score=score, model=model, sort_key=key, number=number)