server.py

#!/usr/bin/env python3

# Grafeno -- Python concept graphs library
# Copyright 2016 Antonio F. G. Sevilla <afgs@ucm.es>
#
# This library is free software; you can redistribute it and/or
# modify it under the terms of the GNU Affero General Public
# License as published by the Free Software Foundation; either
# version 3 of the License, or (at your option) any later version.
#
# This library is distributed in the hope that it will be useful,
# but WITHOUT ANY WARRANTY; without even the implied warranty of
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
# Affero General Public License for more details.
#
# You should have received a copy of the GNU Affero General Public
# License along with this library; if not, write to the Free Software
# Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA 02111-1307 USA

import bottle
from bottle import abort, error, get, post, request, run, static_file, view

bottle.TEMPLATE_PATH = ["res"]

import collections
import json
import re
import unicodedata

import yaml

from grafeno import pipeline

control_chars = "".join(
    [chr(x) for x in range(0, 32)] + [chr(x) for x in range(127, 160)]
)
control_char_re = re.compile("[%s]" % re.escape(control_chars))


def remove_control_chars(s):
    return control_char_re.sub(" ", s)


def run_pipeline(config):
    try:
        result = pipeline.run(config)
    except ValueError as e:
        abort(400, str(e))
    try:
        j = result.to_json()
        result = json.loads(j)
    except AttributeError:
        pass
    bottle.response.content_type = "application/json"
    return json.dumps({"ok": True, "result": result})


def dict_merge(x, y):
    for key, val in y.items():
        if isinstance(val, collections.Mapping):
            r = dict_merge(x.get(key, {}), val)
            x[key] = r
        else:
            x[key] = y[key]
    return x


# ROUTES


@post("/raw")
def raw_request():
    try:
        config = request.json
    except ValueError:
        abort(400, "Invalid json request")
    return run_pipeline(config)


@post("/run/<config_file>")
def stored_config(config_file):
    try:
        reqbody = request.json
    except ValueError:
        abort(400, "Invalid json request")
    try:
        config_file = open("configs/" + config_file + ".yaml")
    except FileNotFoundError:
        abort(400, "Unknown configuration " + config_file)
    config = yaml.load(config_file, Loader=yaml.FullLoader)
    dict_merge(config, reqbody)
    return run_pipeline(config)


@get("/run/<config_file>")
@view("config")
def view_config(config_file):
    try:
        cfile = open("configs/" + config_file + ".yaml")
    except FileNotFoundError:
        abort(404, "Unknown configuration " + config_file)
    try:
        text = open("default.txt").read()
    except FileNotFoundError:
        text = "John loves Mary."
    return dict(name=config_file, config=cfile.read(), default_text=text)


@error(400)
def custom400(error):
    return json.dumps({"ok": False, "error_message": error.body})


@error(500)
def custom500(error):
    return json.dumps({"ok": False, "error_message": error.body})


# OTHER THINGS


@get("/static/logo.svg")
def get_logo():
    return static_file("logo.svg", root="res")


# RUN SERVER

if __name__ == "__main__":
    import argparse

    arg_parser = argparse.ArgumentParser(description="REST server for concept graphs")
    arg_parser.add_argument(
        "-H", "--hostname", help="hostname to bind to", default="localhost"
    )
    arg_parser.add_argument(
        "-P", "--port", type=int, help="port number to bind to", default=9000
    )
    args = arg_parser.parse_args()
    run(host=args.hostname, port=args.port, reloader=True)
else:
    app = bottle.default_app()