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.
156 lines
4.9 KiB
Python
156 lines
4.9 KiB
Python
import os
|
|
import sys
|
|
import imp
|
|
|
|
from flask import current_app
|
|
from collections import defaultdict
|
|
from .plugins import SentimentPlugin, EmotionPlugin
|
|
|
|
try:
|
|
from flask import _app_ctx_stack as stack
|
|
except ImportError:
|
|
from flask import _request_ctx_stack as stack
|
|
|
|
from blueprints import nif_blueprint
|
|
from git import Repo, InvalidGitRepositoryError
|
|
|
|
class Senpy(object):
|
|
|
|
def __init__(self, app=None, plugin_folder="plugins"):
|
|
self.app = app
|
|
base_folder = os.path.join(os.path.dirname(__file__), "plugins")
|
|
|
|
self.search_folders = (folder for folder in (base_folder, plugin_folder, '/tmp/plugins')
|
|
if folder and os.path.isdir(folder))
|
|
|
|
if app is not None:
|
|
self.init_app(app)
|
|
|
|
"""
|
|
|
|
Note: I'm not particularly fond of adding self.app and app.senpy, but
|
|
I can't think of a better way to do it.
|
|
"""
|
|
def init_app(self, app, plugin_folder="plugins"):
|
|
app.senpy = self
|
|
#app.config.setdefault('SQLITE3_DATABASE', ':memory:')
|
|
# Use the newstyle teardown_appcontext if it's available,
|
|
# otherwise fall back to the request context
|
|
if hasattr(app, 'teardown_appcontext'):
|
|
app.teardown_appcontext(self.teardown)
|
|
else:
|
|
app.teardown_request(self.teardown)
|
|
app.register_blueprint(nif_blueprint)
|
|
|
|
def analyse(self, **params):
|
|
algo = None
|
|
print("analysing with params: {}".format(params))
|
|
if "algorithm" in params:
|
|
algo = params["algorithm"]
|
|
elif self.plugins:
|
|
algo = self.default_plugin
|
|
if algo in self.plugins and self.plugins[algo].enabled:
|
|
plug = self.plugins[algo]
|
|
resp = plug.analyse(**params)
|
|
resp.analysis.append(plug.jsonable())
|
|
return resp
|
|
else:
|
|
return {"status": 500, "message": "No valid algorithm"}
|
|
|
|
@property
|
|
def default_plugin(self):
|
|
if self.plugins:
|
|
candidate = self.filter_plugins(enabled=True).keys()[0]
|
|
print("Default: {}".format(candidate))
|
|
return candidate
|
|
else:
|
|
return Exception("No algorithm")
|
|
|
|
def parameters(self, algo):
|
|
return getattr(self.plugins.get(algo or self.default_plugin), "params", {})
|
|
|
|
def enable_plugin(self, plugin):
|
|
self.plugins[plugin].disable()
|
|
|
|
def disable_plugin(self, plugin):
|
|
self.plugins[plugin].disable()
|
|
|
|
def reload_plugin(self, plugin):
|
|
print("Reloading {}".format(plugin))
|
|
plug = self.plugins[plugin]
|
|
nplug = self._load_plugin(plug.module, plug.path)
|
|
del self.plugins[plugin]
|
|
self.plugins[nplug.name] = nplug
|
|
|
|
def _load_plugin(self, plugin, search_folder, enabled=True):
|
|
sys.path.append(search_folder)
|
|
(fp, pathname, desc) = imp.find_module(plugin)
|
|
tmp = imp.load_module(plugin, fp, pathname, desc).plugin
|
|
sys.path.remove(search_folder)
|
|
tmp.path = search_folder
|
|
try:
|
|
repo_path = os.path.join(search_folder, plugin)
|
|
tmp.repo = Repo(repo_path)
|
|
except InvalidGitRepositoryError:
|
|
tmp.repo = None
|
|
if not hasattr(tmp, "enabled"):
|
|
tmp.enabled = enabled
|
|
tmp.module = plugin
|
|
return tmp
|
|
|
|
|
|
def _load_plugins(self):
|
|
plugins = {}
|
|
for search_folder in self.search_folders:
|
|
for item in os.listdir(search_folder):
|
|
if os.path.isdir(os.path.join(search_folder, item)) \
|
|
and os.path.exists(
|
|
os.path.join(search_folder, item, "__init__.py")):
|
|
plugin = self._load_plugin(item, search_folder)
|
|
plugins[plugin.name] = plugin
|
|
|
|
return plugins
|
|
|
|
def teardown(self, exception):
|
|
pass
|
|
|
|
def enable_all(self):
|
|
for plugin in self.plugins:
|
|
self.enable_plugin(plugin)
|
|
|
|
def enable_plugin(self, item):
|
|
self.plugins[item].enabled = True
|
|
|
|
def disable_plugin(self, item):
|
|
self.plugins[item].enabled = False
|
|
|
|
@property
|
|
def plugins(self):
|
|
ctx = stack.top
|
|
if ctx is not None:
|
|
if not hasattr(self, '_plugins'):
|
|
self._plugins = self._load_plugins()
|
|
return self._plugins
|
|
|
|
def filter_plugins(self, **kwargs):
|
|
def matches(plug):
|
|
res = all(getattr(plug, k, None)==v for (k,v) in kwargs.items())
|
|
print("matching {} with {}: {}".format(plug.name, kwargs, res))
|
|
return res
|
|
if not kwargs:
|
|
return self.plugins
|
|
else:
|
|
return {n:p for n,p in self.plugins.items() if matches(p)}
|
|
|
|
def sentiment_plugins(self):
|
|
return (plugin for plugin in self.plugins if
|
|
isinstance(plugin, SentimentPlugin))
|
|
|
|
if __name__ == '__main__':
|
|
from flask import Flask
|
|
app = Flask(__name__)
|
|
sp = Senpy()
|
|
sp.init_app(app)
|
|
with app.app_context():
|
|
sp._load_plugins()
|