From e24eca5bf0747a2a31b49220e304c1f01068a2de Mon Sep 17 00:00:00 2001 From: Jonathan Zernik Date: Thu, 7 Jan 2021 23:17:52 -0800 Subject: [PATCH] Use namedtuple for blockinfo and network sync result (#577) --- squeaknode/bitcoin/block_info.py | 6 ++---- squeaknode/sync/network_task.py | 8 +++----- 2 files changed, 5 insertions(+), 9 deletions(-) diff --git a/squeaknode/bitcoin/block_info.py b/squeaknode/bitcoin/block_info.py index 1585b3b9..6af213bc 100644 --- a/squeaknode/bitcoin/block_info.py +++ b/squeaknode/bitcoin/block_info.py @@ -1,10 +1,8 @@ -from dataclasses import dataclass +from typing import NamedTuple -@dataclass -class BlockInfo: +class BlockInfo(NamedTuple): """Class for getting block info from blockchain.""" - block_height: int block_hash: bytes block_header: bytes diff --git a/squeaknode/sync/network_task.py b/squeaknode/sync/network_task.py index 272e97a0..64ecdbec 100644 --- a/squeaknode/sync/network_task.py +++ b/squeaknode/sync/network_task.py @@ -1,24 +1,22 @@ import logging import queue import threading -from dataclasses import dataclass from typing import Any from typing import List +from typing import NamedTuple from squeaknode.sync.network_sync import NetworkSync logger = logging.getLogger(__name__) -@dataclass -class PeerSyncResult: +class PeerSyncResult(NamedTuple): completed_peer_id: Any = None failed_peer_id: Any = None timeout: Any = None -@dataclass -class NetworkSyncResult: +class NetworkSyncResult(NamedTuple): completed_peer_ids: List[int] failed_peer_ids: List[int] timeout_peer_ids: List[int]