У меня возникли некоторые проблемы с использованием Python для создания HTML-документа. Я пытаюсь создать HTML-список дерева каталогов. Это то, что я до сих пор:

def list_files(startpath):
    for root, dirs, files in os.walk(startpath):
        level = root.replace(startpath, '').count(os.sep)
        if level <= 1:
            print('<li>{}<ul>'.format(os.path.basename(root)))
        else:
            print('<li>{}'.format(os.path.basename(root)))
        for f in files:
            last_file = len(files)-1
            if f == files[last_file]:
                print('<li>{}</li></ul>'.format(f))
            elif f == files[0] and level-1 > 0:
                print('<ul><li>{}</li>'.format(f))
            else:
                print('<li>{}</li>'.format(f))
    print('</li></ul>')

Кажется, это работает хорошо, если есть только корневой каталог, один уровень подкаталогов и файлов. Однако добавление другого уровня подкаталогов вызывает проблемы (поскольку, я думаю, закрывающий тег не вводит достаточное количество раз в конце). Но мне трудно разобраться с этим.

Если это не может быть сделано таким образом, есть ли более простой способ сделать это? Я использую Flask, но я не очень разбираюсь в шаблонах, поэтому, возможно, я что-то упустил.