1
Fork 0
mkdocs-site/hooks/nav.py

25 lines
581 B
Python

import jinja2
def is_list_page(page) -> bool:
if not page.parent:
return False
return page.parent.children[0] == page
def children(page):
if not is_list_page(page):
return []
return page.parent.children[1:]
@jinja2.pass_context
def get_page(context, slug):
nav = context["nav"]
for page in nav.pages:
if page.file.src_uri == slug:
return page
return None
def on_env(env, config, files):
env.filters["is_list_page"] = is_list_page
env.filters["children"] = children
env.globals["get_page"] = get_page