50 lines
1.4 KiB
Python
50 lines
1.4 KiB
Python
import bs4
|
|
import sys
|
|
import os
|
|
from glob import glob
|
|
from .formatters import formatters
|
|
|
|
|
|
def get_required_libraries(soup: bs4.BeautifulSoup) -> list[str]:
|
|
if html := soup.find("html"):
|
|
if isinstance(html, bs4.Tag):
|
|
if html.has_attr("dj-libs"):
|
|
dj_libs = html["dj-libs"]
|
|
if isinstance(dj_libs, str):
|
|
return dj_libs.split(";")
|
|
return []
|
|
|
|
|
|
def convert_file(path: str, overwrite: bool = False) -> None:
|
|
with open(path, "r") as file:
|
|
soup = bs4.BeautifulSoup(file, "html.parser")
|
|
|
|
required_libraries = get_required_libraries(soup)
|
|
|
|
for formatter in formatters:
|
|
formatter.format(soup)
|
|
|
|
if overwrite:
|
|
out_path = path
|
|
else:
|
|
out_path = path.replace(".html", ".n.html")
|
|
with open(out_path, "w") as file:
|
|
for lib in required_libraries:
|
|
file.write(f"{{% load {lib} %}}\n")
|
|
file.write(soup
|
|
.prettify(formatter=bs4.formatter.HTMLFormatter(indent=4))
|
|
.replace(""", '"')) # Could this cause any issues? (maybe in attributes)
|
|
|
|
|
|
def convert_dir(path: str, overwrite: bool = False) -> None:
|
|
for filepath in glob(f"{path}/**/*.html", recursive=True):
|
|
convert_file(filepath, overwrite)
|
|
|
|
|
|
def command_entry() -> None:
|
|
path = os.path.abspath(sys.argv[1])
|
|
if os.path.isdir(path):
|
|
convert_dir(path)
|
|
elif os.path.isfile(path):
|
|
convert_file(path)
|