updated code structure

config.py

@@ -1,3 +1,9 @@
+"""
+Configuration for the pump.fun trading bot.
+"""
+
+from typing import Final
+
 import os
 
 from dotenv import load_dotenv
@@ -6,23 +12,33 @@ from solders.pubkey import Pubkey
 load_dotenv()
 
 # System & pump.fun addresses
-PUMP_PROGRAM = Pubkey.from_string("6EF8rrecthR5Dkzon8Nwu78hRvfCKubJ14M5uBEwF6P")
-PUMP_GLOBAL = Pubkey.from_string("4wTV1YmiEkRvAtNtsSGPtUrqRYQMe5SKy2uB4Jjaxnjf")
-PUMP_EVENT_AUTHORITY = Pubkey.from_string(
+PUMP_PROGRAM: Final[Pubkey] = Pubkey.from_string(
+    "6EF8rrecthR5Dkzon8Nwu78hRvfCKubJ14M5uBEwF6P"
+)
+PUMP_GLOBAL: Final[Pubkey] = Pubkey.from_string(
+    "4wTV1YmiEkRvAtNtsSGPtUrqRYQMe5SKy2uB4Jjaxnjf"
+)
+PUMP_EVENT_AUTHORITY: Final[Pubkey] = Pubkey.from_string(
     "Ce6TQqeHC9p8KetsN6JsjHK7UTZk7nasjjnr7XxXp9F1"
 )
-PUMP_FEE = Pubkey.from_string("CebN5WGQ4jvEPvsVU4EoHEpgzq1VV7AbicfhtW4xC9iM")
-PUMP_LIQUIDITY_MIGRATOR = Pubkey.from_string(
+PUMP_FEE: Final[Pubkey] = Pubkey.from_string(
+    "CebN5WGQ4jvEPvsVU4EoHEpgzq1VV7AbicfhtW4xC9iM"
+)
+PUMP_LIQUIDITY_MIGRATOR: Final[Pubkey] = Pubkey.from_string(
     "39azUYFWPz3VHgKCf3VChUwbpURdCHRxjWVowf5jUJjg"
 )
-SYSTEM_PROGRAM = Pubkey.from_string("11111111111111111111111111111111")
-SYSTEM_TOKEN_PROGRAM = Pubkey.from_string("TokenkegQfeZyiNwAJbNbGKPFXCWuBvf9Ss623VQ5DA")
-SYSTEM_ASSOCIATED_TOKEN_ACCOUNT_PROGRAM = Pubkey.from_string(
+SYSTEM_PROGRAM: Final[Pubkey] = Pubkey.from_string("11111111111111111111111111111111")
+SYSTEM_TOKEN_PROGRAM: Final[Pubkey] = Pubkey.from_string(
+    "TokenkegQfeZyiNwAJbNbGKPFXCWuBvf9Ss623VQ5DA"
+)
+SYSTEM_ASSOCIATED_TOKEN_ACCOUNT_PROGRAM: Final[Pubkey] = Pubkey.from_string(
     "ATokenGPvbdGVxr1b2hvZbsiqW5xWH25efTNsLJA8knL"
 )
-SYSTEM_RENT = Pubkey.from_string("SysvarRent111111111111111111111111111111111")
-SOL = Pubkey.from_string("So11111111111111111111111111111111111111112")
-LAMPORTS_PER_SOL = 1_000_000_000
+SYSTEM_RENT: Final[Pubkey] = Pubkey.from_string(
+    "SysvarRent111111111111111111111111111111111"
+)
+SOL: Final[Pubkey] = Pubkey.from_string("So11111111111111111111111111111111111111112")
+LAMPORTS_PER_SOL: Final[int] = 1_000_000_000
 
 # Trading parameters
 BUY_AMOUNT = 0.0001  # Amount of SOL to spend when buying
@@ -33,11 +49,12 @@ ENABLE_DYNAMIC_PRIORITY_FEE = (
 )
 EXTRA_PRIORITY_FEE = 0.1  # 10% increase in dynamic priority fee
 
-# Your nodes
-# You can also get a trader node https://docs.chainstack.com/docs/solana-trader-nodes
-RPC_ENDPOINT = os.getenv("RPC_ENDPOINT")
-WSS_ENDPOINT = os.getenv("WSS_ENDPOINT")
-MAX_RPS = 25  # RPS of your node to avoid 429 errors
+TOKEN_DECIMALS: Final[int] = 6
+MAX_RETRIES: int = 5
+WAIT_TIME_AFTER_BUY: int = 20
+WAIT_TIME_BEFORE_NEW_TOKEN: int = 5
+WAIT_TIME_AFTER_CREATION: int = 15
 
-# Private key
-PRIVATE_KEY = os.getenv("PRIVATE_KEY")
+# RPS of your node to avoid rate limit errors
+# You can also get a trader node https://docs.chainstack.com/docs/solana-trader-nodes
+MAX_RPS = 25