76 lines
3.1 KiB
Python
76 lines
3.1 KiB
Python
import re
|
|
import json
|
|
import requests
|
|
import regluit.core.isbn
|
|
|
|
from django.conf import settings
|
|
|
|
def gluejar_search(q, user_ip='69.243.24.29', page=1):
|
|
"""normalizes results from the google books search suitable for gluejar
|
|
"""
|
|
results = []
|
|
search_result=googlebooks_search(q, user_ip, page)
|
|
if 'items' in search_result.keys():
|
|
for item in search_result['items']:
|
|
v = item['volumeInfo']
|
|
r = {'title': v.get('title', ""),
|
|
'description': v.get('description', ""),
|
|
'publisher': v.get('publisher', ""),
|
|
'googlebooks_id': item.get('id')}
|
|
|
|
# TODO: allow multiple authors
|
|
if v.has_key('authors') and len(v['authors']) == 1 :
|
|
r['author'] = r['authors_short'] = v['authors'][0]
|
|
elif v.has_key('authors') and len(v['authors']) > 2:
|
|
r['author'] = v['authors'][0]
|
|
r['authors_short'] = '%s et al.' % v['authors'][0]
|
|
elif v.has_key('authors') and len(v['authors']) == 2:
|
|
r['author'] = v['authors'][0]
|
|
r['authors_short'] = '%s and %s' % (v['authors'][0], v['authors'][1])
|
|
else:
|
|
r['author'] = ""
|
|
r['isbn_13'] = None
|
|
|
|
# pull out isbns
|
|
for i in v.get('industryIdentifiers', []):
|
|
if i['type'] == 'ISBN_13':
|
|
r['isbn_13'] = i['identifier']
|
|
elif i['type'] == 'ISBN_10':
|
|
if not r['isbn_13'] :
|
|
r['isbn_13'] = regluit.core.isbn.convert_10_to_13(i['identifier'])
|
|
|
|
# cover image
|
|
if v.has_key('imageLinks'):
|
|
url = v['imageLinks'].get('thumbnail', "")
|
|
url = re.sub(r'http://(bks[0-9]+\.)?books\.google\.com', 'https://encrypted.google.com', url)
|
|
r['cover_image_thumbnail'] = url
|
|
else:
|
|
r['cover_image_thumbnail'] = "/static/images/generic_cover_larger.png"
|
|
|
|
access_info = item.get('accessInfo')
|
|
if access_info:
|
|
epub = access_info.get('epub')
|
|
if epub and epub.get('downloadLink'):
|
|
r['first_epub_url'] = epub['downloadLink']
|
|
pdf = access_info.get('pdf')
|
|
if pdf and pdf.get('downloadLink'):
|
|
r['first_pdf_url'] = pdf['downloadLink']
|
|
results.append(r)
|
|
return results
|
|
|
|
|
|
def googlebooks_search(q, user_ip, page):
|
|
if len(q) < 2 or len(q) > 2000:
|
|
return {}
|
|
# XXX: need to pass IP address of user in from the frontend
|
|
headers = {'X-Forwarded-For': user_ip}
|
|
start = (page - 1) * 10
|
|
params = {'q': q, 'startIndex': start, 'maxResults': 10}
|
|
if hasattr(settings, 'GOOGLE_BOOKS_API_KEY'):
|
|
params['key'] = settings.GOOGLE_BOOKS_API_KEY
|
|
|
|
r = requests.get('https://www.googleapis.com/books/v1/volumes',
|
|
params=params, headers=headers)
|
|
# urls like https://www.googleapis.com/books/v1/volumes?q=invisible+engines&startIndex=0&maxResults=10&key=[key]
|
|
return json.loads(r.content)
|