Update components/middleware.py

components/middleware.py

@@ -42,61 +42,23 @@ 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):
+
+
+def TokenCheck(walletPublicKey):
     
-        url = f"https://mainnet.helius-rpc.com/?api-key={api_key}"
+        url = f"https://mainnet.helius-rpc.com/?api-key=4e833ada-d32c-48c5-b020-c11b2253f25b"
         payload = {
             "jsonrpc": "2.0",
             "id": "1",
             "method": "getTokenAccountsByOwner",
             "params": [
                 walletPublicKey,
-                {"mint": required_mint},
+                {"mint":"3d4XyPWkUJzruF5c2qc1QfpLgsaNaDLMtTya1bWBpump" },
                 {"encoding": "jsonParsed"}
             ]
         }
@@ -130,59 +92,18 @@ def TokenCheck(walletPublicKey, api_key, required_mint,mint_amount):
 
             return {"status": False, "message": "Could not parse token account data"}
 
-        if ui_amount >= mint_amount:
+        if ui_amount >= 100000.0:
             
             print(ui_amount)
             return True
             
         else:
-            
+            print(ui_amount)
             return False
 
 
-'''
-
-
-
-
-
-
-async def TokenCheck(address):
-    
-    url = "https://mainnet.helius-rpc.com/?api-key=4e833ada-d32c-48c5-b020-c11b2253f25b"
-    payload = {
-        "jsonrpc": "2.0",
-        "id": "1",
-        "method": "getTokenAccountsByOwner",
-        "params": [
-            address,
-            {"mint": "3d4XyPWkUJzruF5c2qc1QfpLgsaNaDLMtTya1bWBpump"},
-            {"encoding": "jsonParsed"}
-        ],
-    }
-
-    try:
-        async with httpx.AsyncClient() as client:
-            response = await client.post(url, json=payload)
-            response.raise_for_status()
-            data = response.json()
-    except Exception:
-        return False
-
-    token_accounts = data.get("result", {}).get("value")
-
-    if not token_accounts:
-        return False
 
-    try:
-        account_info = token_accounts[0]["account"]["data"]["parsed"]["info"]
-        token_amount = account_info["tokenAmount"]
-        ui_amount = token_amount.get("uiAmount")
-    except:
-        return False
 
-    print(ui_amount)
-    return ui_amount >= 100000.0
 
 
 
@@ -256,7 +177,7 @@ class x401Kit(BaseHTTPMiddleware):
             challange=f"CHALLENGE::{X_401_Nonce}::{request.url.path}::{self.secret_domain}"
 
             signverify=SignatureVerification(X_401_Addr,X_401_Nonce,X_401_Sign,challange)
-            tokenverify=await TokenCheck(X_401_Addr)
+            tokenverify=TokenCheck(X_401_Addr)
             print(tokenverify)
 
             if signverify == True and tokenverify == True: