main.py

#!/usr/bin/python3
import sys
from pathlib import Path

from fluent import sender

from about import properties
from rest.api.constants.env_constants import EnvConstants
from rest.api.constants.env_init import EnvInit
from rest.api.exception.api_exception_docker import ApiExceptionDocker
from rest.api.exception.api_exception_kubectl import ApiExceptionKubectl
from rest.api.loghelpers.message_dumper import MessageDumper
from rest.api.schedulers.docker_env_expire_scheduler import DockerEnvExpireScheduler
from rest.api.schedulers.kubectl_env_expire_scheduler import KubectlEnvExpireScheduler
from rest.api.views import app
from rest.api.views.docker_view import DockerView
from rest.api.views.kubectl_view import KubectlView
from rest.environment.environment import EnvironmentSingleton
from rest.service.eureka import Eureka
from rest.service.fluentd import Fluentd
from rest.utils.env_startup import EnvStartupSingleton
from rest.utils.io_utils import IOUtils

DockerView.register(app=app)
KubectlView.register(app=app)

app.register_error_handler(ApiExceptionDocker, DockerView.handle_api_error)
app.register_error_handler(ApiExceptionKubectl, KubectlView.handle_api_error)

if __name__ == "__main__":
    cli = sys.modules['flask.cli']
    cli.show_server_banner = lambda *x: None

    fluentd_tag = "startup"
    host = '0.0.0.0'
    port = EnvStartupSingleton.get_instance().get_config_env_vars().get(EnvConstants.PORT)
    message_dumper = MessageDumper()
    io_utils = IOUtils()

    if EnvStartupSingleton.get_instance().get_config_env_vars().get(EnvConstants.EUREKA_SERVER):
        Eureka(EnvStartupSingleton.get_instance().get_config_env_vars().get(EnvConstants.EUREKA_SERVER)).register_app(
            EnvStartupSingleton.get_instance().get_config_env_vars().get(EnvConstants.APP_IP_PORT),
            EnvStartupSingleton.get_instance().get_config_env_vars().get(EnvConstants.APP_APPEND_LABEL))

    io_utils.create_dirs([Path(EnvInit.init.get(EnvConstants.DEPLOY_PATH)),
                          Path(EnvInit.init.get(EnvConstants.TEMPLATES_DIR)),
                          Path(EnvInit.init.get(EnvConstants.VARS_DIR))])

    DockerEnvExpireScheduler(fluentd_utils=DockerView.fluentd,
                             poll_interval=EnvStartupSingleton.get_instance().get_config_env_vars().get(
                                 EnvConstants.SCHEDULER_POLL_INTERVAL),  # seconds
                             env_expire_in=EnvStartupSingleton.get_instance().get_config_env_vars().get(
                                 EnvConstants.ENV_EXPIRE_IN)).start()  # minutes
    KubectlEnvExpireScheduler(fluentd_utils=KubectlView.fluentd,
                              poll_interval=EnvStartupSingleton.get_instance().get_config_env_vars().get(
                                  EnvConstants.SCHEDULER_POLL_INTERVAL),
                              env_expire_in=EnvStartupSingleton.get_instance().get_config_env_vars().get(
                                  EnvConstants.ENV_EXPIRE_IN)).start()

    environ_dump = message_dumper.dump_message(EnvironmentSingleton.get_instance().get_env_and_virtual_env())
    ip_port_dump = message_dumper.dump_message(
        {"host": host, "port": EnvStartupSingleton.get_instance().get_config_env_vars().get(EnvConstants.PORT)})

    app.logger.debug({"msg": environ_dump})
    app.logger.debug({"msg": ip_port_dump})
    app.logger.debug({"msg": EnvStartupSingleton.get_instance().get_config_env_vars()})

    logger = \
        sender.FluentSender(tag=properties.get('name'),
                            host=EnvStartupSingleton.get_instance().get_config_env_vars().get(
                                EnvConstants.FLUENTD_IP_PORT).split(":")[0],
                            port=int(EnvStartupSingleton.get_instance().get_config_env_vars().get(
                                EnvConstants.FLUENTD_IP_PORT).split(":")[1])) \
            if EnvStartupSingleton.get_instance().get_config_env_vars().get(EnvConstants.FLUENTD_IP_PORT) else None
    fluentd_utils = Fluentd(logger)
    fluentd_utils.emit(tag=fluentd_tag, msg=environ_dump)

    is_https = EnvStartupSingleton.get_instance().get_config_env_vars().get(EnvConstants.HTTPS_ENABLE)
    https_cert_path = EnvStartupSingleton.get_instance().get_config_env_vars().get(EnvConstants.HTTPS_CERT)
    https_prv_key_path = EnvStartupSingleton.get_instance().get_config_env_vars().get(EnvConstants.HTTPS_KEY)
    ssl_context = None
    if is_https:
        ssl_context = (https_cert_path, https_prv_key_path)
    app.run(host=host, port=port, ssl_context=ssl_context)