initial dump, sorta broken

6 files changed, 354 insertions, 0 deletions

diff --git a/client/bug.py b/client/bug.py
new file mode 100755
index 0000000..24a523c
--- /dev/null
+++ b/client/bug.py
@@ -0,0 +1,58 @@
+#!../flask/bin/python
+"""
+usage: bug [options] <command> [<args>...]
+
+options:
+  -u, --uri ENDPOINT  Project API endpoint
+  -h, --help          Print this help text
+  -v, --version       Print the client version
+
+commands:
+  open        Open a new ticket
+  list        List all open bugs in the project
+  show        Show a specific ticket by ID
+
+  [Not implemented yet:]
+  resolution  Set the resolution status of a ticket
+  edit        Edit a ticket
+  reopen      Reopen a previously closed ticket
+  close       Close a ticket
+  lock        Lock a ticket, preventing new comments
+  label       Apply a label to a ticket
+
+See 'bug help <command>' for more information on a specific command
+"""
+from importlib import import_module
+from subprocess import call
+from docopt import docopt
+
+commands = ['open', 'delete', 'show', 'list', 'edit']
+def main():
+
+	if args['<command>'] in ['help', None]:
+		if not args['<args>']:
+			print(__doc__.lstrip().rstrip())
+		else:
+			if args['<args>'][0] in commands:
+				bug_mod = import_module('bug_{}'.format(args['<args>'][0]))
+				print(bug_mod.__doc__.lstrip().rstrip())
+			else:
+				exit("'{}' is not a bug.py command. See 'bug help'.".format(args['<args>'][0]))
+	elif args['<command>'] in commands:
+		if not args['--uri']:
+			exit("URI missing")
+
+		bug_mod = import_module('bug_{}'.format(args['<command>']))
+		argv = [args['<command>']] + args['<args>']
+		arguments = args.copy()
+		arguments.update(docopt(bug_mod.__doc__, argv=argv))
+		bug_mod.call(arguments)
+	else:
+		exit("'{}' is not a bug.py command. See 'bug help'.".format(args['<command>']))
+
+if __name__ == '__main__':
+	args = docopt(__doc__,
+	              version='tbt client version 0.0.0.alpha',
+	              options_first=True)
+	#print(args)
+	main()
diff --git a/client/bug_delete.py b/client/bug_delete.py
new file mode 100755
index 0000000..bcce88e
--- /dev/null
+++ b/client/bug_delete.py
@@ -0,0 +1,33 @@
+#!../flask/bin/python
+"""
+usage: bug delete [options] <ticket_id>
+
+If no arguments are given it will open your $EDITOR where the first line is
+the summary following a newline and then the body of the report.  Both are
+required.
+
+    -h, --help            Print this help text
+    -i, --ticket-id       ID of the ticket to delete
+"""
+from docopt import docopt
+import json, requests
+
+if __name__ == '__main__':
+	print(docopt(__doc__))
+
+def call(args):
+	print(args)
+	api_endpoint = args['--uri'] + '/api/1.0/ticket/'
+
+	req = requests.delete(api_endpoint + args['<ticket_id>'])
+
+	res = json.loads(req.text)
+
+	if req.status_code == 404:
+		print("Ticket with ID '{}' could not be deleted: {}".format(args['<ticket_id>'], res['error']))
+	elif req.status_code == 200:
+		print("Ticket with ID '{}' deleted successfully.".format(args['<ticket_id>']))
+	else:
+		exit("ALERT ALERT ALERT")
+
+	#print("{} {}\n    {}".format(t['id'], t['title'], t['uri']))
diff --git a/client/bug_edit.py b/client/bug_edit.py
new file mode 100755
index 0000000..42eca80
--- /dev/null
+++ b/client/bug_edit.py
@@ -0,0 +1,51 @@
+#!../flask/bin/python
+"""
+usage: bug edit [options] ticket <ticket_id>
+       bug edit [options] comment <comment_id>
+
+If both the -s and -b options are given, the summary and body of the ticket or
+comment with the given ID will be updated to the specified strings. If -s is
+given for a comment it is added to the top of the comment instead.
+
+If they are not given, your $EDITOR will be opened where the first line is the
+summary following a newline and then the body of the ticket.  Both are
+required.
+
+options:
+    -h, --help            Print this help text
+    -s, --summary STRING  A short summary of the bug
+    -b, --body STRING     The long description of the bug or the body of the
+                          comment
+"""
+from docopt import docopt
+import json, requests
+
+if __name__ == '__main__':
+	print(docopt(__doc__))
+
+def call(args):
+	print(args)
+	api_endpoint = args['--uri'] + '/api/1.0/ticket'
+
+	ticket = {}
+	if args['--summary']:
+		ticket['summary'] = args['--summary']
+	else:
+		exit("Summary needed, no interactive edit yet")
+
+	if args['--body']:
+		ticket['body'] = args['--body']
+	else:
+		ticket['body'] = None
+
+	headers = {
+		'Content-Type': 'application/json',
+		'Accept': 'text/plain'
+	}
+	payload = json.dumps(ticket)
+
+	r = requests.post(api_endpoint, data=payload, headers=headers)
+
+	t = json.loads(r.text).get('ticket')
+
+	print("{} {}\n    {}".format(t['id'], t['summary'], t['uri']))
diff --git a/client/bug_list.py b/client/bug_list.py
new file mode 100755
index 0000000..6084414
--- /dev/null
+++ b/client/bug_list.py
@@ -0,0 +1,56 @@
+#!../flask/bin/python
+"""
+usage: bug list [options]
+
+If no arguments are given it will open your $EDITOR where the first line is
+the summary following a newline and then the body of the report.  Both are
+required.
+
+    -h, --help            Print this help text
+"""
+from docopt import docopt
+from textwrap import indent
+from datetime import datetime
+import json, requests
+
+if __name__ == '__main__':
+	print(docopt(__doc__))
+
+def call(args):
+	print(args)
+	api_endpoint = args['--uri'] + '/api/1.0/tickets'
+
+	r = requests.get(api_endpoint)
+
+	tickets = json.loads(r.text).get('tickets')
+
+	for ticket in tickets:
+		output = '[TBT#{}] '.format(ticket['id'])
+
+		if 'deleted' in ticket and ticket['deleted'] == True:
+			output += '[DELETED] '
+
+		output += '[{}] '.format(ticket['status'])
+
+		output += '{}\n'.format(ticket['summary'])
+
+		if ticket['status'] != 'open':
+			output += 'Resolution: {}\n'.format(ticket['resolution'])
+
+		if ticket['reason']:
+			output += 'Reason: {}\n'.format(ticket['reason'])
+
+		output += 'Opened by: {} <{}>\n'.format(ticket['opened_by']['nickname'], ticket['opened_by']['email'])
+		output += 'Opened at: {} UTC\n'.format(datetime.strptime( ticket['opened_at'], "%Y-%m-%dT%H:%M:%S" ))
+
+		if ticket['assigned_to']:
+			output += 'Assigned to: {} <{}>\n'.format(ticket['assigned_to']['nickname'], ticket['assigned_to']['email'])
+		else:
+			output += 'Assigned to: Unassigned\n'
+
+		if ticket['updated_at']:
+			output += 'Updated at: {} UTC\n'.format(datetime.strptime(ticket['opened_at'], "%Y-%m-%dT%H:%M:%S" ))
+
+		output += '\n' + indent('{}'.format(ticket['body']), '  ') + '\n'
+
+		print(output)
diff --git a/client/bug_open.py b/client/bug_open.py
new file mode 100755
index 0000000..17a3e4f
--- /dev/null
+++ b/client/bug_open.py
@@ -0,0 +1,101 @@
+#!../flask/bin/python
+"""
+usage: bug open [options]
+
+If no arguments are given it will open your $EDITOR where the first line is
+the summary following a newline and then the body of the report.  Both are
+required.
+
+    -h, --help            Print this help text
+    -s, --summary STRING  A short summary of the bug
+    -b, --body STRING     The long description of the bug
+"""
+import os, re, tempfile, subprocess
+from docopt import docopt
+import json, requests
+from bug_show import show_ticket
+
+if __name__ == '__main__':
+	print(docopt(__doc__))
+
+def call(args):
+	print(args)
+	api_endpoint = args['--uri'] + '/api/1.0/ticket'
+
+	if args['--summary']:
+		summary = args['--summary']
+	else:
+		summary = ''
+
+	if args['--body']:
+		body = args['--body']
+	else:
+		body = ''
+
+	if not(summary and body):
+		(summary, body) = editor_prompt(summary, body)
+
+	ticket = {
+		'summary': summary,
+		'body': body,
+		'user_nickname': 'demi'
+	}
+
+	headers = {
+		'Content-Type': 'application/json',
+		'Accept': 'text/plain'
+	}
+	payload = json.dumps(ticket)
+
+	r = requests.post(api_endpoint, data=payload, headers=headers)
+
+	t = json.loads(r.text).get('ticket')
+
+	print(t)
+	print(show_ticket(t))
+
+
+def editor_prompt(summary, body):
+	editor = os.environ.get('EDITOR','vim')
+	message=''
+
+	if summary:
+		message += summary
+
+	if body:
+		message += '\n\n' + body
+
+	message += """
+
+# Please enter the summary on a single line, followed
+# by an empty line, then followed by the body of the
+# ticket.
+#
+# Both the summary and body are required. If either of
+# them are missing, or they aren't separated properly
+# the submission will be aborted.
+"""
+
+	tmp = tempfile.NamedTemporaryFile()
+	tmp.write(message.encode("utf-8"))
+	tmp.flush()
+
+	regx = re.compile('^(.+?)\n\n(.+)$', re.S)
+
+	subprocess.call([editor, tmp.name])
+
+	tmp.seek(0)
+	data = tmp.read().decode("utf-8")
+	tmp.close()
+
+	data = data[:-263] # Strip the commented out message
+	data = data.lstrip().rstrip() # Strip opening and ending whitespace
+	regmatch = regx.match(data)
+
+	if len(regmatch.groups()) != 2:
+		exit("Error: summary and body not separated properly, aborting")
+
+	summary = regmatch.group(1)
+	body = regmatch.group(2)
+
+	return (summary, body)
diff --git a/client/bug_show.py b/client/bug_show.py
new file mode 100755
index 0000000..0c3c9b8
--- /dev/null
+++ b/client/bug_show.py
@@ -0,0 +1,55 @@
+#!../flask/bin/python
+"""
+usage: bug show [options] <ticket_id>
+
+    -h, --help            Print this help text
+"""
+from docopt import docopt
+from textwrap import indent
+from datetime import datetime
+import json, requests
+
+if __name__ == '__main__':
+	print(docopt(__doc__))
+
+def call(args):
+	print(args)
+	api_endpoint = args['--uri'] + '/api/1.0/ticket/'
+
+	r = requests.get(api_endpoint + args['<ticket_id>'])
+
+	ticket = json.loads(r.text).get('ticket')
+	print(ticket)
+	print(show_ticket(ticket))
+
+def show_ticket(ticket):
+	output = '[TBT#{}] '.format(ticket['id'])
+
+	if 'deleted' in ticket and ticket['deleted'] == True:
+		output += '[DELETED] '
+
+	output += '[{}] '.format(ticket['status'])
+
+	output += '{}\n'.format(ticket['summary'])
+
+
+	if ticket['status'] != 'open':
+		output += 'Resolution: {}\n'.format(ticket['resolution'])
+
+	if ticket['reason']:
+		output += 'Reason: {}\n'.format(ticket['reason'])
+
+	output += 'Opened by: {} <{}>\n'.format(ticket['opened_by']['nickname'], ticket['opened_by']['email'])
+	output += 'Opened at: {} UTC\n'.format(datetime.strptime( ticket['opened_at'], "%Y-%m-%dT%H:%M:%S" ))
+
+	if ticket['assigned_to']:
+		output += 'Assigned to: {} <{}>\n'.format(ticket['assigned_to']['nickname'], ticket['assigned_to']['email'])
+	else:
+		output += 'Assigned to: Unassigned\n'
+
+	if ticket['updated_at']:
+		output += 'Updated at: {} UTC\n'.format(datetime.strptime(ticket['opened_at'], "%Y-%m-%dT%H:%M:%S" ))
+
+	output += '\n' + indent('{}'.format(ticket['body']), '  ') + '\n'
+
+	return output