fix reading cached toctree data

schettino72 · May 4, 2019 · b94f932 · b94f932
1 parent 59b809b
commit b94f932
Show file tree

Hide file tree

Showing 3 changed files with 15 additions and 2 deletions.
diff --git a/CHANGES b/CHANGES
@@ -3,6 +3,12 @@ Changes
 =======
 
 
+0.4.0 (*unreleased*)
+====================
+
+- fix reading cached toctree data
+
+
 0.3.0 (*2019-04-22*)
 ====================
 

diff --git a/setup.py b/setup.py
@@ -5,7 +5,7 @@
 
 setup(
     name='sphinx_press_theme',
-    version='0.3.0',
+    version='0.4.dev0',
     url='https://schettino72.github.io/sphinx_press_site/',
     license='MIT',
     author='Eduardo Naufel Schettino <schetino72>',

diff --git a/sphinx_press_theme/__init__.py b/sphinx_press_theme/__init__.py
@@ -15,11 +15,18 @@ class SimpleTocTreeCollector(EnvironmentCollector):
     """
     def enable(self, app):
         super().enable(app)
-        app.env.toc_dict = {}
+        # env is populated from cache, if not cache create/initalize attibute
+        if not hasattr(app.env, 'toc_dict'):
+            app.env.toc_dict = {}
 
     def clear_doc(self, app, env, docname):
         env.toc_dict.pop(docname, None)
 
+    def merge_other(self, app, env, docnames, other):
+        for docname in docnames:
+            env.toc_dict[docname] = other.toc_dict[docname]
+
+
     def process_doc(self, app, doctree):
         docname = app.env.docname # sphinx mutates this, ouch!!!