forked from Ponysearch/Ponysearch
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:
parent
1b77befe1f
commit
538029dc14
3 changed files with 54 additions and 1 deletions
|
@ -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)
|
||||
|
|
36
searx/plugins/self_useragent.py
Normal file
36
searx/plugins/self_useragent.py
Normal 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
|
|
@ -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)
|
||||
|
|
Loading…
Reference in a new issue