Minor fixes for OpenSearch (#28)

Minor fixes for OpenSearch
google · Dec 20, 2021 · fd2122c · fd2122c
1 parent 561f9d8
commit fd2122c
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 5 deletions.
diff --git a/dfdewey/utils/image_processor.py b/dfdewey/utils/image_processor.py
@@ -433,9 +433,9 @@ def _index_record(self, index_name, string_record):
   def _index_strings(self):
     """Index the extracted strings."""
     if self.config:
-      self.OpenSearch = OpenSearchDataStore(
-          host=self.config.ES_HOST, port=self.config.ES_PORT,
-          url=self.config.ES_URL)
+      self.opensearch = OpenSearchDataStore(
+          host=self.config.OS_HOST, port=self.config.OS_PORT,
+          url=self.config.OS_URL)
     else:
       self.opensearch = OpenSearchDataStore()
     index_name = ''.join(('es', self.image_hash))

diff --git a/dfdewey/utils/index_searcher.py b/dfdewey/utils/index_searcher.py
@@ -83,8 +83,8 @@ def __init__(self, case, image_id, image, config_file=None):
           host=self.config.PG_HOST, port=self.config.PG_PORT,
           db_name=self.config.PG_DB_NAME)
       self.opensearch = OpenSearchDataStore(
-          host=self.config.ES_HOST, port=self.config.ES_PORT,
-          url=self.config.ES_URL)
+          host=self.config.OS_HOST, port=self.config.OS_PORT,
+          url=self.config.OS_URL)
     else:
       self.postgresql = PostgresqlDataStore()
       self.opensearch = OpenSearchDataStore()