mirror of
https://github.com/greflm13/StaticGalleryBuilder.git
synced 2026-02-05 11:09:26 +00:00
switched to Jinja2 templating
This commit is contained in:
Flo Greistorfer
392
generate_html.old.py
Executable file
392
generate_html.old.py
Executable file
@@ -0,0 +1,392 @@
|
||||
#!/usr/bin/env python3
|
||||
import os
|
||||
import argparse
|
||||
import urllib.parse
|
||||
import shutil
|
||||
from multiprocessing import Pool
|
||||
from string import Template
|
||||
from pathlib import Path
|
||||
import numpy as np
|
||||
from tqdm.auto import tqdm
|
||||
|
||||
import cclicense
|
||||
|
||||
environment = Environment(loader=FileSystemLoader("templates/"))
|
||||
|
||||
_ROOT = "/data/pictures/"
|
||||
_WEBROOT = "https://pictures.example.com/"
|
||||
_FOLDERICON = "https://www.svgrepo.com/show/400249/folder.svg"
|
||||
_ROOTTITLE = "Pictures"
|
||||
_FAVICON = "favicon.ico"
|
||||
_AUTHOR = "Author"
|
||||
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"]
|
||||
excludes = [".lock", _FAVICON, "index.html", ".previews"]
|
||||
notlist = ["Galleries", "Archives"]
|
||||
|
||||
thumbnails: list[tuple[str, str]] = []
|
||||
|
||||
HTMLHEADER = """
|
||||
<!DOCTYPE html>
|
||||
<html lang="en">
|
||||
<head>
|
||||
<meta charset="UTF-8">
|
||||
<meta name="viewport" content="width=device-width, initial-scale=1.0">
|
||||
<title>$title</title>
|
||||
<link rel="icon" type="image/x-icon" href="$favicon">
|
||||
<style>
|
||||
* {
|
||||
box-sizing: border-box;
|
||||
}
|
||||
|
||||
body {
|
||||
margin: 0;
|
||||
margin-top: 32px;
|
||||
margin-bottom: 56px;
|
||||
font-family: Arial;
|
||||
}
|
||||
|
||||
.folders {
|
||||
text-align: center;
|
||||
display: -ms-flexbox; /* IE10 */
|
||||
display: flex;
|
||||
-ms-flex-wrap: wrap; /* IE10 */
|
||||
flex-wrap: wrap;
|
||||
justify-content: space-evenly;
|
||||
overflow: hidden;
|
||||
}
|
||||
|
||||
.folders figure {
|
||||
margin-bottom: 32px;
|
||||
margin-top: 50px;
|
||||
}
|
||||
|
||||
.header h1 {
|
||||
font-size: 2.5em;
|
||||
font-weight: bold;
|
||||
text-align: center;
|
||||
}
|
||||
|
||||
.folders img {
|
||||
width: 100px;
|
||||
vertical-align: middle;
|
||||
}
|
||||
|
||||
.folders figcaption {
|
||||
width: 120px;
|
||||
font-size: smaller;
|
||||
text-align: center;
|
||||
}
|
||||
|
||||
.row {
|
||||
display: -ms-flexbox; /* IE10 */
|
||||
display: flex;
|
||||
-ms-flex-wrap: wrap; /* IE10 */
|
||||
flex-wrap: wrap;
|
||||
padding: 0 2px;
|
||||
}
|
||||
|
||||
figure {
|
||||
margin: 0;
|
||||
}
|
||||
|
||||
/* Create four equal columns that sits next to each other */
|
||||
.column {
|
||||
-ms-flex: 12.5%; /* IE10 */
|
||||
flex: 12.5%;
|
||||
max-width: 12.5%;
|
||||
padding: 0 4px;
|
||||
}
|
||||
|
||||
.column img {
|
||||
margin-top: 20px;
|
||||
vertical-align: middle;
|
||||
width: 100%;
|
||||
}
|
||||
|
||||
/* Responsive layout - makes a four column-layout instead of eight columns */
|
||||
@media screen and (max-width: 1000px) {
|
||||
.column {
|
||||
-ms-flex: 25%;
|
||||
flex: 25%;
|
||||
max-width: 25%;
|
||||
}
|
||||
.folders img {
|
||||
width: 80px;
|
||||
}
|
||||
.folders figcaption {
|
||||
width: 100px;
|
||||
font-size: small;
|
||||
}
|
||||
}
|
||||
|
||||
/* Responsive layout - makes a two column-layout instead of four columns */
|
||||
@media screen and (max-width: 800px) {
|
||||
.column {
|
||||
-ms-flex: 50%;
|
||||
flex: 50%;
|
||||
max-width: 50%;
|
||||
}
|
||||
.folders img {
|
||||
width: 60px;
|
||||
}
|
||||
.folders figcaption {
|
||||
width: 80px;
|
||||
font-size: x-small;
|
||||
}
|
||||
}
|
||||
|
||||
/* Responsive layout - makes the two columns stack on top of each other instead of next to each other */
|
||||
@media screen and (max-width: 600px) {
|
||||
.column {
|
||||
-ms-flex: 100%;
|
||||
flex: 100%;
|
||||
max-width: 100%;
|
||||
}
|
||||
.folders img {
|
||||
width: 40px;
|
||||
}
|
||||
.folders figcaption {
|
||||
width: 60px;
|
||||
font-size: xx-small;
|
||||
}
|
||||
}
|
||||
|
||||
.caption {
|
||||
padding-top: 4px;
|
||||
text-align: center;
|
||||
font-style: italic;
|
||||
font-size: 12px;
|
||||
width: 100%;
|
||||
display: block;
|
||||
}
|
||||
|
||||
.license {
|
||||
position: fixed;
|
||||
bottom: 0;
|
||||
width: 100%;
|
||||
background-color: lightgrey;
|
||||
padding: 12px;
|
||||
}
|
||||
|
||||
.navbar {
|
||||
list-style-type: none;
|
||||
margin: 0;
|
||||
padding: 0;
|
||||
overflow: hidden;
|
||||
position: fixed;
|
||||
top: 0;
|
||||
width: 100%;
|
||||
background-color: #333;
|
||||
}
|
||||
|
||||
.navbar li {
|
||||
float: left;
|
||||
}
|
||||
|
||||
.navbar li a {
|
||||
display: block;
|
||||
color: white;
|
||||
text-align: center;
|
||||
padding: 14px 16px;
|
||||
text-decoration: none;
|
||||
}
|
||||
|
||||
.navbar li span {
|
||||
display: block;
|
||||
color: white;
|
||||
text-align: center;
|
||||
padding: 14px 16px;
|
||||
text-decoration: none;
|
||||
}
|
||||
|
||||
/* Change the link color to #111 (black) on hover */
|
||||
.navbar li a:hover {
|
||||
background-color: #111;
|
||||
}
|
||||
</style>
|
||||
</head>
|
||||
<body>
|
||||
"""
|
||||
|
||||
NAVBAR = """
|
||||
<ul class="navbar">
|
||||
<li><a href="$home">Home</a></li>
|
||||
<li><a href="$parent">Parent Directory</a></li>
|
||||
<li style="position: absolute; left: 50%; transform: translateX(-50%);"><span>$title</span></li>
|
||||
$license</ul>
|
||||
"""
|
||||
|
||||
|
||||
def thumbnail_convert(arguments: tuple[str, str]):
|
||||
folder, item = arguments
|
||||
if not os.path.exists(os.path.join(args.root, ".previews", folder.removeprefix(args.root), os.path.splitext(item)[0]) + ".jpg") or args.regenerate:
|
||||
if shutil.which("magick"):
|
||||
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), os.path.splitext(item)[0])}.jpg"')
|
||||
else:
|
||||
os.system(f'convert "{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), os.path.splitext(item)[0])}.jpg"')
|
||||
|
||||
|
||||
def listfolder(folder: str, title: str):
|
||||
if not args.non_interactive:
|
||||
pbar.desc = f"Generating html files - {folder}"
|
||||
pbar.update(0)
|
||||
items: list[str] = os.listdir(folder)
|
||||
items.sort()
|
||||
images: list[str] = []
|
||||
subfolders: list[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)))
|
||||
|
||||
body = Template(HTMLHEADER)
|
||||
navbar = Template(NAVBAR)
|
||||
contains_files = False
|
||||
for item in items:
|
||||
if item not in excludes:
|
||||
if os.path.isdir(os.path.join(folder, item)):
|
||||
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>'])
|
||||
if item not in notlist:
|
||||
listfolder(os.path.join(folder, item), os.path.join(folder, item).removeprefix(args.root))
|
||||
else:
|
||||
if not args.non_interactive:
|
||||
pbar.desc = f"Generating html files - {folder}"
|
||||
pbar.update(0)
|
||||
contains_files = True
|
||||
if os.path.splitext(item)[1].lower() in imgext:
|
||||
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(os.path.splitext(item)[0])}.jpg" alt="{item}"/></a><figcaption class="caption">{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 in (".tif", ".tiff"):
|
||||
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:
|
||||
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())):
|
||||
if raw in (".tif", ".tiff"):
|
||||
image += f': <a href="{args.webroot}{urllib.parse.quote(folder.removeprefix(args.root))}/{urllib.parse.quote(os.path.splitext(item)[0])}{raw.upper()}">TIFF</a>'
|
||||
else:
|
||||
image += f': <a href="{args.webroot}{urllib.parse.quote(folder.removeprefix(args.root))}/{urllib.parse.quote(os.path.splitext(item)[0])}{raw.upper()}">RAW</a>'
|
||||
image += "</figcaption></figure>"
|
||||
images.extend([image])
|
||||
if not args.non_interactive:
|
||||
pbar.desc = f"Generating html files - {folder}"
|
||||
pbar.update(0)
|
||||
if len(images) > 0 or (args.fancyfolders and not contains_files):
|
||||
with open(os.path.join(folder, "index.html"), "w", encoding="utf-8") as f:
|
||||
f.write(body.substitute(title=title, favicon=f"{args.webroot}{_FAVICON}"))
|
||||
f.write(' <div class="header">\n')
|
||||
if folder == args.root:
|
||||
f.write(f" <h1>{os.path.basename(folder)}</h1>\n")
|
||||
else:
|
||||
if args.license:
|
||||
f.write(navbar.substitute(home=args.webroot, parent=f"{args.webroot}{urllib.parse.quote(folder.removeprefix(args.root).removesuffix(folder.split('/')[-1]))}", title=os.path.basename(folder), license=f' <li style="float:right"><a href="{cclicense.licenseurlswitch(args.license)}" target="_blank">License</a></li>\n'))
|
||||
else:
|
||||
f.write(navbar.substitute(home=args.webroot, parent=f"{args.webroot}{urllib.parse.quote(folder.removeprefix(args.root).removesuffix(folder.split('/')[-1]))}", title=os.path.basename(folder), license=""))
|
||||
f.write(' <div class="folders">\n')
|
||||
for subfolder in subfolders:
|
||||
f.write(subfolder)
|
||||
f.write("\n")
|
||||
f.write(" </div>\n")
|
||||
f.write(" </div>\n")
|
||||
if len(images) > 0:
|
||||
f.write(' <div class="row">\n')
|
||||
for chunk in np.array_split(images, 8):
|
||||
f.write(' <div class="column">\n')
|
||||
for image in chunk:
|
||||
f.write(f" {image}\n")
|
||||
f.write(" </div>\n")
|
||||
f.write(" </div>\n")
|
||||
if args.license:
|
||||
f.write(_cclicense.substitute(webroot=args.webroot, title=args.title, author=args.author))
|
||||
f.write(" </body>\n</html>")
|
||||
f.close()
|
||||
else:
|
||||
if os.path.exists(os.path.join(folder, "index.html")):
|
||||
os.remove(os.path.join(folder, "index.html"))
|
||||
if not args.non_interactive:
|
||||
pbar.update(1)
|
||||
|
||||
|
||||
def gettotal(folder):
|
||||
global total
|
||||
|
||||
if not args.non_interactive:
|
||||
pbar.desc = f"Traversing filesystem - {folder}"
|
||||
pbar.update(0)
|
||||
|
||||
items: list[str] = os.listdir(folder)
|
||||
items.sort()
|
||||
|
||||
for item in items:
|
||||
if item not in excludes:
|
||||
if os.path.isdir(os.path.join(folder, item)):
|
||||
total += 1
|
||||
if not args.non_interactive:
|
||||
pbar.update(1)
|
||||
if item not in notlist:
|
||||
gettotal(os.path.join(folder, item))
|
||||
|
||||
|
||||
def main():
|
||||
global total
|
||||
global args
|
||||
global pbar
|
||||
global _cclicense
|
||||
|
||||
total = 0
|
||||
# Parse command-line arguments
|
||||
parser = argparse.ArgumentParser(description="Generate html files for static image host.")
|
||||
parser.add_argument("-p", "--root", help="Root folder", default=_ROOT, required=False, type=str, dest="root")
|
||||
parser.add_argument("-w", "--webroot", help="Webroot url", default=_WEBROOT, required=False, type=str, dest="webroot")
|
||||
parser.add_argument("-i", "--foldericon", help="Foldericon url", default=_FOLDERICON, required=False, type=str, dest="foldericon", metavar="ICON")
|
||||
parser.add_argument("-r", "--regenerate", help="Regenerate thumbnails", action="store_true", default=False, required=False, dest="regenerate")
|
||||
parser.add_argument("-n", "--non-interactive", help="Disable interactive mode", action="store_true", default=False, required=False, dest="non_interactive")
|
||||
parser.add_argument("-l", "--license", help="License", default=None, required=False, choices=["cc-zero", "cc-by", "cc-by-sa", "cc-by-nd", "cc-by-nc", "cc-by-nc-sa", "cc-by-nc-nd"], dest="license")
|
||||
parser.add_argument("-a", "--author", help="Author", default=_AUTHOR, required=False, type=str, dest="author")
|
||||
parser.add_argument("-t", "--title", help="Title", default=_ROOTTITLE, required=False, type=str, dest="title")
|
||||
parser.add_argument("--fancyfolders", help="Use fancy folders instead of default apache ones", action="store_true", default=False, required=False, dest="fancyfolders")
|
||||
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"))
|
||||
|
||||
if args.license:
|
||||
_cclicense = Template(cclicense.licenseswitch(args.license))
|
||||
|
||||
if os.path.exists(os.path.join(args.root, ".lock")):
|
||||
print("Another instance of this program is running.")
|
||||
exit()
|
||||
try:
|
||||
Path(os.path.join(args.root, ".lock")).touch()
|
||||
|
||||
if args.non_interactive:
|
||||
print("Generating html files...")
|
||||
listfolder(args.root, args.title)
|
||||
|
||||
with Pool(os.cpu_count()) as p:
|
||||
print("Generating thumbnails...")
|
||||
p.map(thumbnail_convert, thumbnails)
|
||||
else:
|
||||
pbar = tqdm(desc="Traversing filesystem", unit=" folders", ascii=True, dynamic_ncols=True)
|
||||
gettotal(args.root)
|
||||
pbar.close()
|
||||
|
||||
pbar = tqdm(total=total + 1, desc="Generating html files", unit=" files", ascii=True, dynamic_ncols=True)
|
||||
listfolder(args.root, args.title)
|
||||
pbar.close()
|
||||
|
||||
with Pool(os.cpu_count()) as p:
|
||||
for r in tqdm(p.imap_unordered(thumbnail_convert, thumbnails), total=len(thumbnails), desc="Generating thumbnails", unit=" files", ascii=True, dynamic_ncols=True):
|
||||
pass
|
||||
finally:
|
||||
os.remove(os.path.join(args.root, ".lock"))
|
||||
|
||||
|
||||
if __name__ == "__main__":
|
||||
main()
|
||||
Reference in New Issue
Block a user