From e8cf6f591d4dcaa638fd9e450e6d1f947044da40 Mon Sep 17 00:00:00 2001 From: Justin Barrett Date: Fri, 19 Sep 2025 08:20:09 -0400 Subject: [PATCH] Adding single search --- rorapi/common/matching_single_search.py | 32 +++++----------------- rorapi/management/commands/indexror.py | 4 --- rorapi/management/commands/indexrordump.py | 4 --- 3 files changed, 7 insertions(+), 33 deletions(-) diff --git a/rorapi/common/matching_single_search.py b/rorapi/common/matching_single_search.py index 9c8c4ae..04e67fe 100644 --- a/rorapi/common/matching_single_search.py +++ b/rorapi/common/matching_single_search.py @@ -243,17 +243,11 @@ def choose_candidate(rescored): def match_by_query(text, query, countries): """Match affiliation text using specific ES query.""" - try: - scored_candidates = [] - chosen_candidate = None - chosen_true = None - results = query.execute() - except Exception as e: - return f"query error: {e}", None - try: - candidates = results.hits.hits - except Exception as e: - return f"candidates error: {e}\n{results}", None + scored_candidates = [] + chosen_candidate = None + chosen_true = None + results = query.execute() + candidates = results.hits.hits if candidates: candidates = [c for c in candidates if c["_source"]["status"] == "active"] scored_candidates = [score(text, c) for c in candidates] @@ -310,25 +304,13 @@ def get_output(chosen, all_matched, active_only): def get_candidates(aff, countries, version): qb = ESQueryBuilder(version) - try: - # get the index mappings for an index - curr_v2_index = ES7.indices.get_mapping('organizations-v2') - # return as json string - return json.dumps(curr_v2_index), None - except Exception as e: - return f"query error {version}: {e}", None - try: - qb.add_affiliation_query(aff, 200) - return match_by_query(aff, qb.get_query(), countries) - except Exception as e: - return f"query error: {e}", None + qb.add_affiliation_query(aff, 200) + return match_by_query(aff, qb.get_query(), countries) def match_affiliation(affiliation, active_only, version): countries = get_countries(affiliation) chosen, all_matched = get_candidates(affiliation, countries, version) - if isinstance(chosen, str): - return chosen return get_output(chosen, all_matched, active_only) diff --git a/rorapi/management/commands/indexror.py b/rorapi/management/commands/indexror.py index 2c8c7e7..c86b720 100644 --- a/rorapi/management/commands/indexror.py +++ b/rorapi/management/commands/indexror.py @@ -191,10 +191,6 @@ def index(dataset, version): 'id': n } for n in get_nested_ids_v1(org)] body.append(org) - print("example_1: ", body[0]) - print("example_1: ", body[1]) - print("example_2: ", body[8]) - print("example_2: ", body[9]) ES7.bulk(body) except TransportError: err[index.__name__] = f"Indexing error, reverted index back to previous state" diff --git a/rorapi/management/commands/indexrordump.py b/rorapi/management/commands/indexrordump.py index b810fc3..6719705 100644 --- a/rorapi/management/commands/indexrordump.py +++ b/rorapi/management/commands/indexrordump.py @@ -98,10 +98,6 @@ def index_dump(self, filename, index, dataset): 'id': n } for n in get_nested_ids_v1(org)] body.append(org) - print("example_1: ", body[0]) - print("example_1: ", body[1]) - print("example_2: ", body[8]) - print("example_2: ", body[9]) ES7.bulk(body) except TransportError: self.stdout.write(TransportError)