Refactor, added service driver system

Author: avara1986

.coveragerc

@@ -3,6 +3,6 @@ include=
     *pyms/*
     *tests/*
 omit =
-    *example/*
+    *examples/*
     .tox/*
     venv/*

examples/mininum_microservice/config.yml

@@ -1,6 +1,7 @@
-swagger:
-  path: ""
-  file: "swagger.yaml"
+pyms:
+  swagger:
+    path: ""
+    file: "swagger.yaml"
 my-ms:
   DEBUG: true
   TESTING: false

pyms/constants.py

@@ -2,5 +2,4 @@
 
 LOGGER_NAME = "pyms"
 
-SWAGGER_PATH = "swagger"
-SWAGGER_FILE = "swagger.yaml"
+SERVICE_BASE = "pyms"

pyms/flask/app/__init__.py

@@ -3,11 +3,12 @@
 from typing import Text
 
 import connexion
+from flask import Flask
 from flask_opentracing import FlaskTracer
 
 from pyms.config.conf import get_conf
-from pyms.flask.app.swagger import Swagger
 from pyms.flask.healthcheck import healthcheck_blueprint
+from pyms.flask.services.driver import ServicesManager
 from pyms.logger import CustomJsonFormatter
 from pyms.tracer.main import init_jaeger_tracer
 
@@ -19,8 +20,13 @@ class Microservice:
     def __init__(self, service: Text, path=__file__):
         self.service = service
         self.path = os.path.dirname(path)
-        self.swagger = Swagger()
         self.config = get_conf(service=self.service)
+        self.init_services()
+
+    def init_services(self):
+        service_manager = ServicesManager()
+        for service_name, service in service_manager.get_services():
+            setattr(self, service_name, service)
 
     def init_libs(self):
         return self.application
@@ -39,22 +45,29 @@ def init_logger(self):
         self.application.logger.propagate = False
         self.application.logger.setLevel(logging.INFO)
 
+    def init_app(self):
+        if getattr(self, "swagger", False):
+            app = connexion.App(__name__, specification_dir=os.path.join(self.path, self.swagger.path))
+            app.add_api(self.swagger.file,
+                        arguments={'title': self.config.APP_NAME},
+                        base_path=self.config.APPLICATION_ROOT
+                        )
+
+            application = app.app
+            application._connexion_app = app
+        else:
+            application = Flask(__name__)
+
+        return application
+
     def create_app(self):
         """Initialize the Flask app, register blueprints and initialize
         all libraries like Swagger, database,
         the trace system...
         return the app and the database objects.
         :return:
         """
-
-        app = connexion.App(__name__, specification_dir=os.path.join(self.path, self.swagger.path))
-        app.add_api(self.swagger.file,
-                    arguments={'title': self.config.APP_NAME},
-                    base_path=self.config.APPLICATION_ROOT
-                    )
-
-        self.application = app.app
-        self.application._connexion_app = app
+        self.application = self.init_app()
         self.application.config.from_object(self.config)
         self.application.tracer = None
 

pyms/flask/app/swagger.py

@@ -0,0 +1,35 @@
+import logging
+
+from pyms.config.conf import get_conf
+from pyms.constants import SERVICE_BASE, LOGGER_NAME
+from pyms.utils.utils import import_from
+
+logger = logging.getLogger(LOGGER_NAME)
+
+class DriverService:
+    service = ""
+
+    def __init__(self, service, *args, **kwargs):
+        self.service = ".".join([service, self.service])
+        self.config = get_conf(service=self.service, empty_init=True)
+
+    def __getattr__(self, attr, *args, **kwargs):
+        config_attribute = getattr(self.config, attr)
+        return config_attribute if config_attribute is "" or config_attribute != {} else self.default_values.get(attr,
+                                                                                                                 None)
+
+
+class ServicesManager:
+    service = SERVICE_BASE
+
+    def __init__(self, service=None):
+        self.service = (service if service else SERVICE_BASE)
+        self.config = get_conf(service=self.service, empty_init=True)
+
+    def get_services(self):
+        return ((k, self.get_service(k)) for k in self.config.__dict__.keys() if k not in ['empty_init',])
+
+    def get_service(self, service, *args, **kwargs):
+        service_object = import_from("pyms.flask.services.{}".format(service), "Service")
+        logger.info("Init service {}".format(service))
+        return service_object(service=self.service, *args, **kwargs)

pyms/flask/services/driver.py

@@ -0,0 +1,12 @@
+from pyms.flask.services.driver import DriverService
+
+SWAGGER_PATH = "swagger"
+SWAGGER_FILE = "swagger.yaml"
+
+
+class Service(DriverService):
+    service = "swagger"
+    default_values = {
+        "path": SWAGGER_PATH,
+        "file": SWAGGER_FILE
+    }

pyms/flask/services/swagger.py

@@ -0,0 +1,3 @@
+def import_from(module, name):
+    module = __import__(module, fromlist=[name])
+    return getattr(module, name)

pyms/utils/utils.py

@@ -1,3 +1,7 @@
+pyms:
+  swagger:
+    path: ""
+    file: "swagger.yaml"
 my-ms:
   DEBUG: true
   TESTING: true