diff options
| author | jibe-b <user701@orange.fr> | 2017-09-22 23:43:05 +0200 |
|---|---|---|
| committer | Noémi Ványi <sitbackandwait@gmail.com> | 2017-11-01 14:22:26 +0100 |
| commit | 575159b194440052d7b48aa073d7e03c80799c90 (patch) | |
| tree | 693e1db791842058d11c6ddb2cb181bb64b10da3 /searx/plugins | |
| parent | 7de8b43eb2081853ae15b2a52cc0cae43647320b (diff) | |
[enh] oa_doi_rewrite plugin broadens doai_rewrite
Diffstat (limited to 'searx/plugins')
| -rw-r--r-- | searx/plugins/__init__.py | 4 | ||||
| -rw-r--r-- | searx/plugins/oa_doi_rewrite.py (renamed from searx/plugins/doai_rewrite.py) | 17 |
2 files changed, 17 insertions, 4 deletions
diff --git a/searx/plugins/__init__.py b/searx/plugins/__init__.py index 46c1f8918..4dbcbbd28 100644 --- a/searx/plugins/__init__.py +++ b/searx/plugins/__init__.py @@ -22,7 +22,7 @@ if version_info[0] == 3: logger = logger.getChild('plugins') -from searx.plugins import (doai_rewrite, +from searx.plugins import (oa_doi_rewrite, https_rewrite, infinite_scroll, open_results_on_new_tab, @@ -78,7 +78,7 @@ class PluginStore(): plugins = PluginStore() -plugins.register(doai_rewrite) +plugins.register(oa_doi_rewrite) plugins.register(https_rewrite) plugins.register(infinite_scroll) plugins.register(open_results_on_new_tab) diff --git a/searx/plugins/doai_rewrite.py b/searx/plugins/oa_doi_rewrite.py index b7b814fac..e952c86f5 100644 --- a/searx/plugins/doai_rewrite.py +++ b/searx/plugins/oa_doi_rewrite.py @@ -1,14 +1,19 @@ from flask_babel import gettext import re from searx.url_utils import urlparse, parse_qsl +from flask import request +from searx import settings + regex = re.compile(r'10\.\d{4,9}/[^\s]+') -name = gettext('DOAI rewrite') +name = gettext('Open Access DOI rewrite') description = gettext('Avoid paywalls by redirecting to open-access versions of publications when available') default_on = False preference_section = 'privacy' +doi_resolvers = settings['doi_resolvers'] + def extract_doi(url): match = regex.search(url.path) @@ -21,12 +26,20 @@ def extract_doi(url): return None +def get_doi_resolver(): + doi_resolvers = settings['doi_resolvers'] + doi_resolver = request.args.get('doi_resolver', request.preferences.get_value('doi_resolver'))[0] + if doi_resolver not in doi_resolvers: + doi_resolvers = settings['default_doi_resolver'] + return doi_resolvers[doi_resolver] + + def on_result(request, search, result): doi = extract_doi(result['parsed_url']) if doi and len(doi) < 50: for suffix in ('/', '.pdf', '/full', '/meta', '/abstract'): if doi.endswith(suffix): doi = doi[:-len(suffix)] - result['url'] = 'http://doai.io/' + doi + result['url'] = get_doi_resolver() + doi result['parsed_url'] = urlparse(result['url']) return True |