Files
hykilpikonna/.github/workflows/generator.py
T
2021-12-25 20:35:59 -05:00

51 lines
1.5 KiB
Python

import json
import os
from datetime import datetime, date
from pathlib import Path
import yaml
class EnhancedJSONEncoder(json.JSONEncoder):
def default(self, o: object) -> object:
# Support encoding datetime
if isinstance(o, (datetime, date)):
return o.isoformat()
return super().default(o)
if __name__ == '__main__':
metas = []
for b in os.listdir('content/posts'):
if not b.endswith('.md'):
continue
file = 'content/posts/' + b
with open(file, 'r', encoding='utf-8') as f:
md = f.read().strip()
start = md.index('---\n')
stop = md.index('---\n', start + 1)
yml = md[start + 4:stop]
meta = {**yaml.safe_load(yml), 'file': file}
metas.append(meta)
meta.setdefault('tags', [])
# Parse date
if 'date' not in meta:
try:
meta['date'] = date(int(b[:4]), int(b[5:7]), int(b[8:10]))
except:
pass
if 'title_image' in meta and '/' not in meta['title_image']:
meta['title_image'] = 'content/generated/' + meta['title_image']
metas.sort(key=lambda x: x['date'], reverse=True)
metas_path = Path('content/generated/metas.json')
metas_path.parent.mkdir(exist_ok=True, parents=True)
with open(metas_path, 'w', encoding='utf-8') as f:
f.write('[\n ' + ',\n '.join(
json.dumps(m, cls=EnhancedJSONEncoder, ensure_ascii=False) for m in metas) + '\n]')