[enh] oa_doi_rewrite plugin broadens doai_rewrite

This commit is contained in:
jibe-b 2017-09-22 23:43:05 +02:00 committed by Noémi Ványi
parent 7de8b43eb2
commit 575159b194
6 changed files with 59 additions and 6 deletions

View file

@ -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)

View file

@ -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

View file

@ -15,6 +15,7 @@ LANGUAGE_CODES = [l[0] for l in languages]
LANGUAGE_CODES.append('all')
DISABLED = 0
ENABLED = 1
DOI_RESOLVERS = [r for r in settings['doi_resolvers'].keys()]
class MissingArgumentException(Exception):
@ -266,7 +267,9 @@ class Preferences(object):
'results_on_new_tab': MapSetting(False, map={'0': False,
'1': True,
'False': False,
'True': True})}
'True': True}),
'doi_resolver': MultipleChoiceSetting(['oadoi.org'], choices=DOI_RESOLVERS),
}
self.engines = EnginesSetting('engines', choices=engines)
self.plugins = PluginsSetting('plugins', choices=plugins)

View file

@ -712,3 +712,10 @@ locales:
tr : Türkçe (Turkish)
uk : українська мова (Ukrainian)
zh : 中文 (Chinese)
doi_resolvers :
oadoi.org : 'https://oadoi.org/'
doi.org : 'https://doi.org/'
doai.io : 'http://doai.io/'
default_doi_resolver : 'oadoi.org'

View file

@ -223,6 +223,23 @@
</div>
</div>
{% endfor %}
<div class="panel panel-default">
<div class="panel-heading">
<h3 class="panel-title">{{ _('OA DOI rewrite') }}</h3>
</div>
<div class="panel-body">
<div class="col-xs-6 col-sm-4 col-md-6">{{ _('Avoid paywalls by redirecting to open-access versions of publications when available') }}</div>
<div class="col-xs-6 col-sm-4 col-md-6">
<select class="form-control" id='doi_resolver' name='doi_resolver'>
{% for doi_resolver_name,doi_resolver_url in doi_resolvers.items() %}
<option value="{{ doi_resolver_name }}" {% if doi_resolver_name == current_doi_resolver %}selected="selected"{% endif %}>
{{ doi_resolver_name }} - {{ doi_resolver_url }}
</option>
{% endfor %}
</select>
</div>
</div>
</div>
</div>
</fieldset>
</div>

View file

@ -164,6 +164,14 @@ def get_locale():
return locale
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_resolver
# code-highlighter
@app.template_filter('code_highlighter')
def code_highlighter(codelines, language=None):
@ -695,6 +703,8 @@ def preferences():
shortcuts={y: x for x, y in engine_shortcuts.items()},
themes=themes,
plugins=plugins,
doi_resolvers=settings['doi_resolvers'],
current_doi_resolver=get_doi_resolver(),
allowed_plugins=allowed_plugins,
theme=get_current_theme_name(),
preferences_url_params=request.preferences.get_as_url_params(),
@ -839,7 +849,10 @@ def config():
'autocomplete': settings['search']['autocomplete'],
'safe_search': settings['search']['safe_search'],
'default_theme': settings['ui']['default_theme'],
'version': VERSION_STRING})
'version': VERSION_STRING,
'doi_resolvers': [r for r in search['doi_resolvers']],
'default_doi_resolver': settings['default_doi_resolver'],
})
@app.errorhandler(404)