git.iko.soy/searx
on master help sign in
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.

commit538029dc14ca47afae126e29bf789941d9bffd48 authorLuc Didry <luc@didry.org> dateSat 13 Jun 2015, 17:49 (11 years ago) parent1b77bef
3 files changed+54−1
compare with master
unifiedsplit
searx/plugins/__init__.py+2 searx/plugins/self_useragent.py+36 searx/tests/test_plugins.py+16−1
searx/plugins/__init__.py+2
@@ -21,6 +21,7 @@ logger = logger.getChild('plugins')
2121
2222from searx.plugins import (https_rewrite,
2323 self_ip,
24+ self_useragent,
2425 search_on_category_select)
2526
2627required_attrs = (('name', str),
@@ -72,4 +73,5 @@ class PluginStore():
7273plugins = PluginStore()
7374plugins.register(https_rewrite)
7475plugins.register(self_ip)
76+plugins.register(self_useragent)
7577plugins.register(search_on_category_select)
searx/plugins/self_useragent.pynew file+36
@@ -0,0 +1,36 @@
1+'''
2+searx is free software: you can redistribute it and/or modify
3+it under the terms of the GNU Affero General Public License as published by
4+the Free Software Foundation, either version 3 of the License, or
5+(at your option) any later version.
6+
7+searx is distributed in the hope that it will be useful,
8+but WITHOUT ANY WARRANTY; without even the implied warranty of
9+MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
10+GNU Affero General Public License for more details.
11+
12+You should have received a copy of the GNU Affero General Public License
13+along with searx. If not, see < http://www.gnu.org/licenses/ >.
14+
15+(C) 2015 by Adam Tauber, <asciimoo@gmail.com>
16+'''
17+from flask.ext.babel import gettext
18+import re
19+name = "Self User Agent"
20+description = gettext('Display your own User Agent if the query expression contains "user agent" or "user-agent"')
21+default_on = True
22+
23+
24+# User Agent query regex
25+p = re.compile('user[ -]agent', re.IGNORECASE)
26+
27+
28+# attach callback to the post search hook
29+# request: flask request object
30+# ctx: the whole local context of the pre search hook
31+def post_search(request, ctx):
32+ if p.match(ctx['search'].query):
33+ ua = request.user_agent
34+ ctx['search'].answers.clear()
35+ ctx['search'].answers.add(ua)
36+ return True
searx/tests/test_plugins.py+16−1
@@ -39,9 +39,11 @@ class SelfIPTest(SearxTestCase):
3939 def test_PluginStore_init(self):
4040 store = plugins.PluginStore()
4141 store.register(plugins.self_ip)
42+ store.register(plugins.self_useragent)
4243
43− self.assertTrue(len(store.plugins) == 1)
44+ self.assertTrue(len(store.plugins) == 2)
4445
46+ # IP test
4547 request = Mock(user_plugins=store.plugins,
4648 remote_addr='127.0.0.1')
4749 request.headers.getlist.return_value = []
@@ -49,3 +51,16 @@ class SelfIPTest(SearxTestCase):
4951 query='ip')}
5052 store.call('post_search', request, ctx)
5153 self.assertTrue('127.0.0.1' in ctx['search'].answers)
54+
55+ # User agent test
56+ request = Mock(user_plugins=store.plugins,
57+ user_agent='Mock')
58+ request.headers.getlist.return_value = []
59+ ctx = {'search': Mock(answers=set(),
60+ query='user-agent')}
61+ store.call('post_search', request, ctx)
62+ self.assertTrue('Mock' in ctx['search'].answers)
63+ ctx = {'search': Mock(answers=set(),
64+ query='user agent')}
65+ store.call('post_search', request, ctx)
66+ self.assertTrue('Mock' in ctx['search'].answers)