pdftocgen/tocgen.py

from fitz import Document
from typing import List
from fitzutils import ToCEntry
from .recipe import Recipe, extract_toc

def gen_toc(doc: Document, recipe_dict: dict) -> List[ToCEntry]:
    """Generate the table of content for a document from recipe



    Argument

      doc: a pdf document

      recipe_dict: the recipe dictionary used to generate the toc

    Returns

      a list of ToC entries

    """
    return extract_toc(doc, Recipe(recipe_dict))