sorogon/StaticGalleryBuilder
1
0
Fork 0
mirror of https://github.com/greflm13/StaticGalleryBuilder.git synced 2026-02-05 02:59:27 +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") 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. Takes a screenshot of the given HTML file with the specified CSS applied.
Args: 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. css_file (str): Path to the CSS file to be applied.
output_file (str): Path where the screenshot will be saved. output_file (str): Path where the screenshot will be saved.
driver (webdriver.Chrome): The Chrome WebDriver instance. driver (webdriver.Chrome): The Chrome WebDriver instance.
""" """
try: try:
# Open the HTML file or URL # Open the HTML file or URL
if _html_path.startswith(("http://", "https://")): if html_file_path.startswith(("http://", "https://")):
driver.get(_html_path) driver.get(html_file_path)
else: 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 current theme.css
remove_css_script = """ 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: with open(css_file, "r", encoding="utf-8") as f:
css_content = f.read() css_content = f.read()
# Apply Icon # Extract folder icon content
split = css_content.split(".foldericon {") css_parts = css_content.split(".foldericon {")
split2 = split[1].split("}", maxsplit=1) css_head = css_parts[0]
themehead = split[0] css_tail = css_parts[1].split("}", maxsplit=1)[1]
themetail = split2[1] folder_icon_content = css_parts[1].split("}", maxsplit=1)[0].strip()
foldericon = split2[0].strip() folder_icon_content = re.sub(r"/\*.*\*/", "", folder_icon_content)
foldericon = re.sub(r"/\*.*\*/", "", foldericon)
for match in re.finditer(r"content: (.*);", foldericon): for match in re.finditer(r"content: (.*);", folder_icon_content):
foldericon = match[1] folder_icon_content = match.group(1).replace('"', "")
foldericon = foldericon.replace('"', "")
break 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() svg = f.read()
if "svg.j2" in foldericon: if "svg.j2" in folder_icon_content:
colorscheme = extract_colorscheme(css_file) colorscheme = extract_colorscheme(css_file)
svg = svg.replace("{{ color1 }}", colorscheme["color1"]) for color_key, color_value in colorscheme.items():
svg = svg.replace("{{ color2 }}", colorscheme["color2"]) svg = svg.replace(f"{{{{ {color_key} }}}}", color_value)
svg = svg.replace("{{ color3 }}", colorscheme["color3"])
svg = svg.replace("{{ color4 }}", colorscheme["color4"])
svg = urllib.parse.quote(svg) 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 # Encode CSS content as Base64
encoded_css = base64.b64encode(css_content.encode("utf-8")).decode("utf-8") 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) 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. Writes the README file with previews of included themes.
Args: 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. 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: try:
with open(readme_path, "r", encoding="utf-8") as f: with open(readme_path, "r", encoding="utf-8") as f:
readme = f.read() 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.") logging.info("README.md updated with previews of included themes.")
except FileNotFoundError: 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: except Exception as e:
logging.error("Failed to write README.md: %s", 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. Main function to take screenshots for each CSS file in the folder and update the README.md.
Args: Args:
folder_path (str): Path to the folder containing CSS files. directory_path (str): Path to the folder containing CSS files.
html_path (str): Path to the HTML file or URL for rendering. html_file_path (str): Path to the HTML file or URL for rendering.
""" """
if not os.path.exists(_folder_path): if not os.path.exists(directory_path):
logging.error('Error: Folder path "%s" does not exist.', _folder_path) logging.error('Error: Folder path "%s" does not exist.', directory_path)
return return
# Setup Chrome options # Setup Chrome options
@@ -145,21 +143,21 @@ def main(_folder_path: str, _html_path: str) -> None:
try: try:
themes = [] themes = []
# Iterate over all files in the folder # 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"): if filename.endswith(".css"):
theme_name = os.path.splitext(filename)[0] theme_name = os.path.splitext(filename)[0]
themes.append(theme_name) themes.append(theme_name)
css_file = os.path.join(_folder_path, filename) css_file = os.path.join(directory_path, filename)
output_file = os.path.join(_folder_path, "screenshots", f"{theme_name}.png") output_file = os.path.join(directory_path, "screenshots", f"{theme_name}.png")
# Create screenshots folder if it doesn't exist # Create screenshots folder if it doesn't exist
os.makedirs(os.path.dirname(output_file), exist_ok=True) os.makedirs(os.path.dirname(output_file), exist_ok=True)
# Take screenshot for this CSS file # 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 the README file with the new previews
write_readme(_folder_path, themes) write_readme(directory_path, themes)
finally: finally:
driver.quit() driver.quit()
@@ -167,8 +165,8 @@ def main(_folder_path: str, _html_path: str) -> None:
if __name__ == "__main__": if __name__ == "__main__":
if len(sys.argv) != 3: 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: else:
folder_path = sys.argv[1] dir_path = sys.argv[1]
html_path = sys.argv[2] 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