87 lines
2.8 KiB
Python
87 lines
2.8 KiB
Python
from flask import Flask, render_template, request, jsonify
|
|
from solana.rpc.api import Client
|
|
from dexscreener import DexscreenerClient
|
|
import requests
|
|
import os
|
|
|
|
app = Flask(__name__)
|
|
solana_client = Client("https://api.mainnet-beta.solana.com")
|
|
dexscreener_client = DexscreenerClient()
|
|
|
|
# Replace with the wallet address you want to follow
|
|
FOLLOWED_WALLET = "REPLACE_WITH_WALLET_ADDRESS"
|
|
|
|
# Replace with your wallet address
|
|
YOUR_WALLET = "REPLACE_WITH_YOUR_WALLET_ADDRESS"
|
|
|
|
# Simulated wallet balances (replace with actual balance fetching logic)
|
|
wallet_balances = {
|
|
FOLLOWED_WALLET: {"SOL": 100, "USDC": 1000},
|
|
YOUR_WALLET: {"SOL": 10, "USDC": 100}
|
|
}
|
|
|
|
@app.route('/')
|
|
def index():
|
|
return render_template('index.html')
|
|
|
|
@app.route('/tokens', methods=['GET'])
|
|
def get_tokens():
|
|
# Fetch tokens from your wallet
|
|
return jsonify(list(wallet_balances[YOUR_WALLET].keys()))
|
|
|
|
@app.route('/swap', methods=['POST'])
|
|
def swap_tokens():
|
|
data = request.json
|
|
token_name = data['token_name']
|
|
amount = data['amount']
|
|
# Here you would add logic to perform the token swap
|
|
return jsonify({'status': 'success', 'message': f'Swapped {amount} of {token_name}'})
|
|
|
|
@app.route('/balances', methods=['GET'])
|
|
def get_balances():
|
|
return jsonify(wallet_balances[YOUR_WALLET])
|
|
|
|
@app.route('/followed_wallet_moves', methods=['GET'])
|
|
def get_followed_wallet_moves():
|
|
# In a real-world scenario, you'd use a blockchain explorer API to get recent transactions
|
|
# For this example, we'll simulate a move
|
|
simulated_move = {
|
|
"token": "SOL",
|
|
"action": "swap",
|
|
"amount": 5,
|
|
"to_token": "USDC"
|
|
}
|
|
return jsonify(simulated_move)
|
|
|
|
@app.route('/follow_move', methods=['POST'])
|
|
def follow_move():
|
|
move = request.json
|
|
followed_balance = wallet_balances[FOLLOWED_WALLET][move['token']]
|
|
your_balance = wallet_balances[YOUR_WALLET][move['token']]
|
|
|
|
proportion = your_balance / followed_balance
|
|
amount_to_swap = move['amount'] * proportion
|
|
|
|
# Perform the swap (simulated)
|
|
if wallet_balances[YOUR_WALLET][move['token']] >= amount_to_swap:
|
|
wallet_balances[YOUR_WALLET][move['token']] -= amount_to_swap
|
|
|
|
# Get the current price of the token pair
|
|
pair = dexscreener_client.get_token_pair("solana", move['token'])
|
|
price = float(pair['priceUsd'])
|
|
|
|
received_amount = amount_to_swap * price
|
|
wallet_balances[YOUR_WALLET][move['to_token']] += received_amount
|
|
|
|
return jsonify({
|
|
'status': 'success',
|
|
'message': f"Swapped {amount_to_swap} {move['token']} for {received_amount} {move['to_token']}"
|
|
})
|
|
else:
|
|
return jsonify({
|
|
'status': 'error',
|
|
'message': f"Insufficient balance to swap {amount_to_swap} {move['token']}"
|
|
})
|
|
|
|
if __name__ == '__main__':
|
|
app.run(debug=True) |