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

added lockfile function

Browse Source
This commit is contained in:
Flo Greistorfer
2024-06-27 14:46:15 +02:00
committed by Flo Greistorfer
parent a4174a2b9e
commit 9f9a255d53
1 changed files with 28 additions and 17 deletions
Download Patch File Download Diff File Expand all files Collapse all files

45
generate_html.py
View File

@@ -5,6 +5,7 @@ import urllib.parse
import shutil import shutil
from multiprocessing import Pool from multiprocessing import Pool
from string import Template from string import Template
from pathlib import Path
import numpy as np import numpy as np
from tqdm.auto import tqdm from tqdm.auto import tqdm
@@ -18,7 +19,7 @@ _FOLDERICON = "https://www.svgrepo.com/show/400249/folder.svg"
_ROOTTITLE = "Pictures" _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"]
excludes = ["index.html", "Galleries", ".previews", "Archives", "Wallpaper"] excludes = [".lock", "index.html", "Galleries", ".previews", "Archives", "Wallpaper"]
thumbnails: list[tuple[str, str]] = [] thumbnails: list[tuple[str, str]] = []
@@ -46,6 +47,7 @@ HTMLHEADER = """
-ms-flex-wrap: wrap; /* IE10 */ -ms-flex-wrap: wrap; /* IE10 */
flex-wrap: wrap; flex-wrap: wrap;
justify-content: space-evenly; justify-content: space-evenly;
overflow: hidden;
} }
.folders figure { .folders figure {
@@ -65,6 +67,7 @@ HTMLHEADER = """
.folders figcaption { .folders figcaption {
width: 120px; width: 120px;
text-align: center;
} }
.row { .row {
@@ -269,25 +272,33 @@ def main():
if not os.path.exists(os.path.join(args.root, ".previews")): if not os.path.exists(os.path.join(args.root, ".previews")):
os.mkdir(os.path.join(args.root, ".previews")) os.mkdir(os.path.join(args.root, ".previews"))
if args.non_interactive: if os.path.exists(os.path.join(args.root, ".lock")):
print("Generating html files...") print("Another instance of this program is running.")
listfolder(args.root, _ROOTTITLE) exit()
try:
Path(os.path.join(args.root, ".lock")).touch()
with Pool(os.cpu_count()) as p: if args.non_interactive:
print("Generating thumbnails...") print("Generating html files...")
p.map(thumbnail_convert, thumbnails) listfolder(args.root, _ROOTTITLE)
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) with Pool(os.cpu_count()) as p:
listfolder(args.root, _ROOTTITLE) print("Generating thumbnails...")
pbar.close() p.map(thumbnail_convert, thumbnails)
else:
pbar = tqdm(desc="Traversing filesystem", unit=" folders", ascii=True, dynamic_ncols=True)
gettotal(args.root)
pbar.close()
with Pool(os.cpu_count()) as p: pbar = tqdm(total=total + 1, desc="Generating html files", unit=" files", ascii=True, dynamic_ncols=True)
for r in tqdm(p.imap_unordered(thumbnail_convert, thumbnails), total=len(thumbnails), desc="Generating thumbnails", unit=" files", ascii=True, dynamic_ncols=True): listfolder(args.root, _ROOTTITLE)
... 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__": if __name__ == "__main__":