popov/gogo2
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

refactoring

Browse Source
This commit is contained in:
Dobromir Popov
2024-10-22 02:44:18 +03:00
parent 7a1ef2fb7a
commit 86b3a086b9
2 changed files with 66 additions and 46 deletions
Download Patch File Download Diff File Expand all files Collapse all files

65
crypto/sol/app.py
View File

@ -16,18 +16,18 @@ import requests
import re
import random
from threading import Thread
from solana.keypair import Keypair
from solana.rpc.async_api import AsyncClient
from solana.transaction import VersionedTransaction, TxOpts
from solana.rpc.types import Processed
from jupiter import Jupiter
app = Flask(__name__)
from solders.transaction import VersionedTransaction
from solana.rpc.types import TxOpts
from solana.rpc.commitment import Confirmed, Finalized, Processed
from solders.keypair import Keypair
from jupiter_python_sdk.jupiter import Jupiter
from modules.webui import init_app
from modules.storage import init_db, store_transaction
from modules.utils import telegram_utils, logging, get_pk, send_telegram_message
from modules.SolanaAPI import SAPI, SolanaAPI, get_wallet_balances, get_transaction_details_with_retry, save_token_info
from modules.utils import telegram_utils, logging, get_pk
from modules.SolanaAPI import SAPI
# config = load_config()
load_dotenv()
@ -199,11 +199,11 @@ async def process_log(log_result):
)
await telegram_utils.send_telegram_message(message_text)
await follow_move(tr_details)
await save_token_info()
await SAPI.save_token_info()
except Exception as e:
logging.error(f"Error aquiring log details and following: {e}")
await send_telegram_message(f"Not followed! Error following move.")
await telegram_utils.send_telegram_message(f"Not followed! Error following move.")
@ -244,21 +244,21 @@ def _get_pre_balance(transaction_details: Dict[str, Any], token: str) -> float:
async def follow_move(move):
your_balances = await get_wallet_balances(YOUR_WALLET, doGetTokenName=False)
your_balances = await SAPI.get_wallet_balances(YOUR_WALLET, doGetTokenName=False)
your_balance_info = next((balance for balance in your_balances.values() if balance['address'] == move['token_in']), None)
if your_balance_info is not None:
# Use the balance
print(f"Your balance: {your_balance_info['amount']} {move['symbol_in']}")
else:
print(f"No ballance found for {move['symbol_in']}. Skipping move.")
await send_telegram_message(f"No ballance found for {move['symbol_in']}. Skipping move.")
await telegram_utils.send_telegram_message(f"No ballance found for {move['symbol_in']}. Skipping move.")
return
your_balance = your_balance_info['amount']
token_info = TOKENS_INFO.get(move['token_in'])
token_name_in = token_info.get('symbol') or await get_token_metadata(move['token_in'])
token_name_in = token_info.get('symbol') or await SAPI.get_token_metadata(move['token_in'])
token_name_out = TOKENS_INFO[move['token_out']].get('symbol') or await solanaAPI.get_token_metadata_symbol(move['token_out'])
if not your_balance:
@ -279,7 +279,7 @@ async def follow_move(move):
except ValueError:
msg = f"<b>Move not followed:</b>\nInvalid FOLLOW_AMOUNT '{FOLLOW_AMOUNT}'. Must be 'percentage' or a number."
logging.warning(msg)
await send_telegram_message(msg)
await telegram_utils.send_telegram_message(msg)
return
amount_to_swap = min(amount_to_swap, your_balance) # Ensure we're not trying to swap more than we have
@ -325,7 +325,8 @@ async def follow_move(move):
logging.info(f"Initiating move. Transaction data:\n {transaction_data}")
error_logger.info(f"Initiating move. Transaction data:\n {transaction_data}")
raw_transaction = VersionedTransaction.from_bytes(base64.b64decode(transaction_data))
signature = private_key.sign_message(message.to_bytes_versioned(raw_transaction.message))
message = raw_transaction.message
signature = private_key.sign_message(message.to_bytes_versioned())
signed_txn = VersionedTransaction.populate(raw_transaction.message, [signature])
opts = TxOpts(skip_preflight=False, preflight_commitment=Processed)
@ -338,7 +339,7 @@ async def follow_move(move):
notification += f"\n\n<b>Transaction:</b> <a href='https://solscan.io/tx/{transaction_id}'>{transaction_id}</a>"
await telegram_utils.send_telegram_message(f"Follow Transaction Sent: {transaction_id}")
tx_details = await get_transaction_details_with_retry(transaction_id)
tx_details = await SAPI.get_transaction_details_with_retry(transaction_id)
if tx_details is not None:
break
@ -354,7 +355,7 @@ async def follow_move(move):
await telegram_utils.send_telegram_message(error_message)
amount = amount * 0.75
await get_wallet_balances(YOUR_WALLET, doGetTokenName=False)
await SAPI.get_wallet_balances(YOUR_WALLET, doGetTokenName=False)
try:
if tx_details is None:
@ -377,7 +378,7 @@ async def follow_move(move):
f"\n\n<b>Transaction:</b> <a href='https://solscan.io/tx/{transaction_id}'>{transaction_id}</a>"
)
logging.info(notification)
await send_telegram_message(notification)
await telegram_utils.send_telegram_message(notification)
except Exception as e:
logging.error(f"Error sending notification: {e}")
@ -430,34 +431,38 @@ async def process_messages(websocket):
logger.error(f"An unexpected error occurred: {e}")
pk = get_pk()
pk = None
# Convert Flask app to ASGI
asgi_app = WsgiToAsgi(app)
asgi_app = WsgiToAsgi(init_app)
async def main():
global solanaAPI, bot, PROCESSING_LOG
global solanaAPI, bot, PROCESSING_LOG, pk
pk = await get_pk()
await telegram_utils.initialize()
await telegram_utils.send_telegram_message("Solana Agent Started. Connecting to mainnet...")
# process_transaction
await SAPI.wallet_watch_loop()
async def run_all():
await main()
def run_asyncio_tasks():
asyncio.run(main())
if __name__ == '__main__':
try:
asyncio.run(run_all())
except Exception as e:
logging.error(f"An error occurred: {e}")
import multiprocessing
flask_app = init_app()
# Start the asyncio tasks in a separate process
process = multiprocessing.Process(target=run_asyncio_tasks)
process.start()
# Run the ASGI server
uvicorn.run(
flask_app,
"app:asgi_app",
host="127.0.0.1",
port=3001,
log_level="debug",
reload=True
)
# Wait for the asyncio tasks to complete
process.join()