Files
test/html_to_django/formatters/remove_default_formatter.py
T
2024-05-05 14:31:34 +02:00

20 lines
765 B
Python

from bs4 import BeautifulSoup, Doctype, Tag
def format(soup: BeautifulSoup) -> None:
if html := soup.find("html"):
if isinstance(html, Tag):
if html.has_attr("dj-remove-default"):
if isinstance(soup.contents[0], Doctype):
soup.contents[0].extract()
if head := html.find("head"):
if isinstance(head, Tag):
head.decompose()
if body := html.find("body"):
if isinstance(body, Tag):
if script := body.find("script"):
if isinstance(script, Tag):
script.decompose()
body.unwrap()
html.unwrap()