aboutsummaryrefslogtreecommitdiffstats
path: root/fuzzycat/utils.py
diff options
context:
space:
mode:
authorMartin Czygan <martin.czygan@gmail.com>2020-12-15 02:42:44 +0100
committerMartin Czygan <martin.czygan@gmail.com>2020-12-15 02:42:44 +0100
commitb9b2aafe08abc7f131514a2eba804a569d424efb (patch)
tree3abc5184689f2df9de719bd905c69d0021afbe99 /fuzzycat/utils.py
parentf939949ae3889078ef2c9d77d1cffdd939e11435 (diff)
downloadfuzzycat-b9b2aafe08abc7f131514a2eba804a569d424efb.tar.gz
fuzzycat-b9b2aafe08abc7f131514a2eba804a569d424efb.zip
fix verification
Diffstat (limited to 'fuzzycat/utils.py')
-rw-r--r--fuzzycat/utils.py11
1 files changed, 6 insertions, 5 deletions
diff --git a/fuzzycat/utils.py b/fuzzycat/utils.py
index cf74220..9aa3a4c 100644
--- a/fuzzycat/utils.py
+++ b/fuzzycat/utils.py
@@ -6,6 +6,7 @@ import random
import re
import string
+import requests
from glom import PathAccessError, glom
printable_no_punct = string.digits + string.ascii_letters + string.whitespace
@@ -164,11 +165,11 @@ def random_idents_from_query(query="*",
Return a number of random idents from a search query.
"""
for _ in range(max_retries):
- r = requests.get(es, params={"q": query})
- if r.status_code != 200:
+ resp = requests.get(es, params={"q": query})
+ if resp.status_code != 200:
raise RuntimeError('could not query {} for random item: {}'.format(es, r.url))
- resp = r.json()
- if resp["hits"]["total"] < 2:
+ payload = resp.json()
+ if payload["hits"]["total"] < 2:
continue
- idents = [doc["_source"]["ident"] for doc in resp["hits"]["hits"]]
+ idents = [doc["_source"]["ident"] for doc in payload["hits"]["hits"]]
return random.sample(idents, r)