source: cows_wps/trunk/cows_wps/controllers/submit.py @ 7095

import logging
import urllib2

from pylons import request, response, session, tmpl_context as c
from pylons.controllers.util import abort, redirect_to
from routes import url_for

from cows_wps.renderer.ui_renderer import *
from cows_wps.renderer.proc_config_renderer import *
from cows_wps.controllers import *
from cows_wps.lib.ui.proc_config import *

log = logging.getLogger(__name__)


class SubmitController(BaseController):

    def index(self):
        """
        Submits a job.
        """
        args = {}
        textarea_ids = []
        if "_textarea_ids" in request.params.keys():
            textarea_ids = request.params.getone("_textarea_ids").strip().split()

        for key in request.params.keys():
            # Ignore items prefixed with underscore
            if str(key)[0] == "_": continue

            # Use getall for all args and pipe join them if more than one
            values = request.params.getall(key)
            if values[0] == "-- Please select --":
                del values[0]

            if str(key) in textarea_ids:
                value = request.params.getone(key)
                values = self._handleTextAreaInput(value)

            # Remove if no values provided
            if len(values) == 0: continue

            value = "|".join([str(i) for i in values])
            args[str(key)] = urllib2.quote(value)

        if not args.has_key("proc_id"):
            return "Must provide 'proc_id' arg at the very minimum."

        proc_id = args["proc_id"]
        del args["proc_id"]

        wps_url_format = "/wps?Request=Execute&Format=text/xml&Identifier=%s&Store=false&Status=false&DataInputs="
        wps_request = wps_url_format % proc_id

        for (key, value) in args.items():
            wps_request += "%s,%s," % (key, value)

        # remove comma
        wps_request = wps_request[:-1]
        submitter_url = url_for(controller='/submitter', wps_request_url = wps_request,
                                proc_id = proc_id)
        return redirect_to(submitter_url)

    def _handleTextAreaInput(self, value):
        "Removes any newlines in value and returns as list of strings."
        value = str(value).replace("\r", "").replace("\n", "").strip().split()
        return value

    def choose(self):
        """
        Shows a drop-down list of possible procs to submit - to choose from.
        """
        pc = ProcConfig()
        procs = pc.getProcList()

        resp = """      <form action="/submit/form">
                <p>Please select a process from the drop-down menu below and press 'GO' to generate a form to submit a job.</p>
                <select name="proc_id">"""

        for (proc, long_name) in procs.items():
            resp += ('                  <option name="%s" value="%s">%s</option>\n' % (proc, proc, long_name))

        resp += """             </select>
                <input type="submit" value="GO" />
        </form>"""

        renderer = UIPageRenderer()
        resp = renderer.render("Submission Form",
                               [("Choose a process to submit a job to", resp)])

        return resp


    def form(self):
        """
        Displays a form for submitting a job.
        """
        proc = request.params.get("proc_id", None)
        if proc == None:
            return "Please provide an argument proc_id=&lt;Something&gt;"

        args = {}
        for (k, v) in request.params.items():
            if str(k) == "proc_id": continue
            args[str(k)] = str(v) 

#        pc = ProcConfig()
        pcr = ProcConfigRenderer()
        proc_submission_form_html = pcr.renderProcSubmissionForm(proc, args)
        bbox_required = pcr.bbox_arg_found
        resp = proc_submission_form_html

        renderer = UIPageRenderer()
        resp = renderer.render("Submission Form for: '%s'" % proc,
                               [("Define your inputs", resp)],
                               bbox_required = bbox_required)
        return resp