42 lines
1.5 KiB
Python
42 lines
1.5 KiB
Python
# Based on https://stackoverflow.com/a/46600038 with some modifications
|
|
import re
|
|
from sphinx.directives.other import TocTree
|
|
|
|
|
|
def setup(app):
|
|
app.add_directive('toctree', TocTreeFilt, override=True)
|
|
|
|
return {'parallel_read_safe': True, 'parallel_write_safe': True, 'version': '0.1'}
|
|
|
|
|
|
class TocTreeFilt(TocTree):
|
|
"""
|
|
Override normal toctree directive to support clauses of the kind
|
|
|
|
:filter: Name <link>
|
|
|
|
Where the :filter: part becomes selective to only include the document if
|
|
one of the provided tags is set, same as the logic used by the "only" directive.
|
|
|
|
If no :filter: is supplied, works the same as default Sphinx :toctree:
|
|
|
|
Note that excluding via filter doesn't prevent Sphinx from finding these .rst files
|
|
when it scan the src/ directory, so it's also necessary to make sure that the files
|
|
are covered by the exclude_patterns list in conf.py
|
|
"""
|
|
RE_PATTERN = re.compile(r'^\s*:(.+?):\s*(.+)$')
|
|
|
|
def run(self):
|
|
# Remove all TOC entries that should not be on display
|
|
env = self.state.document.settings.env
|
|
self.content = [self.filter_entry(env, e) for e in self.content if e is not None]
|
|
|
|
return super(TocTreeFilt, self).run()
|
|
|
|
def filter_entry(self, env, entry):
|
|
m = self.RE_PATTERN.match(entry)
|
|
if m is not None:
|
|
tag_filter, entry = m.groups()
|
|
if not env.app.builder.tags.eval_condition(tag_filter):
|
|
return None
|
|
return entry
|