aboutsummaryrefslogtreecommitdiff
path: root/scripts/common.py
diff options
context:
space:
mode:
Diffstat (limited to 'scripts/common.py')
-rw-r--r--scripts/common.py53
1 files changed, 53 insertions, 0 deletions
diff --git a/scripts/common.py b/scripts/common.py
new file mode 100644
index 0000000..9b23816
--- /dev/null
+++ b/scripts/common.py
@@ -0,0 +1,53 @@
+#!/usr/bin/env python3
+
+import os
+import frontmatter
+import dateparser
+import datetime
+from pathlib import Path
+
+BLOG_LIST_FILE = '.bloglist.md'
+
+def get_href(path):
+ path = Path(path)
+ path = path.relative_to(*path.parts[:1])
+ if path.name == 'index.md':
+ return '/{}/'.format(path.parent)
+ else:
+ return '/{}/{}.html'.format(path.parent, path.stem)
+
+def read_metadata(file):
+ post = frontmatter.load(file)
+
+ for field in ['title', 'date']:
+ if post.get(field) is None:
+ raise Exception("{} is missing metadata field '{}'".format(file, field))
+
+ if type(post['date']) not in [datetime.datetime, datetime.date]:
+ date = dateparser.parse(post['date'])
+ else:
+ date = post['date']
+
+ return {
+ 'title': post.get('title'),
+ 'date': date,
+ 'author': post.get('author'),
+ 'description': post.get('description'),
+ 'draft': post.get('draft'),
+ 'href': get_href(file)
+ }
+
+def get_blog_posts(blog_dir):
+ blog_index = os.path.join(blog_dir, 'index.md')
+ posts = []
+
+ for root, dirs, files in os.walk(blog_dir):
+ for file in files:
+ path = os.path.join(root, file)
+ if path.endswith('.md') and path != blog_index and file != BLOG_LIST_FILE:
+ metadata = read_metadata(path)
+ if not metadata['draft']:
+ posts.append(metadata)
+
+ posts.sort(key=lambda p: (p is None, p['date']), reverse=True)
+ return posts