add IndexNameFrequency.NEVER

Add possibility to write index to the exact name specified without any datetime suffixes. Fixes #69
cmanaha · Dec 11, 2019 · 1719218 · 1719218
1 parent e4a16a8
commit 1719218
Show file tree

Hide file tree

Showing 2 changed files with 23 additions and 1 deletion.
diff --git a/cmreslogging/handlers.py b/cmreslogging/handlers.py
@@ -51,11 +51,13 @@ class IndexNameFrequency(Enum):
         - Weekly indices
         - Monthly indices
         - Year indices
+        - Never expiring indices
         """
         DAILY = 0
         WEEKLY = 1
         MONTHLY = 2
         YEARLY = 3
+        NEVER = 4
 
     # Defaults for the class
     __DEFAULT_ELASTICSEARCH_HOST = [{'host': 'localhost', 'port': 9200}]
@@ -115,11 +117,20 @@ def _get_yearly_index_name(es_index_name):
         """
         return "{0!s}-{1!s}".format(es_index_name, datetime.datetime.now().strftime('%Y'))
 
+    @staticmethod
+    def _get_never_index_name(es_index_name):
+        """ Return elasticsearch index name
+        :param: index_name the prefix to be used in the index
+        :return: A srting containing the elasticsearch indexname used which should include just the index name
+        """
+        return "{0!s}".format(es_index_name)
+
     _INDEX_FREQUENCY_FUNCION_DICT = {
         IndexNameFrequency.DAILY: _get_daily_index_name,
         IndexNameFrequency.WEEKLY: _get_weekly_index_name,
         IndexNameFrequency.MONTHLY: _get_monthly_index_name,
-        IndexNameFrequency.YEARLY: _get_yearly_index_name
+        IndexNameFrequency.YEARLY: _get_yearly_index_name,
+        IndexNameFrequency.NEVER: _get_never_index_name,
     }
 
     def __init__(self,

diff --git a/tests/test_cmreshandler.py b/tests/test_cmreshandler.py
@@ -169,6 +169,17 @@ def test_index_name_frequency_functions(self):
             CMRESHandler._get_yearly_index_name(index_name)
         )
 
+        handler = CMRESHandler(hosts=[{'host': self.getESHost(), 'port': self.getESPort()}],
+                               auth_type=CMRESHandler.AuthType.NO_AUTH,
+                               es_index_name=index_name,
+                               use_ssl=False,
+                               index_name_frequency=CMRESHandler.IndexNameFrequency.NEVER,
+                               raise_on_indexing_exceptions=True)
+        self.assertEqual(
+            handler._index_name_func.__func__(index_name),
+            CMRESHandler._get_never_index_name(index_name)
+        )
+
 
 if __name__ == '__main__':
     unittest.main()