Clean up yente.search.base - no longer needed

opensanctions · Jul 18, 2024 · 24c4c04 · 24c4c04
1 parent 25df306
commit 24c4c04
Show file tree

Hide file tree

Showing 4 changed files with 7 additions and 19 deletions.
diff --git a/yente/provider/base.py b/yente/provider/base.py
@@ -1,6 +1,11 @@
+from asyncio import Semaphore
 from typing import Any, Dict, List, Optional
 from typing import AsyncIterator
 
+from yente import settings
+
+query_semaphore = Semaphore(settings.QUERY_CONCURRENCY)
+
 
 class SearchProvider(object):
     async def close(self) -> None:

diff --git a/yente/provider/elastic.py b/yente/provider/elastic.py
@@ -11,9 +11,8 @@
 from yente import settings
 from yente.exc import IndexNotReadyError, YenteIndexError, YenteNotFoundError
 from yente.logs import get_logger
-from yente.search.base import query_semaphore
 from yente.search.mapping import make_entity_mapping, INDEX_SETTINGS
-from yente.provider.base import SearchProvider
+from yente.provider.base import SearchProvider, query_semaphore
 
 log = get_logger(__name__)
 warnings.filterwarnings("ignore", category=ElasticsearchWarning)

diff --git a/yente/provider/opensearch.py b/yente/provider/opensearch.py
@@ -10,9 +10,8 @@
 from yente import settings
 from yente.exc import IndexNotReadyError, YenteIndexError, YenteNotFoundError
 from yente.logs import get_logger
-from yente.search.base import query_semaphore
 from yente.search.mapping import make_entity_mapping, INDEX_SETTINGS
-from yente.provider.base import SearchProvider
+from yente.provider.base import SearchProvider, query_semaphore
 
 log = get_logger(__name__)
 logging.getLogger("opensearch").setLevel(logging.ERROR)

diff --git a/yente/search/base.py b/yente/search/base.py