Ponysearch/searx/plugins/__init__.py

86 lines
2.9 KiB
Python
Raw Normal View History

2015-04-13 00:08:04 +02:00
'''
searx 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.
searx 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 searx. If not, see < http://www.gnu.org/licenses/ >.
(C) 2015 by Adam Tauber, <asciimoo@gmail.com>
'''
from sys import exit
2015-04-13 00:30:12 +02:00
from searx import logger
logger = logger.getChild('plugins')
from searx.plugins import (doai_rewrite,
https_rewrite,
2016-08-11 04:21:28 +02:00
infinite_scroll,
open_results_on_new_tab,
self_info,
search_on_category_select,
2016-04-23 18:26:02 +02:00
tracker_url_remover,
vim_hotkeys)
2015-04-13 00:30:12 +02:00
required_attrs = (('name', (str, unicode)),
('description', (str, unicode)),
2015-03-14 20:22:26 +01:00
('default_on', bool))
optional_attrs = (('js_dependencies', tuple),
('css_dependencies', tuple))
class Plugin():
default_on = False
name = 'Default plugin'
2015-03-14 20:22:26 +01:00
description = 'Default plugin description'
class PluginStore():
def __init__(self):
self.plugins = []
def __iter__(self):
2015-03-10 20:44:02 +01:00
for plugin in self.plugins:
yield plugin
def register(self, *plugins):
for plugin in plugins:
2015-03-14 20:22:26 +01:00
for plugin_attr, plugin_attr_type in required_attrs:
if not hasattr(plugin, plugin_attr) or not isinstance(getattr(plugin, plugin_attr), plugin_attr_type):
logger.critical('missing attribute "{0}", cannot load plugin: {1}'.format(plugin_attr, plugin))
exit(3)
for plugin_attr, plugin_attr_type in optional_attrs:
if not hasattr(plugin, plugin_attr) or not isinstance(getattr(plugin, plugin_attr), plugin_attr_type):
setattr(plugin, plugin_attr, plugin_attr_type())
2015-03-11 01:42:25 +01:00
plugin.id = plugin.name.replace(' ', '_')
self.plugins.append(plugin)
def call(self, plugin_type, request, *args, **kwargs):
ret = True
2015-03-10 20:44:02 +01:00
for plugin in request.user_plugins:
if hasattr(plugin, plugin_type):
ret = getattr(plugin, plugin_type)(request, *args, **kwargs)
if not ret:
break
return ret
plugins = PluginStore()
plugins.register(doai_rewrite)
2015-04-13 00:30:12 +02:00
plugins.register(https_rewrite)
2016-08-11 04:21:28 +02:00
plugins.register(infinite_scroll)
plugins.register(open_results_on_new_tab)
plugins.register(self_info)
plugins.register(search_on_category_select)
plugins.register(tracker_url_remover)
2016-04-23 18:26:02 +02:00
plugins.register(vim_hotkeys)