Listing available nodes

Author: nieznanysprawiciel

examples/low-level-api/list-nodes.py

@@ -0,0 +1,72 @@
+#!/usr/bin/env python3
+import asyncio
+from asyncio import TimeoutError
+from datetime import datetime, timezone
+import json
+import sys
+import pathlib
+from typing import AsyncIterator, List
+
+from yapapi import props as yp
+from yapapi.log import enable_default_logger
+from yapapi.props.builder import DemandBuilder
+from yapapi.rest import Configuration, Market, Activity, Payment  # noqa
+from yapapi.rest.market import OfferProposal
+from examples import utils
+
+examples_dir = pathlib.Path(__file__).resolve().parent.parent
+sys.path.append(str(examples_dir))
+
+
+class NodeInfo(object):
+    def __init__(self, offer: OfferProposal):
+        self.node_id = offer.issuer
+        self.node_name = offer.props["golem.node.id.name"]
+
+
+async def list_offers(conf: Configuration, subnet_tag: str) -> AsyncIterator[OfferProposal]:
+    async with conf.market() as client:
+        market_api = Market(client)
+        dbuild = DemandBuilder()
+        dbuild.add(yp.NodeInfo(name="Scanning Node", subnet_tag=subnet_tag))
+        dbuild.add(yp.Activity(expiration=datetime.now(timezone.utc)))
+
+        async with market_api.subscribe(dbuild.properties, dbuild.constraints) as subscription:
+            async for event in subscription.events():
+                yield event
+
+
+async def list_nodes(conf: Configuration, subnet_tag: str) -> AsyncIterator[List[NodeInfo]]:
+    async for offer in list_offers(conf, subnet_tag):
+        yield NodeInfo(offer)
+
+
+async def print_nodes(conf: Configuration, subnet_tag: str):
+    async for node in list_nodes(conf, subnet_tag=subnet_tag):
+        print(f"{node.node_id}    {node.node_name}")
+
+
+def main():
+    parser = utils.build_parser("List Nodes")
+    args = parser.parse_args()
+
+    subnet = args.subnet_tag
+    sys.stderr.write(f"Using subnet: {utils.TEXT_COLOR_YELLOW}{subnet}{utils.TEXT_COLOR_DEFAULT}\n")
+
+    enable_default_logger()
+    try:
+        asyncio.get_event_loop().run_until_complete(
+            asyncio.wait_for(
+                print_nodes(
+                    Configuration(),
+                    subnet_tag=subnet,
+                ),
+                timeout=4,
+            )
+        )
+    except TimeoutError:
+        pass
+
+
+if __name__ == "__main__":
+    main()