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
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
|
import sphinx.writers.html
import sphinx.ext.mathjax
import re
class HTMLTranslator(sphinx.writers.html.HTMLTranslator):
def visit_displaymath(self, node):
sphinx.ext.mathjax.html_visit_displaymath(self, node)
def depart_displaymath(self, node):
return
def visit_math(self, node):
sphinx.ext.mathjax.html_visit_math(self, node)
def depart_math(self, node):
return
def depart_document(self, node):
methods_by_class = {}
for i, tag in enumerate(self.body):
if re.match(r'<dl class="((py|hoc) )?method">', tag):
method_id = self.body[i + 1].split("id=")[1].split('"')[1].split(".")
method_name = method_id[-1]
method_class = ".".join(method_id[:-1])
if method_class not in methods_by_class:
methods_by_class[method_class] = []
methods_by_class[method_class].append(method_name)
class_names = set(methods_by_class.keys())
for i, tag in enumerate(self.body):
if re.match(r'<dl class="((py|hoc) )?class">', tag):
class_names.add(self.body[i + 1].split("id=")[1].split('"')[1])
functions = []
for i, tag in enumerate(self.body):
if re.match(r'<dl class="((py|hoc) )?function">', tag):
try:
functions.append(self.body[i + 1].split("id=")[1].split('"')[1])
except:
functions.append(self.body[i + 3])
for i, tag in enumerate(self.body):
if re.match(r'<dl class="((py|hoc) )?data">', tag):
data_id = self.body[i + 1].split("id=")[1].split('"')[1].split(".")
data_name = data_id[-1]
data_class = ".".join(data_id[:-1])
if data_class:
if data_class not in methods_by_class:
methods_by_class[data_class] = []
methods_by_class[data_class].append(data_name)
else:
functions.append(data_name)
# sort everything
sort_key = lambda s: s.lower()
functions = sorted(functions, key=sort_key)
class_names = sorted(list(class_names), key=sort_key)
for key, values in zip(methods_by_class.keys(), methods_by_class.values()):
methods_by_class[key] = sorted(values, key=sort_key)
# add any missing class names to methods_by_class
for name in class_names:
if name not in methods_by_class:
methods_by_class[name] = []
jump_table = []
if functions:
fn_jmps = [
'<a href="#%s" title="Link to this definition">%s</a>' % (name, name)
for name in functions
]
jump_table += ["<p>", " · ".join(fn_jmps), "</p>"]
for cl in class_names:
method_jmps = [
'<a href="#%s.%s" title="Link to this definition">%s</a>'
% (cl, name, name)
for name in methods_by_class[cl]
]
jump_table += [
"<p>",
'<dl class="docutils"><dt><a href="#%s" title="Link to this definition">%s</a></dt><dd>%s</dd></dl>'
% (cl, cl, " · ".join(method_jmps)),
"</p>",
]
self.body = jump_table + self.body
sphinx.writers.html.HTMLTranslator.depart_document(self, node)
|
