rawr-tools 0.1.2__py3-none-any.whl

rawr/__init__.py

@@ -0,0 +1,34 @@
+"""RAWR: Risk And Win Ratio — AI Quant Trading Tool."""
+
+from importlib import import_module
+
+from rawr.version import __version__
+
+_STAR_IMPORT_EXPORTS = [
+    "RawrDB",
+    "ApprovalStatus",
+    "TradePlan",
+    "TradePlanRepository",
+    "classify_disclosure",
+    "score_sentiment",
+]
+_OPTIONAL_EXPORTS = [
+    "TradePlanService",
+]
+
+__all__ = [
+    *_STAR_IMPORT_EXPORTS,
+    "__version__",
+]
+
+
+def __getattr__(name: str) -> object:
+    if name == "__version__":
+        return __version__
+    if name in _STAR_IMPORT_EXPORTS or name in _OPTIONAL_EXPORTS:
+        return getattr(import_module("rawr.api"), name)
+    raise AttributeError(f"module {__name__!r} has no attribute {name!r}")
+
+
+def __dir__() -> list[str]:
+    return sorted([*globals(), *__all__])

rawr/api.py

@@ -0,0 +1,39 @@
+"""Supported public Python API for RAWR."""
+
+from importlib import import_module
+from typing import Final
+
+_EXPORTS: Final[dict[str, tuple[str, str]]] = {
+    "RawrDB": ("rawr.core.knowledge.db", "RawrDB"),
+    "TradePlanService": ("rawr.core.plans.service", "TradePlanService"),
+    "ApprovalStatus": ("rawr.core.plans", "ApprovalStatus"),
+    "TradePlan": ("rawr.core.plans", "TradePlan"),
+    "TradePlanRepository": ("rawr.core.plans", "TradePlanRepository"),
+    "classify_disclosure": ("rawr.core.news", "classify_disclosure"),
+    "score_sentiment": ("rawr.core.news", "score_sentiment"),
+}
+_STAR_IMPORT_EXPORTS: Final[tuple[str, ...]] = (
+    "RawrDB",
+    "ApprovalStatus",
+    "TradePlan",
+    "TradePlanRepository",
+    "classify_disclosure",
+    "score_sentiment",
+)
+
+__all__ = list(_STAR_IMPORT_EXPORTS)
+
+
+def __getattr__(name: str) -> object:
+    try:
+        module_name, attr_name = _EXPORTS[name]
+    except KeyError as exc:
+        raise AttributeError(f"module {__name__!r} has no attribute {name!r}") from exc
+
+    value = getattr(import_module(module_name), attr_name)
+    globals()[name] = value
+    return value
+
+
+def __dir__() -> list[str]:
+    return sorted([*globals(), *__all__])