config.py

# pylint: disable=too-few-public-methods

import os
from dotenv import load_dotenv

load_dotenv()

class Config:
    DB_HOST = os.getenv('DB_HOST')
    DB_USER = os.getenv('DB_USER')
    DB_PASSWORD = os.getenv('DB_PASSWORD')
    DB_NAME = os.getenv('DB_NAME')

    REDIS_HOST = os.getenv('REDIS_HOST')
    REDIS_PORT = os.getenv('REDIS_PORT')

    JWT_SECRET = os.getenv('JWT_SECRET')

    HARBOR_HOST = os.getenv("HARBOR_HOST")
    HARBOR_USER = os.getenv("HARBOR_USER")
    HARBOR_PASSWORD = os.getenv("HARBOR_PASSWORD")
    HARBOR_PROJECT = os.getenv("HARBOR_PROJECT")

    LDAP_HOST = os.getenv("LDAP_HOST")
    LDAP_ENABLE_TLS = os.getenv("LDAP_ENABLE_TLS").lower() == 'yes'
    LDAP_CA_PATH = os.getenv("LDAP_CA_PATH")
    LDAP_USER_BASE_DN = os.getenv("LDAP_USER_BASE_DN")
    LDAP_ADMIN_GROUP_DN = os.getenv("LDAP_ADMIN_GROUP_DN")

    WG_LISTEN_IP = os.getenv("WG_LISTEN_IP")

    K8S_NAMESPACE = os.getenv("K8S_NAMESPACE")
    K8S_KUBE_CONFIG = os.getenv("K8S_KUBE_CONFIG")

    # General Parameters
    LOG_PATH = None # None -> stdout
    LOGGING_LEVEL = "info"
    DEBUG = False
    ALLOW_CORS = False
    WORKER_NUM = int(os.getenv("WORKER_NUM", "2"))
    ALLOWED_IP = os.getenv("ALLOWED_IP")

class DevelopmentConfig(Config):
    LOG_PATH = "./judge.log"
    LOGGING_LEVEL = "debug"
    DEBUG = True
    ALLOW_CORS = True

class ProductionConfig(Config):
    LOG_PATH = "/var/log/system-judge.log"
    LOGGING_LEVEL = "info"
    DEBUG = False
    ALLOW_CORS = False

def get_runtime_config():
    return os.getenv("RUNTIME_CONFIG", "Development")