Merge pull request #1412 from d-tux/engines/unsplash
Unsplash image engine
This commit is contained in:
		
						commit
						3e6d5ba35e
					
				
							
								
								
									
										52
									
								
								searx/engines/unsplash.py
									
									
									
									
									
										Normal file
									
								
							
							
						
						
									
										52
									
								
								searx/engines/unsplash.py
									
									
									
									
									
										Normal file
									
								
							| @ -0,0 +1,52 @@ | |||||||
|  | """ | ||||||
|  |  Unsplash | ||||||
|  | 
 | ||||||
|  |  @website     https://unsplash.com | ||||||
|  |  @provide-api yes (https://unsplash.com/developers) | ||||||
|  | 
 | ||||||
|  |  @using-api   no | ||||||
|  |  @results     JSON (using search portal's infiniscroll API) | ||||||
|  |  @stable      no (JSON format could change any time) | ||||||
|  |  @parse       url, title, img_src, thumbnail_src | ||||||
|  | """ | ||||||
|  | 
 | ||||||
|  | from searx.url_utils import urlencode, urlparse, urlunparse, parse_qsl | ||||||
|  | from json import loads | ||||||
|  | 
 | ||||||
|  | url = 'https://unsplash.com/' | ||||||
|  | search_url = url + 'napi/search/photos?' | ||||||
|  | categories = ['images'] | ||||||
|  | page_size = 20 | ||||||
|  | paging = True | ||||||
|  | 
 | ||||||
|  | 
 | ||||||
|  | def clean_url(url): | ||||||
|  |     parsed = urlparse(url) | ||||||
|  |     query = [(k, v) for (k, v) in parse_qsl(parsed.query) if k not in ['ixid', 's']] | ||||||
|  | 
 | ||||||
|  |     return urlunparse((parsed.scheme, | ||||||
|  |                        parsed.netloc, | ||||||
|  |                        parsed.path, | ||||||
|  |                        parsed.params, | ||||||
|  |                        urlencode(query), | ||||||
|  |                        parsed.fragment)) | ||||||
|  | 
 | ||||||
|  | 
 | ||||||
|  | def request(query, params): | ||||||
|  |     params['url'] = search_url + urlencode({'query': query, 'page': params['pageno'], 'per_page': page_size}) | ||||||
|  |     return params | ||||||
|  | 
 | ||||||
|  | 
 | ||||||
|  | def response(resp): | ||||||
|  |     results = [] | ||||||
|  |     json_data = loads(resp.text) | ||||||
|  | 
 | ||||||
|  |     if 'results' in json_data: | ||||||
|  |         for result in json_data['results']: | ||||||
|  |             results.append({'template': 'images.html', | ||||||
|  |                             'url': clean_url(result['links']['html']), | ||||||
|  |                             'thumbnail_src': clean_url(result['urls']['thumb']), | ||||||
|  |                             'img_src': clean_url(result['urls']['raw']), | ||||||
|  |                             'title': result['description'], | ||||||
|  |                             'content': ''}) | ||||||
|  |     return results | ||||||
| @ -616,6 +616,11 @@ engines: | |||||||
| #    content_xpath : //*[@class="meaning"] | #    content_xpath : //*[@class="meaning"] | ||||||
| #    shortcut : ud | #    shortcut : ud | ||||||
| 
 | 
 | ||||||
|  |   - name : unsplash | ||||||
|  |     engine : unsplash | ||||||
|  |     disabled: True | ||||||
|  |     shortcut : us | ||||||
|  | 
 | ||||||
|   - name : yahoo |   - name : yahoo | ||||||
|     engine : yahoo |     engine : yahoo | ||||||
|     shortcut : yh |     shortcut : yh | ||||||
|  | |||||||
							
								
								
									
										38
									
								
								tests/unit/engines/test_unsplash.py
									
									
									
									
									
										Normal file
									
								
							
							
						
						
									
										38
									
								
								tests/unit/engines/test_unsplash.py
									
									
									
									
									
										Normal file
									
								
							| @ -0,0 +1,38 @@ | |||||||
|  | from collections import defaultdict | ||||||
|  | import mock | ||||||
|  | from searx.testing import SearxTestCase | ||||||
|  | from searx.engines import unsplash | ||||||
|  | 
 | ||||||
|  | 
 | ||||||
|  | class TestUnsplashEngine(SearxTestCase): | ||||||
|  |     def test_request(self): | ||||||
|  |         query = 'penguin' | ||||||
|  |         _dict = defaultdict(dict) | ||||||
|  |         _dict['pageno'] = 1 | ||||||
|  |         params = unsplash.request(query, _dict) | ||||||
|  | 
 | ||||||
|  |         self.assertTrue('url' in params) | ||||||
|  |         self.assertTrue(query in params['url']) | ||||||
|  | 
 | ||||||
|  |     def test_response(self): | ||||||
|  |         resp = mock.Mock(text='{}') | ||||||
|  |         result = unsplash.response(resp) | ||||||
|  |         self.assertEqual([], result) | ||||||
|  | 
 | ||||||
|  |         resp.text = '{"results": []}' | ||||||
|  |         result = unsplash.response(resp) | ||||||
|  |         self.assertEqual([], result) | ||||||
|  | 
 | ||||||
|  |         # Sourced from https://unsplash.com/napi/search/photos?query=penguin&xp=&per_page=20&page=2 | ||||||
|  |         with open('./tests/unit/engines/unsplash_fixture.json') as fixture: | ||||||
|  |             resp.text = fixture.read() | ||||||
|  | 
 | ||||||
|  |         result = unsplash.response(resp) | ||||||
|  |         self.assertEqual(len(result), 2) | ||||||
|  |         self.assertEqual(result[0]['title'], 'low angle photography of swimming penguin') | ||||||
|  |         self.assertEqual(result[0]['url'], 'https://unsplash.com/photos/FY8d721UO_4') | ||||||
|  |         self.assertEqual(result[0]['thumbnail_src'], 'https://images.unsplash.com/photo-1523557148507-1b77641c7e7c?ixlib=rb-0.3.5&q=80\ | ||||||
|  | &fm=jpg&crop=entropy&cs=tinysrgb&w=200&fit=max') | ||||||
|  |         self.assertEqual(result[0]['img_src'], 'https://images.unsplash.com/photo-1523557148507-1b77641c7e7c\ | ||||||
|  | ?ixlib=rb-0.3.5') | ||||||
|  |         self.assertEqual(result[0]['content'], '') | ||||||
							
								
								
									
										241
									
								
								tests/unit/engines/unsplash_fixture.json
									
									
									
									
									
										Normal file
									
								
							
							
						
						
									
										241
									
								
								tests/unit/engines/unsplash_fixture.json
									
									
									
									
									
										Normal file
									
								
							| @ -0,0 +1,241 @@ | |||||||
|  | { | ||||||
|  |     "total": 2, | ||||||
|  |     "total_pages": 1, | ||||||
|  |     "results": [ | ||||||
|  |         { | ||||||
|  |             "id": "FY8d721UO_4", | ||||||
|  |             "created_at": "2018-04-12T14:20:35-04:00", | ||||||
|  |             "updated_at": "2018-08-28T20:58:33-04:00", | ||||||
|  |             "width": 3891, | ||||||
|  |             "height": 5829, | ||||||
|  |             "color": "#152C33", | ||||||
|  |             "description": "low angle photography of swimming penguin", | ||||||
|  |             "urls": { | ||||||
|  |                 "raw": "https://images.unsplash.com/photo-1523557148507-1b77641c7e7c?ixlib=rb-0.3.5&ixid=eyJhcHBfaWQiOjEyMDd9&s=095c5fc319c5a77c705f49ad63e0f195", | ||||||
|  |                 "full": "https://images.unsplash.com/photo-1523557148507-1b77641c7e7c?ixlib=rb-0.3.5&q=85&fm=jpg&crop=entropy&cs=srgb&ixid=eyJhcHBfaWQiOjEyMDd9&s=74be977849c173d6929636d491a760c3", | ||||||
|  |                 "regular": "https://images.unsplash.com/photo-1523557148507-1b77641c7e7c?ixlib=rb-0.3.5&q=80&fm=jpg&crop=entropy&cs=tinysrgb&w=1080&fit=max&ixid=eyJhcHBfaWQiOjEyMDd9&s=ad65df26970bd010085f0ca25434de33", | ||||||
|  |                 "small": "https://images.unsplash.com/photo-1523557148507-1b77641c7e7c?ixlib=rb-0.3.5&q=80&fm=jpg&crop=entropy&cs=tinysrgb&w=400&fit=max&ixid=eyJhcHBfaWQiOjEyMDd9&s=5d2edfd073c31eb8ee7b305222bdc5a2", | ||||||
|  |                 "thumb": "https://images.unsplash.com/photo-1523557148507-1b77641c7e7c?ixlib=rb-0.3.5&q=80&fm=jpg&crop=entropy&cs=tinysrgb&w=200&fit=max&ixid=eyJhcHBfaWQiOjEyMDd9&s=a9b9e56e63efc6f4611a87ce7e9a48f8" | ||||||
|  |             }, | ||||||
|  |             "links": { | ||||||
|  |                 "self": "https://api.unsplash.com/photos/FY8d721UO_4", | ||||||
|  |                 "html": "https://unsplash.com/photos/FY8d721UO_4", | ||||||
|  |                 "download": "https://unsplash.com/photos/FY8d721UO_4/download", | ||||||
|  |                 "download_location": "https://api.unsplash.com/photos/FY8d721UO_4/download" | ||||||
|  |             }, | ||||||
|  |             "categories": [], | ||||||
|  |             "sponsored": false, | ||||||
|  |             "likes": 31, | ||||||
|  |             "liked_by_user": false, | ||||||
|  |             "current_user_collections": [], | ||||||
|  |             "slug": null, | ||||||
|  |             "user": { | ||||||
|  |                 "id": "N4gE4mrG8lE", | ||||||
|  |                 "updated_at": "2018-10-03T02:51:19-04:00", | ||||||
|  |                 "username": "gaspanik", | ||||||
|  |                 "name": "Masaaki Komori", | ||||||
|  |                 "first_name": "Masaaki", | ||||||
|  |                 "last_name": "Komori", | ||||||
|  |                 "twitter_username": "cipher", | ||||||
|  |                 "portfolio_url": "https://www.instagram.com/cipher/", | ||||||
|  |                 "bio": null, | ||||||
|  |                 "location": "Tokyo, JAPAN", | ||||||
|  |                 "links": { | ||||||
|  |                     "self": "https://api.unsplash.com/users/gaspanik", | ||||||
|  |                     "html": "https://unsplash.com/@gaspanik", | ||||||
|  |                     "photos": "https://api.unsplash.com/users/gaspanik/photos", | ||||||
|  |                     "likes": "https://api.unsplash.com/users/gaspanik/likes", | ||||||
|  |                     "portfolio": "https://api.unsplash.com/users/gaspanik/portfolio", | ||||||
|  |                     "following": "https://api.unsplash.com/users/gaspanik/following", | ||||||
|  |                     "followers": "https://api.unsplash.com/users/gaspanik/followers" | ||||||
|  |                 }, | ||||||
|  |                 "profile_image": { | ||||||
|  |                     "small": "https://images.unsplash.com/profile-fb-1502270358-e7c86c1011ce.jpg?ixlib=rb-0.3.5&q=80&fm=jpg&crop=faces&cs=tinysrgb&fit=crop&h=32&w=32&s=9fe12f6d177bd6fdbd56d233a80c01a3", | ||||||
|  |                     "medium": "https://images.unsplash.com/profile-fb-1502270358-e7c86c1011ce.jpg?ixlib=rb-0.3.5&q=80&fm=jpg&crop=faces&cs=tinysrgb&fit=crop&h=64&w=64&s=6ad7d156b62e438ae9dc794cba712fff", | ||||||
|  |                     "large": "https://images.unsplash.com/profile-fb-1502270358-e7c86c1011ce.jpg?ixlib=rb-0.3.5&q=80&fm=jpg&crop=faces&cs=tinysrgb&fit=crop&h=128&w=128&s=13a08a2e72e7d11632410e92bd3a9406" | ||||||
|  |                 }, | ||||||
|  |                 "instagram_username": "cipher", | ||||||
|  |                 "total_collections": 0, | ||||||
|  |                 "total_likes": 406, | ||||||
|  |                 "total_photos": 196 | ||||||
|  |             }, | ||||||
|  |             "tags": [ | ||||||
|  |                 { | ||||||
|  |                     "title": "animal" | ||||||
|  |                 }, | ||||||
|  |                 { | ||||||
|  |                     "title": "water" | ||||||
|  |                 }, | ||||||
|  |                 { | ||||||
|  |                     "title": "swim" | ||||||
|  |                 }, | ||||||
|  |                 { | ||||||
|  |                     "title": "aquarium" | ||||||
|  |                 }, | ||||||
|  |                 { | ||||||
|  |                     "title": "wallpaper" | ||||||
|  |                 }, | ||||||
|  |                 { | ||||||
|  |                     "title": "blue" | ||||||
|  |                 }, | ||||||
|  |                 { | ||||||
|  |                     "title": "sealife" | ||||||
|  |                 }, | ||||||
|  |                 { | ||||||
|  |                     "title": "wildlife" | ||||||
|  |                 }, | ||||||
|  |                 { | ||||||
|  |                     "title": "bird" | ||||||
|  |                 }, | ||||||
|  |                 { | ||||||
|  |                     "title": "deep sea" | ||||||
|  |                 }, | ||||||
|  |                 { | ||||||
|  |                     "title": "fish" | ||||||
|  |                 }, | ||||||
|  |                 { | ||||||
|  |                     "title": "water life" | ||||||
|  |                 } | ||||||
|  |             ], | ||||||
|  |             "photo_tags": [ | ||||||
|  |                 { | ||||||
|  |                     "title": "animal" | ||||||
|  |                 }, | ||||||
|  |                 { | ||||||
|  |                     "title": "water" | ||||||
|  |                 }, | ||||||
|  |                 { | ||||||
|  |                     "title": "swim" | ||||||
|  |                 }, | ||||||
|  |                 { | ||||||
|  |                     "title": "aquarium" | ||||||
|  |                 }, | ||||||
|  |                 { | ||||||
|  |                     "title": "wallpaper" | ||||||
|  |                 } | ||||||
|  |             ] | ||||||
|  |         }, | ||||||
|  |         { | ||||||
|  |             "id": "ayKyc01xLWA", | ||||||
|  |             "created_at": "2018-02-16T23:14:31-05:00", | ||||||
|  |             "updated_at": "2018-08-28T20:48:27-04:00", | ||||||
|  |             "width": 4928, | ||||||
|  |             "height": 3264, | ||||||
|  |             "color": "#161618", | ||||||
|  |             "description": "black and white penguins on ice field", | ||||||
|  |             "urls": { | ||||||
|  |                 "raw": "https://images.unsplash.com/photo-1518840801558-9770b4a34eeb?ixlib=rb-0.3.5&ixid=eyJhcHBfaWQiOjEyMDd9&s=4e107a2bc49ab561ba6272eea2ec725d", | ||||||
|  |                 "full": "https://images.unsplash.com/photo-1518840801558-9770b4a34eeb?ixlib=rb-0.3.5&q=85&fm=jpg&crop=entropy&cs=srgb&ixid=eyJhcHBfaWQiOjEyMDd9&s=f9b1e4d4572ab44efb2cf3d601d2b4d9", | ||||||
|  |                 "regular": "https://images.unsplash.com/photo-1518840801558-9770b4a34eeb?ixlib=rb-0.3.5&q=80&fm=jpg&crop=entropy&cs=tinysrgb&w=1080&fit=max&ixid=eyJhcHBfaWQiOjEyMDd9&s=4430cedb63841f1fe055d5005316cc96", | ||||||
|  |                 "small": "https://images.unsplash.com/photo-1518840801558-9770b4a34eeb?ixlib=rb-0.3.5&q=80&fm=jpg&crop=entropy&cs=tinysrgb&w=400&fit=max&ixid=eyJhcHBfaWQiOjEyMDd9&s=ee73c7af22ce445d408e240821ce07af", | ||||||
|  |                 "thumb": "https://images.unsplash.com/photo-1518840801558-9770b4a34eeb?ixlib=rb-0.3.5&q=80&fm=jpg&crop=entropy&cs=tinysrgb&w=200&fit=max&ixid=eyJhcHBfaWQiOjEyMDd9&s=934302390d383cad8c571905e3a80bac" | ||||||
|  |             }, | ||||||
|  |             "links": { | ||||||
|  |                 "self": "https://api.unsplash.com/photos/ayKyc01xLWA", | ||||||
|  |                 "html": "https://unsplash.com/photos/ayKyc01xLWA", | ||||||
|  |                 "download": "https://unsplash.com/photos/ayKyc01xLWA/download", | ||||||
|  |                 "download_location": "https://api.unsplash.com/photos/ayKyc01xLWA/download" | ||||||
|  |             }, | ||||||
|  |             "categories": [], | ||||||
|  |             "sponsored": false, | ||||||
|  |             "likes": 37, | ||||||
|  |             "liked_by_user": false, | ||||||
|  |             "current_user_collections": [], | ||||||
|  |             "slug": null, | ||||||
|  |             "user": { | ||||||
|  |                 "id": "tRb_KGw60Xk", | ||||||
|  |                 "updated_at": "2018-09-20T11:51:54-04:00", | ||||||
|  |                 "username": "ghost_cat", | ||||||
|  |                 "name": "Danielle Barnes", | ||||||
|  |                 "first_name": "Danielle", | ||||||
|  |                 "last_name": "Barnes", | ||||||
|  |                 "twitter_username": null, | ||||||
|  |                 "portfolio_url": null, | ||||||
|  |                 "bio": null, | ||||||
|  |                 "location": null, | ||||||
|  |                 "links": { | ||||||
|  |                     "self": "https://api.unsplash.com/users/ghost_cat", | ||||||
|  |                     "html": "https://unsplash.com/@ghost_cat", | ||||||
|  |                     "photos": "https://api.unsplash.com/users/ghost_cat/photos", | ||||||
|  |                     "likes": "https://api.unsplash.com/users/ghost_cat/likes", | ||||||
|  |                     "portfolio": "https://api.unsplash.com/users/ghost_cat/portfolio", | ||||||
|  |                     "following": "https://api.unsplash.com/users/ghost_cat/following", | ||||||
|  |                     "followers": "https://api.unsplash.com/users/ghost_cat/followers" | ||||||
|  |                 }, | ||||||
|  |                 "profile_image": { | ||||||
|  |                     "small": "https://images.unsplash.com/profile-fb-1508491082-ae77f53e9ac3.jpg?ixlib=rb-0.3.5&q=80&fm=jpg&crop=faces&cs=tinysrgb&fit=crop&h=32&w=32&s=751bf6a557763648d52ffd7e60e79436", | ||||||
|  |                     "medium": "https://images.unsplash.com/profile-fb-1508491082-ae77f53e9ac3.jpg?ixlib=rb-0.3.5&q=80&fm=jpg&crop=faces&cs=tinysrgb&fit=crop&h=64&w=64&s=e46cd1c8713035f045130e1b093b981e", | ||||||
|  |                     "large": "https://images.unsplash.com/profile-fb-1508491082-ae77f53e9ac3.jpg?ixlib=rb-0.3.5&q=80&fm=jpg&crop=faces&cs=tinysrgb&fit=crop&h=128&w=128&s=352eabcf107c3ce95fe51a18485f116b" | ||||||
|  |                 }, | ||||||
|  |                 "instagram_username": null, | ||||||
|  |                 "total_collections": 0, | ||||||
|  |                 "total_likes": 0, | ||||||
|  |                 "total_photos": 21 | ||||||
|  |             }, | ||||||
|  |             "tags": [ | ||||||
|  |                 { | ||||||
|  |                     "title": "ice" | ||||||
|  |                 }, | ||||||
|  |                 { | ||||||
|  |                     "title": "bird" | ||||||
|  |                 }, | ||||||
|  |                 { | ||||||
|  |                     "title": "ice field" | ||||||
|  |                 }, | ||||||
|  |                 { | ||||||
|  |                     "title": "iceberg" | ||||||
|  |                 }, | ||||||
|  |                 { | ||||||
|  |                     "title": "snow" | ||||||
|  |                 }, | ||||||
|  |                 { | ||||||
|  |                     "title": "frozen" | ||||||
|  |                 }, | ||||||
|  |                 { | ||||||
|  |                     "title": "animal" | ||||||
|  |                 }, | ||||||
|  |                 { | ||||||
|  |                     "title": "wildlife" | ||||||
|  |                 }, | ||||||
|  |                 { | ||||||
|  |                     "title": "wild" | ||||||
|  |                 }, | ||||||
|  |                 { | ||||||
|  |                     "title": "antarctica" | ||||||
|  |                 }, | ||||||
|  |                 { | ||||||
|  |                     "title": "sunshine" | ||||||
|  |                 }, | ||||||
|  |                 { | ||||||
|  |                     "title": "daylight" | ||||||
|  |                 }, | ||||||
|  |                 { | ||||||
|  |                     "title": "wilderness" | ||||||
|  |                 }, | ||||||
|  |                 { | ||||||
|  |                     "title": "south pole" | ||||||
|  |                 }, | ||||||
|  |                 { | ||||||
|  |                     "title": "flock" | ||||||
|  |                 } | ||||||
|  |             ], | ||||||
|  |             "photo_tags": [ | ||||||
|  |                 { | ||||||
|  |                     "title": "ice" | ||||||
|  |                 }, | ||||||
|  |                 { | ||||||
|  |                     "title": "bird" | ||||||
|  |                 }, | ||||||
|  |                 { | ||||||
|  |                     "title": "ice field" | ||||||
|  |                 }, | ||||||
|  |                 { | ||||||
|  |                     "title": "iceberg" | ||||||
|  |                 }, | ||||||
|  |                 { | ||||||
|  |                     "title": "snow" | ||||||
|  |                 } | ||||||
|  |             ] | ||||||
|  |         } | ||||||
|  |     ] | ||||||
|  | } | ||||||
		Loading…
	
		Reference in New Issue
	
	Block a user