Move image-fill.py to different repo
This commit is contained in:
parent
0fda2bc258
commit
a50eeeaf79
|
@ -1,55 +0,0 @@
|
|||
# This script automatically inserts img elements into
|
||||
# the html file for the pictures page.
|
||||
|
||||
import glob
|
||||
from pathlib import Path
|
||||
import argparse
|
||||
from fileinput import FileInput
|
||||
from sys import exit
|
||||
|
||||
parser = argparse.ArgumentParser(
|
||||
prog="ImgInserter",
|
||||
description="Fills img elements into the html file for the pictures page"
|
||||
)
|
||||
|
||||
parser.add_argument("html_file",
|
||||
help="HTML file to be modified")
|
||||
parser.add_argument("img_dir",
|
||||
help="""Directory where images are stored;
|
||||
must be in the same directory as the HTML file""")
|
||||
|
||||
args = parser.parse_args()
|
||||
|
||||
# Find the path to the HTML file
|
||||
html_path = glob.glob(args.html_file)
|
||||
if len(html_path) == 0:
|
||||
print("HTML file provided cannot be found")
|
||||
exit(1)
|
||||
html_path = Path(html_path[0]) # Shadow original html_path, now a Path
|
||||
|
||||
# Find the image directory
|
||||
image_dir = glob.glob(args.img_dir)
|
||||
if len(image_dir) == 0:
|
||||
print("Image directory cannot be found")
|
||||
exit(1)
|
||||
image_dir = Path(image_dir[0]) # Shadow original image_dir, now a Path
|
||||
|
||||
# Find all images in the image directory
|
||||
images = list(image_dir.rglob("*.png")) + \
|
||||
list(image_dir.rglob("*.jpg")) + \
|
||||
list(image_dir.rglob("*.jpeg"))
|
||||
|
||||
# Generate the img tag for each file
|
||||
imgs = []
|
||||
for i in images:
|
||||
src = str(i.relative_to(i.parents[1]))
|
||||
imgs.append(f'<img src="{src}" />')
|
||||
|
||||
with FileInput(html_path, inplace=True) as html:
|
||||
for line in html:
|
||||
if "Images go below" in line: # Insert imgs here
|
||||
print(line, end='')
|
||||
for img in sorted(imgs):
|
||||
print(img)
|
||||
else:
|
||||
print(line, end='')
|
Loading…
Reference in New Issue