From 4493e31a1afaf49187fc9a151982279d83b71f65 Mon Sep 17 00:00:00 2001 From: Mark Powers Date: Tue, 14 Sep 2021 21:42:15 -0500 Subject: Refactor commands to use keyword arguments --- commands.py | 113 +++++++++++++++++++++++++++++++++++++ journal.py | 40 ------------- main.py | 185 +++--------------------------------------------------------- util.py | 63 +++++++++++++++++++++ 4 files changed, 184 insertions(+), 217 deletions(-) create mode 100644 commands.py delete mode 100644 journal.py create mode 100644 util.py diff --git a/commands.py b/commands.py new file mode 100644 index 0000000..4d941cf --- /dev/null +++ b/commands.py @@ -0,0 +1,113 @@ +import sys + +import graphql_queries +from datetime import datetime, timedelta +from util import clean_filename, get_tree, open_editor, get_single_page, print_item, args_for_date + +def create(path, title, content=None): + page = get_single_page(path) + if page is not None: + print("Page already exists with path: %s" % path) + if input("Edit it? (y/n) ") == "y": + edit(path) + return + if not content: + content = open_editor("create", path, "") + response = graphql_queries.create_page(content, title, path) + result = response["data"]["pages"]["create"]["responseResult"] + if not result["succeeded"]: + print("Error!", result["message"]) + sys.exit(1) + print(result["message"]) + +def tree(regex): + """ + Finds pages based on a path search + """ + for item in get_tree(regex): + print_item(item) + +def single(path, raw=False): + """ + View a page with the given path + """ + page = get_single_page(path) + if page is None: + print("No page with path: %s" % path) + sys.exit(1) + if raw: + print("-" * 80) + print_item(page) + print("-" * 80) + print(page["content"]) + +def move(source, dest): + """ + Move a page from one path to another + """ + page = get_single_page(source) + if page is None: + print("Source page %s does not exist" % source) + sys.exit(1) + response = graphql_queries.move_page(page["id"], dest) + result = response["data"]["pages"]["move"]["responseResult"] + if not result["succeeded"]: + print("Error!", result["message"]) + sys.exit(1) + print(result["message"]) + +def edit(path, save=False): + """ + Edit a page + """ + page = get_single_page(path) + if page is None: + print("No page with path: %s" % path) + if input("Create it? (y/n) ") == "y": + title = input("Enter the title: ").strip() + create(path, title) + return + body = page["content"] + + # Open it in editor + new_body = open_editor("edit", path, body) + + # Prompt user to save it to the wiki + print_item(page) + print("-" * 80) + print(new_body) + print("-" * 80) + if save or input("Save changes? (y/n) ") == "y": + response = graphql_queries.edit_page(page["id"], new_body, page["title"], page["path"]) + result = response["data"]["pages"]["update"]["responseResult"] + if not result["succeeded"]: + print("Error!", result["message"]) + sys.exit(1) + print(result["message"]) + +def fill_in_pages(): + last_date = None + for page in get_tree("journal"): + try: + date = datetime.strptime(page["path"], "journal/%Y/%b/%d") + if last_date is None or date > last_date: + last_date = date + except ValueError: + continue + today = datetime.now().date() + if last_date is None: + last_date = today + pending_date = last_date.date() + while pending_date < today: + pending_date += timedelta(days=1) + create(**args_for_date(pending_date)) + +def today(): + """ + Creates a journal page with the path "journal/YYYY/MM/DD" + """ + args = args_for_date(datetime.now().date()) + if get_single_page(args["path"]) is not None: + edit(args["path"]) + else: + create(**args) diff --git a/journal.py b/journal.py deleted file mode 100644 index 484c78d..0000000 --- a/journal.py +++ /dev/null @@ -1,40 +0,0 @@ -from datetime import datetime, timedelta - -from main import get_tree, get_single_page, create - -def args_for_date(date): - return { - "path": date.strftime("journal/%Y/%b/%d").lower(), - "title": date.strftime("%B %-d"), - } - - -def fill_in_pages(args): - last_date = None - for page in get_tree(["journal"]): - try: - date = datetime.strptime(page["path"], "journal/%Y/%b/%d") - if last_date is None or date > last_date: - last_date = date - except ValueError: - continue - today = datetime.now().date() - if last_date is None: - last_date = today - pending_date = last_date.date() - while pending_date < today: - pending_date += timedelta(days=1) - create(args_for_date(pending_date)) - - -def today(args): - """ - Creates a journal page with the path "journal/YYYY/MM/DD" - - args: used - """ - args = args_for_date(datetime.now().date()) - if get_single_page(args["path"]) is not None: - edit(args) - else: - create(args) diff --git a/main.py b/main.py index 361361a..c91fc1c 100755 --- a/main.py +++ b/main.py @@ -1,179 +1,10 @@ #!/usr/bin/env python3 import sys -import os -import subprocess -import re -import datetime import argparse from config import config -import graphql_queries -import journal - -def print_item(item): - trimmmed_path = item["path"][:17]+"..." if len(item["path"]) > 20 else item["path"] - print("| %6s | %20s | %44s |" % (item["id"], trimmmed_path, item["title"]) ) - - -def create(args): - """ - Create a page with the given args. - - args is a dictionary with the following keys: - path: the path to create a page with - title: the title for the page - content (optional): the content for the page. If not present, editor is opened - """ - page = get_single_page(args["path"]) - if page is not None: - print("Page already exists with path: %s" % args["path"]) - if input("Edit it? (y/n) ") == "y": - edit(args) - return - title = args["title"] - path = args["path"] - if "content" not in args: - content = open_editor("create", path, "") - else: - content = args["content"] - response = graphql_queries.create_page(content, title, path) - result = response["data"]["pages"]["create"]["responseResult"] - if not result["succeeded"]: - print("Error!", result["message"]) - sys.exit(1) - print(result["message"]) - -def get_tree(regex): - response = graphql_queries.get_tree() - regex = " ".join(regex) - pages = [] - for item in response["data"]["pages"]["list"]: - if not re.search(regex, item["path"]): - continue - pages.append(item) - return pages - -def tree(args): - """ - Finds pages based on a path search - - args is a dictionary with the following keys: - regex: the regex to search paths with - """ - for item in get_tree(args["regex"]): - print_item(item) - -def get_single_page(path): - """ - Gets the page from the wiki with the given path - """ - if path.startswith("/"): - path = path[1:] - for item in graphql_queries.get_tree()["data"]["pages"]["list"]: - if path == item["path"]: - page_id = int(item["id"]) - response = graphql_queries.get_single_page(page_id) - return response["data"]["pages"]["single"] - return None - -def single(args): - """ - View a page with the given path - - args is a dictionary with the following keys: - path: the path of the page - """ - page = get_single_page(args["path"]) - if page is None: - print("No page with path: %s" % args["path"]) - sys.exit(1) - if False: # not args["skip-header"]: - print("-" * 80) - print_item(page) - print("-" * 80) - print(page["content"]) - -def move(args): - """ - Move a page from one path to another - - args is a dictionary with the following keys: - src_path: page of existing path - dst_path: path page should be moved to - """ - source = args["src_path"] - dest = args["dst_path"] - page = get_single_page(source) - if page is None: - print("Source page %s does not exist" % source) - sys.exit(1) - response = graphql_queries.move_page(page["id"], dest) - result = response["data"]["pages"]["move"]["responseResult"] - if not result["succeeded"]: - print("Error!", result["message"]) - sys.exit(1) - print(result["message"]) - -def clean_filename(pathname): - """ - Clean the path so that it can be used as a filename - """ - pathname = str(pathname).strip().replace('/', '_') - pathname = re.sub(r'\W', '', pathname) - return pathname[:200] - -def open_editor(action, pathname, initial_body): - """ - Open a page with the given pathname and intial_body in an editor, using - action in the filename. Returns the content of the edited file. - """ - if "VISUAL" in os.environ: - editor = os.environ['VISUAL'] - else: - editor = os.environ['EDITOR'] - filename = "/tmp/wikijscmd-"+action+"-"+clean_filename(pathname)+".md" - if len(initial_body) > 0: - with open(filename, "w") as f: - f.write(initial_body) - subprocess.run([editor, filename]) - with open(filename, "r") as f: - new_body = f.read() - os.remove(filename) - return new_body - -def edit(args): - """ - Edit a page - - args is a dictionary with the following keys: - path: the path of the page to edit - save (optional): include with any value to save without any prompt - """ - page = get_single_page(args["path"]) - if page is None: - print("No page with path: %s" % args["path"]) - if input("Create it? (y/n) ") == "y": - title = input("Enter the title: ").strip() - create({"path": args["path"], "title": title}) - return - body = page["content"] - - # Open it in editor - new_body = open_editor("edit", args["path"], body) - - # Prompt user to save it to the wiki - print_item(page) - print("-" * 80) - print(new_body) - print("-" * 80) - if "save" in args or input("Save changes? (y/n) ") == "y": - response = graphql_queries.edit_page(page["id"], new_body, page["title"], page["path"]) - result = response["data"]["pages"]["update"]["responseResult"] - if not result["succeeded"]: - print("Error!", result["message"]) - sys.exit(1) - print(result["message"]) +from commands import create, edit, single, tree, today, move, fill_in_pages def main(): parser = argparse.ArgumentParser("wikijscmd") @@ -183,11 +14,11 @@ def main(): parser_create = subparsers.add_parser("create", help="create a page") parser_create.add_argument("path", type=str, help="the path of the new page") parser_create.add_argument("title", type=str, help="the title of the new page") - parser_create.add_argument("content", nargs='*', type=str, help="optional page content") + parser_create.add_argument("content", nargs="?", type=str, help="optional page content") parser_create.set_defaults(command=create) parser_tree = subparsers.add_parser("tree", help="search in the page tree") - parser_tree.add_argument("regex", nargs='*', type=str, help="optional regex to search paths with") + parser_tree.add_argument("regex", type=str, help="optional regex to search paths with") parser_tree.set_defaults(command=tree) parser_single = subparsers.add_parser("single", help="view a single page") @@ -199,15 +30,15 @@ def main(): parser_edit.set_defaults(command=edit) parser_today = subparsers.add_parser("today", help="create/edit the journal page for today") - parser_today.set_defaults(command=journal.today) + parser_today.set_defaults(command=today) parser_move = subparsers.add_parser("move", help="move a page") - parser_move.add_argument("src_path", type=str, help="the path of the page to move") - parser_move.add_argument("dst_path", type=str, help="the destination path") + parser_move.add_argument("source", type=str, help="the path of the page to move") + parser_move.add_argument("dest", type=str, help="the destination path") parser_move.set_defaults(command=move) parser_journal = subparsers.add_parser("journal", help="create journal pages") - parser_journal.set_defaults(command=journal.fill_in_pages) + parser_journal.set_defaults(command=fill_in_pages) args = vars(parser.parse_args()) callback = args["command"] @@ -215,7 +46,7 @@ def main(): parser.print_help() else: del args["command"] - callback(args) + callback(**args) if __name__ == "__main__": main() diff --git a/util.py b/util.py new file mode 100644 index 0000000..a3b9fbd --- /dev/null +++ b/util.py @@ -0,0 +1,63 @@ +import graphql_queries +import subprocess +import re +import os + +def print_item(item): + trimmmed_path = item["path"][:17]+"..." if len(item["path"]) > 20 else item["path"] + print("| %6s | %20s | %44s |" % (item["id"], trimmmed_path, item["title"]) ) + +def get_single_page(path): + """ + Gets the page from the wiki with the given path + """ + if path.startswith("/"): + path = path[1:] + for item in graphql_queries.get_tree()["data"]["pages"]["list"]: + if path == item["path"]: + page_id = int(item["id"]) + response = graphql_queries.get_single_page(page_id) + return response["data"]["pages"]["single"] + return None + +def get_tree(regex): + response = graphql_queries.get_tree() + pages = [] + for item in response["data"]["pages"]["list"]: + if not re.search(regex, item["path"]): + continue + pages.append(item) + return pages + +def clean_filename(pathname): + """ + Clean the path so that it can be used as a filename + """ + pathname = str(pathname).strip().replace('/', '_') + pathname = re.sub(r'\W', '', pathname) + return pathname[:200] + +def open_editor(action, pathname, initial_body): + """ + Open a page with the given pathname and intial_body in an editor, using + action in the filename. Returns the content of the edited file. + """ + if "VISUAL" in os.environ: + editor = os.environ['VISUAL'] + else: + editor = os.environ['EDITOR'] + filename = "/tmp/wikijscmd-"+action+"-"+clean_filename(pathname)+".md" + if len(initial_body) > 0: + with open(filename, "w") as f: + f.write(initial_body) + subprocess.run([editor, filename]) + with open(filename, "r") as f: + new_body = f.read() + os.remove(filename) + return new_body + +def args_for_date(date): + return { + "path": date.strftime("journal/%Y/%b/%d").lower(), + "title": date.strftime("%B %-d"), + } -- cgit v1.2.3