👌 IMPROVE: Handling relative path of master_doc to populate frontmatter (#46)

* amended filename function

* minor changes

Author: AakashGfude

jupyterbook_latex/transforms.py

@@ -8,7 +8,7 @@
 from sphinx import addnodes
 from sphinx.builders.latex.nodes import thebibliography
 
-from .utils import getFilename, removeExtension
+from .utils import getFilenameWithSubpath, removeExtension
 from .nodes import HiddenCellNode, H2Node, H3Node
 
 # Utility functions
@@ -103,7 +103,12 @@ def alterNodes(sectLevelsDict, parentSect):
                     replaceWithNode(sect, HiddenCellNode, False)
 
         # check if the document is the masterdoc
-        if getFilename(self.document["source"]) == self.app.config.master_doc:
+        numbSlashes = self.app.config.master_doc.count("/")
+
+        if (
+            getFilenameWithSubpath(self.document["source"], numbSlashes)
+            == self.app.config.master_doc
+        ):
             # pull the toctree-wrapper and append it later to the topmost document level
             for node in self.document.traverse(docutils.nodes.compound):
                 if "toctree-wrapper" in node["classes"]:
@@ -122,7 +127,7 @@ class handleSubSections(SphinxPostTransform):
 
     def apply(self, **kwargs: Any) -> None:
         docname = self.document["source"]
-        if getFilename(docname) == self.app.config.master_doc:
+        if getFilenameWithSubpath(docname, 0) == self.app.config.master_doc:
             for compound in self.document.traverse(docutils.nodes.compound):
                 if "toctree-wrapper" in compound["classes"]:
                     nodecopy = compound
@@ -149,7 +154,7 @@ def checkNodeIsInPart(part, node):
             return False
 
         docname = self.document["source"]
-        if getFilename(docname) == self.app.config.master_doc:
+        if getFilenameWithSubpath(docname, 0) == self.app.config.master_doc:
             TOC_PATH = Path(self.app.confdir).joinpath("_toc.yml")
             tocfile = yaml.safe_load(TOC_PATH.read_text("utf8"))
 

jupyterbook_latex/utils.py

@@ -1,18 +1,32 @@
-from pathlib import Path
-
-
-def getFilename(source):
-    name = Path(source).stem
-    return name
-
-
 def removeExtension(filename):
     index = filename.find(".")
     if index > 0:
         filename = filename[0:index]
     return filename
 
 
+def getFilenameWithSubpath(source, slashes):
+    """Gets the filename along with the relative path upto the number of slashes.
+    :param source: full source path
+    :param slashes: denotes the level of subpath
+    """
+    filename = ""
+    original = str(source)
+
+    while slashes >= 0:
+        index = original.rfind("/")
+        if index == -1:
+            return filename
+        filename = original[index:] + filename
+        original = original[:index]
+        slashes -= 1
+
+    if filename[0] == "/":
+        filename = filename[1:]
+
+    return removeExtension(filename)
+
+
 def sphinxEncode(string):
     return (
         string.replace("~", "\\textasciitilde{}")