moose-lib 0.4.228__py3-none-any.whl → 0.4.229__py3-none-any.whl

moose_lib/config/config_file.py

@@ -0,0 +1,82 @@
+"""
+Configuration file handling for Moose.
+
+This module provides functionality for reading and parsing the moose.config.toml file,
+which contains project-wide configuration settings.
+"""
+import os
+import tomllib
+from dataclasses import dataclass
+from typing import Optional
+
+@dataclass
+class ClickHouseConfig:
+    """ClickHouse configuration settings from moose.config.toml."""
+    host: str
+    host_port: int
+    user: str
+    password: str
+    db_name: str
+    use_ssl: bool = False
+    native_port: Optional[int] = None
+
+@dataclass
+class ProjectConfig:
+    """Project configuration from moose.config.toml."""
+    language: str
+    clickhouse_config: ClickHouseConfig
+
+def find_config_file(start_dir: str = os.getcwd()) -> Optional[str]:
+    """Find moose.config.toml by walking up directory tree.
+
+    Args:
+        start_dir: Directory to start searching from. Defaults to current directory.
+
+    Returns:
+        Path to moose.config.toml if found, None otherwise.
+    """
+    current_dir = os.path.abspath(start_dir)
+    while True:
+        config_path = os.path.join(current_dir, "moose.config.toml")
+        if os.path.exists(config_path):
+            return config_path
+
+        parent_dir = os.path.dirname(current_dir)
+        # Reached root directory
+        if parent_dir == current_dir:
+            break
+        current_dir = parent_dir
+    return None
+
+def read_project_config() -> ProjectConfig:
+    """Read and parse moose.config.toml.
+
+    Returns:
+        ProjectConfig object containing parsed configuration.
+    """
+    config_path = find_config_file()
+    if not config_path:
+        raise FileNotFoundError(
+            "moose.config.toml not found in current directory or any parent directory"
+        )
+
+    try:
+        with open(config_path, "rb") as f:
+            config_data = tomllib.load(f)
+
+        clickhouse_config = ClickHouseConfig(
+            host=config_data["clickhouse_config"]["host"],
+            host_port=config_data["clickhouse_config"]["host_port"],
+            user=config_data["clickhouse_config"]["user"],
+            password=config_data["clickhouse_config"]["password"],
+            db_name=config_data["clickhouse_config"]["db_name"],
+            use_ssl=config_data["clickhouse_config"].get("use_ssl", False),
+            native_port=config_data["clickhouse_config"].get("native_port")
+        )
+
+        return ProjectConfig(
+            language=config_data["language"],
+            clickhouse_config=clickhouse_config
+        )
+    except Exception as e:
+        raise RuntimeError(f"Failed to parse moose.config.toml: {e}")

moose_lib/config/runtime.py

@@ -0,0 +1,83 @@
+"""
+Runtime configuration management for Moose.
+
+This module provides a singleton registry for managing runtime configuration settings,
+particularly for ClickHouse connections.
+"""
+from dataclasses import dataclass
+from typing import Optional
+
+@dataclass
+class RuntimeClickHouseConfig:
+    """Runtime ClickHouse configuration settings."""
+    host: str
+    port: str
+    username: str
+    password: str
+    database: str
+    use_ssl: bool
+
+class ConfigurationRegistry:
+    """Singleton registry for managing runtime configuration.
+
+    This class provides a centralized way to manage and access runtime configuration
+    settings, with fallback to file-based configuration when runtime settings are not set.
+    """
+    _instance: Optional['ConfigurationRegistry'] = None
+    _clickhouse_config: Optional[RuntimeClickHouseConfig] = None
+
+    @classmethod
+    def get_instance(cls) -> 'ConfigurationRegistry':
+        """Get the singleton instance of ConfigurationRegistry.
+
+        Returns:
+            The singleton ConfigurationRegistry instance.
+        """
+        if not cls._instance:
+            cls._instance = cls()
+        return cls._instance
+
+    def set_clickhouse_config(self, config: RuntimeClickHouseConfig) -> None:
+        """Set the runtime ClickHouse configuration.
+
+        Args:
+            config: The ClickHouse configuration to use.
+        """
+        self._clickhouse_config = config
+
+    def get_clickhouse_config(self) -> RuntimeClickHouseConfig:
+        """Get the current ClickHouse configuration.
+
+        If runtime configuration is not set, falls back to reading from moose.config.toml.
+
+        Returns:
+            The current ClickHouse configuration.
+        """
+        if self._clickhouse_config:
+            return self._clickhouse_config
+
+        # Fallback to reading from config file
+        from .config_file import read_project_config
+        try:
+            config = read_project_config()
+            return RuntimeClickHouseConfig(
+                host=config.clickhouse_config.host,
+                port=str(config.clickhouse_config.host_port),
+                username=config.clickhouse_config.user,
+                password=config.clickhouse_config.password,
+                database=config.clickhouse_config.db_name,
+                use_ssl=config.clickhouse_config.use_ssl
+            )
+        except Exception as e:
+            raise RuntimeError(f"Failed to get ClickHouse configuration: {e}")
+
+    def has_runtime_config(self) -> bool:
+        """Check if runtime configuration is set.
+
+        Returns:
+            True if runtime configuration is set, False otherwise.
+        """
+        return self._clickhouse_config is not None
+
+# Create singleton instance
+config_registry = ConfigurationRegistry.get_instance()

{moose_lib-0.4.228.dist-info → moose_lib-0.4.229.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: moose_lib
-Version: 0.4.228
+Version: 0.4.229
 Home-page: https://www.fiveonefour.com/moose
 Author: Fiveonefour Labs Inc.
 Author-email: support@fiveonefour.com

{moose_lib-0.4.228.dist-info → moose_lib-0.4.229.dist-info}/RECORD

@@ -9,6 +9,9 @@ moose_lib/query_param.py,sha256=AB5BKu610Ji-h1iYGMBZKfnEFqt85rS94kzhDwhWJnc,6288
 moose_lib/tasks.py,sha256=6MXA0j7nhvQILAJVTQHCAsquwrSOi2zAevghAc_7kXs,1554
 moose_lib/clients/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 moose_lib/clients/redis_client.py,sha256=UBCdxwgZpIOIOy2EnPyxJIAYjw_qmNwGsJQCQ66SxUI,8117
+moose_lib/config/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
+moose_lib/config/config_file.py,sha256=NyjY6YFraBel7vBk18lLkpGaPR1viKMAEv4ZldyfLIA,2585
+moose_lib/config/runtime.py,sha256=gF1wrUCbp0MxRU92k0vyItvWX2x6P1FuBRzM61nJE_U,2776
 moose_lib/dmv2/__init__.py,sha256=zyGK2YDN4zQ2SMn9-SAiOuhpU46STePuJjK18LEserg,2269
 moose_lib/dmv2/_registry.py,sha256=agdZ7xzS99Caou60Q2pEErzEwyNYHqwy6XqV79eEmwg,504
 moose_lib/dmv2/consumption.py,sha256=71wdv6ZuEi8Om7aX3Lq-d6bAoc1-iw3Wudb8dHESJKI,4072
@@ -28,7 +31,7 @@ tests/__init__.py,sha256=0Gh4yzPkkC3TzBGKhenpMIxJcRhyrrCfxLSfpTZnPMQ,53
 tests/conftest.py,sha256=ZVJNbnr4DwbcqkTmePW6U01zAzE6QD0kNAEZjPG1f4s,169
 tests/test_moose.py,sha256=mBsx_OYWmL8ppDzL_7Bd7xR6qf_i3-pCIO3wm2iQNaA,2136
 tests/test_redis_client.py,sha256=d9_MLYsJ4ecVil_jPB2gW3Q5aWnavxmmjZg2uYI3LVo,3256
-moose_lib-0.4.228.dist-info/METADATA,sha256=01yY_cIhFC-MaaKFT7nmXNG1gvT6qlhz0yuf7BL_u14,729
-moose_lib-0.4.228.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
-moose_lib-0.4.228.dist-info/top_level.txt,sha256=XEns2-4aCmGp2XjJAeEH9TAUcGONLnSLy6ycT9FSJh8,16
-moose_lib-0.4.228.dist-info/RECORD,,
+moose_lib-0.4.229.dist-info/METADATA,sha256=ZblYKIznk4YNqvuBDBn_3LIreuxEl_y1ERY7vZCh3yk,729
+moose_lib-0.4.229.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
+moose_lib-0.4.229.dist-info/top_level.txt,sha256=XEns2-4aCmGp2XjJAeEH9TAUcGONLnSLy6ycT9FSJh8,16
+moose_lib-0.4.229.dist-info/RECORD,,