Das deutsche Python-Forum

Code: Alles auswählen

flat_list = [
    (1, u'headline 1 level 1'),
    (2, u'headline 2 level 2'),
    (2, u'headline 3 level 2'),
    (4, u'headline 4 level 4'),
    (1, u'headline 5 level 1'),
    (3, u'headline 6 level 3')
]

Code: Alles auswählen

[ # Ebene 1
    (1, u'headline 1 level 1'),
    [ # Ebene 2
        (2, u'headline 2 level 2'),
        (2, u'headline 3 level 2'), 
        [ # Ebene 3, leer
            [ # Ebene 4
                (4, u'headline 4 level 4'),
            ],
        ],
    ],
    (1, u'headline 5 level 1'),
    [ # Ebene 2, leer
        [ # Ebene 3
            (3, u'headline 6 level 3'),
        ],
    ],
]

Code: Alles auswählen

[ # Ebene 1
    (1, u'headline 1 level 1', [
        # Ebene 2
        (2, u'headline 2 level 2', []),
        (2, u'headline 3 level 2', [
            # Ebene 3, leer
            (3, None, [
                # Ebene 4
                (4, u'headline 4 level 4', []),
            ]),
        ]),
    ]),
    (1, u'headline 5 level 1', [
        # Ebene 2, leer
        (2, None, [
            # Ebene 3
            (3, u'headline 6 level 3', []),
        ]),
    ]),
]

Code: Alles auswählen

= headline 1 level 1
== headline 2 level 2
== headline 3 level 2
==== headline 4 level 4
= headline 5 level 1
=== headline 6 level 3

Code: Alles auswählen

def nest(flat):
    tree = []
    stack = [tree]
    
    for element in flat_list:
        index, _ = element
        stack_length = len(stack)
        
        if index > stack_length:
            for _ in range(stack_length, index):
                l = []
                stack[-1].append(l)
                stack.append(l)
        elif index < stack_length:
            stack = stack[:index]
            
        stack[-1].append(element)
    
    return tree

Code: Alles auswählen

class TableOfContent(object):
    def __init__(self):
        self.max_depth = None
        self.headlines = []
        self._created = False
 
    def __call__(self, depth=None, **kwargs):
        """Called when if the macro <<toc>> is defined when it is emitted."""
        if self._created:
            return "<<toc>>"
    
        self._created = True
        if depth is not None:
            self.max_depth = depth
    
        return '<<toc>>'
 
    def add_headline(self, level, content):
        """Add the current header to the toc."""
        if self.max_depth is None or level <= self.max_depth:
            self.headlines.append(
                (level, content)
            )
 
    def emit(self, document):
        """Emit the toc where the <<toc>> macro was."""
        cur_level = 0
        result = []
        for level, content in self.headlines:
            while level>cur_level:
                result.append('\t'*cur_level+'<ul>')
                cur_level += 1
            while level<cur_level:
                cur_level -= 1
                result.append('\t'*cur_level+'</ul>')
            result.append('{0}<li><a href="#{1}">{1}</a></li>'.format('\t'*cur_level, content))
        while cur_level:
            cur_level -= 1
            result.append('\t'*cur_level+'</ul>')
        
        # FIXME: We should not use <p> here, because it doesn't match
        #        if no newline was made before <<toc>>
        substitution = "<p><<toc>></p>" if "<p><<toc>></p>" in document else "<<toc>>"
        return document.replace(substitution, '\n'.join(result), 1)