axsdb 0.0.2__py3-none-any.whl

axsdb/__init__.py

@@ -0,0 +1,27 @@
+from .core import (
+    AbsorptionDatabase,
+    CKDAbsorptionDatabase,
+    MonoAbsorptionDatabase,
+)
+from .factory import AbsorptionDatabaseFactory
+from .error import (
+    get_error_handling_config,
+    set_error_handling_config,
+    ErrorHandlingAction,
+    ErrorHandlingPolicy,
+    ErrorHandlingConfiguration,
+)
+from ._version import version as __version__
+
+__all__ = [
+    "AbsorptionDatabase",
+    "AbsorptionDatabaseFactory",
+    "ErrorHandlingAction",
+    "ErrorHandlingConfiguration",
+    "ErrorHandlingPolicy",
+    "CKDAbsorptionDatabase",
+    "MonoAbsorptionDatabase",
+    "get_error_handling_config",
+    "set_error_handling_config",
+    "__version__",
+]

axsdb/_version.py

@@ -0,0 +1,8 @@
+import sys
+
+if sys.version_info >= (3, 8):
+    from importlib import metadata
+else:
+    import importlib_metadata as metadata
+
+version = metadata.version("axsdb")

axsdb/cli.py

@@ -0,0 +1,75 @@
+from __future__ import annotations
+
+import logging
+from enum import Enum
+from typing import Annotated
+
+from pathlib import Path
+import typer
+from rich.logging import RichHandler
+
+logger = logging.getLogger("axsdb")
+app = typer.Typer()
+
+
+class LogLevel(str, Enum):
+    CRITICAL = "CRITICAL"
+    ERROR = "ERROR"
+    WARNING = "WARNING"
+    INFO = "INFO"
+    DEBUG = "DEBUG"
+    NOTSET = "NOTSET"
+
+
+@app.callback()
+def cli(
+    log_level: Annotated[
+        LogLevel, typer.Option(help="Set log level.")
+    ] = LogLevel.WARNING,
+    debug: Annotated[
+        bool,
+        typer.Option(
+            help="Enable debug mode. This will notably print exceptions with locals."
+        ),
+    ] = False,
+):
+    if debug:
+        app.pretty_exceptions_enable = True
+
+    logging.basicConfig(
+        level=log_level.name,
+        format="%(message)s",
+        datefmt="[%X]",
+        handlers=[RichHandler(rich_tracebacks=True)],
+    )
+
+
+@app.command()
+def check(
+    path: Annotated[Path, typer.Argument(help="Path to the checked database.")],
+    mode: Annotated[str, typer.Option("--mode", "-m", help="Database spectral mode.")],
+    fix: Annotated[
+        bool, typer.Option("--fix", "-f", help="Fix issues that can be.")
+    ] = False,
+):
+    """
+    Check data for availability and integrity, optionally fix them.
+    """
+    from axsdb.core import get_absdb_type
+
+    try:
+        cls = get_absdb_type(mode)
+    except ValueError:
+        logger.critical(f"Unsupported mode '{mode}'")
+        exit(1)
+
+    logger.info(f"Opening '{path}'")
+    try:
+        cls.from_directory(path, fix=fix)
+        logger.info("Success!")
+    except FileNotFoundError:
+        pass
+
+
+def main():
+    app()