mirror of https://github.com/gsi-upm/senpy
You cannot select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
115 lines
3.8 KiB
Python
115 lines
3.8 KiB
Python
import logging
|
|
import ConfigParser
|
|
|
|
logger = logging.getLogger(__name__)
|
|
|
|
PARAMS = {"input": {"aliases": ["i", "input"],
|
|
"required": True,
|
|
"help": "Input text"
|
|
},
|
|
"informat": {"aliases": ["f", "informat"],
|
|
"required": False,
|
|
"default": "text",
|
|
"options": ["turtle", "text"],
|
|
},
|
|
"intype": {"aliases": ["intype", "t"],
|
|
"required": False,
|
|
"default": "direct",
|
|
"options": ["direct", "url", "file"],
|
|
},
|
|
"outformat": {"aliases": ["outformat", "o"],
|
|
"default": "json-ld",
|
|
"required": False,
|
|
"options": ["json-ld"],
|
|
},
|
|
"language": {"aliases": ["language", "l"],
|
|
"required": False,
|
|
"options": ["es", "en"],
|
|
},
|
|
"prefix": {"aliases": ["prefix", "p"],
|
|
"required": True,
|
|
"default": "",
|
|
},
|
|
"urischeme": {"aliases": ["urischeme", "u"],
|
|
"required": False,
|
|
"default": "RFC5147String",
|
|
"options": "RFC5147String"
|
|
},
|
|
}
|
|
|
|
|
|
class SenpyPlugin(object):
|
|
def __init__(self, info=None):
|
|
if not info:
|
|
raise ValueError("You need to provide configuration information for the plugin.")
|
|
logger.debug("Initialising {}".format(info))
|
|
self.name = info["name"]
|
|
self.version = info["version"]
|
|
self.params = info.get("params", PARAMS.copy())
|
|
self.extra_params = info.get("extra_params", {})
|
|
self.params.update(self.extra_params)
|
|
self.is_activated = False
|
|
self.info = info
|
|
|
|
def analyse(self, *args, **kwargs):
|
|
logger.debug("Analysing with: {} {}".format(self.name, self.version))
|
|
pass
|
|
|
|
def activate(self):
|
|
pass
|
|
|
|
def deactivate(self):
|
|
pass
|
|
|
|
|
|
@property
|
|
def id(self):
|
|
return "{}_{}".format(self.name, self.version)
|
|
|
|
def jsonable(self, parameters=False):
|
|
resp = {
|
|
"@id": "{}_{}".format(self.name, self.version),
|
|
"is_activated": self.is_activated,
|
|
}
|
|
if hasattr(self, "repo") and self.repo:
|
|
resp["repo"] = self.repo.remotes[0].url
|
|
if parameters:
|
|
resp["parameters"] = self.params
|
|
elif self.extra_params:
|
|
resp["extra_parameters"] = self.extra_params
|
|
return resp
|
|
|
|
|
|
class SentimentPlugin(SenpyPlugin):
|
|
def __init__(self,
|
|
min_polarity_value=0,
|
|
max_polarity_value=1,
|
|
**kwargs):
|
|
super(SentimentPlugin, self).__init__(**kwargs)
|
|
self.minPolarityValue = min_polarity_value
|
|
self.maxPolarityValue = max_polarity_value
|
|
|
|
def jsonable(self, *args, **kwargs):
|
|
resp = super(SentimentPlugin, self).jsonable(*args, **kwargs)
|
|
resp["marl:maxPolarityValue"] = self.maxPolarityValue
|
|
resp["marl:minPolarityValue"] = self.minPolarityValue
|
|
return resp
|
|
|
|
|
|
class EmotionPlugin(SenpyPlugin):
|
|
def __init__(self,
|
|
min_emotion_value=0,
|
|
max_emotion_value=1,
|
|
emotion_category=None,
|
|
**kwargs):
|
|
super(EmotionPlugin, self).__init__(**kwargs)
|
|
self.minEmotionValue = min_emotion_value
|
|
self.maxEmotionValue = max_emotion_value
|
|
self.emotionCategory = emotion_category
|
|
|
|
def jsonable(self, *args, **kwargs):
|
|
resp = super(EmotionPlugin, self).jsonable(*args, **kwargs)
|
|
resp["onyx:minEmotionValue"] = self.minEmotionValue
|
|
resp["onyx:maxEmotionValue"] = self.maxEmotionValue
|
|
return resp
|