caspervk/dailyreleases
Archived
1
0
Fork 0
Code Releases Activity

Make string matching in Steam and GOG store search case insensitive.

Browse source
This commit is contained in:
Casper V. Kristensen 2018-08-03 03:24:57 +02:00
parent e62e608c48
commit 419333fac3
Signed by: caspervk
GPG key ID: B1156723DB3BDDA8
3 changed files with 22 additions and 18 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

16
dailyreleases/gog.py
View file

@ -1,6 +1,7 @@
import difflib
import logging
from dailyreleases import util
logger = logging.getLogger(__name__)
@ -14,18 +15,15 @@ class GOG(object):
"limit": 5,
"search": query
}
results = self.cache.get("https://www.gog.com/games/ajax/filtered", params=payload).json()["products"]
products = {p["title"]: p
for p in self.cache.get("https://www.gog.com/games/ajax/filtered", params=payload).json()["products"]
if p["isGame"]}
best_match = difflib.get_close_matches(query,
[r["title"] for r in results if r["isGame"]],
n=1,
cutoff=0.90)
best_match = util.case_insensitive_close_matches(query, products, n=1, cutoff=0.90)
if not best_match:
logger.debug("Unable to find %s in GOG search results", query)
return
logger.debug("Best match is '%s'", best_match[0])
return next("https://gog.com{}".format(r["url"])
for r in results
if r["title"] == best_match[0])
return "https://gog.com{url}".format(**products[best_match[0]])

13
dailyreleases/steam.py
View file

@ -1,4 +1,3 @@
import difflib
import logging
logger = logging.getLogger(__name__)
@ -51,18 +50,14 @@ class Steam(object):
payload = {
"term": query
}
results = self.cache.get("https://store.steampowered.com/api/storesearch", params=payload).json()["items"]
items = {i["name"]: i for i in self.cache.get("https://store.steampowered.com/api/storesearch",
params=payload).json()["items"]}
best_match = difflib.get_close_matches(query,
[r["name"] for r in results],
n=1,
cutoff=0.90)
best_match = util.case_insensitive_close_matches(query, items, n=1, cutoff=0.90)
if not best_match:
logger.debug("Unable to find %s in Steam search results", query)
return
logger.debug("Best match is '%s'", best_match[0])
return next("https://store.steampowered.com/{}/{}".format(r["type"], r["id"])
for r in results
if r["name"] == best_match[0])
return "https://store.steampowered.com/{type}/{id}".format(**items[best_match[0]])

11
dailyreleases/util.py Normal file
View file

@ -0,0 +1,11 @@
import difflib
def case_insensitive_close_matches(word, possibilities, n=3, cutoff=0.6):
"""
Python's difflib.get_close_matches does case sensitive sequence matching, this function decorates the library
function to make it case insensitive.
"""
possibilities = {sequence.lower(): sequence for sequence in possibilities}
close_matches = difflib.get_close_matches(word.lower(), possibilities, n=n, cutoff=cutoff)
return [possibilities[m] for m in close_matches]