Update components/middleware.py

components/middleware.py

@@ -40,6 +40,52 @@ def SecretNonceGenerator():
 
 
 
+
+
+
+def TokenCheck(walletPublicKey, api_key, required_mint, mint_amount):
+
+    url = f"https://mainnet.helius-rpc.com/?api-key={api_key}"
+    payload = {
+        "jsonrpc": "2.0",
+        "id": "1",
+        "method": "getTokenAccountsByOwner",
+        "params": [
+            walletPublicKey,
+            {"mint": required_mint},
+            {"encoding": "jsonParsed"}
+        ]
+    }
+
+    try:
+        response = requests.post(url, json=payload)
+        response.raise_for_status()
+        data = response.json()
+    except:
+        return False  # always boolean
+
+    token_accounts = data.get("result", {}).get("value", [])
+    if not token_accounts:
+        return False
+
+    try:
+        token_amount = token_accounts[0]["account"]["data"]["parsed"]["info"]["tokenAmount"]
+
+        # reliable extraction
+        if token_amount["uiAmount"] is not None:
+            ui_amount = float(token_amount["uiAmount"])
+        else:
+            ui_amount = float(token_amount["uiAmountString"])
+
+    except:
+        return False  # ALWAYS return strict boolean
+
+    # compare as float
+    return ui_amount >= float(mint_amount)
+
+
+
+'''
 def TokenCheck(walletPublicKey, api_key, required_mint,mint_amount):
     
         url = f"https://mainnet.helius-rpc.com/?api-key={api_key}"
@@ -93,7 +139,7 @@ def TokenCheck(walletPublicKey, api_key, required_mint,mint_amount):
             return False
 
 
-
+'''