diff --git a/generate_html.py b/generate_html.py index 31255f6..d0c404c 100755 --- a/generate_html.py +++ b/generate_html.py @@ -151,49 +151,50 @@ def listfolder(folder: str, title: str): if not os.path.exists(os.path.join(args.root, ".previews", folder.removeprefix(args.root))): os.mkdir(os.path.join(args.root, ".previews", folder.removeprefix(args.root))) - with open(os.path.join(folder, "index.html"), "w", encoding="utf-8") as f: - temp_obj = Template(HTMLHEADER) - f.write(temp_obj.substitute(title=title)) - for item in items: - if item != "Galleries" and item != ".previews": - if os.path.isdir(os.path.join(folder, item)): - subfolders.extend([f'
Folder icon
{item}
']) - listfolder(os.path.join(folder, item), item) - else: - if os.path.splitext(item)[1].lower() in imgext: - image = f'
{item}
{item}' - if not os.path.exists(os.path.join(args.root, ".previews", folder.removeprefix(args.root), item)): - thumbnails.append((folder, item)) - for raw in rawext: - if os.path.exists(os.path.join(folder, os.path.splitext(item)[0] + raw)): - if raw == ".tif" or raw == ".tiff": - image += f': TIFF' - else: - image += f': RAW' - elif os.path.exists(os.path.join(folder, os.path.splitext(item)[0] + raw.upper())): - if raw == ".tif" or raw == ".tiff": - image += f': TIFF' - else: - image += f': RAW' - image += "
" - images.extend([image]) - f.write('
\n') - f.write(f"

{title}

\n") - f.write('
\n') - for subfolder in subfolders: - f.write(subfolder) - f.write("\n") - f.write("
\n") - f.write("
\n") - f.write('
\n') - for chunk in np.array_split(images, 8): - f.write('
\n') - for image in chunk: - f.write(f" {image}\n") + temp_obj = Template(HTMLHEADER) + for item in items: + if item != "Galleries" and item != ".previews": + if os.path.isdir(os.path.join(folder, item)): + subfolders.extend([f'
Folder icon
{item}
']) + listfolder(os.path.join(folder, item), item) + else: + if os.path.splitext(item)[1].lower() in imgext: + image = f'
{item}
{item}' + if not os.path.exists(os.path.join(args.root, ".previews", folder.removeprefix(args.root), item)): + thumbnails.append((folder, item)) + for raw in rawext: + if os.path.exists(os.path.join(folder, os.path.splitext(item)[0] + raw)): + if raw == ".tif" or raw == ".tiff": + image += f': TIFF' + else: + image += f': RAW' + elif os.path.exists(os.path.join(folder, os.path.splitext(item)[0] + raw.upper())): + if raw == ".tif" or raw == ".tiff": + image += f': TIFF' + else: + image += f': RAW' + image += "
" + images.extend([image]) + if len(images) > 0: + with open(os.path.join(folder, "index.html"), "w", encoding="utf-8") as f: + f.write(temp_obj.substitute(title=title)) + f.write('
\n') + f.write(f"

{title}

\n") + f.write('
\n') + for subfolder in subfolders: + f.write(subfolder) + f.write("\n") f.write("
\n") - f.write("
\n") - f.write(" \n") - f.close() + f.write("
\n") + f.write('
\n') + for chunk in np.array_split(images, 8): + f.write('
\n') + for image in chunk: + f.write(f" {image}\n") + f.write("
\n") + f.write("
\n") + f.write(" \n") + f.close() def main():