New script to create a new post for Jekyll

jekyll_new_post.py

@@ -0,0 +1,120 @@
+#!/usr/bin/env python3
+
+import argparse
+import datetime
+from hashlib import sha1
+from pathlib import Path
+import subprocess
+from textwrap import fill
+
+from git import Repo
+import pytz
+import sarge
+from slugify import slugify
+import yaml
+
+
+def is_jekyll_root(path: Path):
+    return (Path / "_config.yml").exists()
+
+
+def hash_file(filename: Path) -> str:
+
+    # Git-like SHA1 hashing
+
+    filesize = filename.stat().st_size
+
+    with filename.open() as handle:
+        data = handle.read()
+
+    contents = "blob " + str(filesize) + "\0" + data
+    file_hash = sha1(contents.encode())
+
+    return file_hash.hexdigest()
+
+
+def perform_commit(filepath: Path, title: str=None) -> None:
+
+    repo = Repo("./")  # FIXME
+    repo.index.add(str(filepath))
+    message = "New post: {}".format(title)
+    message = fill(message, width=78)
+
+    repo.commit(message)
+    repo.remotes.origin.push()
+
+
+def create_new_post(title: str, categories: list=None, tags: list=None,
+                    post_metadata: dict=None, comments: bool=True) -> Path:
+
+    title_slug = slugify(title)
+    current_time = datetime.datetime.now(pytz.timezone("Europe/Rome"))
+    date_slug = current_time.date().isoformat()
+    filename = "{0}-{1}.markdown".format(date_slug, title_slug)
+    formatted_date = current_time.strftime("%Y-%m-%d %H:%M:%S%z")
+
+    metadata = dict(title=title, comments=comments,
+                    date=formatted_date)
+
+    if categories is not None:
+        metadata["categories"] = categories
+    else:
+        metadata["categories"] = ["General"]
+
+    if tags is not None:
+        metadata["tags"] = tags
+
+    if post_metadata is not None:
+        for key, value in post_metadata.items():
+            metadata[key] = value
+
+    final_path = Path("_posts") / filename
+
+    with final_path.open("w") as handle:
+        handle.write("---\n")
+        yaml.safe_dump(metadata, handle, default_flow_style=False)
+        handle.write("---\n")
+
+    pre_edit_hash = hash_file(final_path)
+    subprocess.check_call("/usr/bin/kate {}".format(Path("_posts") / filename),
+                          shell=True)
+    post_edit_hash = hash_file(final_path)
+
+    if pre_edit_hash == post_edit_hash:
+        print("No post content. Aborting.")
+        return
+
+    return (final_path)
+
+
+def main():
+
+    parser = argparse.ArgumentParser()
+    parser.add_argument("-t", "--tags", nargs="+",
+                        help="Post tags (space separated)")
+    parser.add_argument("-c", "--categories", nargs="+",
+                        help="Post categories (space separated)")
+    parser.add_argument("--disable_comments", action="store_false",
+                        help="Disable comments for the post")
+    parser.add_argument("--commit", action="store_true",
+                        help="Commit and push to the remote repository")
+    parser.add_argument("title", help="Title of the new post")
+
+    options = parser.parse_args()
+
+    path = create_new_post(options.title, options.categories, options.tags,
+                           comments=options.disable_comments)
+
+    if path is None:
+        exit(1)
+
+    print("Created new post {}".format(path.name))
+
+    if options.commit:
+        print("Committing to upstream repository...")
+        perform_commit(path, options.title)
+    print("Done.")
+
+
+if __name__ == "__main__":
+    main()