chore: clean up token airdrop example for pylint compliance (#1094)

Signed-off-by: prishajaiswal75 <prisha.jaiswal.btech2023@sitpune.edu.in>

Author: prishajaiswal75

CHANGELOG.md

@@ -49,6 +49,7 @@ This changelog is based on [Keep a Changelog](https://keepachangelog.com/en/1.1.
 - Renamed workflow files for consistent PR check naming:
   `examples.yml` → `pr-check-examples.yml`,
   `test.yml` → `pr-check-test.yml` (#1043)
+- Cleaned up `token_airdrop_claim_auto` example for pylint compliance (no functional changes). (#1079)
 
 
 ### Fixed

examples/tokens/token_airdrop_claim_auto.py

@@ -1,7 +1,8 @@
 """
 Hedera Token Airdrop Example Script
 
-This script demonstrates and end-to-end example for an account to automatically (no user action required) claim a set of airdrops.
+This script demonstrates an end-to-end example for an account to
+automatically (no user action required) claim a set of airdrops.
 
 Unique configurations of this account:
 - 10 auto-association slots.
@@ -19,12 +20,17 @@
 uv run examples/tokens/token_airdrop_claim_auto.py
 python examples/tokens/token_airdrop_claim_auto.py
 """
+# pylint: disable=import-error,
+# pylint: disable=too-many-arguments,
+# pylint: disable=too-many-positional-arguments,
+# pylint: disable=protected-access,
+# pylint: disable=broad-exception-caught
+
 import os
 import sys
 from typing import Iterable
 from dotenv import load_dotenv
-from hiero_sdk_python import (
-    Client,
+from hiero_sdk_python import (Client,
     Network,
     AccountId,
     PrivateKey,
@@ -45,6 +51,7 @@
 load_dotenv()
 
 def setup_client():
+    """Set up and return a Hedera client using environment configuration."""
     network_name = os.getenv("NETWORK", "testnet")
 
     # Validate environment variables
@@ -63,7 +70,9 @@ def setup_client():
         print(f"Client set up with operator id {client.operator_account_id}")
 
     except Exception as e:
-        raise ConnectionError(f"Error initializing client: {e}")
+        raise ConnectionError(
+            f"Error initializing client: {e}"
+        ) from e
 
     print(f"✅ Connected to Hedera {network_name} network as operator: {operator_id}")
     return client, operator_id, operator_key
@@ -73,6 +82,8 @@ def create_receiver(
         signature_required: bool =False,
         max_auto_assoc: int =10
     ):
+
+    """Create and return a configured Hedera client."""
     receiver_key = PrivateKey.generate()
     receiver_public_key = receiver_key.public_key()
 
@@ -110,8 +121,9 @@ def create_fungible_token(
         initial_supply: int =50,
         max_supply: int = 1000,
     ):
+    """Create and return a fungible token on the Hedera network."""
     try:
-        receipt = ( 
+        receipt = (
             TokenCreateTransaction()
             .set_token_name(name)
             .set_token_symbol(symbol)
@@ -143,8 +155,9 @@ def create_nft_token(
         symbol: str ="MNT",
         max_supply: int = 100
     ):
+    """Create and return a non-fungible (NFT) token on the Hedera network."""
     try:
-        receipt = ( 
+        receipt = (
             TokenCreateTransaction()
             .set_token_name(name)
             .set_token_symbol(symbol)
@@ -174,8 +187,9 @@ def mint_nft_token(
         operator_key: PrivateKey,
         nft_token_id: TokenId,
     ):
+    """Mint a new NFT for the given NFT token and return its serial number."""
     try:
-        receipt = ( 
+        receipt = (
             TokenMintTransaction()
             .set_token_id(nft_token_id)
             .set_metadata([b"NFT Metadata Example"])
@@ -189,8 +203,8 @@ def mint_nft_token(
         if receipt.status != ResponseCode.SUCCESS:
             status_message = ResponseCode(receipt.status).name
             raise RuntimeError(f"❌ NFT token mint failed: {status_message}")
-
-        print(f"✅ NFT {nft_token_id} serial {serial} minted with NFT id of {nft_id}. Total NFT supply is {total_supply} ")
+        print(f"✅ NFT {nft_token_id} serial {serial} minted with NFT id of {nft_id}.")
+        print(f"Total NFT supply is {total_supply}")
         return nft_id
     except Exception as e:
         raise RuntimeError(f"❌ Error minting NFT token: {e}") from e
@@ -202,16 +216,25 @@ def log_balances(
     nft_ids: Iterable[NftId],
     prefix: str = ""
 ):
+    """Fetch and log token balances for operator and receiver accounts."""
     print(f"\n===== {prefix} Balances =====")
 
     try:
-        operator_balance = CryptoGetAccountBalanceQuery().set_account_id(operator_id).execute(client)
-        receiver_balance = CryptoGetAccountBalanceQuery().set_account_id(receiver_id).execute(client)
-    except Exception as e:
+        operator_balance = (
+            CryptoGetAccountBalanceQuery()
+            .set_account_id(operator_id)
+            .execute(client)
+        )
+        receiver_balance = (
+            CryptoGetAccountBalanceQuery()
+            .set_account_id(receiver_id)
+            .execute(client)
+        )
+    except Exception as e: # pylint: disable=broad-exception-caught
         print(f"❌ Failed to fetch balances: {e}")
         return
 
-    def log_fungible(account_id: AccountId, balances: dict, token_ids: Iterable[TokenId]):
+    def log_fungible(_account_id: AccountId, balances: dict, token_ids: Iterable[TokenId]):
         print("  Fungible tokens:")
         for token_id in token_ids:
             print(f"    {token_id}: {balances.get(token_id, 0)}")
@@ -249,16 +272,23 @@ def perform_airdrop(
         receiver_id: AccountId,
         fungible_ids: Iterable[TokenId],
         nft_ids: Iterable[NftId],
-        ft_amount: int = 100
+        ft_amount: int = 100,
     ):
-
+    """Perform a token airdrop from operator to receiver."""
     try:
         tx = TokenAirdropTransaction()
 
         for fungible_id in fungible_ids:
             tx.add_token_transfer(fungible_id, operator_id, -ft_amount)
             tx.add_token_transfer(fungible_id, receiver_id, ft_amount)
-            print(f"📤 Transferring {ft_amount} of fungible token {fungible_id} from {operator_id} → {receiver_id}")
+
+            message = (
+                f"📤 Transferring {ft_amount} of fungible token {fungible_id} "
+                f"from {operator_id} → {receiver_id}"
+            )
+            print(message)
+
+
 
         for nft_id in nft_ids:
             tx.add_nft_transfer(nft_id, operator_id, receiver_id)
@@ -278,15 +308,24 @@ def perform_airdrop(
         raise RuntimeError("Airdrop execution failed") from e
 
 def main():
+    """Run the token airdrop auto-claim example workflow."""
     # Set up client and return client, operator_id, operator_key
     client, operator_id, operator_key = setup_client()
 
-    # Create and return a fungible token to airdrop 
+    # Create and return a fungible token to airdrop
     print("Create 50 fungible tokens and 1 NFT to airdrop")
-    fungible_id = create_fungible_token(client, operator_id, operator_key, name="My Fungible Token", symbol="123", initial_supply=50, max_supply = 2000)
-
-    # Create and return an nft token to airdrop 
-    nft_token_id = create_nft_token(client, operator_id, operator_key, name="My NFT Token", symbol = "MNFT", max_supply=1000)
+    fungible_id = create_fungible_token(client, operator_id,
+                                        operator_key,
+                                        name="My Fungible Token",
+                                        symbol="123", initial_supply=50,
+                                        max_supply = 2000
+                                    )
+
+    # Create and return an nft token to airdrop
+    nft_token_id = create_nft_token(client, operator_id,
+                                    operator_key, name="My NFT Token",
+                                    symbol = "MNFT", max_supply=1000
+                                )
 
     # Mint and return an nft to airdrop
     nft_serial = mint_nft_token(client, operator_key, nft_token_id)
@@ -300,16 +339,38 @@ def main():
 
     # Check pre-airdrop balances
     print("\n🔍 Verifying sender has tokens to airdrop and receiver neither:")
-    log_balances(client, operator_id, receiver_id, [fungible_id], [nft_serial], prefix="Before airdrop")
+    log_balances(client, operator_id, receiver_id, [fungible_id], [nft_serial],
+                prefix="Before airdrop")
 
     # Initiate airdrop of 20 fungible tokens and 1 nft token id
-    perform_airdrop(client, operator_id, operator_key, receiver_id, [fungible_id], [nft_serial], 20)
-
-    print("\n🔍 Verifying receiver has received airdrop contents automatically and sender has sent:")
-    log_balances(client, operator_id, receiver_id, [fungible_id], [nft_serial], prefix="After airdrop")
-
-    print("✅ Auto-association successful: Receiver accepted airdropped tokens without pre-association.")
-    print("✅ Airdrop successful: Receiver accepted new fungible tokens without pre-association.")
+    perform_airdrop(client,
+                    operator_id,
+                    operator_key,
+                    receiver_id,
+                    [fungible_id],
+                    [nft_serial],
+                    20
+                )
+
+    print("\n🔍 Verifying receiver has received airdrop contents automatically"
+        "and sender has sent:"
+    )
+    log_balances(
+    client,
+    operator_id,
+    receiver_id,
+    [fungible_id],
+    [nft_serial],
+    prefix="After airdrop",
+    )
+
+
+    print("✅ Auto-association successful: Receiver accepted airdropped tokens "
+        "without pre-association."
+    )
+    print("✅ Airdrop successful: Receiver accepted new fungible tokens "
+        "without pre-association."
+    )
 
 if __name__ == "__main__":
     main()