538029d · searx · git.iko.soy

searx/plugins/__init__.py+2

compare with master

unifiedsplit

@@ -21,6 +21,7 @@ logger = logger.getChild('plugins')

21

22from searx.plugins import (https_rewrite,

23 self_ip,

24+ self_useragent,

24 search_on_category_select)

25 search_on_category_select)

25

26

26required_attrs = (('name', str),

27required_attrs = (('name', str),

@@ -72,4 +73,5 @@ class PluginStore():

72plugins = PluginStore()

73plugins = PluginStore()

73plugins.register(https_rewrite)

74plugins.register(https_rewrite)

74plugins.register(self_ip)

75plugins.register(self_ip)

76+plugins.register(self_useragent)

75plugins.register(search_on_category_select)

77plugins.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+

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

39 def test_PluginStore_init(self):

40 store = plugins.PluginStore()

41 store.register(plugins.self_ip)

42+ store.register(plugins.self_useragent)

42

43

43− self.assertTrue(len(store.plugins) == 1)

44+ self.assertTrue(len(store.plugins) == 2)

44

45

46+ # IP test

45 request = Mock(user_plugins=store.plugins,

47 request = Mock(user_plugins=store.plugins,

46 remote_addr='127.0.0.1')

48 remote_addr='127.0.0.1')

47 request.headers.getlist.return_value = []

49 request.headers.getlist.return_value = []

@@ -49,3 +51,16 @@ class SelfIPTest(SearxTestCase):

49 query='ip')}

51 query='ip')}

50 store.call('post_search', request, ctx)

52 store.call('post_search', request, ctx)

51 self.assertTrue('127.0.0.1' in ctx['search'].answers)

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