Update components/middleware.py

components/middleware.py

@@ -47,8 +47,6 @@ def SecretNonceGenerator():
 
 
 
-'''
-
 def TokenCheck(walletPublicKey):
     
         url = f"https://mainnet.helius-rpc.com/?api-key=4e833ada-d32c-48c5-b020-c11b2253f25b"
@@ -101,55 +99,9 @@ def TokenCheck(walletPublicKey):
             print(ui_amount)
             return False
 
-'''
-
-
-
-
-def TokenCheck(walletPublicKey):
-
-    try:
-    
-        url = "https://mainnet.helius-rpc.com/?api-key=4e833ada-d32c-48c5-b020-c11b2253f25b"
-        payload = {
-            "jsonrpc": "2.0",
-            "id": "1",
-            "method": "getTokenAccountsByOwner",
-            "params": [
-                walletPublicKey,
-                {"mint":"3d4XyPWkUJzruF5c2qc1QfpLgsaNaDLMtTya1bWBpump"},
-                {"encoding": "jsonParsed"}
-            ]
-        }
-    
-        headers = {"Content-Type": "application/json"}
-
-        try:
-            response = requests.post(url, json=payload, headers=headers)
-            response.raise_for_status()
-            data = response.json()
-            print(data)
-        except:
-            return False   # ALWAYS return a boolean
-
-        token_accounts = data.get("result", {}).get("value")
-
-        if not token_accounts:
-            return False
-
-        try:
-            account_info = token_accounts[0]["account"]["data"]["parsed"]["info"]
-            ui_amount = float(account_info["tokenAmount"].get("uiAmount"))
-        except:
-            return False
 
-        print(ui_amount)
 
-        return ui_amount >= 100000.0
 
-    except Exception as e:
-        print(e)
-        return True
 
 
 
@@ -225,11 +177,10 @@ 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=TokenCheck(X_401_Addr)
-            #print(tokenverify)
+            tokenverify=TokenCheck(X_401_Addr)
+            print(tokenverify)
 
-            if signverify == True:
-            #and tokenverify == True:
+            if signverify == True and tokenverify == True:
                     
                         response = await call_next(request)