refactor: improve modularity of transfer_transaction_fungible example (#1023)

Signed-off-by: undefinedIsMyLife <shinetina169@gmail.com>

Author: undefinedIsMyLife

CHANGELOG.md

@@ -7,6 +7,7 @@ This changelog is based on [Keep a Changelog](https://keepachangelog.com/en/1.1.
 ## [Unreleased]
 
 ### Added
+- Modularized `transfer_transaction_fungible` example by introducing `account_balance_query()` & `transfer_transaction()`.Renamed `transfer_tokens()` → `main()`
 - Phase 2 of the inactivity-unassign bot:Automatically detects stale open pull requests (no commit activity for 21+ days), comments with a helpful InactivityBot message, closes the stale PR, and unassigns the contributor from the linked issue.
 - Added **str**() to CustomFixedFee and updated examples and tests accordingly.
 - Added a github template for good first issues

examples/transaction/transfer_transaction_fungible.py

@@ -23,6 +23,9 @@
 load_dotenv()
 network_name = os.getenv('NETWORK', 'testnet').lower()
 
+# --------------------------
+#  CLIENT SETUP
+# --------------------------
 def setup_client():
     """Initialize and set up the client with operator account"""
     network = Network(network_name)
@@ -40,7 +43,9 @@ def setup_client():
         print("❌ Error: Creating client, Please check your .env file")
         sys.exit(1)
 
-
+# --------------------------
+#  ACCOUNT CREATION
+# --------------------------
 def create_account(client, operator_key):
     """Create a new recipient account"""
     print("\nSTEP 1: Creating a new recipient account...")
@@ -59,7 +64,10 @@ def create_account(client, operator_key):
     except Exception as e:
         print(f"Error creating new account: {e}")
         sys.exit(1)
-
+        
+# --------------------------
+#  TOKEN CREATION
+# --------------------------
 def create_token(client, operator_id, operator_key):
     print("\nSTEP 2: Creating a new token...")
     try:
@@ -81,6 +89,9 @@ def create_token(client, operator_id, operator_key):
         print(f"❌ Error creating token: {e}")
         sys.exit(1)
 
+# --------------------------
+#  TOKEN ASSOCIATION
+# --------------------------
 def associate_token(client, recipient_id, recipient_key, token_id):
     print("\nSTEP 3: Associating Token...")
     try:
@@ -96,7 +107,44 @@ def associate_token(client, recipient_id, recipient_key, token_id):
         print(f"❌ Error associating token: {e}")
         sys.exit(1)
 
-def transfer_tokens():
+# --------------------------
+#  ACCOUNT BALANCE QUERY 
+# --------------------------
+def account_balance_query(client, account_id):
+    """Query and return token balances for an account."""
+    try:
+        result = CryptoGetAccountBalanceQuery(account_id=account_id).execute(client)
+        return result.token_balances
+    except Exception as e:
+        print(f"❌ Error fetching account balance: {e}")
+        sys.exit(1)
+
+# --------------------------
+#  TRANSFER TRANSACTION 
+# --------------------------
+def transfer_transaction(client, operator_id, operator_key, recipient_id, token_id):
+    """Execute a token transfer transaction."""
+    print("\nSTEP 4: Transfering Token...")
+    try:
+        tx = (
+            TransferTransaction()
+            .add_token_transfer(token_id, operator_id, -1)
+            .add_token_transfer(token_id, recipient_id, 1)
+            .freeze_with(client)
+            .sign(operator_key)
+        )
+
+        tx.execute(client)
+        print("✅ Success! Token transfer complete.\n")
+
+    except Exception as e:
+        print(f"❌ Error transferring token: {e}")
+        sys.exit(1)
+
+# --------------------------
+#  MAIN ORCHESTRATOR (RENAMED)
+# --------------------------
+def main():
     """
     A full example to create a new recipent account, a fungible token, and
     transfer the token to that account
@@ -113,40 +161,20 @@ def transfer_tokens():
     # Associate Token
     associate_token(client, recipient_id, recipient_key, token_id)
 
-    # Transfer Token
-    print("\nSTEP 4: Transfering Token...")
-    try:
-        # Check balance before transfer
-        balance_before = (
-            CryptoGetAccountBalanceQuery(account_id=recipient_id)
-            .execute(client)
-            .token_balances
-        )
-        print("Token balance before token transfer:")
-        print(f"{token_id}: {balance_before.get(token_id)}")
+    # Balance before transfer
+    balance_before = account_balance_query(client, recipient_id)
+    print("Token balance BEFORE transfer:")
+    print(f"{token_id}: {balance_before.get(token_id)}")
 
-        transfer_tx = (
-            TransferTransaction()
-            .add_token_transfer(token_id, operator_id, -1)
-            .add_token_transfer(token_id, recipient_id, 1)
-            .freeze_with(client)
-            .sign(operator_key)
-        )
-        transfer_tx.execute(client)
+    # Transfer
+    transfer_transaction(client, operator_id, operator_key, recipient_id, token_id)
+
+    # Balance after transfer
+    balance_after = account_balance_query(client, recipient_id)
+    print("Token balance AFTER transfer:")
+    print(f"{token_id}: {balance_after.get(token_id)}")
 
-        print("\n✅ Success! Token transfer complete.\n")
 
-        # Check balance after transfer
-        balance_after = (
-            CryptoGetAccountBalanceQuery(account_id=recipient_id)
-            .execute(client)
-            .token_balances
-        )
-        print("Token balance after token transfer:")
-        print(f"{token_id}: {balance_after.get(token_id)}")
-    except Exception as e:
-        print(f"❌ Error transferring token: {str(e)}")
-        sys.exit(1)
 
 if __name__ == "__main__":
-    transfer_tokens()
+    main()