summaryrefslogtreecommitdiff
path: root/docs/ext/docstrings.py
blob: 5e5f82fa226ce85d165f9fe610047cabf0fa577a (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
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
import itertools
import os

import jinja2
import six
import sphinx
import sphinx.ext.napoleon as napoleon
from sphinx.ext.napoleon.docstring import GoogleDocstring


def classref(value, short=True):
    tilde = '~' if short else ''
    return ':class:`%sgitlab.objects.%s`' % (tilde, value.__name__)


def setup(app):
    app.connect('autodoc-process-docstring', _process_docstring)
    app.connect('autodoc-skip-member', napoleon._skip_member)

    conf = napoleon.Config._config_values

    for name, (default, rebuild) in six.iteritems(conf):
        app.add_config_value(name, default, rebuild)
    return {'version': sphinx.__display_version__, 'parallel_read_safe': True}


def _process_docstring(app, what, name, obj, options, lines):
    result_lines = lines
    docstring = GitlabDocstring(result_lines, app.config, app, what, name, obj,
                                options)
    result_lines = docstring.lines()
    lines[:] = result_lines[:]


class GitlabDocstring(GoogleDocstring):
    def _build_doc(self, tmpl, **kwargs):
        env = jinja2.Environment(loader=jinja2.FileSystemLoader(
            os.path.dirname(__file__)), trim_blocks=False)
        env.filters['classref'] = classref
        template = env.get_template(tmpl)
        output = template.render(**kwargs)

        return output.split('\n')

    def __init__(self, *args, **kwargs):
        super(GitlabDocstring, self).__init__(*args, **kwargs)

        if getattr(self._obj, '__name__', None) == 'Gitlab':
            mgrs = []
            gl = self._obj('http://dummy', private_token='dummy')
            for item in vars(gl).items():
                if hasattr(item[1], 'obj_cls'):
                    mgrs.append(item)
            self._parsed_lines.extend(self._build_doc('gl_tmpl.j2',
                                                      mgrs=sorted(mgrs)))
        elif hasattr(self._obj, 'obj_cls') and self._obj.obj_cls is not None:
            self._parsed_lines.extend(self._build_doc('manager_tmpl.j2',
                                                      cls=self._obj.obj_cls))
        elif hasattr(self._obj, 'canUpdate') and self._obj.canUpdate:
            self._parsed_lines.extend(self._build_doc('object_tmpl.j2',
                                                      obj=self._obj))