ravenscroftj
/
brainsteam.co.uk
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

add page meta tool for bstools

This commit is contained in:
James Ravenscroft 2023-07-09 11:33:28 +01:00
parent 0a7840fdfc
commit 58fb76fafb
1 changed files with 21 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

22
bstools/bstools.py
View File

@ -19,7 +19,7 @@ def cli():
def fix_post_types(folder: str, old_type: str, new_type: str): def fix_post_types(folder: str, old_type: str, new_type: str):
"""Fix post type metadata""" """Fix post type metadata"""
for root, dirs, files in os.walk(folder): for root, _, files in os.walk(folder):
for file in files: for file in files:
if file.endswith(".md"): if file.endswith(".md"):
@ -40,7 +40,27 @@ def fix_post_types(folder: str, old_type: str, new_type: str):
frontmatter.dump(data, f) 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)