59 lines
1.7 KiB
Python
59 lines
1.7 KiB
Python
|
import time
|
||
|
|
from queue import Queue
|
||
|
|
from loguru import logger
|
||
|
|
from web3 import Web3
|
||
|
|
from util import fetch_abi
|
||
|
|
|
||
|
|
|
||
|
|
class EventMonitor:
|
||
|
|
"""Monitor Ethereum event logs"""
|
||
|
|
def __init__(self, config: dict, queue: Queue):
|
||
|
|
self.config = config
|
||
|
|
self.queue = queue
|
||
|
|
|
||
|
|
self.web3 = Web3(Web3.HTTPProvider(self.config['infura_url']))
|
||
|
|
|
||
|
|
pool_address = self.config['pool_address']
|
||
|
|
pool_abi = fetch_abi(pool_address)
|
||
|
|
self.contract = self.web3.eth.contract(
|
||
|
|
address=pool_address,
|
||
|
|
abi=pool_abi
|
||
|
|
)
|
||
|
|
self.event_filter = self.contract.events.PairCreated.create_filter(fromBlock='latest')
|
||
|
|
|
||
|
|
def log_loop(self, interval: int):
|
||
|
|
while True:
|
||
|
|
try:
|
||
|
|
if not self._check_connection():
|
||
|
|
logger.error("No connection to Ethereum RPC")
|
||
|
|
continue
|
||
|
|
for event in self.event_filter.get_new_entries():
|
||
|
|
self._handle_event(event)
|
||
|
|
except Exception as e:
|
||
|
|
logger.error(e)
|
||
|
|
|
||
|
|
time.sleep(interval)
|
||
|
|
|
||
|
|
def _check_connection(self):
|
||
|
|
if self.web3.is_connected():
|
||
|
|
logger.debug("Connected to Ethereum RPC")
|
||
|
|
return True
|
||
|
|
else:
|
||
|
|
logger.warning("Failed to connect to Ethereum RPC")
|
||
|
|
return False
|
||
|
|
|
||
|
|
def _handle_event(self, event):
|
||
|
|
# TODO: Get token names
|
||
|
|
# TODO: Resolve token names
|
||
|
|
# TODO: Create data structure for the event
|
||
|
|
|
||
|
|
token0 = event.args['token0']
|
||
|
|
token1 = event.args['token1']
|
||
|
|
|
||
|
|
logger.info(f"New pair: {token0} + {token1}")
|
||
|
|
|
||
|
|
self.queue.put(
|
||
|
|
f"New Uniswap pair: https://etherscan.io/address/{token0} + https://etherscan.io/address/{token1}"
|
||
|
|
)
|
||
|
|
|