def create_task(args):
# INITIALIZE DRIVERS AND HELPERS
wrike_gateway = Wrike(wait_for_redirect=True, data_filepath=Helper.get_data_filepath("wrike"))
github_helper = GithubAPIHelper()
# GET FOLDER ID
id_folder = args.id_folder
should_check_folder_id_validity = True
if id_folder is None:
res = WrikeAPIHelper(wrike=wrike_gateway).get_folders_list(args.title_folder)
if res["id"] is None:
print 'Folder "{0}" could not be found'.format(res["prepend"])
sys.exit(-1)
else:
id_folder = res["id"]
should_check_folder_id_validity = False
if should_check_folder_id_validity:
f = wrike_gateway.get_folder(id_folder)
if f is None:
print "Wrong folder id"
sys.exit(-1)
# CREATE TASK
create_task_params = {"description": args.description}
if args.self_assign:
contact = wrike_gateway.get_current_contact()
create_task_params.update({"responsibles": "['{0}']".format(contact["id"])})
task = wrike_gateway.create_task(id_folder, args.title, create_task_params)
print task["permalink"]
# CREATE ISSUE
if args.github_issue:
issue = github_helper.issue_from_task_object(task, args.self_assign)
print issue["html_url"]
wrike_gateway.create_task_comment(task["id"], issue["html_url"])
branch_name = Helper.branch_name(issue)
if not args.nobranch:
Helper.create_branch(branch_name)
else:
print branch_name
wrike_gateway.redirect(task["permalink"])
def issue_from_wrike(args):
# GET TASKS
tasks = []
wrike_gateway = Wrike(data_filepath=Helper.get_data_filepath('wrike'), wait_for_redirect=True)
github_gateway = GithubAPIGateway(*Helper.owner_and_repo())
for taskid in args.taskids:
task = wrike_gateway.get_task(taskid)
if task is None:
print "'{0}' is not a valid taskid or it cannot be found".format(taskid)
sys.exit(-1)
tasks.append(task)
# BODY OF ISSUE
body = ''
for task in tasks:
body += '### {0}\n___\n\n{1}\n'.format(task['permalink'].encode('utf-8'), task['description'].encode('utf-8'))
# TITLE OF ISSUE
title = tasks[0]['title']
if len(tasks) > 1:
title += ' (+{0} Wrike tasks)'.format(len(tasks) - 1)
# CREATE ISSUE
issue = github_gateway.create_issue(title, True, {'body': body})
print issue['html_url']
branch_name = Helper.branch_name(issue)
if args.nobranch == False:
Helper.create_branch(branch_name)
else:
print branch_name
# WRITE LINK TO ISSUE ON EVERY TASK AND SET ASIGNEE
wrike_gateway.redirect(issue['html_url'])
contact = wrike_gateway.get_current_contact()
for task in tasks:
wrike_gateway.create_task_comment(task['id'], issue['html_url'])
if contact['id'] not in task['responsibleIds']:
wrike_gateway.change_task(task['id'], {
'addResponsibles': "['{0}']".format(contact['id'])
})