linter fixes

Signed-off-by: Shivam Kumar <shivamkm07@gmail.com>
dapr · Jan 2, 2024 · a342f73 · a342f73
1 parent ebde05a
commit a342f73
Show file tree

Hide file tree

Showing 4 changed files with 11 additions and 8 deletions.
diff --git a/ext/dapr-ext-workflow/dapr/ext/workflow/dapr_workflow_client.py b/ext/dapr-ext-workflow/dapr/ext/workflow/dapr_workflow_client.py
@@ -61,11 +61,12 @@ def __init__(
         metadata = tuple()
         if settings.DAPR_API_TOKEN:
             metadata = ((DAPR_API_TOKEN_HEADER, settings.DAPR_API_TOKEN),)
+        options = self._logger.get_options()
         self.__obj = client.TaskHubGrpcClient(host_address=uri.endpoint,
                                               metadata=metadata,
                                               secure_channel=uri.tls,
-                                              log_handler=self._logger.get_options().log_handler,
-                                              log_formatter=self._logger.get_options().log_formatter)
+                                              log_handler=options.log_handler,
+                                              log_formatter=options.log_formatter)
 
     def schedule_new_workflow(self, workflow: Workflow, *, input: Optional[TInput] = None,
                               instance_id: Optional[str] = None,

diff --git a/ext/dapr-ext-workflow/dapr/ext/workflow/logger/logger.py b/ext/dapr-ext-workflow/dapr/ext/workflow/logger/logger.py
@@ -2,9 +2,10 @@
 from typing import Union
 from dapr.ext.workflow.logger.options import LoggerOptions
 
+
 class Logger:
-    def __init__(self, 
-                 name: str, 
+    def __init__(self,
+                 name: str,
                  options: Union[LoggerOptions, None] = None):
         # If options is None, then create a new LoggerOptions object
         if options is None:
@@ -16,7 +17,7 @@ def __init__(self,
         logger.handlers.append(log_handler)
         self._logger_options = options
         self._logger = logger
-    
+
     def get_options(self) -> LoggerOptions:
         return self._logger_options
 

diff --git a/ext/dapr-ext-workflow/dapr/ext/workflow/logger/options.py b/ext/dapr-ext-workflow/dapr/ext/workflow/logger/options.py
@@ -37,4 +37,4 @@ def __init__(
                 datefmt='%Y-%m-%d %H:%M:%S')
         self.log_level = log_level
         self.log_handler = log_handler
-        self.log_formatter = log_formatter
+        self.log_formatter = log_formatter
diff --git a/ext/dapr-ext-workflow/dapr/ext/workflow/workflow_runtime.py b/ext/dapr-ext-workflow/dapr/ext/workflow/workflow_runtime.py
@@ -53,11 +53,12 @@ def __init__(
         except ValueError as error:
             raise DaprInternalError(f'{error}') from error
 
+        options = self._logger.get_options()
         self.__worker = worker.TaskHubGrpcWorker(host_address=uri.endpoint,
                                                  metadata=metadata,
                                                  secure_channel=uri.tls,
-                                                 log_handler=self._logger.get_options().log_handler,
-                                                 log_formatter=self._logger.get_options().log_formatter)
+                                                 log_handler=options.log_handler,
+                                                 log_formatter=options.log_formatter)
 
     def register_workflow(self, fn: Workflow):
         self._logger.info(f"Registering workflow '{fn.__name__}' with runtime")