rafaelcaricio/moviewyrm
1
0
Fork 0
forked from mirrors/bookwyrm
Code Issues 4 Pull requests Projects Releases Wiki Activity

Expand inventaire book data

Browse source
This commit is contained in:
Mouse Reeve 2021-04-06 18:34:55 -07:00
parent fec3d63e46
commit 29e7659b76
1 changed files with 22 additions and 5 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

27
bookwyrm/connectors/inventaire.py
View file

@ -1,4 +1,5 @@
""" inventaire data connector """ """ inventaire data connector """
from bookwyrm import models
from .abstract_connector import AbstractConnector, SearchResult, Mapping from .abstract_connector import AbstractConnector, SearchResult, Mapping
from .abstract_connector import get_data from .abstract_connector import get_data
from .connector_manager import ConnectorException from .connector_manager import ConnectorException
@ -95,12 +96,15 @@ class Connector(AbstractConnector):
def is_work_data(self, data): def is_work_data(self, data):
return data.get("type") == "work" return data.get("type") == "work"
def get_edition_from_work_data(self, data): def load_edition_data(self, work_uri):
value = data.get("uri") """ get a list of editions for a work """
url = "{:s}?action=reverse-claims&property=P629&value={:s}".format( url = "{:s}?action=reverse-claims&property=P629&value={:s}".format(
self.books_url, value self.books_url, work_uri
) )
data = get_data(url) return get_data(url)
def get_edition_from_work_data(self, data):
data = self.load_edition_data(data.get("uri"))
try: try:
uri = data["uris"][0] uri = data["uris"][0]
except KeyError: except KeyError:
@ -120,7 +124,20 @@ class Connector(AbstractConnector):
yield self.get_or_create_author(self.get_remote_id(author)) yield self.get_or_create_author(self.get_remote_id(author))
def expand_book_data(self, book): def expand_book_data(self, book):
return work = book
# go from the edition to the work, if necessary
if isinstance(book, models.Edition):
work = book.parent_work
try:
edition_options = self.load_edition_data(work.inventaire_id)
except ConnectorException:
# who knows, man
return
for edition_uri in edition_options.get("uris"):
remote_id = self.get_remote_id(edition_uri)
self.get_or_create_book(remote_id)
def get_cover_url(self, cover_blob, *_): def get_cover_url(self, cover_blob, *_):
"""format the relative cover url into an absolute one: """format the relative cover url into an absolute one: