|
@@ -0,0 +1,46 @@
|
|
|
|
|
+import json
|
|
|
|
|
+import requests
|
|
|
|
|
+import logging
|
|
|
|
|
+
|
|
|
|
|
+logger = logging.getLogger(__name__)
|
|
|
|
|
+
|
|
|
|
|
+SEARCH_URL = "https://openlibrary.org/search.json?title={title}"
|
|
|
|
|
+ISBN_URL = "https://openlibrary.org/isbn/{isbn}.json"
|
|
|
|
|
+
|
|
|
|
|
+
|
|
|
|
|
+def get_first(key: str, result: dict) -> str:
|
|
|
|
|
+ obj = ""
|
|
|
|
|
+ if obj_list := result.get(key):
|
|
|
|
|
+ obj = obj_list[0]
|
|
|
|
|
+ return obj
|
|
|
|
|
+
|
|
|
|
|
+
|
|
|
|
|
+def lookup_book_from_openlibrary(title: str, author: str = None) -> dict:
|
|
|
|
|
+ search_url = SEARCH_URL.format(title=title)
|
|
|
|
|
+ response = requests.get(search_url)
|
|
|
|
|
+
|
|
|
|
|
+ if response.status_code != 200:
|
|
|
|
|
+ logger.warn(f"Bad response from OL: {response.status_code}")
|
|
|
|
|
+ return {}
|
|
|
|
|
+
|
|
|
|
|
+ results = json.loads(response.content)
|
|
|
|
|
+
|
|
|
|
|
+ if len(results.get("docs")) == 0:
|
|
|
|
|
+ logger.warn(f"No results found from OL for {title}")
|
|
|
|
|
+ return {}
|
|
|
|
|
+
|
|
|
|
|
+ top = results.get("docs")[0]
|
|
|
|
|
+ if author and author not in top["author_name"]:
|
|
|
|
|
+ logger.warn(
|
|
|
|
|
+ f"Lookup for {title} found top result with mismatched author"
|
|
|
|
|
+ )
|
|
|
|
|
+
|
|
|
|
|
+ return {
|
|
|
|
|
+ "title": top.get("title"),
|
|
|
|
|
+ "isbn": top.get("isbn")[0],
|
|
|
|
|
+ "openlibrary_id": top.get("cover_edition_key"),
|
|
|
|
|
+ "author_name": get_first("author_name", top),
|
|
|
|
|
+ "author_openlibrary_id": get_first("author_key", top),
|
|
|
|
|
+ "goodreads_id": get_first("id_goodreads", top),
|
|
|
|
|
+ "first_publish_year": top.get("first_publish_year"),
|
|
|
|
|
+ }
|
