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

eeeeh

Browse Source
This commit is contained in:
Flo Greistorfer
2024-07-13 21:48:39 +02:00
parent e540353a3b
commit bf16204395
2 changed files with 39 additions and 40 deletions
Download Patch File Download Diff File Expand all files Collapse all files

78
generate_previews.py
View File

@@ -16,22 +16,22 @@ from modules.svg_handling import extract_colorscheme
logging.basicConfig(level=logging.INFO, format="%(asctime)s - %(levelname)s - %(message)s")
def take_screenshot(_html_path: str, css_file: str, output_file: str, driver: webdriver.Chrome) -> None:
def take_screenshot(html_file_path: str, css_file: str, output_file: str, driver: webdriver.Chrome) -> None:
"""
Takes a screenshot of the given HTML file with the specified CSS applied.
Args:
html_path (str): Path to the HTML file or URL.
html_file_path (str): Path to the HTML file or URL.
css_file (str): Path to the CSS file to be applied.
output_file (str): Path where the screenshot will be saved.
driver (webdriver.Chrome): The Chrome WebDriver instance.
"""
try:
# Open the HTML file or URL
if _html_path.startswith(("http://", "https://")):
driver.get(_html_path)
if html_file_path.startswith(("http://", "https://")):
driver.get(html_file_path)
else:
driver.get(f"file://{os.path.abspath(_html_path)}")
driver.get(f"file://{os.path.abspath(html_file_path)}")
# Remove current theme.css
remove_css_script = """
@@ -47,28 +47,26 @@ def take_screenshot(_html_path: str, css_file: str, output_file: str, driver: we
with open(css_file, "r", encoding="utf-8") as f:
css_content = f.read()
# Apply Icon
split = css_content.split(".foldericon {")
split2 = split[1].split("}", maxsplit=1)
themehead = split[0]
themetail = split2[1]
foldericon = split2[0].strip()
foldericon = re.sub(r"/\*.*\*/", "", foldericon)
for match in re.finditer(r"content: (.*);", foldericon):
foldericon = match[1]
foldericon = foldericon.replace('"', "")
# Extract folder icon content
css_parts = css_content.split(".foldericon {")
css_head = css_parts[0]
css_tail = css_parts[1].split("}", maxsplit=1)[1]
folder_icon_content = css_parts[1].split("}", maxsplit=1)[0].strip()
folder_icon_content = re.sub(r"/\*.*\*/", "", folder_icon_content)
for match in re.finditer(r"content: (.*);", folder_icon_content):
folder_icon_content = match.group(1).replace('"', "")
break
if not "url" in foldericon:
with open(foldericon, "r", encoding="utf-8") as f:
if "url" not in folder_icon_content:
with open(folder_icon_content, "r", encoding="utf-8") as f:
svg = f.read()
if "svg.j2" in foldericon:
if "svg.j2" in folder_icon_content:
colorscheme = extract_colorscheme(css_file)
svg = svg.replace("{{ color1 }}", colorscheme["color1"])
svg = svg.replace("{{ color2 }}", colorscheme["color2"])
svg = svg.replace("{{ color3 }}", colorscheme["color3"])
svg = svg.replace("{{ color4 }}", colorscheme["color4"])
for color_key, color_value in colorscheme.items():
svg = svg.replace(f"{{{{ {color_key} }}}}", color_value)
svg = urllib.parse.quote(svg)
css_content = f'{themehead}\n.foldericon {{\n content: url("data:image/svg+xml,{svg}");\n}}\n{themetail}'
css_content = f'{css_head}\n.foldericon {{\n content: url("data:image/svg+xml,{svg}");\n}}\n{css_tail}'
# Encode CSS content as Base64
encoded_css = base64.b64encode(css_content.encode("utf-8")).decode("utf-8")
@@ -92,15 +90,15 @@ def take_screenshot(_html_path: str, css_file: str, output_file: str, driver: we
logging.error("Failed to take screenshot for %s: %s", css_file, e)
def write_readme(_folder_path: str, themes: List[str]) -> None:
def write_readme(directory_path: str, themes: List[str]) -> None:
"""
Writes the README file with previews of included themes.
Args:
folder_path (str): Path to the folder containing the themes and README.md.
directory_path (str): Path to the folder containing the themes and README.md.
themes (List[str]): List of theme names.
"""
readme_path = os.path.join(_folder_path, "README.md")
readme_path = os.path.join(directory_path, "README.md")
try:
with open(readme_path, "r", encoding="utf-8") as f:
readme = f.read()
@@ -115,21 +113,21 @@ def write_readme(_folder_path: str, themes: List[str]) -> None:
logging.info("README.md updated with previews of included themes.")
except FileNotFoundError:
logging.error("README.md not found in %s", _folder_path)
logging.error("README.md not found in %s", directory_path)
except Exception as e:
logging.error("Failed to write README.md: %s", e)
def main(_folder_path: str, _html_path: str) -> None:
def main(directory_path: str, html_file_path: str) -> None:
"""
Main function to take screenshots for each CSS file in the folder and update the README.md.
Args:
folder_path (str): Path to the folder containing CSS files.
html_path (str): Path to the HTML file or URL for rendering.
directory_path (str): Path to the folder containing CSS files.
html_file_path (str): Path to the HTML file or URL for rendering.
"""
if not os.path.exists(_folder_path):
logging.error('Error: Folder path "%s" does not exist.', _folder_path)
if not os.path.exists(directory_path):
logging.error('Error: Folder path "%s" does not exist.', directory_path)
return
# Setup Chrome options
@@ -145,21 +143,21 @@ def main(_folder_path: str, _html_path: str) -> None:
try:
themes = []
# Iterate over all files in the folder
for filename in sorted(os.listdir(_folder_path)):
for filename in sorted(os.listdir(directory_path)):
if filename.endswith(".css"):
theme_name = os.path.splitext(filename)[0]
themes.append(theme_name)
css_file = os.path.join(_folder_path, filename)
output_file = os.path.join(_folder_path, "screenshots", f"{theme_name}.png")
css_file = os.path.join(directory_path, filename)
output_file = os.path.join(directory_path, "screenshots", f"{theme_name}.png")
# Create screenshots folder if it doesn't exist
os.makedirs(os.path.dirname(output_file), exist_ok=True)
# Take screenshot for this CSS file
take_screenshot(_html_path, css_file, output_file, driver)
take_screenshot(html_file_path, css_file, output_file, driver)
# Write the README file with the new previews
write_readme(_folder_path, themes)
write_readme(directory_path, themes)
finally:
driver.quit()
@@ -167,8 +165,8 @@ def main(_folder_path: str, _html_path: str) -> None:
if __name__ == "__main__":
if len(sys.argv) != 3:
logging.error("Usage: python script_name.py folder_path html_path")
logging.error("Usage: python script_name.py directory_path html_file_path")
else:
folder_path = sys.argv[1]
dir_path = sys.argv[1]
html_path = sys.argv[2]
main(folder_path, html_path)
main(dir_path, html_path)

1
themes/icons/folder-4.svg Normal file
View File

@@ -0,0 +1 @@
<svg xmlns="http://www.w3.org/2000/svg" viewBox="0 0 48 48" width="48px" height="48px"><linearGradient id="WQEfvoQAcpQgQgyjQQ4Hqa" x1="24" x2="24" y1="6.708" y2="14.977" gradientUnits="userSpaceOnUse"><stop offset="0" stop-color="#eba600"/><stop offset="1" stop-color="#c28200"/></linearGradient><path fill="url(#WQEfvoQAcpQgQgyjQQ4Hqa)" d="M24.414,10.414l-2.536-2.536C21.316,7.316,20.553,7,19.757,7L5,7C3.895,7,3,7.895,3,9l0,30 c0,1.105,0.895,2,2,2l38,0c1.105,0,2-0.895,2-2V13c0-1.105-0.895-2-2-2l-17.172,0C25.298,11,24.789,10.789,24.414,10.414z"/><linearGradient id="WQEfvoQAcpQgQgyjQQ4Hqb" x1="24" x2="24" y1="10.854" y2="40.983" gradientUnits="userSpaceOnUse"><stop offset="0" stop-color="#ffd869"/><stop offset="1" stop-color="#fec52b"/></linearGradient><path fill="url(#WQEfvoQAcpQgQgyjQQ4Hqb)" d="M21.586,14.414l3.268-3.268C24.947,11.053,25.074,11,25.207,11H43c1.105,0,2,0.895,2,2v26 c0,1.105-0.895,2-2,2H5c-1.105,0-2-0.895-2-2V15.5C3,15.224,3.224,15,3.5,15h16.672C20.702,15,21.211,14.789,21.586,14.414z"/></svg>
Side by Side

After

Width:  |  Height:  |  Size: 1020 B