mirror of
https://github.com/chainstacklabs/pumpfun-bonkfun-bot.git
synced 2026-08-16 16:58:04 +00:00
fixed formatting
This commit is contained in:
@@ -3,36 +3,41 @@ import hashlib
|
||||
import json
|
||||
import os
|
||||
import sys
|
||||
|
||||
import websockets
|
||||
|
||||
sys.path.append(os.path.abspath(os.path.join(os.path.dirname(__file__), '..')))
|
||||
from config import WSS_ENDPOINT, PUMP_PROGRAM
|
||||
sys.path.append(os.path.abspath(os.path.join(os.path.dirname(__file__), "..")))
|
||||
from config import PUMP_PROGRAM, WSS_ENDPOINT
|
||||
|
||||
|
||||
async def save_transaction(tx_data, tx_signature):
|
||||
os.makedirs("blockSubscribe-transactions", exist_ok=True)
|
||||
hashed_signature = hashlib.sha256(tx_signature.encode()).hexdigest()
|
||||
file_path = os.path.join("blockSubscribe-transactions", f"{hashed_signature}.json")
|
||||
with open(file_path, 'w') as f:
|
||||
with open(file_path, "w") as f:
|
||||
json.dump(tx_data, f, indent=2)
|
||||
print(f"Saved transaction: {hashed_signature[:8]}...")
|
||||
|
||||
|
||||
async def listen_for_transactions():
|
||||
async with websockets.connect(WSS_ENDPOINT) as websocket:
|
||||
subscription_message = json.dumps({
|
||||
"jsonrpc": "2.0",
|
||||
"id": 1,
|
||||
"method": "blockSubscribe",
|
||||
"params": [
|
||||
{"mentionsAccountOrProgram": str(PUMP_PROGRAM)},
|
||||
{
|
||||
"commitment": "confirmed",
|
||||
"encoding": "base64",
|
||||
"showRewards": False,
|
||||
"transactionDetails": "full",
|
||||
"maxSupportedTransactionVersion": 0
|
||||
}
|
||||
]
|
||||
})
|
||||
subscription_message = json.dumps(
|
||||
{
|
||||
"jsonrpc": "2.0",
|
||||
"id": 1,
|
||||
"method": "blockSubscribe",
|
||||
"params": [
|
||||
{"mentionsAccountOrProgram": str(PUMP_PROGRAM)},
|
||||
{
|
||||
"commitment": "confirmed",
|
||||
"encoding": "base64",
|
||||
"showRewards": False,
|
||||
"transactionDetails": "full",
|
||||
"maxSupportedTransactionVersion": 0,
|
||||
},
|
||||
],
|
||||
}
|
||||
)
|
||||
await websocket.send(subscription_message)
|
||||
print(f"Subscribed to blocks mentioning program: {PUMP_PROGRAM}")
|
||||
|
||||
@@ -40,27 +45,36 @@ async def listen_for_transactions():
|
||||
try:
|
||||
response = await websocket.recv()
|
||||
data = json.loads(response)
|
||||
|
||||
if 'method' in data and data['method'] == 'blockNotification':
|
||||
if 'params' in data and 'result' in data['params']:
|
||||
block_data = data['params']['result']
|
||||
if 'value' in block_data and 'block' in block_data['value']:
|
||||
block = block_data['value']['block']
|
||||
if 'transactions' in block:
|
||||
transactions = block['transactions']
|
||||
|
||||
if "method" in data and data["method"] == "blockNotification":
|
||||
if "params" in data and "result" in data["params"]:
|
||||
block_data = data["params"]["result"]
|
||||
if "value" in block_data and "block" in block_data["value"]:
|
||||
block = block_data["value"]["block"]
|
||||
if "transactions" in block:
|
||||
transactions = block["transactions"]
|
||||
for tx in transactions:
|
||||
if isinstance(tx, dict) and 'transaction' in tx:
|
||||
if isinstance(tx['transaction'], list) and len(tx['transaction']) > 0:
|
||||
tx_signature = tx['transaction'][0]
|
||||
elif isinstance(tx['transaction'], dict) and 'signatures' in tx['transaction']:
|
||||
tx_signature = tx['transaction']['signatures'][0]
|
||||
if isinstance(tx, dict) and "transaction" in tx:
|
||||
if (
|
||||
isinstance(tx["transaction"], list)
|
||||
and len(tx["transaction"]) > 0
|
||||
):
|
||||
tx_signature = tx["transaction"][0]
|
||||
elif (
|
||||
isinstance(tx["transaction"], dict)
|
||||
and "signatures" in tx["transaction"]
|
||||
):
|
||||
tx_signature = tx["transaction"][
|
||||
"signatures"
|
||||
][0]
|
||||
else:
|
||||
continue
|
||||
await save_transaction(tx, tx_signature)
|
||||
elif 'result' in data:
|
||||
elif "result" in data:
|
||||
print(f"Subscription confirmed")
|
||||
except Exception as e:
|
||||
print(f"An error occurred: {str(e)}")
|
||||
|
||||
|
||||
if __name__ == "__main__":
|
||||
asyncio.run(listen_for_transactions())
|
||||
asyncio.run(listen_for_transactions())
|
||||
|
||||
Reference in New Issue
Block a user