1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
|
#!/usr/bin/python3
import sys
import os
import subprocess
from config import config
import graphql_queries
import re
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(argv):
if len(argv) != 2 or len(argv) != 3:
print("Usage: ./main.py create <path> <title> <content?>")
sys.exit(1)
path = argv[0]
title = argv[1]
if len(argv) == 2:
content = open_editor("create", path, "")
else:
content = argv[2]
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(argv):
response = graphql_queries.get_tree()
for item in response["data"]["pages"]["list"]:
if len(argv) == 1 and argv[0] not in item["path"]:
continue
print_item(item)
def get_single_page(argv):
argument = argv[0]
if not argument.isdigit():
# Strip leading slash
if argument.startswith("/"):
argument = argument[1:]
found = False
for item in graphql_queries.get_tree()["data"]["pages"]["list"]:
if argument == item["path"]:
argument = item["id"]
found = True
if not found:
print("No page with path: %s" % argument)
sys.exit(0)
page_id = int(argument)
response = graphql_queries.get_single_page(page_id)
return response["data"]["pages"]["single"]
def single(argv):
if len(argv) != 1:
print("Usage: ./main.py single <id|path>")
sys.exit(0)
page = get_single_page(argv)
print("-" * 80)
print_item(page)
print("-" * 80)
print(page["content"])
def clean_filename(pathname):
pathname = str(pathname).strip().replace('/', '_')
pathname = re.sub(r'\W', '', pathname)
return pathname[:200]
def open_editor(action, pathname, initial_body):
if "VISUAL" in os.environ:
editor = os.environ['VISUAL']
else:
editor = os.environ['EDITOR']
filename = "/tmp/wikijscmd-"+action+"-"+clean_filename(pathname)
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(argv):
# Load content to edit
if len(argv) != 1:
print("Usage: ./main.py edit <id|path>")
sys.exit(0)
page = get_single_page(argv)
body = page["content"]
# Open it in editor
print(argv)
new_body = open_editor("edit", argv[0], body)
# Prompt user to save it to the wiki
print_item(page)
print("-" * 80)
print(new_body)
print("-" * 80)
if 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 main():
if len(sys.argv) < 2:
print("Usage: ./main.py <command> <args>")
print("Commands:")
print("\tcreate <path> <title> <content?>")
print("\ttree <contains?>")
print("\tsingle <id|path>")
print("\tedit <id|path>")
sys.exit(0)
commands = {
"create": create,
"tree": tree,
"single": single,
"edit": edit
}
command = sys.argv[1]
if command in commands:
# Pass in arguments after the command
commands[command](sys.argv[2:])
else:
print("Unknown command: %s" % sys.argv[1])
if __name__ == "__main__":
main()
|
