Moved task stuff into a new attract.tasks package

attract/tasks/routes.py

@@ -0,0 +1,91 @@
+import logging
+
+from flask import Blueprint, render_template, request
+import flask
+
+import pillarsdk
+from pillar.web.system_util import pillar_api
+import pillar.api.utils
+
+from attract.modules import attract_project_view
+from attract.node_types.task import node_type_task
+from attract import current_attract
+
+blueprint = Blueprint('attract.tasks', __name__, url_prefix='/tasks')
+perproject_blueprint = Blueprint('attract.tasks.perproject', __name__,
+                                 url_prefix='/<project_url>/tasks')
+log = logging.getLogger(__name__)
+
+
+@blueprint.route('/')
+def index():
+    return render_template('attract/tasks/index.html')
+
+
+@perproject_blueprint.route('/', endpoint='index')
+@attract_project_view()
+def for_project(project):
+    api = pillar_api()
+
+    tasks = pillarsdk.Node.all({
+        'where': {
+            'project': project['_id'],
+            'node_type': node_type_task['name'],
+        }}, api=api)
+
+    return render_template('attract/tasks/for_project.html',
+                           tasks=tasks['_items'],
+                           project=project)
+
+
+@perproject_blueprint.route('/<task_id>')
+@attract_project_view(extension_props=True)
+def view_embed_task(project, attract_props, task_id):
+    api = pillar_api()
+    task = pillarsdk.Node.find(task_id, api=api)
+    node_type = project.get_node_type(node_type_task['name'])
+    # Fetch project users so that we can assign them tasks
+    users = project.get_users(api=api)
+    project.users = users['_items']
+
+    log.info('Attract properties: %s', attract_props)
+
+    return render_template('attract/tasks/view_task_embed.html',
+                           task=task,
+                           project=project,
+                           task_node_type=node_type,
+                           attract_props=attract_props.to_dict())
+
+
+@perproject_blueprint.route('/<task_id>', methods=['POST'])
+@attract_project_view()
+def save(project, task_id):
+    log.info('Saving task %s', task_id)
+    log.debug('Form data: %s', request.form)
+
+    task_dict = request.form.to_dict()
+    task_dict['users'] = request.form.getlist('users')
+
+    task = current_attract.task_manager.edit_task(task_id, **task_dict)
+
+    return pillar.api.utils.jsonify(task.to_dict())
+
+
+@perproject_blueprint.route('/create', methods=['POST'])
+@attract_project_view()
+def create_task(project):
+
+    task_type = request.form['task_type']
+    parent = request.form.get('parent', None)
+
+    task = current_attract.task_manager.create_task(project,
+                                                    task_type=task_type,
+                                                    parent=parent)
+
+    resp = flask.make_response()
+    resp.headers['Location'] = flask.url_for('.view_embed_task',
+                                             project_url=project['url'],
+                                             task_id=task['_id'])
+    resp.status_code = 201
+
+    return flask.make_response(flask.jsonify({'task_id': task['_id']}), 201)