[executors] feat: add enum for logger types

Author: AndreyMarkinPPC

libs/executors/garf_executors/__init__.py

@@ -48,4 +48,4 @@ def setup_executor(
   'ApiExecutionContext',
 ]
 
-__version__ = '0.0.9'
+__version__ = '0.0.10'

libs/executors/garf_executors/entrypoints/utils.py

@@ -15,6 +15,7 @@
 
 from __future__ import annotations
 
+import enum
 import logging
 import sys
 from collections.abc import Sequence
@@ -89,8 +90,15 @@ class GarfParamsException(Exception):
   """Defines exception for incorrect parameters."""
 
 
+class LoggerEnum(str, enum.Enum):
+  local = 'local'
+  rich = 'rich'
+
+
 def init_logging(
-  loglevel: str = 'INFO', logger_type: str = 'local', name: str = __name__
+  loglevel: str = 'INFO',
+  logger_type: str | LoggerEnum = 'local',
+  name: str = __name__,
 ) -> logging.Logger:
   if logger_type == 'rich':
     logging.basicConfig(

libs/executors/tests/unit/entrypoints/test_utils.py

@@ -79,3 +79,11 @@ def test_parse(self, param_parser):
       },
       'template': {},
     }
+
+
+@pytest.mark.parametrize(
+  'logger_type',
+  ['local', 'rich', utils.LoggerEnum.local, utils.LoggerEnum.rich],
+)
+def test_init_logging(logger_type):
+  assert utils.init_logging(logger_type=logger_type)