cleanup

crypto/sol/app.py

@@ -1,4 +1,3 @@
-
 from flask import Flask, render_template, request, jsonify
 from solana.rpc.async_api import AsyncClient
 from solana.rpc.commitment import Confirmed
@@ -7,37 +6,42 @@ from dexscreener import DexscreenerClient
 import asyncio
 from telegram import Bot
 from telegram.constants import ParseMode
-import os
 import datetime
 
 app = Flask(__name__)
 
+# Use the production Solana RPC endpoint
 solana_client = AsyncClient("https://api.mainnet-beta.solana.com")
 dexscreener_client = DexscreenerClient()
 
-
+# Configuration
-# This can be your own ID, or one for a developer group/channel.
-# You can use the /start command of this bot to see your chat id.
 DEVELOPER_CHAT_ID = "777826553"
-# Replace with the wallet address you want to follow
 FOLLOWED_WALLET = "9U7D916zuQ8qcL9kQZqkcroWhHGho5vD8VNekvztrutN"
-# Replace with your wallet address
 YOUR_WALLET = "65nzyZXTLC81MthTo52a2gRJjqryTizWVqpK2fDKLye5"
-
-# Channel DBot: 
-# TELEGRAM_CHAT_ID = "777826553"
-# Telegram Bot Token
-# t.me/kevin_ai_robot
-# TOKEN = '6805059978:AAHNJKuOeazMSJHc3-BXRCsFfEVyFHeFnjw'
-# t.me/artitherobot 6749075936:AAHUHiPTDEIu6JH7S2fQdibwsu6JVG3FNG0
-# Replace with your Telegram Bot Token
 TELEGRAM_BOT_TOKEN = "6805059978:AAHNJKuOeazMSJHc3-BXRCsFfEVyFHeFnjw"
 
-
-
 # Initialize Telegram Bot
 bot = Bot(token=TELEGRAM_BOT_TOKEN)
-t send_telegram_message(message)
+
+# Token addresses (to be populated dynamically)
+TOKEN_ADDRESSES = {}
+
+async def get_token_balance(wallet_address, token_address):
+    try:
+        balance = await solana_client.get_token_account_balance(
+            Pubkey.from_string(token_address),
+            commitment=Confirmed
+        )
+        return float(balance['result']['value']['uiAmount'])
+    except Exception as e:
+        print(f"Error getting balance for {token_address}: {str(e)}")
+        return 0
+
+async def send_telegram_message(message):
+    try:
+        await bot.send_message(chat_id=DEVELOPER_CHAT_ID, text=message, parse_mode=ParseMode.HTML)
+    except Exception as e:
+        print(f"Error sending Telegram message: {str(e)}")
 
 @app.route('/')
 def index():
@@ -53,7 +57,6 @@ async def get_balances():
     balances = await get_wallet_balances(YOUR_WALLET)
     return jsonify(balances)
 
-
 @app.route('/follow_move', methods=['POST'])
 async def follow_move():
     move = request.json
@@ -70,8 +73,7 @@ async def follow_move():
     amount_to_swap = move['amount'] * proportion
 
     if your_balance >= amount_to_swap:
-        # Here you would implement the actual swap logic
+        # Implement actual swap logic here
-        # For now, we'll just simulate the swap
         pair = dexscreener_client.get_token_pair("solana", move['token'])
         price = float(pair['priceUsd'])
         received_amount = amount_to_swap * price
@@ -97,79 +99,12 @@ async def follow_move():
             'message': f"Insufficient balance to swap {amount_to_swap:.6f} {move['token']}"
         })
 
-
-@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"
-    }
-    
-    # Send Telegram notification about the detected move
-    asyncio.run(send_telegram_message(
-        f"<b>Move Detected:</b>\n"
-        f"Followed wallet swapped {simulated_move['amount']} {simulated_move['token']} "
-        f"to {simulated_move['to_token']}"
-    ))
-    
-    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
-
-    if wallet_balances[YOUR_WALLET][move['token']] >= amount_to_swap:
-        wallet_balances[YOUR_WALLET][move['token']] -= amount_to_swap
-        
-        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
-
-        # Send Telegram notification about the followed move
-        asyncio.run(send_telegram_message(
-            f"<b>Move Followed:</b>\n"
-            f"Swapped {amount_to_swap:.4f} {move['token']} "
-            f"for {received_amount:.4f} {move['to_token']}"
-        ))
-
-        return jsonify({
-            'status': 'success',
-            'message': f"Swapped {amount_to_swap:.4f} {move['token']} for {received_amount:.4f} {move['to_token']}"
-        })
-    else:
-        # Send Telegram notification about the failed move
-        asyncio.run(send_telegram_message(
-            f"<b>Move Failed:</b>\n"
-            f"Insufficient balance to swap {amount_to_swap:.4f} {move['token']}"
-        ))
-
-        return jsonify({
-            'status': 'error',
-            'message': f"Insufficient balance to swap {amount_to_swap:.4f} {move['token']}"
-        })
-
-
 async def get_wallet_balances(wallet_address):
     balances = {}
     for token, address in TOKEN_ADDRESSES.items():
         balances[token] = await get_token_balance(wallet_address, address)
     return balances
 
-async def send_telegram_message(message):
-    return
-    # await bot.send_message(chat_id=TELEGRAM_CHAT_ID, text=message, parse_mode=ParseMode.HTML)
-
 async def get_non_zero_token_balances(wallet_address):
     non_zero_balances = {}
     for token, address in TOKEN_ADDRESSES.items():
@@ -187,7 +122,6 @@ async def list_initial_wallet_states():
     followed_non_zero = await get_non_zero_token_balances(FOLLOWED_WALLET)
     your_non_zero = await get_non_zero_token_balances(YOUR_WALLET)
     
-    # Update TOKEN_ADDRESSES with non-zero balances from both wallets
     TOKEN_ADDRESSES = {**followed_non_zero, **your_non_zero}
     
     followed_wallet_state = "\n".join([f"{token}: {amount:.6f}" for token, amount in followed_wallet_balances.items() if amount > 0])
@@ -204,10 +138,6 @@ async def list_initial_wallet_states():
     )
     
     await send_telegram_message(message)
-    
-
-
-
 
 async def send_startup_message():
     current_time = datetime.datetime.now().strftime("%Y-%m-%d %H:%M:%S")
@@ -224,4 +154,4 @@ if __name__ == '__main__':
     asyncio.run(send_startup_message())
     asyncio.run(list_initial_wallet_states())
     print(f"Monitored Tokens: {', '.join(TOKEN_ADDRESSES.keys())}")
-    app.run(debug=True, port=3009)
+    app.run(host='0.0.0.0', port=3009)