HD-Icons/_generate.py
2022-11-25 00:51:58 +08:00

50 lines
1.9 KiB
Python

import pathlib
from pathlib import Path
root = pathlib.Path(__file__).parent.resolve()
template_path = root / "_TEMPLATE.md"
readme_path = root / "README.md"
# def generate_img_tag(file,folder,radius):
# return f'<div style="width: 105px;text-align: center;display: inline-block;font-size: 12px;margin: 5px;"><img style="border: 1.5px dashed white;border-radius: {radius};" src="{folder}/{file.name}" alt="{file.stem}" /><br/>{file.stem}</div>'
def generate_img_tag(file,folder,radius):
return f'<div style=""><img style="width: 105px;text-align: center;display: inline-block;font-size: 12px;margin: 5px;" src="{folder}/{file.name}" alt="{file.stem}" /><br/>{file.stem}</div>'
imgs = sorted(Path("./border-radius").glob("*.png"))
img_tags = [generate_img_tag(x,"border-radius","25px") for x in imgs]
circle_imgs = sorted(Path("./circle").glob("*.png"))
circle_img_tags = [generate_img_tag(x,"circle","50px") for x in circle_imgs]
line_number = 0
all_nums = len(img_tags) + len(circle_img_tags)
# Read the template file
with open(template_path, "r", encoding="UTF-8") as f:
lines = f.readlines()
# Find the line that starts with "<!-- ICONS -->"
for line in lines:
if line.startswith("<!-- START BORDER-RADIUS ICONS -->"):
line_number = lines.index(line)
break
for line in lines:
if line.startswith("<!-- START CIRCLE ICONS -->"):
circle_line_number = lines.index(line)
break
# Insert the icons after the line
lines.insert(line_number + 1, " ".join(img_tags))
lines.insert(circle_line_number + 1, " ".join(circle_img_tags))
lines = ["# 图标预览(当前共计 " + str(all_nums) + " 个)" if '# 图标预览(当前共计 0 个)' in i else i for i in lines]
# Write the new file
with open(readme_path, "w", encoding="UTF-8") as f:
f.write("".join(lines))
f.write("\n")
print("Done!")
print("Please commit the new README.md file.")