diff --git a/src/qlever/commands/extract_queries.py b/src/qlever/commands/extract_queries.py index 3d78aa5b..a9802029 100644 --- a/src/qlever/commands/extract_queries.py +++ b/src/qlever/commands/extract_queries.py @@ -31,6 +31,12 @@ def additional_arguments(self, subparser) -> None: help="Base name for the query descriptions" " (default: `Log extract`)", ) + subparser.add_argument( + "--log-file", + type=str, + help="Name of the log file to extract queries from" + " (default: `.server-log.txt`)", + ) subparser.add_argument( "--output-file", type=str, @@ -40,7 +46,10 @@ def additional_arguments(self, subparser) -> None: def execute(self, args) -> bool: # Show what the command does. - log_file_name = f"{args.name}.server-log.txt" + if args.log_file is not None: + log_file_name = args.log_file + else: + log_file_name = f"{args.name}.server-log.txt" self.show( f"Extract SPARQL queries from `{log_file_name}`" f" and write them to `{args.output_file}`", @@ -74,7 +83,9 @@ def execute(self, args) -> bool: # A new query in the log. if "Processing the following SPARQL query" in line: query = [] - query_index = description_base_count.get(description_base, 0) + 1 + query_index = ( + description_base_count.get(description_base, 0) + 1 + ) description_base_count[description_base] = query_index continue # If we have started a query: extend until we meet the next log diff --git a/src/qlever/commands/settings.py b/src/qlever/commands/settings.py index ee79b1ae..f5a5976c 100644 --- a/src/qlever/commands/settings.py +++ b/src/qlever/commands/settings.py @@ -28,6 +28,7 @@ def relevant_qleverfile_arguments(self) -> dict[str : list[str]]: def additional_arguments(self, subparser) -> None: all_keys = [ + "always-multiply-unions", "cache-max-num-entries", "cache-max-size", "cache-max-size-single-entry",