sibi-flux 2026.1.2__py3-none-any.whl → 2026.1.4__py3-none-any.whl

sibi_flux/__init__.py

@@ -28,7 +28,6 @@ from sibi_flux import dask_cluster
 from sibi_flux.utils import clickhouse_writer
 from sibi_flux import parquet
 
-
 __all__ = [
     "__version__",
     "Logger",

sibi_flux/cli.py

@@ -7,39 +7,62 @@ from sibi_flux.init.core import initialize_project
 app = typer.Typer(help="Sibi Flux CLI")
 console = Console()
 
+
 @app.callback()
 def callback():
     """
     Sibi Flux CLI
     """
 
+
 @app.command()
 def init(
     project_name: str = typer.Argument(..., help="Name of the project to create"),
-    lib: bool = typer.Option(False, "--lib", help="Initialize as a library project (passed to uv init)"),
-    app: bool = typer.Option(False, "--app", help="Initialize as an application project (passed to uv init)")
+    lib: bool = typer.Option(
+        False, "--lib", help="Initialize as a library project (passed to uv init)"
+    ),
+    app: bool = typer.Option(
+        False, "--app", help="Initialize as an application project (passed to uv init)"
+    ),
 ):
     """
     Initialize a new Sibi Flux project.
-    
+
     Creates a new directory <project_name>, initializes it with 'uv',
     and adds 'sibi-flux' as a dependency.
     """
     initialize_project(project_name, lib, app)
 
+
 @app.command()
 def env(
     project_path: Path = typer.Argument(Path("."), help="Project root directory"),
-    env_file: Optional[Path] = typer.Option(None, "--env-file", "-e", help="Path to environment file (defaults to .env)"),
-    cleanup: bool = typer.Option(False, "--cleanup", help="Remove existing configuration files"),
-    production: bool = typer.Option(False, "--production", "-p", help="Generate production skeleton (no hardcoded values)"),
+    env_file: Optional[Path] = typer.Option(
+        None, "--env-file", "-e", help="Path to environment file (defaults to .env)"
+    ),
+    cleanup: bool = typer.Option(
+        False, "--cleanup", help="Remove existing configuration files"
+    ),
+    production: bool = typer.Option(
+        False,
+        "--production",
+        "-p",
+        help="Generate production skeleton (no hardcoded values)",
+    ),
 ):
     """
     Initialize configuration files (settings.py, credentials) based on .env
     """
     from sibi_flux.init.env import init_env
+
     init_env(project_path, env_file, cleanup=cleanup, production_mode=production)
 
 
+# Mount Datacube CLI
+from sibi_flux.datacube.cli import app as dc_app
+
+app.add_typer(dc_app, name="dc", help="Datacube generation and management commands")
+
+
 if __name__ == "__main__":
     app()

sibi_flux/config/settings.py

@@ -9,7 +9,7 @@ class SibiBaseSettings(BaseSettings):
     model_config = SettingsConfigDict(
         env_file=".env", env_file_encoding="utf-8", extra="ignore"
     )
-    
+
     conf_name: ClassVar[str] = ""
 
 
@@ -103,6 +103,25 @@ class DatabaseSettings(SibiBaseSettings):
         }
 
 
+class ClickhouseBaseSettings(SibiBaseSettings):
+    """Base settings for ClickHouse connection."""
+    
+    interface: str = "http"
+    host: str = "localhost"
+    port: int = 8123
+    database: str = "default"
+    username: str = "default"
+    password: SecretStr = SecretStr("")
+
+    def to_legacy_dict(self) -> dict[str, Any]:
+        return {
+            "interface": self.interface,
+            "host": self.host,
+            "port": self.port,
+            "database": self.database,
+            "username": self.username,
+            "password": self.password.get_secret_value() if self.password else None,
+        }
 
 
 class RedisBaseSettings(SibiBaseSettings):

sibi_flux/datacube/_data_cube.py

@@ -210,6 +210,7 @@ class Datacube(DfHelper):
         Offloads synchronous validation (CPU bound) to a thread.
         """
         import asyncio
+
         return await asyncio.to_thread(self.validate_data, df)
 
     def get_ddl(self, table_name: Optional[str] = None) -> str: