Add -n to list

This commit is contained in:
2022-02-15 18:02:31 +01:00
parent 47675509f1
commit 235f5ae8cb
2 changed files with 19 additions and 4 deletions

View File

@@ -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="")

View File

@@ -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)