feat: Added missing fields for AccountCreateTransaction (hiero-ledger#779)

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

Author: manishdait

CHANGELOG.md

@@ -17,7 +17,9 @@ This changelog is based on [Keep a Changelog](https://keepachangelog.com/en/1.1.
 - Add `examples/account_info.py` to demonstrate `AccountInfo` opeartions
 - Added `HbarUnit` class and Extend `Hbar` class to handle floating-point numbers
 - feat: Allow `PrivateKey` to be used for keys in `TopicCreateTransaction` for consistency.
-
+- EvmAddress class
+- `alias`, `staked_account_id`, `staked_node_id` and `decline_staking_reward` fields to AccountCreateTransaction
+- `staked_account_id`, `staked_node_id` and `decline_staking_reward` fields to AccountInfo
 
 ### Changed
 - Refactored token-related example scripts (`token_delete.py`, `token_dissociate.py`, etc.) for improved readability and modularity. [#370]

src/hiero_sdk_python/__init__.py

@@ -15,6 +15,7 @@
 # Crypto
 from .crypto.private_key import PrivateKey
 from .crypto.public_key import PublicKey
+from .crypto.evm_address import EvmAddress
 
 # Tokens
 from .tokens.token_create_transaction import TokenCreateTransaction
@@ -164,6 +165,7 @@
     # Crypto
     "PrivateKey",
     "PublicKey",
+    "EvmAddress",
 
     # Tokens
     "TokenCreateTransaction",

src/hiero_sdk_python/account/account_create_transaction.py

@@ -3,8 +3,11 @@
 """
 
 from typing import Optional, Union
+import warnings
 
+from hiero_sdk_python.account.account_id import AccountId
 from hiero_sdk_python.channels import _Channel
+from hiero_sdk_python.crypto.evm_address import EvmAddress
 from hiero_sdk_python.crypto.public_key import PublicKey
 from hiero_sdk_python.Duration import Duration
 from hiero_sdk_python.executable import _Method
@@ -31,7 +34,11 @@ def __init__(
         receiver_signature_required: Optional[bool] = None,
         auto_renew_period: Optional[Duration] = AUTO_RENEW_PERIOD,
         memo: Optional[str] = None,
-        max_automatic_token_associations: Optional[int] = 0 
+        max_automatic_token_associations: Optional[int] = 0,
+        alias: Optional[EvmAddress] = None,
+        staked_account_id: Optional[AccountId] = None,
+        staked_node_id: Optional[int] = None,
+        decline_staking_reward: Optional[bool] = False
     ) -> None:
         """
         Initializes a new AccountCreateTransaction instance with default values
@@ -43,6 +50,13 @@ def __init__(
             receiver_signature_required (Optional[bool]): Whether receiver signature is required.
             auto_renew_period (Duration): Auto-renew period in seconds (default is ~90 days).
             memo (Optional[str]): Memo for the account.
+            max_automatic_token_associations (Optional[int]): The maximum number of tokens that 
+                can be auto-associated.
+            alias (Optional[EvmAddress]): The 20-byte EVM address to be used as the account's alias.
+            staked_account_id (Optional[AccountId]): The account to which this account will stake.
+            staked_node_id (Optional[int]): ID of the node this account is staked to.
+            decline_staking_reward (Optional[bool]): If true, the account declines receiving a 
+                staking reward (default is False).
         """
         super().__init__()
         self.key: Optional[PublicKey] = key
@@ -52,6 +66,10 @@ def __init__(
         self.account_memo: Optional[str] = memo
         self.max_automatic_token_associations: Optional[int] = max_automatic_token_associations
         self._default_transaction_fee = DEFAULT_TRANSACTION_FEE
+        self.alias: Optional[EvmAddress] = alias
+        self.staked_account_id: Optional[AccountId] = staked_account_id
+        self.staked_node_id: Optional[int] = staked_node_id
+        self.decline_staking_reward = decline_staking_reward
 
     def set_key(self, key: PublicKey) -> "AccountCreateTransaction":
         """
@@ -63,8 +81,50 @@ def set_key(self, key: PublicKey) -> "AccountCreateTransaction":
         Returns:
             AccountCreateTransaction: The current transaction instance for method chaining.
         """
+        warnings.warn(
+            "The 'set_key' method is deprecated, Use `set_key_without_alias` instead.",
+            DeprecationWarning,
+        )
+        self._require_not_frozen()
+        self.key = key
+        return self
+
+    def set_key_without_alias(self, key: PublicKey) -> "AccountCreateTransaction":
+        """
+        Sets the public key for the new account without alias.
+
+        Args:
+            key (PublicKey): The public key to assign to the account.
+
+        Returns:
+            AccountCreateTransaction: The current transaction instance for method chaining.
+        """
+        self._require_not_frozen()
+        self.key = key
+        return self
+
+    def set_key_with_alias(
+        self,
+        key: PublicKey,
+        ecdsa_key: Optional[PublicKey]=None
+    ) -> "AccountCreateTransaction":
+        """
+        Sets the public key for the new account and assigns an alias derived from an ECDSA key.
+        
+        If `ecdsa_key` is provided, its corresponding EVM address will be used as the account alias.
+        Otherwise, the alias will be derived from the provided `key`.
+
+        Args:
+            key (PublicKey): The public key to assign to the account.
+            ecdsa_key (Optional[PublicKey]): An optional ECDSA public key used 
+                to derive the account alias.
+
+        Returns:
+            AccountCreateTransaction: The current transaction instance to allow method chaining.
+        """
         self._require_not_frozen()
         self.key = key
+        self.alias = ecdsa_key.to_evm_address() if ecdsa_key is not None else key.to_evm_address()
         return self
 
     def set_initial_balance(self, balance: Union[Hbar, int]) -> "AccountCreateTransaction":
@@ -133,15 +193,111 @@ def set_account_memo(self, memo: str) -> "AccountCreateTransaction":
         return self
 
     def set_max_automatic_token_associations(self, max_assoc: int) -> "AccountCreateTransaction":
-        """Sets the maximum number of automatic token associations for the account."""
+        """
+        Sets the maximum number of automatic token associations for the account.
+
+        Args:
+            max_assoc (int): The maximum number of automatic 
+                token associations to allow (default 0).
+
+        Returns:
+            AccountCreateTransaction: The current transaction instance for method chaining.
+        """
         self._require_not_frozen()
         # FIX
         if max_assoc < -1:
             raise ValueError("max_automatic_token_associations must be -1 (unlimited) or a non-negative integer.")
         self.max_automatic_token_associations = max_assoc
         return self
 
-    def _build_proto_body(self):
+    def set_alias(self, alias_evm_address: Union[EvmAddress, str]) -> "AccountCreateTransaction":
+        """
+        Sets the EVM Address alias for the account.
+
+        Args:
+            alias_evm_address (Union[EvmAddress, str]): The 20-byte EVM address to 
+                be used as the account's alias.
+
+        Returns:
+            AccountCreateTransaction: The current transaction instance for method chaining.
+        """
+        self._require_not_frozen()
+        if isinstance(alias_evm_address, str):
+            if len(alias_evm_address.removeprefix("0x")) == 40:
+                self.alias = EvmAddress.from_string(alias_evm_address)
+            else:
+                raise ValueError("alias_evm_address must be a valid 20-byte EVM address")
+
+        elif isinstance(alias_evm_address, EvmAddress):
+            self.alias = alias_evm_address
+
+        else:
+            raise TypeError("alias_evm_address must be of type str or EvmAddress")
+
+        return self
+
+    def set_staked_account_id(
+        self,
+        account_id: Union[AccountId, str]
+    ) -> "AccountCreateTransaction":
+        """
+        Sets the staked account id for the account.
+
+        Args:
+            account_id (Union[AccountId, str]): The account to which this account will stake.
+
+        Returns:
+            AccountCreateTransaction: The current transaction instance for method chaining.
+        """
+        self._require_not_frozen()
+        if isinstance(account_id, str):
+            self.staked_account_id = AccountId.from_string(account_id)
+        elif isinstance(account_id, AccountId):
+            self.staked_account_id = account_id
+        else:
+            raise TypeError("account_id must be of type str or AccountId")
+
+        return self
+
+    def set_staked_node_id(self, node_id: int) -> "AccountCreateTransaction":
+        """
+        Sets the staked node id for the account.
+
+        Args:
+            node_id (int): The node to which this account will stake.
+
+        Returns:
+            AccountCreateTransaction: The current transaction instance for method chaining.
+        """
+        self._require_not_frozen()
+        if not isinstance(node_id, int):
+            raise TypeError("node_id must be of type int")
+
+        self.staked_node_id = node_id
+        return self
+
+    def set_decline_staking_reward(
+        self,
+        decline_staking_reward: bool
+    ) -> "AccountCreateTransaction":
+        """
+        Sets the decline staking reward for the account.
+
+        Args:
+            decline_staking_reward (bool): If true, the account declines 
+            receiving a staking reward (default is False)
+
+        Returns:
+            AccountCreateTransaction: The current transaction instance for method chaining.
+        """
+        self._require_not_frozen()
+        if not isinstance(decline_staking_reward, bool):
+            raise TypeError("decline_staking_reward must be of type bool")
+
+        self.decline_staking_reward = decline_staking_reward
+        return self
+
+    def _build_proto_body(self) -> crypto_create_pb2.CryptoCreateTransactionBody:
         """
         Returns the protobuf body for the account create transaction.
 
@@ -162,15 +318,24 @@ def _build_proto_body(self):
         else:
             raise TypeError("initial_balance must be Hbar or int (tinybars).")
 
-        return crypto_create_pb2.CryptoCreateTransactionBody(
+        proto_body = crypto_create_pb2.CryptoCreateTransactionBody(
             key=self.key._to_proto(),
             initialBalance=initial_balance_tinybars,
             receiverSigRequired=self.receiver_signature_required,
             autoRenewPeriod=duration_pb2.Duration(seconds=self.auto_renew_period.seconds),
             memo=self.account_memo,
-            max_automatic_token_associations=self.max_automatic_token_associations
+            max_automatic_token_associations=self.max_automatic_token_associations,
+            alias=self.alias.address_bytes if self.alias else None,
+            decline_reward=self.decline_staking_reward
         )
 
+        if self.staked_account_id:
+            proto_body.staked_account_id.CopyFrom(self.staked_account_id._to_proto())
+        elif self.staked_node_id:
+            proto_body.staked_node_id = self.staked_node_id
+
+        return proto_body
+
     def build_transaction_body(self) -> transaction_pb2.TransactionBody:
         """
         Builds and returns the protobuf transaction body for account creation.

src/hiero_sdk_python/account/account_info.py

@@ -9,6 +9,7 @@
 from hiero_sdk_python.account.account_id import AccountId
 from hiero_sdk_python.crypto.public_key import PublicKey
 from hiero_sdk_python.Duration import Duration
+from hiero_sdk_python.hapi.services.basic_types_pb2 import StakingInfo
 from hiero_sdk_python.hapi.services.crypto_get_info_pb2 import CryptoGetInfoResponse
 from hiero_sdk_python.hbar import Hbar
 from hiero_sdk_python.timestamp import Timestamp
@@ -37,6 +38,9 @@ class AccountInfo:
             associated with this account.
         account_memo (Optional[str]): The memo associated with this account.
         owned_nfts (Optional[int]): The number of NFTs owned by this account.
+        staked_account_id (Optional[AccountId]): The account to which this account is staked.
+        staked_node_id (Optional[int]): The node to which this account is staked.
+        decline_staking_reward (bool): Whether this account declines receiving staking rewards. 
     """
 
     account_id: Optional[AccountId] = None
@@ -52,6 +56,9 @@ class AccountInfo:
     account_memo: Optional[str] = None
     owned_nfts: Optional[int] = None
     max_automatic_token_associations: Optional[int] = None
+    staked_account_id: Optional[AccountId] = None
+    staked_node_id: Optional[int] = None
+    decline_staking_reward: Optional[bool] = None
 
     @classmethod
     def _from_proto(cls, proto: CryptoGetInfoResponse.AccountInfo) -> "AccountInfo":
@@ -72,7 +79,7 @@ def _from_proto(cls, proto: CryptoGetInfoResponse.AccountInfo) -> "AccountInfo":
         if proto is None:
             raise ValueError("Account info proto is None")
 
-        return cls(
+        account_info: "AccountInfo" = cls(
             account_id=AccountId._from_proto(proto.accountID) if proto.accountID else None,
             contract_account_id=proto.contractAccountID,
             is_deleted=proto.deleted,
@@ -95,6 +102,21 @@ def _from_proto(cls, proto: CryptoGetInfoResponse.AccountInfo) -> "AccountInfo":
             max_automatic_token_associations=proto.max_automatic_token_associations,
         )
 
+        staking_info = proto.staking_info if proto.HasField('staking_info') else None
+
+        if staking_info:
+            account_info.staked_account_id = (
+                AccountId._from_proto(staking_info.staked_account_id) 
+                if staking_info.HasField('staked_account_id') else None
+            )
+            account_info.staked_node_id = (
+                staking_info.staked_node_id 
+                if staking_info.HasField('staked_node_id') else None
+            )
+            account_info.decline_staking_reward = staking_info.decline_reward
+
+        return account_info
+
     def _to_proto(self) -> CryptoGetInfoResponse.AccountInfo:
         """Converts this AccountInfo object to its protobuf representation.
         Serializes this `AccountInfo` instance into a
@@ -125,4 +147,9 @@ def _to_proto(self) -> CryptoGetInfoResponse.AccountInfo:
             memo=self.account_memo,
             ownedNfts=self.owned_nfts,
             max_automatic_token_associations=self.max_automatic_token_associations,
+            staking_info=StakingInfo(
+                staked_account_id=self.staked_account_id._to_proto() if self.staked_account_id else None,
+                staked_node_id=self.staked_node_id if self.staked_node_id else None,
+                decline_reward=self.decline_staking_reward
+            ),
         )

src/hiero_sdk_python/crypto/evm_address.py

@@ -0,0 +1,55 @@
+class EvmAddress:
+    """
+    Represents a 20-byte EVM address derived from the rightmost 20 bytes of 
+    32 byte Keccak-256 hash of an ECDSA public key.
+    """
+    def __init__(self, address_bytes: bytes) -> None:
+        """
+        Initialize an EvmAddress instance from bytes.
+        
+        Args:
+        address_bytes (bytes): A 20-byte sequence representing the EVM address.
+        """
+        if len(address_bytes) != 20:
+            raise ValueError("EvmAddress must be exactly 20 bytes long.")
+
+        self.address_bytes: bytes = address_bytes
+
+    @classmethod
+    def from_string(cls, evm_address: str) -> "EvmAddress":
+        """
+        Create an EvmAddress from a hex string (with or without '0x' prefix).
+        """
+        if not isinstance(evm_address, str):
+            raise TypeError("evm_address must be a of type string.")
+
+        address = evm_address[2:] if evm_address.startswith('0x') else evm_address
+
+        if len(address) == 40:
+            return cls(address_bytes=bytes.fromhex(address))
+
+        raise ValueError("Invalid hex string for evm_address.")
+
+    @classmethod
+    def from_bytes(cls, address_bytes: "bytes") -> "EvmAddress":
+        """Create an EvmAddress from raw bytes."""
+        return cls(address_bytes)
+
+    def to_string(self) -> str:
+        """Return the EVM address as a hex string"""
+        return bytes.hex(self.address_bytes)
+
+    def __str__(self) -> str:
+        return self.to_string()
+
+    def __repr__(self) -> str:
+        return f"<EvmAddress hex={self.to_string()}>"
+
+    def __eq__(self, obj: object) -> bool:
+        if not isinstance(obj, EvmAddress):
+            return False
+
+        return self.address_bytes == obj.address_bytes
+
+    def __hash__(self) -> int:
+        return hash(self.address_bytes)