Use custom data extractor for inventaire connector

This commit is contained in:
Mouse Reeve
2021-04-06 12:29:06 -07:00
parent fba44206ac
commit 22ebe60c0a
2 changed files with 18 additions and 7 deletions

View File

@ -1,8 +1,6 @@
""" inventaire data connector """
from bookwyrm import models
from .abstract_connector import AbstractConnector, SearchResult, Mapping
from .abstract_connector import get_data
from .connector_manager import ConnectorException
class Connector(AbstractConnector):
@ -26,6 +24,11 @@ class Connector(AbstractConnector):
Mapping("goodreadsKey", remote_field="wdt:P2969", formatter=get_claim),
]
def get_book_data(self, remote_id):
data = get_data(remote_id)
extracted = list(data.get("entities").values())
return extracted[0] if extracted else {}
def parse_search_data(self, data):
return data.get('results')
@ -49,7 +52,7 @@ class Connector(AbstractConnector):
""" beep bloop """
def is_work_data(self, data):
return True
return data.get("type") == "work"
def get_edition_from_work_data(self, data):
return {}