Fix mypy type checking errors and add stricter type annotations (#165)

Signed-off-by: Anirudh Sengar <[email protected]>

Author: anirudhsengar

src/hiero_sdk_python/executable.py

@@ -1,13 +1,13 @@
 from os import error
 import time
-import typing
+from typing import Callable, Optional, Any, TYPE_CHECKING
 import grpc
 from abc import ABC, abstractmethod
 from enum import IntEnum
 
 from hiero_sdk_python.channels import _Channel
 from hiero_sdk_python.exceptions import MaxAttemptsError
-if typing.TYPE_CHECKING:
+if TYPE_CHECKING:
     from hiero_sdk_python.client.client import Client
 
 # Default values for retry and backoff configuration in miliseconds
@@ -28,8 +28,8 @@ class _Method:
 
     def __init__(
         self,
-        query_func: typing.Callable = None,
-        transaction_func: typing.Callable = None,
+        query_func: Callable = None,
+        transaction_func: Optional[Callable[..., Any]] = None,
     ):
         """
         Initialize a Method instance with the appropriate callable functions.

src/hiero_sdk_python/query/account_balance_query.py

@@ -1,3 +1,4 @@
+from typing import Optional, Any
 from hiero_sdk_python.query.query import Query
 from hiero_sdk_python.hapi.services import crypto_get_account_balance_pb2, query_pb2
 from hiero_sdk_python.account.account_id import AccountId
@@ -14,15 +15,15 @@ class CryptoGetAccountBalanceQuery(Query):
     including hbars and tokens.
     """
 
-    def __init__(self, account_id: AccountId = None) -> None:
+    def __init__(self, account_id: Optional[AccountId] = None) -> None:
         """
         Initializes a new instance of the CryptoGetAccountBalanceQuery class.
 
         Args:
             account_id (AccountId, optional): The ID of the account to retrieve the balance for.
         """
         super().__init__()
-        self.account_id = account_id
+        self.account_id: Optional[AccountId] = account_id
 
     def set_account_id(self, account_id: AccountId) -> "CryptoGetAccountBalanceQuery":
         """
@@ -112,7 +113,7 @@ def execute(self, client) -> AccountBalance:
 
         return AccountBalance._from_proto(response.cryptogetAccountBalance)
 
-    def _get_query_response(self, response: any) -> crypto_get_account_balance_pb2.CryptoGetAccountBalanceResponse:
+    def _get_query_response(self, response: Any) -> crypto_get_account_balance_pb2.CryptoGetAccountBalanceResponse:
         """
         Extracts the account balance response from the full response.
         

src/hiero_sdk_python/query/query.py

@@ -1,8 +1,8 @@
 import time
 
-from typing import List, Optional
+from typing import List, Optional, Any, Union
 
-from hiero_sdk_python.exceptions import PrecheckError
+from hiero_sdk_python.exceptions import PrecheckError, ReceiptStatusError
 from hiero_sdk_python.executable import _Method
 from hiero_sdk_python.channels import _Channel
 from hiero_sdk_python.hapi.services import query_header_pb2, query_pb2
@@ -49,15 +49,15 @@ def __init__(self) -> None:
         self.node_index: int = 0
         self.payment_amount: Optional[Hbar] = None
 
-    def _get_query_response(self, response: any) -> query_pb2.Query:
+    def _get_query_response(self, response: Any) -> query_pb2.Query:
         """
         Extracts the query-specific response object from the full response.
         
         Subclasses must implement this method to properly extract their
         specific response object.
         
         Args:
-            response (any): The full response from the network
+            response (Any): The full response from the network
             
         Returns:
             The query-specific response object
@@ -257,7 +257,7 @@ def _make_request(self) -> query_pb2.Query:
         """
         raise NotImplementedError("_make_request must be implemented by subclasses.")
 
-    def _map_response(self, response: any, node_id: int, proto_request: query_pb2.Query) -> query_pb2.Query:
+    def _map_response(self, response: Any, node_id: int, proto_request: query_pb2.Query) -> query_pb2.Query:
         """
         Maps the network response to the appropriate response object.
         
@@ -271,7 +271,7 @@ def _map_response(self, response: any, node_id: int, proto_request: query_pb2.Qu
         """
         return response
 
-    def _should_retry(self, response: any) -> _ExecutionState:
+    def _should_retry(self, response: Any) -> _ExecutionState:
         """
         Determines whether the query should be retried based on the response.
         
@@ -300,7 +300,7 @@ def _should_retry(self, response: any) -> _ExecutionState:
         else:
             return _ExecutionState.ERROR
 
-    def _map_status_error(self, response: any) -> PrecheckError:
+    def _map_status_error(self, response: Any) -> Union[PrecheckError,ReceiptStatusError]:
         """
         Maps a response status code to an appropriate error object.
         

src/hiero_sdk_python/query/token_info_query.py

@@ -1,3 +1,4 @@
+from typing import Optional
 from hiero_sdk_python.query.query import Query
 from hiero_sdk_python.hapi.services import query_pb2, token_get_info_pb2, response_pb2
 from hiero_sdk_python.executable import _Method
@@ -16,15 +17,15 @@ class TokenInfoQuery(Query):
     including the token's properties and settings.
     
     """
-    def __init__(self, token_id: TokenId = None) -> None:
+    def __init__(self, token_id: Optional[TokenId] = None) -> None:
         """
         Initializes a new TokenInfoQuery instance with an optional token_id.
 
         Args:
             token_id (TokenId, optional): The ID of the token to query.
         """
         super().__init__()
-        self.token_id: TokenId = token_id
+        self.token_id: Optional[TokenId] = token_id
 
     def set_token_id(self, token_id: TokenId) -> "TokenInfoQuery":
         """

src/hiero_sdk_python/query/token_nft_info_query.py

@@ -1,4 +1,4 @@
-from typing import Optional
+from typing import Optional, Any
 from hiero_sdk_python.query.query import Query
 from hiero_sdk_python.hapi.services import query_pb2, response_pb2, token_get_nft_info_pb2
 from hiero_sdk_python.executable import _Method

src/hiero_sdk_python/query/topic_info_query.py

@@ -1,4 +1,4 @@
-from typing import Optional
+from typing import Optional, Any
 from hiero_sdk_python.query.query import Query
 from hiero_sdk_python.hapi.services import query_pb2, consensus_get_topic_info_pb2, response_pb2
 from hiero_sdk_python.client.client import Client
@@ -121,7 +121,7 @@ def _get_method(self, channel: _Channel) -> _Method:
             query_func=channel.topic.getTopicInfo
         )
 
-    def _should_retry(self, response: any) -> _ExecutionState:
+    def _should_retry(self, response: Any) -> _ExecutionState:
         """
         Determines whether the query should be retried based on the response.
         
@@ -174,7 +174,7 @@ def execute(self, client: Client) -> TopicInfo:
 
         return TopicInfo._from_proto(response.consensusGetTopicInfo.topicInfo)
 
-    def _get_query_response(self, response: any) -> consensus_get_topic_info_pb2.ConsensusGetTopicInfoResponse:
+    def _get_query_response(self, response: Any) -> consensus_get_topic_info_pb2.ConsensusGetTopicInfoResponse:
         """
         Extracts the topic info response from the full response.
         

src/hiero_sdk_python/query/topic_message_query.py

@@ -1,7 +1,7 @@
 import time
 import threading
 from datetime import datetime
-from typing import Optional, Callable, Union, Dict, List
+from typing import Optional, Callable, Union, Dict, List, Any
 
 from hiero_sdk_python.hapi.mirror import consensus_service_pb2 as mirror_proto
 from hiero_sdk_python.hapi.services import basic_types_pb2, timestamp_pb2
@@ -30,15 +30,15 @@ def __init__(
         """
         Initializes a TopicMessageQuery.
         """
-        self._topic_id: TopicId = self._parse_topic_id(topic_id) if topic_id else None
-        self._start_time: timestamp_pb2.Timestamp = self._parse_timestamp(start_time) if start_time else None
-        self._end_time: timestamp_pb2.Timestamp = self._parse_timestamp(end_time) if end_time else None
-        self._limit: int = limit
+        self._topic_id: Optional[TopicId] = self._parse_topic_id(topic_id) if topic_id else None
+        self._start_time: Optional[timestamp_pb2.Timestamp] = self._parse_timestamp(start_time) if start_time else None
+        self._end_time: Optional[timestamp_pb2.Timestamp] = self._parse_timestamp(end_time) if end_time else None
+        self._limit: Optional[int] = limit
         self._chunking_enabled: bool = chunking_enabled
         self._completion_handler: Optional[Callable[[], None]] = None
 
         self._max_attempts: int = 10
-        self._max_backoff: int = 8.0
+        self._max_backoff: float = 8.0
 
     def set_max_attempts(self, attempts: int) -> "TopicMessageQuery":
         """Sets the maximum number of attempts to reconnect on failure."""

src/hiero_sdk_python/query/transaction_get_receipt_query.py

@@ -1,4 +1,4 @@
-from typing import Optional,Union
+from typing import Optional,Union,Any
 
 from hiero_sdk_python.client.client import Client
 from hiero_sdk_python.exceptions import PrecheckError, ReceiptStatusError
@@ -129,7 +129,7 @@ def _get_method(self, channel: _Channel) -> _Method:
             query_func=channel.crypto.getTransactionReceipts
         )
 
-    def _should_retry(self, response: any) -> _ExecutionState:
+    def _should_retry(self, response: Any) -> _ExecutionState:
         """
         Determines whether the query should be retried based on the response.
         
@@ -167,7 +167,7 @@ def _should_retry(self, response: any) -> _ExecutionState:
         else:
             return _ExecutionState.FINISHED
 
-    def _map_status_error(self, response: any) -> Union[PrecheckError,ReceiptStatusError]:
+    def _map_status_error(self, response: Any) -> Union[PrecheckError,ReceiptStatusError]:
         """
         Maps a response status code to an appropriate error object.
         

src/hiero_sdk_python/query/transaction_record_query.py

@@ -1,3 +1,4 @@
+from typing import Optional, Any, Union
 from hiero_sdk_python.hapi.services import query_header_pb2, transaction_get_record_pb2, query_pb2
 from hiero_sdk_python.query.query import Query
 from hiero_sdk_python.response_code import ResponseCode
@@ -15,12 +16,12 @@ class TransactionRecordQuery(Query):
     Represents a query for a transaction record on the Hedera network.
     """
 
-    def __init__(self, transaction_id: TransactionId = None):
+    def __init__(self, transaction_id: Optional[TransactionId] = None):
         """
         Initializes the TransactionRecordQuery with the provided transaction ID.
         """
         super().__init__()
-        self.transaction_id : TransactionId = transaction_id
+        self.transaction_id : Optional[TransactionId] = transaction_id
 
     def set_transaction_id(self, transaction_id: TransactionId):
         """
@@ -86,7 +87,7 @@ def _get_method(self, channel: _Channel) -> _Method:
             query_func=channel.crypto.getTxRecordByTxID
         )
 
-    def _should_retry(self, response):
+    def _should_retry(self, response: Any) -> _ExecutionState:
         """
         Determines whether the query should be retried based on the response.
         
@@ -128,7 +129,7 @@ def _should_retry(self, response):
         else:
             return _ExecutionState.ERROR
 
-    def _map_status_error(self, response):
+    def _map_status_error(self, response: Any) -> Union[PrecheckError,ReceiptStatusError]:
         """
         Maps a response status code to an appropriate error object.
         
@@ -158,7 +159,7 @@ def _map_status_error(self, response):
         receipt = response.transactionGetRecord.transactionRecord.receipt
 
         return ReceiptStatusError(status, self.transaction_id, TransactionReceipt._from_proto(receipt))
-        
+     
     def execute(self, client):
         """
         Executes the transaction record query.
@@ -184,7 +185,7 @@ def execute(self, client):
 
         return TransactionRecord._from_proto(response.transactionGetRecord.transactionRecord, self.transaction_id)
 
-    def _get_query_response(self, response):
+    def _get_query_response(self, response: Any):
         """
         Extracts the transaction record response from the full response.