Adding logging to pyflyte launch plan activation command (#117)

flytekit/__init__.py

@@ -2,4 +2,4 @@
 
 import flytekit.plugins
 
-__version__ = '0.8.0'
+__version__ = '0.8.1'

flytekit/clis/sdk_in_container/launch_plan.py

@@ -2,6 +2,7 @@
 
 import click
 import six as _six
+import logging as _logging
 
 from flytekit.clis.helpers import construct_literal_map_from_parameter_map as _construct_literal_map_from_parameter_map
 from flytekit.clis.sdk_in_container import constants as _constants
@@ -159,6 +160,7 @@ def activate_all_impl(project, domain, version, pkgs, ignore_schedules=False):
     # TODO: We should optionally allow deactivation of missing launch plans
 
     # Discover all launch plans by loading the modules
+    _logging.info(f"Setting this version's {version} launch plans active in {project} {domain}")
     for m, k, lp in iterate_registerable_entities_in_order(
         pkgs, include_entities={_SdkLaunchPlan}, detect_unreferenced_entities=False
     ):
@@ -170,6 +172,7 @@ def activate_all_impl(project, domain, version, pkgs, ignore_schedules=False):
             version
         )
         if not (lp.is_scheduled and ignore_schedules):
+            _logging.info(f"Setting active {_utils.fqdn(m.__name__, k, entity_type=lp.resource_type)}")
             lp.update(_launch_plan_model.LaunchPlanState.ACTIVE)
 
 

flytekit/clis/sdk_in_container/pyflyte.py

@@ -2,7 +2,7 @@
 from __future__ import print_function
 
 import os as _os
-
+import logging as _logging
 import click
 
 try:
@@ -15,7 +15,7 @@
 from flytekit.clis.sdk_in_container.serialize import serialize
 from flytekit.clis.sdk_in_container.constants import CTX_PROJECT, CTX_DOMAIN, CTX_VERSION
 from flytekit.clis.sdk_in_container.launch_plan import launch_plans
-from flytekit.configuration import internal as _internal_config, platform as _platform_config
+from flytekit.configuration import internal as _internal_config, platform as _platform_config, sdk as _sdk_config
 
 from flytekit.configuration.internal import CONFIGURATION_PATH
 from flytekit.configuration.platform import URL as _URL
@@ -43,6 +43,11 @@ def main(ctx, project, domain, config=None, pkgs=None, version=None, insecure=No
     """
     update_configuration_file(config)
 
+    # Update the logger if it's set
+    log_level = _internal_config.LOGGING_LEVEL.get() or _sdk_config.LOGGING_LEVEL.get()
+    if log_level is not None:
+        _logging.getLogger().setLevel(log_level)
+
     ctx.obj = dict()
     ctx.obj[CTX_PROJECT] = project
     ctx.obj[CTX_DOMAIN] = domain