chore: added type var to chunked transaction

Signed-off-by: Manish Dait <daitmanish88@gmail.com>

Author: manishdait

src/hiero_sdk_python/transaction/chunked_transaction.py

@@ -1,7 +1,7 @@
 from __future__ import annotations
 
 from abc import ABC, abstractmethod
-from typing import Literal, overload
+from typing import Literal, TypeVar, overload
 
 from hiero_sdk_python.client.client import Client
 from hiero_sdk_python.transaction.transaction import Transaction
@@ -10,6 +10,9 @@
 from hiero_sdk_python.transaction.transaction_response import TransactionResponse
 
 
+T = TypeVar("T", bound="ChunkedTransaction")
+
+
 class ChunkedTransaction(Transaction, ABC):
     """
     Abstract base class for transactions that support chunking.
@@ -36,7 +39,7 @@ def __init__(self) -> None:
         self.max_chunks: int = 20
 
     @abstractmethod
-    def _build_proto_body(self):
+    def _build_proto_body(self: T):
         """
         Builds the protobuf body for the current chunk.
 
@@ -52,7 +55,7 @@ def _build_proto_body(self):
         """
         pass
 
-    def set_chunk_size(self, chunk_size: int) -> ChunkedTransaction:
+    def set_chunk_size(self: T, chunk_size: int) -> T:
         """
         Sets the chunk size for this transaction.
 
@@ -73,7 +76,7 @@ def set_chunk_size(self, chunk_size: int) -> ChunkedTransaction:
         self._total_chunks = self.get_required_chunks()
         return self
 
-    def set_max_chunks(self, max_chunks: int) -> ChunkedTransaction:
+    def set_max_chunks(self: T, max_chunks: int) -> T:
         """
         Sets the maximum number of chunks allowed.
 
@@ -93,7 +96,7 @@ def set_max_chunks(self, max_chunks: int) -> ChunkedTransaction:
         self.max_chunks = max_chunks
         return self
 
-    def _validate_chunking(self) -> None:
+    def _validate_chunking(self: T) -> None:
         """
         Validates that the transaction doesn't exceed the maximum number of chunks.
 
@@ -106,7 +109,7 @@ def _validate_chunking(self) -> None:
                 f"Required: {self.get_required_chunks()} Increase limit with set_max_chunks()."
             )
 
-    def freeze_with(self, client: Client) -> ChunkedTransaction:
+    def freeze_with(self: T, client: Client) -> T:
         """
         Freezes the transaction by building transaction bodies for all chunks.
 
@@ -156,7 +159,7 @@ def freeze_with(self, client: Client) -> ChunkedTransaction:
 
     @overload
     def execute(
-        self,
+        self: T,
         client: Client,
         timeout: int | float | None = None,
         wait_for_receipt: Literal[True] = True,
@@ -165,15 +168,15 @@ def execute(
 
     @overload
     def execute(
-        self,
+        self: T,
         client: Client,
         timeout: int | float | None = None,
         wait_for_receipt: Literal[False] = False,
         validate_status: bool = False,
     ) -> TransactionResponse: ...
 
     def execute(
-        self,
+        self: T,
         client: Client,
         timeout: int | float | None = None,
         wait_for_receipt: bool = True,
@@ -200,7 +203,7 @@ def execute(
 
     @overload
     def execute_all(
-        self,
+        self: T,
         client: Client,
         timeout: int | float | None = None,
         wait_for_receipt: Literal[True] = True,
@@ -209,15 +212,15 @@ def execute_all(
 
     @overload
     def execute_all(
-        self,
+        self: T,
         client: Client,
         timeout: int | float | None = None,
         wait_for_receipt: Literal[False] = False,
         validate_status: bool = False,
     ) -> list[TransactionResponse]: ...
 
     def execute_all(
-        self,
+        self: T,
         client: Client,
         timeout: int | float | None = None,
         wait_for_receipt: bool = True,
@@ -258,7 +261,7 @@ def execute_all(
         return responses
 
     @property
-    def body_size_all_chunks(self) -> list[int]:
+    def body_size_all_chunks(self: T) -> list[int]:
         """
         Returns an array of body sizes for each chunk in the transaction.