summaryrefslogtreecommitdiff
path: root/doc/docstring.py
blob: 5652bd2cb426efdc9e50bd1d3c706903050c5d8d (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
from docutils import nodes
from docutils.statemachine import ViewList
from nose.util import resolve_name


def docstring_directive(dirname, arguments, options, content, lineno,
                        content_offset, block_text, state, state_machine):
    obj_name = arguments[0]
    obj = resolve_name(obj_name)
    rst = ViewList()
    rst.append(obj.__doc__, '<docstring>')
    print "CALLED", obj_name, obj, rst
    node = nodes.section()
    surrounding_title_styles = state.memo.title_styles
    surrounding_section_level = state.memo.section_level
    state.memo.title_styles = []
    state.memo.section_level = 0
    state.nested_parse(rst, 0, node, match_titles=1)
    state.memo.title_styles = surrounding_title_styles
    state.memo.section_level = surrounding_section_level
    return node.children


def setup(app):
    app.add_directive('docstring', docstring_directive, 1, (1, 0, 1))