mirror of
https://github.com/greflm13/StaticGalleryBuilder.git
synced 2026-02-05 11:09:26 +00:00
removed old generator
This commit is contained in:
Flo Greistorfer
126
cclicense.py
126
cclicense.py
@@ -6,132 +6,6 @@ class License:
|
|||||||
pics: list[str]
|
pics: list[str]
|
||||||
|
|
||||||
|
|
||||||
def licenseswitch(cclicense: str) -> str:
|
|
||||||
switch = {
|
|
||||||
"cc-zero": """
|
|
||||||
<div class="license" xmlns:cc="http://creativecommons.org/ns#" xmlns:dct="http://purl.org/dc/terms/">
|
|
||||||
<a property="dct:title" rel="cc:attributionURL" href="$webroot">$title</a> by <span property="cc:attributionName">$author</span> is marked with
|
|
||||||
<a href="https://creativecommons.org/publicdomain/zero/1.0/" target="_blank" rel="license noopener noreferrer" style="display: inline-block"
|
|
||||||
>CC0 1.0<img
|
|
||||||
style="height: 22px !important; margin-left: 3px; vertical-align: text-bottom"
|
|
||||||
src="https://mirrors.creativecommons.org/presskit/icons/cc.svg"
|
|
||||||
alt="" /><img
|
|
||||||
style="height: 22px !important; margin-left: 3px; vertical-align: text-bottom"
|
|
||||||
src="https://mirrors.creativecommons.org/presskit/icons/zero.svg"
|
|
||||||
alt=""
|
|
||||||
/></a>
|
|
||||||
</div>
|
|
||||||
""",
|
|
||||||
"cc-by": """
|
|
||||||
<div class="license" xmlns:cc="http://creativecommons.org/ns#" xmlns:dct="http://purl.org/dc/terms/">
|
|
||||||
<a property="dct:title" rel="cc:attributionURL" href="$webroot">$title</a> by <span property="cc:attributionName">$author</span> is licensed under
|
|
||||||
<a href="https://creativecommons.org/licenses/by/4.0/" target="_blank" rel="license noopener noreferrer" style="display: inline-block"
|
|
||||||
>CC BY 4.0<img
|
|
||||||
style="height: 22px !important; margin-left: 3px; vertical-align: text-bottom"
|
|
||||||
src="https://mirrors.creativecommons.org/presskit/icons/cc.svg"
|
|
||||||
alt="" /><img
|
|
||||||
style="height: 22px !important; margin-left: 3px; vertical-align: text-bottom"
|
|
||||||
src="https://mirrors.creativecommons.org/presskit/icons/by.svg"
|
|
||||||
alt=""
|
|
||||||
/></a>
|
|
||||||
</div>
|
|
||||||
""",
|
|
||||||
"cc-by-sa": """
|
|
||||||
<div class="license" xmlns:cc="http://creativecommons.org/ns#" xmlns:dct="http://purl.org/dc/terms/">
|
|
||||||
<a property="dct:title" rel="cc:attributionURL" href="$webroot">$title</a> by <span property="cc:attributionName">$author</span> is licensed under
|
|
||||||
<a href="https://creativecommons.org/licenses/by-sa/4.0/" target="_blank" rel="license noopener noreferrer" style="display: inline-block"
|
|
||||||
>CC BY-SA 4.0<img
|
|
||||||
style="height: 22px !important; margin-left: 3px; vertical-align: text-bottom"
|
|
||||||
src="https://mirrors.creativecommons.org/presskit/icons/cc.svg"
|
|
||||||
alt="" /><img
|
|
||||||
style="height: 22px !important; margin-left: 3px; vertical-align: text-bottom"
|
|
||||||
src="https://mirrors.creativecommons.org/presskit/icons/by.svg"
|
|
||||||
alt="" /><img
|
|
||||||
style="height: 22px !important; margin-left: 3px; vertical-align: text-bottom"
|
|
||||||
src="https://mirrors.creativecommons.org/presskit/icons/sa.svg"
|
|
||||||
alt=""
|
|
||||||
/></a>
|
|
||||||
</div>
|
|
||||||
""",
|
|
||||||
"cc-by-nd": """
|
|
||||||
<div class="license" xmlns:cc="http://creativecommons.org/ns#" xmlns:dct="http://purl.org/dc/terms/">
|
|
||||||
<a property="dct:title" rel="cc:attributionURL" href="$webroot">$title</a> by <span property="cc:attributionName">$author</span> is licensed under
|
|
||||||
<a href="https://creativecommons.org/licenses/by-nd/4.0/" target="_blank" rel="license noopener noreferrer" style="display: inline-block"
|
|
||||||
>CC BY-ND 4.0<img
|
|
||||||
style="height: 22px !important; margin-left: 3px; vertical-align: text-bottom"
|
|
||||||
src="https://mirrors.creativecommons.org/presskit/icons/cc.svg"
|
|
||||||
alt="" /><img
|
|
||||||
style="height: 22px !important; margin-left: 3px; vertical-align: text-bottom"
|
|
||||||
src="https://mirrors.creativecommons.org/presskit/icons/by.svg"
|
|
||||||
alt="" /><img
|
|
||||||
style="height: 22px !important; margin-left: 3px; vertical-align: text-bottom"
|
|
||||||
src="https://mirrors.creativecommons.org/presskit/icons/nd.svg"
|
|
||||||
alt=""
|
|
||||||
/></a>
|
|
||||||
</div>
|
|
||||||
""",
|
|
||||||
"cc-by-nc": """
|
|
||||||
<div class="license" xmlns:cc="http://creativecommons.org/ns#" xmlns:dct="http://purl.org/dc/terms/">
|
|
||||||
<a property="dct:title" rel="cc:attributionURL" href="$webroot">$title</a> by <span property="cc:attributionName">$author</span> is licensed under
|
|
||||||
<a href="https://creativecommons.org/licenses/by-nc/4.0/" target="_blank" rel="license noopener noreferrer" style="display: inline-block"
|
|
||||||
>CC BY-NC 4.0<img
|
|
||||||
style="height: 22px !important; margin-left: 3px; vertical-align: text-bottom"
|
|
||||||
src="https://mirrors.creativecommons.org/presskit/icons/cc.svg"
|
|
||||||
alt="" /><img
|
|
||||||
style="height: 22px !important; margin-left: 3px; vertical-align: text-bottom"
|
|
||||||
src="https://mirrors.creativecommons.org/presskit/icons/by.svg"
|
|
||||||
alt="" /><img
|
|
||||||
style="height: 22px !important; margin-left: 3px; vertical-align: text-bottom"
|
|
||||||
src="https://mirrors.creativecommons.org/presskit/icons/nc.svg"
|
|
||||||
alt=""
|
|
||||||
/></a>
|
|
||||||
</div>
|
|
||||||
""",
|
|
||||||
"cc-by-nc-sa": """
|
|
||||||
<div class="license" xmlns:cc="http://creativecommons.org/ns#" xmlns:dct="http://purl.org/dc/terms/">
|
|
||||||
<a property="dct:title" rel="cc:attributionURL" href="$webroot">$title</a> by <span property="cc:attributionName">$author</span> is licensed under
|
|
||||||
<a href="https://creativecommons.org/licenses/by-nc-sa/4.0/" target="_blank" rel="license noopener noreferrer" style="display: inline-block"
|
|
||||||
>CC BY-NC-SA 4.0<img
|
|
||||||
style="height: 22px !important; margin-left: 3px; vertical-align: text-bottom"
|
|
||||||
src="https://mirrors.creativecommons.org/presskit/icons/cc.svg"
|
|
||||||
alt="" /><img
|
|
||||||
style="height: 22px !important; margin-left: 3px; vertical-align: text-bottom"
|
|
||||||
src="https://mirrors.creativecommons.org/presskit/icons/by.svg"
|
|
||||||
alt="" /><img
|
|
||||||
style="height: 22px !important; margin-left: 3px; vertical-align: text-bottom"
|
|
||||||
src="https://mirrors.creativecommons.org/presskit/icons/nc.svg"
|
|
||||||
alt="" /><img
|
|
||||||
style="height: 22px !important; margin-left: 3px; vertical-align: text-bottom"
|
|
||||||
src="https://mirrors.creativecommons.org/presskit/icons/sa.svg"
|
|
||||||
alt=""
|
|
||||||
/></a>
|
|
||||||
</div>
|
|
||||||
""",
|
|
||||||
"cc-by-nc-nd": """
|
|
||||||
<div class="license" xmlns:cc="http://creativecommons.org/ns#" xmlns:dct="http://purl.org/dc/terms/">
|
|
||||||
<a property="dct:title" rel="cc:attributionURL" href="$webroot">$title</a> by <span property="cc:attributionName">$author</span> is licensed under
|
|
||||||
<a href="https://creativecommons.org/licenses/by-nc-nd/4.0/" target="_blank" rel="license noopener noreferrer" style="display: inline-block"
|
|
||||||
>CC BY-NC-ND 4.0<img
|
|
||||||
style="height: 22px !important; margin-left: 3px; vertical-align: text-bottom"
|
|
||||||
src="https://mirrors.creativecommons.org/presskit/icons/cc.svg"
|
|
||||||
alt="" /><img
|
|
||||||
style="height: 22px !important; margin-left: 3px; vertical-align: text-bottom"
|
|
||||||
src="https://mirrors.creativecommons.org/presskit/icons/by.svg"
|
|
||||||
alt="" /><img
|
|
||||||
style="height: 22px !important; margin-left: 3px; vertical-align: text-bottom"
|
|
||||||
src="https://mirrors.creativecommons.org/presskit/icons/nc.svg"
|
|
||||||
alt="" /><img
|
|
||||||
style="height: 22px !important; margin-left: 3px; vertical-align: text-bottom"
|
|
||||||
src="https://mirrors.creativecommons.org/presskit/icons/nd.svg"
|
|
||||||
alt=""
|
|
||||||
/></a>
|
|
||||||
</div>
|
|
||||||
""",
|
|
||||||
}
|
|
||||||
|
|
||||||
return switch.get(cclicense, "")
|
|
||||||
|
|
||||||
|
|
||||||
def licenseurlswitch(cclicense: str) -> str:
|
def licenseurlswitch(cclicense: str) -> str:
|
||||||
switch = {
|
switch = {
|
||||||
"cc-zero": "https://creativecommons.org/publicdomain/zero/1.0/",
|
"cc-zero": "https://creativecommons.org/publicdomain/zero/1.0/",
|
||||||
|
|||||||
@@ -1,392 +0,0 @@
|
|||||||
#!/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