add missing fields for AccountUpdateTransaction

Author: emiliyank

CHANGELOG.md

@@ -41,7 +41,6 @@ This changelog is based on [Keep a Changelog](https://keepachangelog.com/en/1.1.
 - 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
-- Added `examples/token_create_transaction_supply_key.py` to demonstrate token creation with and without a supply key.
 - Added `examples/token_create_transaction_kyc_key.py` to demonstrate KYC key functionality, including creating tokens with/without KYC keys, granting/revoking KYC status, and understanding KYC requirements for token transfers.
 - Add `set_token_ids`, `_from_proto`, `_validate_checksum` to TokenAssociateTransaction [#795]
 - Added BatchTransaction class
@@ -52,6 +51,7 @@ This changelog is based on [Keep a Changelog](https://keepachangelog.com/en/1.1.
 - Added `docs\sdk_developers\training\receipts.md` as a training guide for users to understand hedera receipts.
 - Add `set_token_ids`, `_from_proto`, `_validate_checksum` to TokenAssociateTransaction [#795]
 - docs: added `network_and_client.md` with a table of contents, and added external example scripts (`client.py`).
+- Add `max_automatic_token_associations`, `staked_account_id`, `staked_node_id` and `decline_staking_reward` fields to `AccountUpdateTransaction` (#801)
 
 ### Changed
 

src/hiero_sdk_python/account/account_update_transaction.py

@@ -6,7 +6,7 @@
 from dataclasses import dataclass
 from typing import Optional
 
-from google.protobuf.wrappers_pb2 import BoolValue, StringValue
+from google.protobuf.wrappers_pb2 import BoolValue, Int32Value, StringValue
 
 from hiero_sdk_python.account.account_id import AccountId
 from hiero_sdk_python.channels import _Channel
@@ -35,6 +35,14 @@ class AccountUpdateParams:
         account_memo (Optional[str]): The new memo for the account.
         receiver_signature_required (Optional[bool]): Whether receiver signature is required.
         expiration_time (Optional[Timestamp]): The new expiration time for the account.
+        max_automatic_token_associations (Optional[int]): The maximum number of tokens that
+            can be auto-associated with this account. Use -1 for unlimited, 0 for none.
+        staked_account_id (Optional[AccountId]): The account to which this account is staking
+            its balances. Mutually exclusive with staked_node_id.
+        staked_node_id (Optional[int]): The node ID to which this account is staking
+            its balances. Mutually exclusive with staked_account_id.
+        decline_staking_reward (Optional[bool]): If true, the account declines receiving
+            staking rewards.
     """
 
     account_id: Optional[AccountId] = None
@@ -43,6 +51,10 @@ class AccountUpdateParams:
     account_memo: Optional[str] = None
     receiver_signature_required: Optional[bool] = None
     expiration_time: Optional[Timestamp] = 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
 
 
 class AccountUpdateTransaction(Transaction):
@@ -70,6 +82,10 @@ def __init__(self, account_params: Optional[AccountUpdateParams] = None):
         self.account_memo = params.account_memo
         self.receiver_signature_required = params.receiver_signature_required
         self.expiration_time = params.expiration_time
+        self.max_automatic_token_associations = params.max_automatic_token_associations
+        self.staked_account_id = params.staked_account_id
+        self.staked_node_id = params.staked_node_id
+        self.decline_staking_reward = params.decline_staking_reward
 
     def set_account_id(self, account_id: Optional[AccountId]) -> "AccountUpdateTransaction":
         """
@@ -161,6 +177,121 @@ def set_expiration_time(
         self.expiration_time = expiration_time
         return self
 
+    def set_max_automatic_token_associations(
+        self, max_automatic_token_associations: Optional[int]
+    ) -> "AccountUpdateTransaction":
+        """
+        Sets the maximum number of tokens that can be auto-associated with this account.
+
+        Args:
+            max_automatic_token_associations (Optional[int]): The maximum number of tokens
+                that can be auto-associated. Use -1 for unlimited, 0 for none.
+                Must be >= -1.
+
+        Returns:
+            AccountUpdateTransaction: This transaction instance.
+
+        Raises:
+            ValueError: If max_automatic_token_associations is less than -1.
+        """
+        self._require_not_frozen()
+        if max_automatic_token_associations is not None and max_automatic_token_associations < -1:
+            raise ValueError(
+                "max_automatic_token_associations must be -1 (unlimited) or a non-negative integer."
+            )
+        self.max_automatic_token_associations = max_automatic_token_associations
+        return self
+
+    def set_staked_account_id(
+        self, staked_account_id: Optional[AccountId]
+    ) -> "AccountUpdateTransaction":
+        """
+        Sets the account to which this account is staking its balances.
+
+        This field is mutually exclusive with staked_node_id. Setting this will
+        clear any previously set staked_node_id. Passing ``None`` (or calling
+        :func:`clear_staked_account_id`) removes staking and sends the sentinel
+        AccountId (0.0.0) to the network.
+
+        Args:
+            staked_account_id (Optional[AccountId]): The account to which this account
+                will stake its balances. ``None`` clears the staking configuration.
+
+        Returns:
+            AccountUpdateTransaction: This transaction instance.
+        """
+        self._require_not_frozen()
+        if staked_account_id is None:
+            return self.clear_staked_account_id()
+        self.staked_account_id = staked_account_id
+        self.staked_node_id = None  # Clear the other field in the oneOf
+        return self
+
+    def set_staked_node_id(
+        self, staked_node_id: Optional[int]
+    ) -> "AccountUpdateTransaction":
+        """
+        Sets the node ID to which this account is staking its balances.
+
+        This field is mutually exclusive with staked_account_id. Setting this will
+        clear any previously set staked_account_id. Passing ``None`` (or calling
+        :func:`clear_staked_node_id`) removes staking and sends the sentinel value (-1).
+
+        Args:
+            staked_node_id (Optional[int]): The node ID to which this account will stake
+                its balances. ``None`` clears the staking configuration.
+
+        Returns:
+            AccountUpdateTransaction: This transaction instance.
+        """
+        self._require_not_frozen()
+        if staked_node_id is None:
+            return self.clear_staked_node_id()
+        self.staked_node_id = staked_node_id
+        self.staked_account_id = None  # Clear the other field in the oneOf
+        return self
+
+    def clear_staked_account_id(self) -> "AccountUpdateTransaction":
+        """
+        Clears staking to an account by setting the sentinel AccountId (0.0.0).
+
+        Returns:
+            AccountUpdateTransaction: This transaction instance.
+        """
+        self._require_not_frozen()
+        self.staked_account_id = AccountId(0, 0, 0)
+        self.staked_node_id = None
+        return self
+
+    def clear_staked_node_id(self) -> "AccountUpdateTransaction":
+        """
+        Clears staking to a node by setting the sentinel node ID (-1).
+
+        Returns:
+            AccountUpdateTransaction: This transaction instance.
+        """
+        self._require_not_frozen()
+        self.staked_node_id = -1
+        self.staked_account_id = None
+        return self
+
+    def set_decline_staking_reward(
+        self, decline_staking_reward: Optional[bool]
+    ) -> "AccountUpdateTransaction":
+        """
+        Sets whether the account declines receiving staking rewards.
+
+        Args:
+            decline_staking_reward (Optional[bool]): If True, the account declines receiving
+                staking rewards. If False or None, the account will receive rewards.
+
+        Returns:
+            AccountUpdateTransaction: This transaction instance.
+        """
+        self._require_not_frozen()
+        self.decline_staking_reward = decline_staking_reward
+        return self
+
     def _build_proto_body(self):
         """
         Returns the protobuf body for the account update transaction.
@@ -174,7 +305,7 @@ def _build_proto_body(self):
         if self.account_id is None:
             raise ValueError("Missing required AccountID to update")
 
-        return CryptoUpdateTransactionBody(
+        proto_body = CryptoUpdateTransactionBody(
             accountIDToUpdate=self.account_id._to_proto(),
             key=self.key._to_proto() if self.key else None,
             memo=StringValue(value=self.account_memo) if self.account_memo is not None else None,
@@ -187,8 +318,26 @@ def _build_proto_body(self):
                 if self.receiver_signature_required is not None
                 else None
             ),
+            max_automatic_token_associations=(
+                Int32Value(value=self.max_automatic_token_associations)
+                if self.max_automatic_token_associations is not None
+                else None
+            ),
+            decline_reward=(
+                BoolValue(value=self.decline_staking_reward)
+                if self.decline_staking_reward is not None
+                else None
+            ),
         )
 
+        # Handle staked_id oneOf: only one can be set
+        if self.staked_account_id is not None:
+            proto_body.staked_account_id.CopyFrom(self.staked_account_id._to_proto())
+        elif self.staked_node_id is not None:
+            proto_body.staked_node_id = self.staked_node_id
+
+        return proto_body
+
     def build_transaction_body(self):
         """
         Builds the transaction body for this account update transaction.
@@ -227,3 +376,4 @@ def _get_method(self, channel: _Channel) -> _Method:
             _Method: An object containing the transaction function to update an account.
         """
         return _Method(transaction_func=channel.crypto.updateAccount, query_func=None)
+