Skip to content

Commit

Permalink
add nanorpc as library, include requests dependency
Browse files Browse the repository at this point in the history
  • Loading branch information
gr0vity-dev committed Apr 16, 2024
1 parent 3ff06b2 commit 03e11aa
Show file tree
Hide file tree
Showing 9 changed files with 21 additions and 523 deletions.
3 changes: 3 additions & 0 deletions .gitignore
Original file line number Diff line number Diff line change
Expand Up @@ -14,3 +14,6 @@ venv_py/
**/._.DS_Store
debug
unit_tests/testcases
snippets/
testcases/
nl_config.toml
12 changes: 7 additions & 5 deletions nanolab/publisher/block_asserts.py
Original file line number Diff line number Diff line change
@@ -1,5 +1,5 @@
from abc import ABC, abstractmethod
from nanolab.src.nano_rpc import NanoRpcV2
from nanorpc.client import NanoRpcTyped
from nanolab.publisher.block_event import BlockConfirmationEvent
from nanolab.src.utils import get_config_parser
import asyncio
Expand Down Expand Up @@ -44,8 +44,9 @@ def __init__(self, event_bus=None, node_name=None):
self.event_bus = event_bus
conf_p = get_config_parser()
node_name = node_name if node_name else conf_p.get_nodes_name()[:-1]
self.nano_rpc_default = NanoRpcV2(conf_p.get_node_rpc(node_name))
self.nano_rpc_all = [NanoRpcV2(url) for url in conf_p.get_nodes_rpc()]
self.nano_rpc_default = NanoRpcTyped(conf_p.get_node_rpc(node_name))
self.nano_rpc_all = [NanoRpcTyped(url)
for url in conf_p.get_nodes_rpc()]

async def _fetch_block_info(self, block_hash):
return await self.nano_rpc_default.block_info(block_hash,
Expand Down Expand Up @@ -101,7 +102,8 @@ async def assert_blocks_confirmed(self, block_hashes):

async def _publish_block_confirmation_event(self, wait_s, block_hash,
timeout):
if not self.event_bus: return
if not self.event_bus:
return
event = BlockConfirmationEvent(block_hash, timeout, wait_s)
await self.event_bus.publish('block_confirmed', event)

Expand Down Expand Up @@ -135,4 +137,4 @@ async def assert_single_block_confirmed(self, block_hash: str):

async def assert_single_block_confirmed_wait(self, block_hash, wait_s,
interval):
await self.assert_blocks_confirmed_wait([block_hash], wait_s, interval)
await self.assert_blocks_confirmed_wait([block_hash], wait_s, interval)
5 changes: 2 additions & 3 deletions nanolab/publisher/block_generator.py
Original file line number Diff line number Diff line change
@@ -1,4 +1,3 @@
#from nanolab.src.nano_rpc_v2 import NanoRpcV2
from nanomock.modules.nl_rpc import NanoRpc, NanoLibTools
from nanolab.src.utils import get_config_parser
from abc import ABC, abstractmethod
Expand Down Expand Up @@ -67,7 +66,7 @@ def set_broadcast_blocks(self, broadcast):
def create_send_and_open_block(self, send_amount_raw, source_seed,
source_index, destination_seed,
destination_index, representative):
#destination = self.nano_rpc_default.generate_account(destination_seed, destination_index)
# destination = self.nano_rpc_default.generate_account(destination_seed, destination_index)
return self.blockgen_single_account_opener(
representative=representative,
source_seed=source_seed,
Expand Down Expand Up @@ -121,4 +120,4 @@ def blockgen_single_account_opener(
open_block["account_data"]["source_index"] = destination_index

res = [send_block, open_block]
return res
return res
10 changes: 5 additions & 5 deletions nanolab/publisher/confirmation_stats.py
Original file line number Diff line number Diff line change
@@ -1,4 +1,4 @@
from nanolab.src.nano_rpc import NanoRpcV2
from nanorpc.client import NanoRpcTyped
from nanolab.src.utils import get_config_parser
from time import strftime, gmtime, time
from math import ceil
Expand All @@ -12,16 +12,16 @@ def __init__(self, block_timeout_s):
self.formatter = ConfirmationTableFormatter()

async def initialize(self, node_name):
self.rpc_v2 = NanoRpcV2(get_config_parser().get_node_rpc(node_name))
self.rpc_v2.create_session()
start_block_count = await self.rpc_v2.block_count()
self.nanorpc = NanoRpcTyped(
get_config_parser().get_node_rpc(node_name))
start_block_count = await self.nanorpc.block_count()
self.calculator.set_start_block_count(start_block_count)

def set_end_block_count(self, end_block_count: dict) -> None:
self.calculator.set_end_block_count(end_block_count)

async def print_stats(self) -> None:
end_block_count = await self.rpc_v2.block_count()
end_block_count = await self.nanorpc.block_count()
self.calculator.set_end_block_count(end_block_count)
stats = self.calculator.compute_stats()
self.printer.print_stats(stats, self.formatter)
Expand Down
1 change: 0 additions & 1 deletion nanolab/publisher/test_case.py
Original file line number Diff line number Diff line change
@@ -1,5 +1,4 @@
from abc import ABC, abstractmethod
from nanolab.src.nano_rpc import NanoRpcV2
from nanolab.publisher.block_asserts import BlockAsserts
from nanolab.publisher.block_generator import BlockGenerator
from nanolab.publisher.confirmation_stats import ConfirmationStatsManager
Expand Down
Loading

0 comments on commit 03e11aa

Please sign in to comment.