#

# Copyright 2015-2016 IBM Corporation

#

# Licensed under the Apache License, Version 2.0 (the "License");

# you may not use this file except in compliance with the License.

# You may obtain a copy of the License at

#

# http://www.apache.org/licenses/LICENSE-2.0

#

# Unless required by applicable law or agreed to in writing, software

# distributed under the License is distributed on an "AS IS" BASIS,

# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.

# See the License for the specific language governing permissions and

# limitations under the License.

#

import sys

import os

import json

import subprocess

import codecs

import traceback

import flask

from gevent.wsgi import WSGIServer

proxy = flask.Flask(__name__)

proxy.debug = False

with open("function.py") as f:

flask.g = f.read()

@proxy.route("/run", methods=['POST'])

def run():

message = flask.request.get_json(force=True,silent=True)

if not message or not isinstance(message, dict):

flask.abort(403)

if not "value" in message:

flask.abort(403)

value = message["value"]

if not isinstance(value, dict):

flask.abort(403)

# initialize the namespace for the execution

namespace = {}

result = None

try:

exec(flask.g, namespace)

exec("param = " + json.dumps(value), namespace)

exec("fun = main(param)", namespace)

result = namespace['fun']

except Exception:

traceback.print_exc(file = sys.stderr)

sys.stdout.flush()

sys.stderr.flush()

if result and isinstance(result, dict):

response = flask.jsonify(result)

response.status_code = 200

return response

else:

response = flask.jsonify({ "error": "the action did not return a dictionary", "action_output": result })

response.status_code = 502

return response

# start server in a forever loop

if __name__ == "__main__":

server = WSGIServer(('', int(os.getenv("PORT", 8080))), proxy, log=None)

server.serve_forever()