exc yaml

Process yaml file generated by exc utility.

https://github.com/a-services/jbang-catalog/blob/main/exc.java

Extract line numbers for exceptions and print then in chunks of chunk_length

import yaml
import argparse

parser = argparse.ArgumentParser(description="exc yaml")

parser.add_argument("log_path", help="Log path")
parser.add_argument("-c", "--chunk-length", help="Length of exception stack trace chunk", type=int, default=10)

args = parser.parse_args()

log_path = args.log_path
chunk_length = args.chunk_length

def read_yaml(file_path):
    with open(file_path, 'r') as file:
        # Load all documents from the YAML file
        yaml_content = yaml.safe_load(file)
    return [item['lno'] for item in yaml_content]

def extract_lines(log_path, line_numbers):
    lines_to_extract = set()
    # Include the three following lines for each line number
    for line in line_numbers:
        lines_to_extract.update(range(line, line + chunk_length))

    extracted_lines = {}
    with open(log_path, 'r') as file:
        for i, line in enumerate(file, 1):
            if i in lines_to_extract:
                extracted_lines[i] = line.rstrip()
    return extracted_lines

def write_output(extracted_lines, output_path):
    k1 = 0
    with open(output_path, 'w') as file:
        for k2 in sorted(extracted_lines.keys()):
            if k2 > k1 + 1:
                file.write("\n\n")
            k1 = k2

            file.write(f"{k2}: {extracted_lines[k2]}\n")


yaml_path = log_path + '.yml'

output_path = log_path + '.txt'

line_numbers = read_yaml(yaml_path)
extracted_lines = extract_lines(log_path, line_numbers)
write_output(extracted_lines, output_path)
print(f"[exc_yaml] Extraction complete with chunk length: {chunk_length}")
print(f"[exc_yaml] Output file: {output_path}")