source: branches/branch-1.8/zoo-project/zoo-services/wcps-proxy/cgi-env/WCPS.py @ 981

# -*- coding: utf-8 -*-
#
# Author : Vlad Merticariu <merticariu@rasdaman.com>
#
# Copyright 2015 rasdaman GmbH. All rights reserved.
#
# Permission is hereby granted, free of charge, to any person obtaining a
# copy of this software and associated documentation files (the
# "Software"), to deal in the Software without restriction, including with
# out limitation the rights to use, copy, modify, merge, publish,
# distribute, sublicense, and/or sell copies of the Software, and to
# permit persons to whom the Software is furnished to do so, subject to
# the following conditions:
#
# The above copyright notice and this permission notice shall be included
# in all copies or substantial portions of the Software.
#
# THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS
# OR IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF
# MERCHANTABILITY, FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT.
# IN NO EVENT SHALL THE AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY
# CLAIM, DAMAGES OR OTHER LIABILITY, WHETHER IN AN ACTION OF CONTRACT,
# TORT OR OTHERWISE, ARISING FROM, OUT OF OR IN CONNECTION WITH THE
# SOFTWARE OR THE USE OR OTHER DEALINGS IN THE SOFTWARE.
#
import zoo
import requests

WCPS_REQUEST_TEMPLATE = "%endPoint%?service=WCS&version=2.0.1&request=ProcessCoverages&query=%query%"

def WCPS(conf,inputs,outputs):
        end_point = validate_end_point(inputs["endPoint"]["value"])
        query = get_correct_query(inputs["query"]["value"],outputs["Response"]["mimeType"])
        response_data_type = get_data_type(query)
        status, response = execute_wcps_request(end_point, query)
        import sys    
        print >> sys.stderr,query
        #handle errors
        if status:
                conf["lenv"]["message"] = zoo._("The end point at " + end_point + " could not be reached.")
                return zoo.SERVICE_FAILED
        if "ows:ExceptionReport" in response:
                conf["lenv"]["message"] = parse_error_message(response)
                return zoo.SERVICE_FAILED
                
        #no errors, everything ok       
        outputs["Response"]["value"] = remove_trailing_characters(response, response_data_type)
        #outputs["Response"]["mimeType"] = response_data_type
        #outputs["Response"]["encoding"] = get_encoding(outputs["Response"]["mimeType"])
        return zoo.SERVICE_SUCCEEDED

def parse_error_message(response):
        return response.split("<ows:ExceptionText>")[1].split("</ows:ExceptionText>")[0]

def validate_end_point(end_point):
        ret = end_point
        if not end_point.startswith("http"):
                ret = "http://" + end_point
        return ret

def get_data_type(query):
        ret = "text/plain"
        if "\"png\"" in query:
                ret = "image/png"
        if "\"csv\"" in query:
                ret = "text/csv"
        if "\"tiff\"" in query:
                ret = "image/tiff"
        if "\"jpeg\"" in query or "\"jpg\"" in query:
                ret = "image/jpeg"
        return ret

def get_correct_query(query,mimeType):
    return query.replace("[format]",get_data_ext(mimeType))

def get_data_ext(mimeType):
        ret = "csv"
        for i in ["png","csv","tiff","jpeg"]:
                if i in mimeType:
                        ret = i
                        break
        return ret

def get_encoding(data_type):
        ret = "utf-8"
        if data_type.startswith("image"):
                ret = "base64"
        return ret

def remove_trailing_characters(response, data_type):
        response_start = 27
        if data_type.startswith("image"):
                response_start += len(data_type)
        else:
                response_start += 10
        ret = response[response_start:-17]
        return ret

def execute_wcps_request(end_point, query):
        composed_url = WCPS_REQUEST_TEMPLATE.replace("%endPoint%", end_point).replace("%query%", query) 
        try:
                response = requests.get(composed_url)
        except:
                #can not open url
                return 1, ""
        #all ok
        return 0, response.content