tonutils 2.0.1b2__py3-none-any.whl → 2.0.1b4__py3-none-any.whl

tonutils/__init__.py

@@ -1,7 +1,6 @@
 from . import (
     clients,
     contracts,
-    protocols,
     exceptions,
     types,
     utils,
@@ -12,7 +11,6 @@ __all__ = [
     "__version__",
     "clients",
     "contracts",
-    "protocols",
     "exceptions",
     "types",
     "utils",

tonutils/__meta__.py

@@ -1 +1 @@
-__version__ = "2.0.1b2"
+__version__ = "2.0.1b4"

tonutils/cli.py

@@ -0,0 +1,111 @@
+import argparse
+import asyncio
+import typing as t
+
+from tonutils.__meta__ import __version__
+from tonutils.clients.adnl.provider.config import (
+    get_mainnet_global_config,
+    get_testnet_global_config,
+    load_global_config,
+)
+from tonutils.tools.status_monitor import LiteServerMonitor
+from tonutils.types import NetworkGlobalID
+
+NETWORK_MAP: t.Dict[str, NetworkGlobalID] = {
+    "mainnet": NetworkGlobalID.MAINNET,
+    "testnet": NetworkGlobalID.TESTNET,
+}
+
+
+def parse_network(value: str) -> NetworkGlobalID:
+    value = value.lower().strip()
+    if value not in NETWORK_MAP:
+        raise argparse.ArgumentTypeError(f"Unknown network: {value}")
+    return NETWORK_MAP[value]
+
+
+def cmd_status(args: argparse.Namespace) -> None:
+    if args.config:
+        config = load_global_config(args.config)
+    else:
+        config_getter = {
+            NetworkGlobalID.MAINNET: get_mainnet_global_config,
+            NetworkGlobalID.TESTNET: get_testnet_global_config,
+        }
+        config = config_getter[args.network]()
+
+    async def _run() -> None:
+        monitor = LiteServerMonitor.from_config(
+            config=config,
+            network=args.network,
+            rps_limit=args.rps,
+        )
+        try:
+            await monitor.run()
+        finally:
+            await monitor.stop()
+
+    try:
+        asyncio.run(_run())
+    except KeyboardInterrupt:
+        pass
+
+
+def _create_parser() -> argparse.ArgumentParser:
+    parser = argparse.ArgumentParser(
+        prog="tonutils",
+        description="Tonutils CLI.",
+    )
+    parser.add_argument(
+        "-v",
+        "--version",
+        action="version",
+        version=f"tonutils {__version__}",
+    )
+
+    subs = parser.add_subparsers(dest="command", metavar="command")
+
+    status = subs.add_parser(
+        "status",
+        help="Monitor lite-servers status.",
+    )
+    status.add_argument(
+        "-n",
+        "--network",
+        type=parse_network,
+        metavar="NET",
+        default=NetworkGlobalID.MAINNET,
+        help="mainnet (default) or testnet",
+    )
+    status.add_argument(
+        "-c",
+        "--config",
+        type=str,
+        metavar="PATH",
+        help="Config file path or URL",
+    )
+    status.add_argument(
+        "-r",
+        "--rps",
+        type=int,
+        metavar="N",
+        default=100,
+        help="Requests per second (default: 100)",
+    )
+    status.set_defaults(func=cmd_status)
+
+    return parser
+
+
+def main() -> None:
+    parser = _create_parser()
+    args = parser.parse_args()
+
+    if hasattr(args, "func"):
+        args.func(args)
+    else:
+        parser.print_help()
+
+
+if __name__ == "__main__":
+    main()

tonutils/clients/__init__.py

@@ -1,13 +1,13 @@
-from .adnl import AdnlBalancer, AdnlClient, AdnlProvider
-from .base import BaseClient
+from .adnl import (
+    LiteBalancer,
+    LiteClient,
+    AdnlProvider,
+)
 from .http import (
     HttpBalancer,
     ChainstackHttpClient,
-    ChainstackHttpProvider,
     QuicknodeHttpClient,
-    QuicknodeHttpProvider,
     TatumHttpClient,
-    TatumHttpProvider,
     TonapiHttpClient,
     TonapiHttpProvider,
     ToncenterHttpClient,
@@ -15,17 +15,13 @@ from .http import (
 )
 
 __all__ = [
-    "BaseClient",
-    "AdnlBalancer",
-    "AdnlClient",
+    "LiteBalancer",
+    "LiteClient",
     "AdnlProvider",
     "HttpBalancer",
     "ChainstackHttpClient",
-    "ChainstackHttpProvider",
     "QuicknodeHttpClient",
-    "QuicknodeHttpProvider",
     "TatumHttpClient",
-    "TatumHttpProvider",
     "TonapiHttpClient",
     "TonapiHttpProvider",
     "ToncenterHttpClient",

tonutils/clients/adnl/__init__.py

@@ -1,5 +1,9 @@
-from .balancer import AdnlBalancer
-from .client import AdnlClient
+from .balancer import LiteBalancer
+from .client import LiteClient
 from .provider import AdnlProvider
 
-__all__ = ["AdnlBalancer", "AdnlClient", "AdnlProvider"]
+__all__ = [
+    "LiteBalancer",
+    "LiteClient",
+    "AdnlProvider",
+]