aboutsummaryrefslogtreecommitdiffstats
path: root/docs/utils.py
diff options
context:
space:
mode:
Diffstat (limited to 'docs/utils.py')
-rw-r--r--docs/utils.py10
1 files changed, 7 insertions, 3 deletions
diff --git a/docs/utils.py b/docs/utils.py
index ba6c8e34..3f58e767 100644
--- a/docs/utils.py
+++ b/docs/utils.py
@@ -160,9 +160,13 @@ def cleanup() -> None:
content = file.read_text(encoding="utf-8").splitlines(keepends=True)
# Rename the extension to be less wordy
- # Example: pydis_core.exts -> pydis_core Exts
- title = content[0].split()[0].strip().replace("pydis_core.", "").replace(".", " ").title()
- title = f"{title}\n{'=' * len(title)}\n\n"
+ # Example: pydis_core.exts -> Exts
+ title = content[0].split()[0].strip().replace("\\", "").replace("pydis_core.", "").replace(".", " ")
+ if "pydis_core" in title:
+ # Root title: pydis_core -> pydis core
+ title = title.replace("_", " ")
+
+ title = f"{title.title()}\n{'=' * len(title)}\n\n"
content = title, *content[3:]
file.write_text("".join(content), encoding="utf-8")