diff --git a/slo_generator/constants.py b/slo_generator/constants.py index cd47f62a..5c18a366 100644 --- a/slo_generator/constants.py +++ b/slo_generator/constants.py @@ -28,6 +28,7 @@ COLORED_OUTPUT: int = int(os.environ.get("COLORED_OUTPUT", "0")) DRY_RUN: bool = bool(int(os.environ.get("DRY_RUN", "0"))) DEBUG: int = int(os.environ.get("DEBUG", "0")) +LOGGING_TIMESTAMP_ENABLED: int = int(os.environ.get("LOGGING_TIMESTAMP_ENABLED", "0")) # Exporters supporting v2 SLO report format V2_EXPORTERS: Tuple[str, ...] = ("Pubsub", "Cloudevent") diff --git a/slo_generator/utils.py b/slo_generator/utils.py index b88534fc..1cdec92a 100644 --- a/slo_generator/utils.py +++ b/slo_generator/utils.py @@ -33,7 +33,7 @@ import yaml from dateutil import tz -from slo_generator.constants import DEBUG +from slo_generator.constants import DEBUG, LOGGING_TIMESTAMP_ENABLED try: # pytype: disable=import-error @@ -181,8 +181,12 @@ def setup_logging(): else: level = logging.INFO format_str = "%(levelname)s - %(message)s" + + if LOGGING_TIMESTAMP_ENABLED == "1": + format_str = "%(asctime)s " + format_str + logging.basicConfig( - stream=sys.stdout, level=level, format=format_str, datefmt="%m/%d/%Y %I:%M:%S" + stream=sys.stdout, level=level, format=format_str, datefmt="%m/%d/%Y %H:%M:%S" ) logging.getLogger("googleapiclient").setLevel(logging.ERROR)