#1061 Cannot create custom index in python bingo-elastic (#1165)

epam · Jul 13, 2023 · dd67530 · dd67530
1 parent c449ff9
commit dd67530
Showing 1 changed file with 17 additions and 8 deletions.
diff --git a/bingo/bingo-elastic/python/bingo_elastic/elastic.py b/bingo/bingo-elastic/python/bingo_elastic/elastic.py
@@ -38,15 +38,24 @@
 
 
 class IndexName(Enum):
+    def __init__(self, value):
+        self._value_ = value
+
     BINGO_MOLECULE = "bingo-molecules"
     BINGO_REACTION = "bingo-reactions"
+    BINGO_CUSTOM = "custom-index"
+
+    def set_value(self, new_value):
+        self._value_ = new_value
 
 
 def get_index_name(record: IndigoRecord) -> IndexName:
     if isinstance(record, IndigoRecordMolecule):
         return IndexName.BINGO_MOLECULE
     if isinstance(record, IndigoRecordReaction):
         return IndexName.BINGO_REACTION
+    if isinstance(record, str):
+        return IndexName.BINGO_CUSTOM
     raise AttributeError(f"Unknown IndigoRecord type {record}")
 
 
@@ -143,14 +152,14 @@ async def a_create_index(
 
 
 def prepare(
-    index_name: str, records: Generator[IndigoRecord, None, None]
+    records: Generator[IndigoRecord, None, None]
 ) -> Generator[Dict, None, None]:
     for record in records:
-        if get_index_name(record).value != index_name:
-            raise ValueError(
-                f"Index {index_name} doesn't support store value "
-                f"of type {type(record)}"
-            )
+        # if get_index_name(record).value != index_name:
+        #     raise ValueError(
+        #         f"Index {index_name} doesn't support store value "
+        #         f"of type {type(record)}"
+        #     )
         yield record.as_dict()
 
 
@@ -217,7 +226,7 @@ async def index_records(self, records: Generator, chunk_size: int = 500):
         # pylint: disable=unused-variable
         async for is_ok, action in async_streaming_bulk(
             self.el_client,
-            prepare(self.index_name, records),
+            prepare(records),
             index=self.index_name,
             chunk_size=chunk_size,
         ):
@@ -307,7 +316,7 @@ def index_records(self, records: Generator, chunk_size: int = 500):
         # pylint: disable=unused-variable
         for is_ok, action in streaming_bulk(
             self.el_client,
-            prepare(self.index_name, records),
+            prepare(records),
             index=self.index_name,
             chunk_size=chunk_size,
         ):