fix: resharding (#20)

frolvanya · web-flow · commit c946ca860331 · 2025-02-22T09:29:23.000-08:00
* fix: resharding

* chore: silence linting error
diff --git a/src/near_lake_framework/__init__.py b/src/near_lake_framework/__init__.py
@@ -50,7 +50,7 @@ class LakeConfig:
     start_block_height: near_primitives.BlockHeight
     blocks_preload_pool_size: int
 
-    def __init__(  # pylint: disable=too-many-arguments
+    def __init__(  # pylint: disable=too-many-arguments,too-many-positional-arguments
         self,
         network: Network,
         aws_access_key_id: str,
diff --git a/src/near_lake_framework/s3_fetchers.py b/src/near_lake_framework/s3_fetchers.py
@@ -41,8 +41,8 @@ async def fetch_streamer_message(
     block = near_primitives.Block.from_json(body)
 
     shards_fetching = [
-        fetch_shard_or_retry(s3_client, s3_bucket_name, block_height, shard_id)
-        for shard_id in range(len(block.chunks))
+        fetch_shard_or_retry(s3_client, s3_bucket_name, block_height, chunk.shard_id)
+        for chunk in block.chunks
     ]
     shards = await asyncio.gather(*shards_fetching)
 

Original file line number	Diff line number	Diff line change
`@@ -41,8 +41,8 @@ async def fetch_streamer_message(`
`41`	`41`	`block = near_primitives.Block.from_json(body)`
`42`	`42`
`43`	`43`	`shards_fetching = [`
`44`		`- fetch_shard_or_retry(s3_client, s3_bucket_name, block_height, shard_id)`
`45`		`- for shard_id in range(len(block.chunks))`
	`44`	`+ fetch_shard_or_retry(s3_client, s3_bucket_name, block_height, chunk.shard_id)`
	`45`	`+ for chunk in block.chunks`
`46`	`46`	`]`
`47`	`47`	`shards = await asyncio.gather(*shards_fetching)`
`48`	`48`