From 58fb76fafb0e856a2913458928df39341ec9e380 Mon Sep 17 00:00:00 2001
From: James Ravenscroft <ravenscroftj@gmail.com>
Date: Sun, 9 Jul 2023 11:33:28 +0100
Subject: [PATCH] add page meta tool for bstools

---
 bstools/bstools.py | 22 +++++++++++++++++++++-
 1 file changed, 21 insertions(+), 1 deletion(-)

diff --git a/bstools/bstools.py b/bstools/bstools.py
index 0d56f71..facf69c 100644
--- a/bstools/bstools.py
+++ b/bstools/bstools.py
@@ -19,7 +19,7 @@ def cli():
 def fix_post_types(folder: str, old_type: str, new_type: str):
     """Fix post type metadata"""
 
-    for root, dirs, files in os.walk(folder):
+    for root, _, files in os.walk(folder):
 
         for file in files:
             if file.endswith(".md"):
@@ -40,7 +40,27 @@ def fix_post_types(folder: str, old_type: str, new_type: str):
                         frontmatter.dump(data, f)
 
 
+@cli.command()
+@click.option("--folder", type=click.Path(dir_okay=True, file_okay=False), required=True)
+@click.option("--page_meta", type=str, help="comma separated list of fields to include in page meta", required=True)
+def set_page_meta(folder: str, page_meta: str):
 
+    meta = page_meta.split(",")
+
+    for root, _, files in os.walk(folder):
+
+        for file in files:
+            if file.endswith(".md"):
+                full_path = os.path.join(root,file)
+                data = frontmatter.load(full_path)
+
+                print(f"Update page_meta for {full_path}: {meta}")
+                if 'page_meta' in data:
+                    del data['page_meta']
+                data['post_meta'] = meta
+
+                with open(full_path,'wb') as f:
+                    frontmatter.dump(data, f)