如何将包含HTML标签的文本剪切到预期长度？

from bs4 import BeautifulSoup

def truncate_html_text(html_text, length):
    soup = BeautifulSoup(html_text, 'html.parser')
    truncated_text = ''
    count = 0

    def traverse(node):
        nonlocal truncated_text, count

        if count >= length:
            return

        if isinstance(node, str):
            remaining_length = length - count
            if len(node) <= remaining_length:
                truncated_text += node
                count += len(node)
            else:
                truncated_text += node[:remaining_length]
                count += remaining_length
        else:
            for child in node.contents:
                traverse(child)

    traverse(soup)

    # Remove the last incomplete HTML tag if necessary
    if count > length:
        truncated_text = truncated_text[:-(count - length)]
        last_tag_index = truncated_text.rfind('<')
        last_tag_close_index = truncated_text.rfind('>')
        if last_tag_index > last_tag_close_index:
            truncated_text = truncated_text[:last_tag_index]

    return truncated_text