Update components/paymenthandling.py

components/paymenthandling.py

@@ -1,20 +1,60 @@
 from solders.keypair import Keypair
 from solana.rpc.async_api import AsyncClient
+from .mongodbconnection import provideClient
+
+
+
+client=provideClient()
+db=client["agentpaymentdb"]
+coll=db["agentpaymentcol"]
+
+
+
 
 
 class PaymentHandler():
     def __init__(self,agent):
         self.agent=agent
+        self.rpc=AsyncClient("https://mainnet.helius-rpc.com/?api-key=4e833ada-d32c-48c5-b020-c11b2253f25b")
 
 
-    def create_keypair(self, agent_id=None):
+    def create_keypair(self):
         """Create a new keypair and return it as MongoDB-friendly format"""
         keypair = Keypair()
         keypair_data = {
-            "agent_id": agent_id,
+            "agent_id": agent,
             "public_key": str(keypair.pubkey())
             "secret_key": bytes(keypair).hex()
         }
-        
+        coll.update_one(
+                {"owner": data.owner},
+                {"$push": {"agents": keypair_data}}
+                )
+
+
+        async def transfer_sol(self, from_keypair, to_pubkey, amount_sol):
+            """Transfer SOL from one account to another"""
+            try:
+                lamports = int(amount_sol * 1e9)
+            
+                transfer_ix = transfer(
+                    TransferParams(
+                        from_pubkey=from_keypair.pubkey(),
+                        to_pubkey=to_pubkey,
+                        lamports=lamports
+                    ))
     
-        
+                recent_blockhash =self.rpc.get_latest_blockhash()
+            
+                txn = Transaction(recent_blockhash=recent_blockhash.value.blockhash)
+                txn.add(transfer_ix)
+                txn.sign(from_keypair)
+                result = await self.client.send_transaction(txn, from_keypair)
+                print(f"✅ SOL Transfer successful!")
+                print(f"📝 Signature: {result.value}")
+                return True
+            
+            except Exception as e:
+                print(f"❌ Transfer failed: {e}")
+                return False
+