Add a self user agent plugin

Just like with the "ip" query, duckduckgo gives the server's
information with the "user agent" query.
This corrects this behavior by adding a plugin based on self_ip.py plugin.
This commit is contained in:
Luc Didry 2015-06-13 17:49:33 +02:00
parent 1b77befe1f
commit 538029dc14
3 changed files with 54 additions and 1 deletions

View File

@ -21,6 +21,7 @@ logger = logger.getChild('plugins')
from searx.plugins import (https_rewrite,
self_ip,
self_useragent,
search_on_category_select)
required_attrs = (('name', str),
@ -72,4 +73,5 @@ class PluginStore():
plugins = PluginStore()
plugins.register(https_rewrite)
plugins.register(self_ip)
plugins.register(self_useragent)
plugins.register(search_on_category_select)

View File

@ -0,0 +1,36 @@
'''
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 flask.ext.babel import gettext
import re
name = "Self User Agent"
description = gettext('Display your own User Agent if the query expression contains "user agent" or "user-agent"')
default_on = True
# User Agent query regex
p = re.compile('user[ -]agent', re.IGNORECASE)
# attach callback to the post search hook
# request: flask request object
# ctx: the whole local context of the pre search hook
def post_search(request, ctx):
if p.match(ctx['search'].query):
ua = request.user_agent
ctx['search'].answers.clear()
ctx['search'].answers.add(ua)
return True

View File

@ -39,9 +39,11 @@ class SelfIPTest(SearxTestCase):
def test_PluginStore_init(self):
store = plugins.PluginStore()
store.register(plugins.self_ip)
store.register(plugins.self_useragent)
self.assertTrue(len(store.plugins) == 1)
self.assertTrue(len(store.plugins) == 2)
# IP test
request = Mock(user_plugins=store.plugins,
remote_addr='127.0.0.1')
request.headers.getlist.return_value = []
@ -49,3 +51,16 @@ class SelfIPTest(SearxTestCase):
query='ip')}
store.call('post_search', request, ctx)
self.assertTrue('127.0.0.1' in ctx['search'].answers)
# User agent test
request = Mock(user_plugins=store.plugins,
user_agent='Mock')
request.headers.getlist.return_value = []
ctx = {'search': Mock(answers=set(),
query='user-agent')}
store.call('post_search', request, ctx)
self.assertTrue('Mock' in ctx['search'].answers)
ctx = {'search': Mock(answers=set(),
query='user agent')}
store.call('post_search', request, ctx)
self.assertTrue('Mock' in ctx['search'].answers)