mirror of
https://github.com/bookwyrm-social/bookwyrm.git
synced 2024-06-10 09:19:27 +00:00
52 lines
1.8 KiB
Python
52 lines
1.8 KiB
Python
''' activitystream api and books '''
|
|
from django.http import HttpResponse, HttpResponseRedirect, HttpResponseBadRequest
|
|
from django.core.exceptions import ObjectDoesNotExist
|
|
from django.core import serializers
|
|
from fedireads.models import Author, Book, Work
|
|
import requests
|
|
|
|
openlibrary_url = 'https://openlibrary.org'
|
|
|
|
def get_book(request, olkey):
|
|
# check if this is a valid open library key, and a book
|
|
response = requests.get(openlibrary_url + '/book/' + olkey + '.json')
|
|
|
|
# get the existing entry from our db, if it exists
|
|
try:
|
|
book = Book.objects.get(openlibary_key=olkey)
|
|
except ObjectDoesNotExist:
|
|
book = Book(openlibary_key=olkey)
|
|
data = response.json()
|
|
book.data = data
|
|
if request and request.user and request.user.is_authenticated:
|
|
book.added_by = request.user
|
|
book.save()
|
|
for work_id in data['works']:
|
|
work_id = work_id['key']
|
|
book.works.add(get_or_create_work(work_id))
|
|
for author_id in data['authors']:
|
|
author_id = author_id['key']
|
|
book.authors.add(get_or_create_author(author_id))
|
|
return HttpResponse(serializers.serialize('json', [book]))
|
|
|
|
def get_or_create_work(olkey):
|
|
try:
|
|
work = Work.objects.get(openlibary_key=olkey)
|
|
except ObjectDoesNotExist:
|
|
response = requests.get(openlibrary_url + olkey + '.json')
|
|
data = response.json()
|
|
work = Work(openlibary_key=olkey, data=data)
|
|
work.save()
|
|
return work
|
|
|
|
def get_or_create_author(olkey):
|
|
try:
|
|
author = Author.objects.get(openlibary_key=olkey)
|
|
except ObjectDoesNotExist:
|
|
response = requests.get(openlibrary_url + olkey + '.json')
|
|
data = response.json()
|
|
author = Author(openlibary_key=olkey, data=data)
|
|
author.save()
|
|
return author
|
|
|