71 lines
1.9 KiB
Python
Executable file
71 lines
1.9 KiB
Python
Executable file
#!/usr/bin/env python3
|
|
# SPDX-FileCopyrightText: 2021 Luca Beltrame <lbeltrame@kde.org>
|
|
# SPDX-License-Identifier: BSD-3-Clause
|
|
|
|
import argparse
|
|
from datetime import datetime
|
|
from pathlib import Path
|
|
from typing import List
|
|
|
|
import frontmatter
|
|
import sarge
|
|
from slugify import slugify
|
|
|
|
POST_PATH = "content/post"
|
|
|
|
|
|
def new_post_title(title: str) -> str:
|
|
suffix = "md"
|
|
current_date = datetime.now().date().isoformat()
|
|
slugified_title = slugify(title)
|
|
post_title = f"{current_date}-{slugified_title}.{suffix}"
|
|
|
|
return post_title
|
|
|
|
|
|
def create_post(post_title: str, tags: List[str] = None,
|
|
categories: List[str] = None, comments: bool = True) -> bool:
|
|
cmd = ["hugo", "new", "-k", "post"]
|
|
title = new_post_title(post_title)
|
|
path = Path(POST_PATH) / title
|
|
cmd.append(str(path))
|
|
|
|
pid = sarge.run(cmd)
|
|
if pid.returncode != 0:
|
|
return False
|
|
|
|
header = frontmatter.load(path)
|
|
|
|
if categories is not None:
|
|
header["categories"] = categories
|
|
if tags is not None:
|
|
header["tags"] = tags
|
|
|
|
header["comments"] = comments
|
|
|
|
with path.open("w") as handle:
|
|
frontmatter.dump(header, handle)
|
|
|
|
return True
|
|
|
|
|
|
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("--draft", action="store_true",
|
|
help="Create the post as draft")
|
|
parser.add_argument("title", help="Title of the new post")
|
|
|
|
pass
|
|
|
|
|
|
if __name__ == "__main__":
|
|
main()
|