eTree美化筆記

#!/usr/bin/env python
# -*- coding: UTF-8 -*-
# Time      :  15:35
# Email     : [email protected]
# File      : eTreeIndent.py
__author__ = 'ChenLiang.Miao'


# import --+--+--+--+--+--+--+--+--+--+--+--+--+--+--+--+--+--+--+--+--+--+--+--+--+ #

# function +--+--+--+--+--+--+--+--+--+--+--+--+--+--+--+--+--+--+--+--+--+--+--+--+ #

# +--+--+--+--+--+--+--+--+--+--+--+--+--+--+--+--+--+--+--+--+--+--+--+--+--+--+--+ #
def indent(element, level=0):
    """

    :param element: 分支
    :param level: 第幾層
    :return: 編輯分支
    """
    i = "\n" + level * "  "
    if len(element):
        if not element.text or not element.text.strip():
            element.text = i + "  "
        if not element.tail or not element.tail.strip():
            element.tail = i
            
        for element in element:  # 逐層編寫
            indent(element, level + 1)
            
        if not element.tail or not element.tail.strip():  # 結尾
            element.tail = i
    else:
        if level and (not element.tail or not element.tail.strip()):
            element.tail = i