From 575159b194440052d7b48aa073d7e03c80799c90 Mon Sep 17 00:00:00 2001 From: jibe-b Date: Fri, 22 Sep 2017 23:43:05 +0200 Subject: [enh] oa_doi_rewrite plugin broadens doai_rewrite --- searx/plugins/oa_doi_rewrite.py | 45 +++++++++++++++++++++++++++++++++++++++++ 1 file changed, 45 insertions(+) create mode 100644 searx/plugins/oa_doi_rewrite.py (limited to 'searx/plugins/oa_doi_rewrite.py') diff --git a/searx/plugins/oa_doi_rewrite.py b/searx/plugins/oa_doi_rewrite.py new file mode 100644 index 000000000..e952c86f5 --- /dev/null +++ b/searx/plugins/oa_doi_rewrite.py @@ -0,0 +1,45 @@ +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('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) + if match: + return match.group(0) + for _, v in parse_qsl(url.query): + match = regex.search(v) + if match: + return match.group(0) + 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'] = get_doi_resolver() + doi + result['parsed_url'] = urlparse(result['url']) + return True -- cgit v1.2.3