Update generate-home-buttons.py

Tyler887 · Apr 13, 2022 · 315daa1 · 315daa1
1 parent 13ff059
commit 315daa1
Showing 1 changed file with 6 additions and 3 deletions.
diff --git a/generate-home-buttons.py b/generate-home-buttons.py
@@ -1,11 +1,14 @@
 import os, glob
 from colorama import *
-def addhome(page):
+def addhome(page, entry=False):
    try:
       if os.path.isfile(page):
        with open(page, "a+") as file:
         if page.endswith(".md"):
-         file.write(f"\n\n[<div align=\"center\"><sub><i><< Back to home...</i></sub></div>](/typoqsuat)")
+         if entry:
+          file.write(f"\n\n[<sub><i><< Back to home...</i></sub>](/typoqsuat) <sub>|</sub> [<sub><i>YAML source...</i></sub>](https://github.com/Tyler887/typoqsuat/blob/main/api/entry/{os.path.basename(page).replace('.md','.yaml')})")
+         else:
+          file.write(f"\n\n[<sub><i><< Back to home...</i></sub>](/typoqsuat)")
          print(f"{Fore.GREEN}Added{Style.RESET_ALL} home button to {file}!\n")
         else:
          print(f"{Fore.YELLOW}Skipped{Style.RESET_ALL} {file}.\n")
@@ -18,6 +21,6 @@ def addhome(page):
 print(f"{Fore.BLUE}>>>{Style.RESET_ALL} Generating home buttons for entries.")
 
 for i in glob.glob(f"{os.getcwd()}/typo/*"):
-  addhome(i)
+  addhome(i, entry=True)
 
 print("Generated home buttons!")