Merge pull request #13 from cloudblue/CP-7

allow loggers configuration from remote

Author: marcserrat

connect/eaas/dataclasses.py

@@ -64,6 +64,8 @@ class ConfigurationPayload:
     configuration: dict = None
     logging_api_key: str = None
     environment_type: str = None
+    log_level: str = None
+    runner_log_level: str = None
 
     def to_json(self):
         return dataclasses.asdict(self)

connect/eaas/main.py

@@ -7,7 +7,6 @@
 import asyncio
 import logging
 import logging.config
-import pathlib
 import signal
 
 from connect.eaas.worker import Worker
@@ -72,7 +71,6 @@ def main():
     parser = argparse.ArgumentParser(prog='cextrun')
     parser.add_argument('-u', '--unsecure', action='store_true')
     parser.add_argument('-d', '--debug', action='store_true', default=False)
-    parser.add_argument('-e', '--extension-dir', default='/extension', type=pathlib.Path)
     data = parser.parse_args()
     configure_logger(data.debug)
     start(data)

connect/eaas/worker.py

@@ -217,9 +217,20 @@ async def configuration(self, data):
         It will stop the tasks manager so the extension can be
         reconfigured, then restart the tasks manager.
         """
-        self.extension_config = data.configuration
-        self.logging_api_key = data.logging_api_key
-        self.environment_type = data.environment_type
+        if data.configuration:
+            self.extension_config = data.configuration
+        if data.logging_api_key:
+            self.logging_api_key = data.logging_api_key
+        if data.environment_type:
+            self.environment_type = data.environment_type
+        if data.log_level:
+            logging.getLogger('eaas.extension').setLevel(
+                getattr(logging, data.log_level),
+            )
+        if data.runner_log_level:
+            logging.getLogger('connect.eaas').setLevel(
+                getattr(logging, data.runner_log_level),
+            )
         logger.info('Extension configuration has been updated.')
 
     async def pause(self):