Merge branch 'feature/add-app-more-reusable' into feature/add-swagger-conf

Author: avara1986

pyms/flask/app/__init__.py

@@ -25,6 +25,20 @@ def __init__(self, service: Text, path=__file__):
     def init_libs(self):
         return self.application
 
+    def init_tracer(self):
+        self.application.tracer = FlaskTracer(init_jaeger_tracer(), True, self.application)
+
+    def init_logger(self):
+        formatter = CustomJsonFormatter('(timestamp) (level) (name) (module) (funcName) (lineno) (message)')
+        formatter.add_service_name(self.application.config["APP_NAME"])
+        if getattr(self.application, "tracer", False):
+            formatter.add_trace_span(self.application.tracer)
+        log_handler = logging.StreamHandler()
+        log_handler.setFormatter(formatter)
+        self.application.logger.addHandler(log_handler)
+        self.application.logger.propagate = False
+        self.application.logger.setLevel(logging.INFO)
+
     def create_app(self):
         """Initialize the Flask app, register blueprints and initialize
         all libraries like Swagger, database,
@@ -50,18 +64,10 @@ def create_app(self):
         self.init_libs()
         self.add_error_handlers()
 
-        # Inject Modules
-        formatter = CustomJsonFormatter('(timestamp) (level) (name) (module) (funcName) (lineno) (message)')
         if not self.application.config["TESTING"]:
-            log_handler = logging.StreamHandler()
+            self.init_tracer()
 
-            self.application.tracer = FlaskTracer(init_jaeger_tracer(), True, self.application)
-            formatter.add_service_name(self.application.config["APP_NAME"])
-            formatter.add_trace_span(self.application.tracer)
-            log_handler.setFormatter(formatter)
-            self.application.logger.addHandler(log_handler)
-            self.application.logger.propagate = False
-            self.application.logger.setLevel(logging.INFO)
+        self.init_logger()
 
         return self.application