generate_files_json.py

#!/usr/bin/python3

import os
import json
import glob

data = {}
data['comparisonfiles'] = {}

for subset in next(os.walk("comparisonfiles/"))[1]:
    data['comparisonfiles'][subset] = {}
    data['comparisonfiles'][subset]["format"] = []
    format_list = [
        format
        for format in next(os.walk("comparisonfiles/" + subset + "/large"))[1]
    ]
    for format in format_list:
        extension = [
            os.path.splitext(os.path.basename(fn))[1][1:]
            for fn in glob.glob(
                "comparisonfiles/" + subset + "/large/" + format + "/*")
            if os.path.splitext(os.path.basename(fn))[1] != ".png"
        ][0]
        data['comparisonfiles'][subset]["format"].append({
            "extension": extension,
            "name": format
        })

    data['comparisonfiles'][subset]["format"].append({
        "extension": "png",
        "name": "Original"
    })

    filenames_list = [
        os.path.splitext(os.path.basename(files))[0]
        for files in next(
            os.walk("comparisonfiles/" + subset + "/Original/"))[2]
    ]
    data['comparisonfiles'][subset]["files"] = []
    for filename in filenames_list:
        data['comparisonfiles'][subset]["files"].append({
            "title": "",
            "filename": filename
        })

with open('comparisonfiles.json', 'w') as outfile:
    json.dump(data, outfile, indent=4)