From df3838480d2ad884d67d65f4cd58d4a510de57a7 Mon Sep 17 00:00:00 2001
From: Son NK <nguyenkims@hotmail.com>
Date: Sun, 8 Dec 2019 16:16:28 +0100
Subject: [PATCH] add COLOR_LOG config

---
 app/config.py | 1 +
 app/log.py    | 4 +++-
 2 files changed, 4 insertions(+), 1 deletion(-)

diff --git a/app/config.py b/app/config.py
index f412d801..cc5833e2 100644
--- a/app/config.py
+++ b/app/config.py
@@ -25,6 +25,7 @@ else:
     load_dotenv()
 
 RESET_DB = "RESET_DB" in os.environ
+COLOR_LOG = "COLOR_LOG" in os.environ
 
 # Allow user to have 1 year of premium: set the expiration_date to 1 year more
 PROMO_CODE = "SIMPLEISBETTER"
diff --git a/app/log.py b/app/log.py
index 5582a0a1..14b10a83 100644
--- a/app/log.py
+++ b/app/log.py
@@ -13,6 +13,7 @@ from app.config import (
     CLOUDWATCH_LOG_GROUP,
     ENABLE_CLOUDWATCH,
     CLOUDWATCH_LOG_STREAM,
+    COLOR_LOG,
 )
 
 _log_format = "%(asctime)s - %(name)s - %(levelname)s - %(process)d - %(module)s:%(lineno)d - %(funcName)s - %(message)s"
@@ -66,7 +67,8 @@ def _get_logger(name):
     # no propagation to avoid propagating to root logger
     logger.propagate = False
 
-    coloredlogs.install(level="DEBUG", logger=logger)
+    if COLOR_LOG:
+        coloredlogs.install(level="DEBUG", logger=logger)
 
     return logger