From 2d3aca7e6977e5378b7db7c13089d0cce1e853d1 Mon Sep 17 00:00:00 2001 From: Rasmus Wriedt Larsen Date: Thu, 2 Nov 2023 09:30:18 +0100 Subject: [PATCH] Python: Allow single file processing --- python/ql/src/meta/ClassHierarchy/process-mrva-results.py | 7 ++++++- 1 file changed, 6 insertions(+), 1 deletion(-) diff --git a/python/ql/src/meta/ClassHierarchy/process-mrva-results.py b/python/ql/src/meta/ClassHierarchy/process-mrva-results.py index 17b0eb03565..6d6b83e4b70 100644 --- a/python/ql/src/meta/ClassHierarchy/process-mrva-results.py +++ b/python/ql/src/meta/ClassHierarchy/process-mrva-results.py @@ -84,7 +84,12 @@ def parse_from_file(path: Path) -> set: def gather_from_bqrs_results(): with CodeQL() as codeql: - for f in glob.glob(f"{sys.argv[1]}/**.bqrs", recursive=True): + if os.path.exists(sys.argv[1]) and not os.path.isdir(sys.argv[1]) and sys.argv[1].endswith(".bqrs"): + files = [sys.argv[1]] + else: + files = glob.glob(f"{sys.argv[1]}/**.bqrs", recursive=True) + + for f in files: print(f"Processing {f}") json_data = codeql.command(["bqrs", "decode", "--format=json", f])