Message

Complexity

Total Complexity

3

Size/Duplication

Total Lines	11
Duplicated Lines	0 %

Importance

Changes	1
Bugs	0	Features	0

3 Methods

Rating	Name	Duplication	Size	Complexity
A	to_JSON()	0	2	1
A	to_JSON_dict()	0	4	1
A	__init__()	0	2	1

#!/usr/bin/env python
# -*- coding: utf-8 -*-

# use data structures
from .ds import Document, Sentence, Dependencies
from .utils import post_json
import json


class Processor(object):

    def __init__(self, address):
        self.service = "{}/annotate".format(address)

    def annotate(self, text):
        try:
            # load json and build Sentences and Document
            msg = Message(text)
            annotated_text = post_json(self.service, msg.to_JSON())
            return Document.load_from_JSON(annotated_text)

        except Exception as e:
            #print(e)
            return None
            #raise Exception("Connection refused!  Is the server running?")


class FastNLPProcessor(Processor):

    def __init__(self, address):
        self.service = "{}/fastnlp/annotate".format(address)

    def annotate(self, text):
        return super(FastNLPProcessor, self).annotate(text)


class BioNLPProcessor(Processor):

    def __init__(self, address):
        self.service = "{}/bionlp/annotate".format(address)

    def annotate(self, text):
        return super(BioNLPProcessor, self).annotate(text)


class Message(object):
    def __init__(self, text):
        self.text = text

    def to_JSON_dict(self):
        jdict = dict()
        jdict["text"] = self.text
        return jdict

    def to_JSON(self):
        return json.dumps(self.to_JSON_dict(), sort_keys=True, indent=4)