DinQuant/backend_api_python/app/config/database.py

"""
Database and cache configuration
"""

import os


class MetaRedisConfig(type):
    """Redis configuration"""

    @property
    def HOST(cls):
        return os.getenv("REDIS_HOST", "localhost")

    @property
    def PORT(cls):
        return int(os.getenv("REDIS_PORT", 6379))

    @property
    def PASSWORD(cls):
        return os.getenv("REDIS_PASSWORD", None)

    @property
    def DB(cls):
        return int(os.getenv("REDIS_DB", 0))

    @property
    def CONNECT_TIMEOUT(cls):
        return int(os.getenv("REDIS_CONNECT_TIMEOUT", 5))

    @property
    def SOCKET_TIMEOUT(cls):
        return int(os.getenv("REDIS_SOCKET_TIMEOUT", 5))

    @property
    def MAX_CONNECTIONS(cls):
        return int(os.getenv("REDIS_MAX_CONNECTIONS", 10))


class RedisConfig(metaclass=MetaRedisConfig):
    """Redis cache configuration."""

    @classmethod
    def get_url(cls) -> str:
        """Get the Redis connection URL."""
        if cls.PASSWORD:
            return f"redis://:{cls.PASSWORD}@{cls.HOST}:{cls.PORT}/{cls.DB}"
        return f"redis://{cls.HOST}:{cls.PORT}/{cls.DB}"


class MetaCacheConfig(type):
    """Cache business configuration."""

    def ENABLED(cls):
        # Forced to be off by default unless explicitly enabled by an environment variable
        return os.getenv("CACHE_ENABLED", "False").lower() == "true"

    @property
    def DEFAULT_EXPIRE(cls):
        return int(os.getenv("CACHE_EXPIRE", 300))

    @property
    def KLINE_CACHE_TTL(cls):
        return {
            "1m": 5,  # K-line cache for 1 minute and 5 seconds
            "3m": 30,  # 3 minutes K-line cache 30 seconds
            "5m": 60,  # 5 minutes K-line cache 1 minute
            "15m": 300,  # 15 minutes K-line cache 5 minutes
            "30m": 300,  # 30 minutes K-line cache 5 minutes
            "1H": 300,  # 1 hour K-line cache for 5 minutes
            "4H": 300,  # 4 hours K-line cache for 5 minutes
            "1D": 300,  # Daily K-line cache for 5 minutes
            # Compatible with lowercase
            "1h": 300,
            "4h": 300,
            "1d": 300,
        }

    @property
    def ANALYSIS_CACHE_TTL(cls):
        return 3600

    @property
    def PRICE_CACHE_TTL(cls):
        return 10


class CacheConfig(metaclass=MetaCacheConfig):
    """Cache configuration."""

    pass