forked from Ponysearch/Ponysearch
add Frinkiac engine & tests
This commit is contained in:
parent
0491b78f70
commit
33fa3da2e8
3 changed files with 94 additions and 0 deletions
42
searx/engines/frinkiac.py
Normal file
42
searx/engines/frinkiac.py
Normal file
|
@ -0,0 +1,42 @@
|
||||||
|
"""
|
||||||
|
Frinkiac (Images)
|
||||||
|
|
||||||
|
@website https://www.frinkiac.com
|
||||||
|
@provide-api no
|
||||||
|
@using-api no
|
||||||
|
@results JSON
|
||||||
|
@stable no
|
||||||
|
@parse url, title, img_src
|
||||||
|
"""
|
||||||
|
|
||||||
|
from json import loads
|
||||||
|
from urllib import urlencode
|
||||||
|
|
||||||
|
categories = ['images']
|
||||||
|
|
||||||
|
BASE = 'https://frinkiac.com/'
|
||||||
|
SEARCH_URL = '{base}api/search?{query}'
|
||||||
|
RESULT_URL = '{base}?{query}'
|
||||||
|
THUMB_URL = '{base}img/{episode}/{timestamp}/medium.jpg'
|
||||||
|
IMAGE_URL = '{base}img/{episode}/{timestamp}.jpg'
|
||||||
|
|
||||||
|
def request(query, params):
|
||||||
|
params['url'] = SEARCH_URL.format(base=BASE, query=urlencode({'q': query}))
|
||||||
|
return params
|
||||||
|
|
||||||
|
def response(resp):
|
||||||
|
results = []
|
||||||
|
response_data = loads(resp.text)
|
||||||
|
for result in response_data:
|
||||||
|
episode = result['Episode']
|
||||||
|
timestamp = result['Timestamp']
|
||||||
|
|
||||||
|
results.append({'template': 'images.html',
|
||||||
|
'url': RESULT_URL.format(base=BASE,
|
||||||
|
query=urlencode({'p': 'caption', 'e': episode, 't': timestamp})),
|
||||||
|
'title': episode,
|
||||||
|
'content': '',
|
||||||
|
'thumbnail_src': THUMB_URL.format(base=BASE, episode=episode, timestamp=timestamp),
|
||||||
|
'img_src': IMAGE_URL.format(base=BASE, episode=episode, timestamp=timestamp)})
|
||||||
|
|
||||||
|
return results
|
|
@ -110,6 +110,11 @@ engines:
|
||||||
# Or you can use the html non-stable engine, activated by default
|
# Or you can use the html non-stable engine, activated by default
|
||||||
engine : flickr_noapi
|
engine : flickr_noapi
|
||||||
|
|
||||||
|
- name : frinkiac
|
||||||
|
engine : frinkiac
|
||||||
|
shortcut : frk
|
||||||
|
disabled : True
|
||||||
|
|
||||||
- name : gigablast
|
- name : gigablast
|
||||||
engine : gigablast
|
engine : gigablast
|
||||||
shortcut : gb
|
shortcut : gb
|
||||||
|
|
47
tests/unit/engines/test_frinkiac.py
Normal file
47
tests/unit/engines/test_frinkiac.py
Normal file
|
@ -0,0 +1,47 @@
|
||||||
|
# -*- coding: utf-8 -*-
|
||||||
|
from collections import defaultdict
|
||||||
|
import mock
|
||||||
|
from json import dumps
|
||||||
|
from searx.engines import frinkiac
|
||||||
|
from searx.testing import SearxTestCase
|
||||||
|
|
||||||
|
|
||||||
|
class TestFrinkiacEngine(SearxTestCase):
|
||||||
|
|
||||||
|
def test_request(self):
|
||||||
|
query = 'test_query'
|
||||||
|
request_dict = defaultdict(dict)
|
||||||
|
params = frinkiac.request(query, request_dict)
|
||||||
|
self.assertTrue('url' in params)
|
||||||
|
|
||||||
|
def test_response(self):
|
||||||
|
self.assertRaises(AttributeError, frinkiac.response, None)
|
||||||
|
self.assertRaises(AttributeError, frinkiac.response, [])
|
||||||
|
self.assertRaises(AttributeError, frinkiac.response, '')
|
||||||
|
self.assertRaises(AttributeError, frinkiac.response, '[]')
|
||||||
|
|
||||||
|
text = dumps([{'Id': 654234,
|
||||||
|
'Episode': 'S05E21',
|
||||||
|
'Timestamp': 3453455,
|
||||||
|
'Filename': ''},
|
||||||
|
{'Id': 435354,
|
||||||
|
'Episode': 'S05E22',
|
||||||
|
'Timestamp': 3453456,
|
||||||
|
'Filename': ''},
|
||||||
|
{'Id': 435333,
|
||||||
|
'Episode': 'S05E23',
|
||||||
|
'Timestamp': 3453457,
|
||||||
|
'Filename': ''},
|
||||||
|
{'Id': 477234,
|
||||||
|
'Episode': 'S05E24',
|
||||||
|
'Timestamp': 3453458,
|
||||||
|
'Filename': ''}])
|
||||||
|
|
||||||
|
response = mock.Mock(text=text)
|
||||||
|
results = frinkiac.response(response)
|
||||||
|
self.assertEqual(type(results), list)
|
||||||
|
self.assertEqual(len(results), 4)
|
||||||
|
self.assertEqual(results[0]['title'], u'S05E21')
|
||||||
|
self.assertEqual(results[0]['url'], 'https://frinkiac.com/?p=caption&e=S05E21&t=3453455')
|
||||||
|
self.assertEqual(results[0]['thumbnail_src'], 'https://frinkiac.com/img/S05E21/3453455/medium.jpg')
|
||||||
|
self.assertEqual(results[0]['img_src'], 'https://frinkiac.com/img/S05E21/3453455.jpg')
|
Loading…
Reference in a new issue