diff --git a/html_to_django/converter.py b/html_to_django/converter.py index a92a6b7..2161f89 100644 --- a/html_to_django/converter.py +++ b/html_to_django/converter.py @@ -5,14 +5,12 @@ 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 get_required_libraries(soup: bs4.BeautifulSoup) -> set[str]: + libs = set() + for element in soup.find_all(attrs={"dj-libs": True}): + libs |= set(element.get("dj-libs").split(";")) + del element.attrs["dj-libs"] + return libs def convert_file(path: str, overwrite: bool = False) -> None: diff --git a/tests/test_project/test.html b/tests/test_project/test.html index 54df5de..02b30b0 100644 --- a/tests/test_project/test.html +++ b/tests/test_project/test.html @@ -9,7 +9,7 @@
Else
-