def traverse_cand_strs(cand_strs, cand, no_other=True):
res = []
query_count = 0
for cs in cand_strs:
query_count += 1
t = db_api.lookup(cs, no_other)
for r in t:
res.append(Res_Entry(int(r["id"]), r["type"], r["aliases"][0], len(r["aliases"]), cs, cand))
return (res, query_count)
def look_for_entities(words, entities):
res = []
query_count = 0
for candidate in slice(words, 2):
for candidate_as_str in generate_candidate_variants(candidate):
query_count += 1
# t = lookup(candidate_as_str, entities)
t = db_api.lookup(candidate_as_str)
if t:
# res.append((t, candidate, candidate_as_str))
res.append((t['id'], candidate, candidate_as_str))
print("number of queries: {}".format(query_count))
return res
def look_for_entities(words, entities):
res = []
query_count = 0
for candidate in slice(words, 2):
for candidate_str in generate_candidate_variants(candidate):
# print(candidate_as_str)
query_count += 1
# t = lookup(candidate_as_str, entities)
t = db_api.lookup(candidate_str)
# if t:
# res.append((t, candidate, candidate_as_str))
for r in t:
res.append((r['id'], r["aliases"][0], candidate, candidate_str))
print("number of queries: {}".format(query_count))
return res
