added dj-libs tag; do not overwrite input files by default again

html_to_django/converter.py

@@ -5,10 +5,22 @@ 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)
 
@@ -17,7 +29,8 @@ def convert_file(path: str, overwrite: bool = False) -> None:
     else:
         out_path = path.replace(".html", ".n.html")
     with open(out_path, "w") as file:
-        file.write("{% load static %}\n")
+        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)
@@ -31,6 +44,6 @@ def convert_dir(path: str, overwrite: bool = False) -> None:
 def command_entry() -> None:
     path = os.path.abspath(sys.argv[1])
     if os.path.isdir(path):
-        convert_dir(path, True)
+        convert_dir(path)
     elif os.path.isfile(path):
         convert_file(path)