D-Net

#!/usr/bin/python

from flask import Flask, jsonify, request

from flask.json import JSONEncoder

from eu.dnetlib.es_connector import DLIESConnector

from flask_prometheus import monitor

from logger import dlilogger

DLI_INDEX = None

ES_HOST = None

def read_properies():

    f = open('api.properties')

    for line in f:

        s = line.strip().split('=')

        if s[0] == 'api.index':

            DLI_INDEX = s[1]

        if s[1] == 'es_index':

            ES_HOST = [x.strip() for x in s[1].split(',')]

class MyJSONEncoder(JSONEncoder):

    def default(self, obj):

        return obj.__dict__

app = Flask(__name__)

app.debug = False

app.json_encoder = MyJSONEncoder

base_dnet_url = "http://aggregator-dli.openaire.eu/dli/"

class InvalidUsage(Exception):

    status_code = 400

    def __init__(self, message, status_code=None, payload=None):

        Exception.__init__(self)

        self.message = message

        if status_code is not None:

            self.status_code = status_code

        self.payload = payload

    def to_dict(self):

        rv = dict(self.payload or ())

        rv['message'] = self.message

        return rv

# Routes

@app.route('/')

def root():

    return app.send_static_file('index.html')

@app.route('/<path:path>')

def static_proxy(path):

    # send_static_file will guess the correct MIME type

    return app.send_static_file(path)

@app.route('/api/item/<path:identifier>', methods=['post', 'get'])

def get_item(identifier):

    if identifier:

        object_type = request.form.get('type')

        start = request.form.get('from')

        if len(object_type) == 0:

            object_type = None

        if len(start) == 0:

            start = None

        else:

            start = int(start)

        connector = DLIESConnector(ES_HOST, DLI_INDEX)

        return jsonify(result=connector.item_by_id(identifier, object_type, start))

    else:

        raise InvalidUsage('This view is gone', status_code=410)

@app.route('/api/stats/', methods=['post', 'get'])

def stats():

    raise Exception("Method to be implemented")

    # q = QueryResolver(host, base_dnet_url)

    # return jsonify(stats=q.get_stats())

@app.route('/api/ds_info/', methods=['post', 'get'])

def info():

    raise Exception("Method to be implemented")

@app.route('/api/stats_detail/', methods=['post', 'get'])

def stats_detail():

    return ""

@app.route('/api/post/', methods=['post', 'get'])

def query_post():

    action = None

    query = None

    filter_key = None

    result = {}

    start = 0

    if 'action' in request.form:

        action = request.form['action']

    if 'query' in request.form:

        query = request.form['query']

    if 'start' in request.form:

        start = int(request.form['start'])

    if 'filter' in request.form:

        filter_key = request.form['filter']

    if action == 'query' and query is not None:

        connector = DLIESConnector()

        try:

            result = connector.simple_query(query, start=start, user_filter=filter_key)

        except Exception as e:

            print e

        return jsonify(result=result)

    else:

        return jsonify(error={'error_code': 'an error occur during query on server'})

if __name__ == '__main__':

    app.logger.addHandler(dlilogger)

    monitor(app, port=8000)

    app.run()