ruff

Author: richardkiss

chia/consensus/block_store_protocol.py

@@ -42,7 +42,9 @@ async def is_fully_compactified(self, header_hash: bytes32) -> Optional[bool]: .
     async def replace_proof(self, header_hash: bytes32, block: FullBlock) -> None: ...
     async def count_compactified_blocks(self) -> int: ...
     async def count_uncompactified_blocks(self) -> int: ...
-    async def get_block_records_close_to_peak(self, blocks_n: int) -> tuple[dict[bytes32, BlockRecord], Optional[bytes32]]: ...
+    async def get_block_records_close_to_peak(
+        self, blocks_n: int
+    ) -> tuple[dict[bytes32, BlockRecord], Optional[bytes32]]: ...
     def get_host_parameter_limit(self) -> int: ...
     async def get_prev_hash(self, header_hash: bytes32) -> bytes32: ...
-    def transaction(self) -> AbstractAsyncContextManager[None]: ...
+    def transaction(self) -> AbstractAsyncContextManager[None]: ...

chia/full_node/block_store.py

@@ -3,6 +3,7 @@
 import dataclasses
 import logging
 import sqlite3
+from contextlib import AbstractAsyncContextManager
 from typing import Optional
 
 import typing_extensions
@@ -396,25 +397,6 @@ async def get_block_records_by_hash(self, header_hashes: list[bytes32]) -> list[
             ret.append(all_blocks[hh])
         return ret
 
-    async def get_prev_hash(self, header_hash: bytes32) -> bytes32:
-        """
-        Returns the header hash preceeding the input header hash.
-        Throws an exception if the block is not present
-        """
-        cached = self.block_cache.get(header_hash)
-        if cached is not None:
-            return cached.prev_header_hash
-
-        async with self.db_wrapper.reader_no_transaction() as conn:
-            async with conn.execute(
-                "SELECT prev_hash FROM full_blocks WHERE header_hash=?",
-                (header_hash,),
-            ) as cursor:
-                row = await cursor.fetchone()
-                if row is None:
-                    raise KeyError("missing block in chain")
-                return bytes32(row[0])
-
     async def get_block_bytes_by_hash(self, header_hashes: list[bytes32]) -> list[bytes]:
         """
         Returns a list of Full Blocks block blobs, ordered by the same order in which header_hashes are passed in.
@@ -545,30 +527,6 @@ async def get_peak(self) -> Optional[tuple[bytes32, uint32]]:
             return None
         return bytes32(peak_row[0]), uint32(peak_height[0])
 
-    async def get_block_records_close_to_peak(
-        self, blocks_n: int
-    ) -> tuple[dict[bytes32, BlockRecord], Optional[bytes32]]:
-        """
-        Returns a dictionary with all blocks that have height >= peak height - blocks_n, as well as the
-        peak header hash. Only blocks that are part of the main chain/current peak are included.
-        """
-
-        peak = await self.get_peak()
-        if peak is None:
-            return {}, None
-
-        ret: dict[bytes32, BlockRecord] = {}
-        async with self.db_wrapper.reader_no_transaction() as conn:
-            async with conn.execute(
-                "SELECT header_hash, block_record FROM full_blocks WHERE height >= ? AND in_main_chain=1",
-                (peak[1] - blocks_n,),
-            ) as cursor:
-                for row in await cursor.fetchall():
-                    header_hash = bytes32(row[0])
-                    ret[header_hash] = BlockRecord.from_bytes(row[1])
-
-        return ret, peak[0]
-
     async def set_peak(self, header_hash: bytes32) -> None:
         # We need to be in a sqlite transaction here.
         # Note: we do not commit this to the database yet, as we need to also change the coin store