sorogon/StaticGalleryBuilder
1
0
Fork 0
mirror of https://github.com/greflm13/StaticGalleryBuilder.git synced 2026-02-05 02:59:27 +00:00
Code Issues Packages Projects Releases Wiki Activity

added command line arguments

Browse Source
This commit is contained in:
Flo Greistorfer
2024-06-26 23:42:04 +02:00
parent 624cfe23bc
commit 535bf04d40
1 changed files with 56 additions and 34 deletions
Download Patch File Download Diff File Expand all files Collapse all files

90
generate-html.py → generate_html.py Normal file → Executable file
View File

@@ -1,17 +1,19 @@
#!/usr/bin/env python3 #!/usr/bin/env python3
import sys
import os import os
import argparse
import urllib.parse import urllib.parse
from multiprocessing import Pool from multiprocessing import Pool
from string import Template
import numpy as np import numpy as np
""" """
root and webroot must point to the same folder, one on filesystem and one on the webserver. Use absolut paths, e.g. /data/pictures/ and https://pictures.example.com/ root and webroot must point to the same folder, one on filesystem and one on the webserver. Use absolut paths, e.g. /data/pictures/ and https://pictures.example.com/
""" """
ROOT = "/mnt/nfs/pictures/" _ROOT = "/mnt/nfs/pictures/"
WEBROOT = "https://pictures.sorogon.eu/" _WEBROOT = "https://pictures.sorogon.eu/"
FOLDERICON = "https://www.svgrepo.com/show/400249/folder.svg" _FOLDERICON = "https://www.svgrepo.com/show/400249/folder.svg"
_ROOTTITLE = "Pictures"
imgext = [".jpg", ".jpeg"] imgext = [".jpg", ".jpeg"]
rawext = [".3fr", ".ari", ".arw", ".bay", ".braw", ".crw", ".cr2", ".cr3", ".cap", ".data", ".dcs", ".dcr", ".dng", ".drf", ".eip", ".erf", ".fff", ".gpr", ".iiq", ".k25", ".kdc", ".mdc", ".mef", ".mos", ".mrw", ".nef", ".nrw", ".obm", ".orf", ".pef", ".ptx", ".pxn", ".r3d", ".raf", ".raw", ".rwl", ".rw2", ".rwz", ".sr2", ".srf", ".srw", ".tif", ".tiff", ".x3f"] rawext = [".3fr", ".ari", ".arw", ".bay", ".braw", ".crw", ".cr2", ".cr3", ".cap", ".data", ".dcs", ".dcr", ".dng", ".drf", ".eip", ".erf", ".fff", ".gpr", ".iiq", ".k25", ".kdc", ".mdc", ".mef", ".mos", ".mrw", ".nef", ".nrw", ".obm", ".orf", ".pef", ".ptx", ".pxn", ".r3d", ".raf", ".raw", ".rwl", ".rw2", ".rwz", ".sr2", ".srf", ".srw", ".tif", ".tiff", ".x3f"]
@@ -23,7 +25,7 @@ HTMLHEADER = """
<head> <head>
<meta charset="UTF-8"> <meta charset="UTF-8">
<meta name="viewport" content="width=device-width, initial-scale=1.0"> <meta name="viewport" content="width=device-width, initial-scale=1.0">
<title>Pictures</title> <title>$title</title>
<style> <style>
* { * {
box-sizing: border-box; box-sizing: border-box;
@@ -34,18 +36,26 @@ HTMLHEADER = """
font-family: Arial; font-family: Arial;
} }
.header { .folders {
text-align: center; text-align: center;
padding: 32px;
display: -ms-flexbox; /* IE10 */ display: -ms-flexbox; /* IE10 */
display: flex; display: flex;
-ms-flex-wrap: wrap; /* IE10 */ -ms-flex-wrap: wrap; /* IE10 */
flex-wrap: wrap; flex-wrap: wrap;
justify-content: space-evenly; justify-content: space-evenly;
} }
.header img { .folders figure {
margin-bottom: 32px;
}
.header h1 {
font-size: 2.5em;
font-weight: bold;
text-align: center;
}
.folders img {
width: 100px; width: 100px;
vertical-align: middle; vertical-align: middle;
} }
@@ -83,7 +93,7 @@ HTMLHEADER = """
flex: 25%; flex: 25%;
max-width: 25%; max-width: 25%;
} }
.header img { .folders img {
width: 80px; width: 80px;
} }
} }
@@ -95,7 +105,7 @@ HTMLHEADER = """
flex: 50%; flex: 50%;
max-width: 50%; max-width: 50%;
} }
.header img { .folders img {
width: 60px; width: 60px;
} }
} }
@@ -107,7 +117,7 @@ HTMLHEADER = """
flex: 100%; flex: 100%;
max-width: 100%; max-width: 100%;
} }
.header img { .folders img {
width: 50px; width: 50px;
} }
} }
@@ -128,48 +138,53 @@ HTMLHEADER = """
def thumbnail_convert(arguments: tuple[str, str]): def thumbnail_convert(arguments: tuple[str, str]):
folder, item = arguments folder, item = arguments
os.system(f'magick "{os.path.join(folder, item)}" -quality 75% -define jpeg:size=1024x1024 -define jpeg:extent=100kb -thumbnail 512x512 -auto-orient "{os.path.join(ROOT, ".previews", folder.removeprefix(ROOT), item)}"') if not os.path.exists(os.path.join(args.root, ".previews", folder.removeprefix(args.root), item)) and args.regenerate is False:
os.system(f'magick "{os.path.join(folder, item)}" -quality 75% -define jpeg:size=1024x1024 -define jpeg:extent=100kb -thumbnail 512x512 -auto-orient "{os.path.join(args.root, ".previews", folder.removeprefix(args.root), item)}"')
def listfolder(folder: str): def listfolder(folder: str, title: str):
items: list[str] = os.listdir(folder) items: list[str] = os.listdir(folder)
items.sort() items.sort()
images: list[str] = [] images: list[str] = []
subfolders: list[str] = [] subfolders: list[str] = []
if not os.path.exists(os.path.join(ROOT, ".previews", folder.removeprefix(ROOT))): if not os.path.exists(os.path.join(args.root, ".previews", folder.removeprefix(args.root))):
os.mkdir(os.path.join(ROOT, ".previews", folder.removeprefix(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: with open(os.path.join(folder, "index.html"), "w", encoding="utf-8") as f:
f.write(HTMLHEADER) temp_obj = Template(HTMLHEADER)
f.write(temp_obj.substitute(title=title))
for item in items: for item in items:
if item != "Galleries" and item != ".previews": if item != "Galleries" and item != ".previews":
if os.path.isdir(os.path.join(folder, item)): if os.path.isdir(os.path.join(folder, item)):
subfolders.extend([f'<figure><a href="{WEBROOT}{urllib.parse.quote(folder.removeprefix(ROOT))}/{urllib.parse.quote(item)}"><img src="{FOLDERICON}" alt="Folder icon"/></a><figcaption><a href="{WEBROOT}{urllib.parse.quote(folder.removeprefix(ROOT))}/{urllib.parse.quote(item)}">{item}</a></figcaption></figure>']) subfolders.extend([f'<figure><a href="{args.webroot}{urllib.parse.quote(folder.removeprefix(args.root))}/{urllib.parse.quote(item)}"><img src="{args.foldericon}" alt="Folder icon"/></a><figcaption><a href="{args.webroot}{urllib.parse.quote(folder.removeprefix(args.root))}/{urllib.parse.quote(item)}">{item}</a></figcaption></figure>'])
listfolder(os.path.join(folder, item)) listfolder(os.path.join(folder, item), item)
else: else:
if os.path.splitext(item)[1].lower() in imgext: if os.path.splitext(item)[1].lower() in imgext:
image = f'<figure><a href="{WEBROOT}{urllib.parse.quote(folder.removeprefix(ROOT))}/{urllib.parse.quote(item)}"><img src="{WEBROOT}.previews/{urllib.parse.quote(folder.removeprefix(ROOT))}/{urllib.parse.quote(item)}" alt="{item}"/></a><figcaption class="caption">{item}' image = f'<figure><a href="{args.webroot}{urllib.parse.quote(folder.removeprefix(args.root))}/{urllib.parse.quote(item)}"><img src="{args.webroot}.previews/{urllib.parse.quote(folder.removeprefix(args.root))}/{urllib.parse.quote(item)}" alt="{item}"/></a><figcaption class="caption">{item}'
if not os.path.exists(os.path.join(ROOT, ".previews", folder.removeprefix(ROOT), item)): if not os.path.exists(os.path.join(args.root, ".previews", folder.removeprefix(args.root), item)):
thumbnails.append((folder, item)) thumbnails.append((folder, item))
for raw in rawext: for raw in rawext:
if os.path.exists(os.path.join(folder, os.path.splitext(item)[0] + raw)): if os.path.exists(os.path.join(folder, os.path.splitext(item)[0] + raw)):
if raw == ".tif" or raw == ".tiff": if raw == ".tif" or raw == ".tiff":
image += f': <a href="{WEBROOT}{urllib.parse.quote(folder.removeprefix(ROOT))}/{urllib.parse.quote(os.path.splitext(item)[0])}{raw}">TIFF</a>' image += f': <a href="{args.webroot}{urllib.parse.quote(folder.removeprefix(args.root))}/{urllib.parse.quote(os.path.splitext(item)[0])}{raw}">TIFF</a>'
else: else:
image += f': <a href="{WEBROOT}{urllib.parse.quote(folder.removeprefix(ROOT))}/{urllib.parse.quote(os.path.splitext(item)[0])}{raw}">RAW</a>' image += f': <a href="{args.webroot}{urllib.parse.quote(folder.removeprefix(args.root))}/{urllib.parse.quote(os.path.splitext(item)[0])}{raw}">RAW</a>'
elif os.path.exists(os.path.join(folder, os.path.splitext(item)[0] + raw.upper())): elif os.path.exists(os.path.join(folder, os.path.splitext(item)[0] + raw.upper())):
if raw == ".tif" or raw == ".tiff": if raw == ".tif" or raw == ".tiff":
image += f': <a href="{WEBROOT}{urllib.parse.quote(folder.removeprefix(ROOT))}/{urllib.parse.quote(os.path.splitext(item)[0])}{raw}">TIFF</a>' image += f': <a href="{args.webroot}{urllib.parse.quote(folder.removeprefix(args.root))}/{urllib.parse.quote(os.path.splitext(item)[0])}{raw}">TIFF</a>'
else: else:
image += f': <a href="{WEBROOT}{urllib.parse.quote(folder.removeprefix(ROOT))}/{urllib.parse.quote(os.path.splitext(item)[0])}{raw}">RAW</a>' image += f': <a href="{args.webroot}{urllib.parse.quote(folder.removeprefix(args.root))}/{urllib.parse.quote(os.path.splitext(item)[0])}{raw}">RAW</a>'
image += "</figcaption></figure>" image += "</figcaption></figure>"
images.extend([image]) images.extend([image])
f.write(' <div class="header">\n') f.write(' <div class="header">\n')
f.write(f" <h1>{title}</h1>\n")
f.write(' <div class="folders">\n')
for subfolder in subfolders: for subfolder in subfolders:
f.write(subfolder) f.write(subfolder)
f.write("\n") f.write("\n")
f.write(" </div>\n") f.write(" </div>\n")
f.write(" </div>\n")
f.write(' <div class="row">\n') f.write(' <div class="row">\n')
for chunk in np.array_split(images, 8): for chunk in np.array_split(images, 8):
f.write(' <div class="column">\n') f.write(' <div class="column">\n')
@@ -182,16 +197,23 @@ def listfolder(folder: str):
def main(): def main():
global ROOT global args
global WEBROOT # Parse command-line arguments
if not ROOT.endswith("/"): parser = argparse.ArgumentParser(description="Generate html files for static image host.")
ROOT += "/" parser.add_argument("-f", "--root", help="Root folder", default=_ROOT, required=False, type=str, dest="root")
if not WEBROOT.endswith("/"): parser.add_argument("-w", "--webroot", help="Webroot url", default=_WEBROOT, required=False, type=str, dest="webroot")
WEBROOT += "/" parser.add_argument("-i", "--foldericon", help="Foldericon url", default=_FOLDERICON, required=False, type=str, dest="foldericon", metavar="ICON")
if not os.path.exists(os.path.join(ROOT, ".previews")): parser.add_argument("-r", "--regenerate", help="Regenerate thumbnails", action="store_true", default=False, required=False, dest="regenerate")
os.mkdir(os.path.join(ROOT, ".previews")) args = parser.parse_args()
if not args.root.endswith("/"):
args.root += "/"
if not args.webroot.endswith("/"):
args.webroot += "/"
if not os.path.exists(os.path.join(args.root, ".previews")):
os.mkdir(os.path.join(args.root, ".previews"))
print("Generating html files...") print("Generating html files...")
listfolder(ROOT) listfolder(args.root, _ROOTTITLE)
with Pool(os.cpu_count()) as p: with Pool(os.cpu_count()) as p:
print("Generating thumbnails...") print("Generating thumbnails...")