87 lines
3.0 KiB
Python
87 lines
3.0 KiB
Python
# -*- coding: utf-8 -*-
|
|
import sphinx.roles
|
|
import sphinx.environment
|
|
from sphinx.writers.html import HTMLTranslator
|
|
from docutils.writers.html4css1 import HTMLTranslator as DocutilsTranslator
|
|
|
|
def patch():
|
|
# navify toctree (oh god)
|
|
@monkey(sphinx.environment.BuildEnvironment)
|
|
def resolve_toctree(old_resolve, self, *args, **kwargs):
|
|
""" If navbar, bootstrapify TOC to yield a navbar
|
|
|
|
"""
|
|
navbar = kwargs.pop('navbar', None)
|
|
toc = old_resolve(self, *args, **kwargs)
|
|
if toc is None:
|
|
return None
|
|
|
|
navbarify(toc[0], navbar=navbar)
|
|
return toc
|
|
|
|
# monkeypatch visit_table to remove border and add .table
|
|
HTMLTranslator.visit_table = visit_table
|
|
# disable colspec crap
|
|
HTMLTranslator.write_colspecs = lambda self: None
|
|
# copy data- attributes straight from source to dest
|
|
HTMLTranslator.starttag = starttag_data
|
|
|
|
def navbarify(node, navbar=None):
|
|
"""
|
|
:param node: toctree node to navbarify
|
|
:param navbar: Whether this toctree is a 'main' navbar, a 'side' navbar or
|
|
not a navbar at all
|
|
"""
|
|
if navbar == 'side':
|
|
for n in node.traverse():
|
|
if n.tagname == 'bullet_list':
|
|
n['classes'].append('nav')
|
|
elif navbar == 'main':
|
|
# add classes to just toplevel
|
|
node['classes'].extend(['nav', 'navbar-nav', 'navbar-right'])
|
|
for list_item in node.children:
|
|
# bullet_list
|
|
# list_item
|
|
# compact_paragraph
|
|
# reference
|
|
# bullet_list
|
|
# list_item
|
|
# compact_paragraph
|
|
# reference
|
|
list_item['classes'].append('dropdown')
|
|
# list_item.compact_paragraph.reference
|
|
link = list_item.children[0].children[0]
|
|
link['classes'].append('dropdown-toggle')
|
|
link.attributes['data-toggle'] = 'dropdown'
|
|
# list_item.bullet_list
|
|
list_item.children[1]['classes'].append('dropdown-menu')
|
|
|
|
def visit_table(self, node):
|
|
"""
|
|
* remove border
|
|
* add table class
|
|
"""
|
|
self._table_row_index = 0
|
|
self.context.append(self.compact_p)
|
|
self.compact_p = True
|
|
classes = ' '.join({'table', self.settings.table_style}).strip()
|
|
self.body.append(self.starttag(node, 'table', CLASS=classes))
|
|
|
|
def starttag_data(self, node, tagname, suffix='\n', empty=False, **attributes):
|
|
attributes.update(
|
|
(k, v) for k, v in node.attributes.iteritems()
|
|
if k.startswith('data-')
|
|
)
|
|
# oh dear
|
|
return DocutilsTranslator.starttag(
|
|
self, node, tagname, suffix=suffix, empty=empty, **attributes)
|
|
|
|
class monkey(object):
|
|
def __init__(self, obj):
|
|
self.obj = obj
|
|
def __call__(self, fn):
|
|
name = fn.__name__
|
|
old = getattr(self.obj, name)
|
|
setattr(self.obj, name, lambda self_, *args, **kwargs: \
|
|
fn(old, self_, *args, **kwargs))
|