on master help sign in
Merge pull request #504 from kvch/frinkiac-engine

Simpson screenshot search engine

commit4a0ccabefa185a898d9430785cf4ae84fa92d14f authorAdam Tauber <asciimoo@gmail.com> dateThu 4 Feb 2016, 23:11 (11 years ago) parent0491b78baf16b8
3 files changed+98
searx/engines/frinkiac.py+44 searx/settings.yml+5 tests/unit/engines/test_frinkiac.py+49
searx/engines/frinkiac.pynew file+44
@@ -0,0 +1,44 @@
1+"""
2+Frinkiac (Images)
3+
4+@website https://www.frinkiac.com
5+@provide-api no
6+@using-api no
7+@results JSON
8+@stable no
9+@parse url, title, img_src
10+"""
11+
12+from json import loads
13+from urllib import urlencode
14+
15+categories = ['images']
16+
17+BASE = 'https://frinkiac.com/'
18+SEARCH_URL = '{base}api/search?{query}'
19+RESULT_URL = '{base}?{query}'
20+THUMB_URL = '{base}img/{episode}/{timestamp}/medium.jpg'
21+IMAGE_URL = '{base}img/{episode}/{timestamp}.jpg'
22+
23+
24+def request(query, params):
25+ params['url'] = SEARCH_URL.format(base=BASE, query=urlencode({'q': query}))
26+ return params
27+
28+
29+def response(resp):
30+ results = []
31+ response_data = loads(resp.text)
32+ for result in response_data:
33+ episode = result['Episode']
34+ timestamp = result['Timestamp']
35+
36+ results.append({'template': 'images.html',
37+ 'url': RESULT_URL.format(base=BASE,
38+ query=urlencode({'p': 'caption', 'e': episode, 't': timestamp})),
39+ 'title': episode,
40+ 'content': '',
41+ 'thumbnail_src': THUMB_URL.format(base=BASE, episode=episode, timestamp=timestamp),
42+ 'img_src': IMAGE_URL.format(base=BASE, episode=episode, timestamp=timestamp)})
43+
44+ return results
searx/settings.yml+5
@@ -110,6 +110,11 @@ engines:
110110# Or you can use the html non-stable engine, activated by default
111111 engine : flickr_noapi
112112
113+ - name : frinkiac
114+ engine : frinkiac
115+ shortcut : frk
116+ disabled : True
117+
113118 - name : gigablast
114119 engine : gigablast
115120 shortcut : gb
tests/unit/engines/test_frinkiac.pynew file+49
@@ -0,0 +1,49 @@
1+# -*- coding: utf-8 -*-
2+from collections import defaultdict
3+import mock
4+from json import dumps
5+from searx.engines import frinkiac
6+from searx.testing import SearxTestCase
7+
8+
9+class TestFrinkiacEngine(SearxTestCase):
10+
11+ def test_request(self):
12+ query = 'test_query'
13+ request_dict = defaultdict(dict)
14+ params = frinkiac.request(query, request_dict)
15+ self.assertTrue('url' in params)
16+
17+ def test_response(self):
18+ self.assertRaises(AttributeError, frinkiac.response, None)
19+ self.assertRaises(AttributeError, frinkiac.response, [])
20+ self.assertRaises(AttributeError, frinkiac.response, '')
21+ self.assertRaises(AttributeError, frinkiac.response, '[]')
22+
23+ text = """
24+[{"Id":770931,
25+ "Episode":"S06E18",
26+ "Timestamp":534616,
27+ "Filename":""},
28+ {"Id":1657080,
29+ "Episode":"S12E14",
30+ "Timestamp":910868,
31+ "Filename":""},
32+ {"Id":1943753,
33+ "Episode":"S14E21",
34+ "Timestamp":773439,
35+ "Filename":""},
36+ {"Id":107835,
37+ "Episode":"S02E03",
38+ "Timestamp":531709,
39+ "Filename":""}]
40+ """
41+
42+ response = mock.Mock(text=text)
43+ results = frinkiac.response(response)
44+ self.assertEqual(type(results), list)
45+ self.assertEqual(len(results), 4)
46+ self.assertEqual(results[0]['title'], u'S06E18')
47+ self.assertEqual(results[0]['url'], 'https://frinkiac.com/?p=caption&e=S06E18&t=534616')
48+ self.assertEqual(results[0]['thumbnail_src'], 'https://frinkiac.com/img/S06E18/534616/medium.jpg')
49+ self.assertEqual(results[0]['img_src'], 'https://frinkiac.com/img/S06E18/534616.jpg')