mirrors/bookwyrm
1
0
Fork 1
mirror of https://github.com/bookwyrm-social/bookwyrm.git synced 2024-07-02 11:45:50 +00:00
Code Issues Projects Releases Wiki Activity

Merge pull request #1069 from bookwyrm-social/inventaire-confidence

Browse source 
Adds confidence ratings to inventaire results
This commit is contained in:
Mouse Reeve 2021-05-10 16:10:30 -07:00 committed by GitHub
parent dad0161844 07c84a6e1a
commit 9e9f94753d
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23
1 changed files with 12 additions and 0 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

12
bookwyrm/connectors/inventaire.py
View file

@ -74,6 +74,14 @@ class Connector(AbstractConnector):
**{k: data.get(k) for k in ["uri", "image", "labels", "sitelinks"]},
}
def search(self, query, min_confidence=None):
"""overrides default search function with confidence ranking"""
results = super().search(query)
if min_confidence:
# filter the search results after the fact
return [r for r in results if r.confidence >= min_confidence]
return results
def parse_search_data(self, data):
return data.get("results")
@ -84,6 +92,9 @@ class Connector(AbstractConnector):
if images
else None
)
# a deeply messy translation of inventaire's scores
confidence = float(search_result.get("_score", 0.1))
confidence = 0.1 if confidence < 150 else 0.999
return SearchResult(
title=search_result.get("label"),
key=self.get_remote_id(search_result.get("uri")),
@ -92,6 +103,7 @@ class Connector(AbstractConnector):
self.base_url, search_result.get("uri")
),
cover=cover,
confidence=confidence,
connector=self,
)