p1otek/btc-trading
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

feat: add heartbeat and robust error handling to main loop (v1.3.5)

Browse Source
This commit is contained in:
Gemini CLI
2026-03-05 22:34:21 +01:00
parent fcfa2244f2
commit f1d7c7138c
1 changed files with 39 additions and 28 deletions
Download Patch File Download Diff File Expand all files Collapse all files

21
src/strategies/ping_pong_bot.py
View File

@ -69,7 +69,7 @@ class DatabaseManager:
class PingPongBot: class PingPongBot:
def __init__(self, config_path="config/ping_pong_config.yaml"): def __init__(self, config_path="config/ping_pong_config.yaml"):
self.version = "1.3.4" self.version = "1.3.5"
with open(config_path, 'r') as f: with open(config_path, 'r') as f:
self.config = yaml.safe_load(f) self.config = yaml.safe_load(f)
@ -108,6 +108,7 @@ class PingPongBot:
self.status_msg = "Initializing..." self.status_msg = "Initializing..."
self.last_signal = None self.last_signal = None
self.start_time = datetime.now() self.start_time = datetime.now()
self.last_refresh_time = datetime.now()
self.console = Console() self.console = Console()
# Parameters # Parameters
@ -236,13 +237,13 @@ class PingPongBot:
logger.error(f"Trade Error: {e}") logger.error(f"Trade Error: {e}")
def generate_dashboard(self): def generate_dashboard(self):
"""Generates the dashboard layout without printing directly""" """Generates the dashboard layout"""
layout = Layout() layout = Layout()
layout.split_column( layout.split_column(
Layout(name="header", size=3), Layout(name="header", size=3),
Layout(name="indicators", size=7), Layout(name="indicators", size=7),
Layout(name="position", size=6), Layout(name="position", size=6),
Layout(name="footer", size=2) Layout(name="footer", size=3)
) )
# Header # Header
@ -269,7 +270,8 @@ class PingPongBot:
layout["position"].update(Panel(pos_table, title="[bold green]PORTFOLIO STATUS[/]")) layout["position"].update(Panel(pos_table, title="[bold green]PORTFOLIO STATUS[/]"))
# Footer # Footer
footer_text = f"Status: [bold blue]{self.status_msg}[/] | Last Signal: [bold yellow]{self.last_signal or 'N/A'}[/] | Last Candle: {self.last_candle_time or 'N/A'}" refresh_str = self.last_refresh_time.strftime("%H:%M:%S")
footer_text = f"Status: [bold blue]{self.status_msg}[/]\nLast Signal: [bold yellow]{self.last_signal or 'N/A'}[/] | Last Candle: {self.last_candle_time or 'N/A'} | [dim italic]Heartbeat: {refresh_str}[/]"
layout["footer"].update(Panel(footer_text)) layout["footer"].update(Panel(footer_text))
return layout return layout
@ -278,9 +280,12 @@ class PingPongBot:
await self.db.connect() await self.db.connect()
last_exchange_update = 0 last_exchange_update = 0
with Live(self.generate_dashboard(), refresh_per_second=2) as live: with Live(self.generate_dashboard(), refresh_per_second=1) as live:
while True: while True:
try:
now = time.time() now = time.time()
self.last_refresh_time = datetime.now()
# 1. Exchange Sync (15s) # 1. Exchange Sync (15s)
if now - last_exchange_update >= 15: if now - last_exchange_update >= 15:
await self.update_exchange_data() await self.update_exchange_data()
@ -304,6 +309,12 @@ class PingPongBot:
# 3. Update Dashboard # 3. Update Dashboard
live.update(self.generate_dashboard()) live.update(self.generate_dashboard())
except Exception as e:
logger.exception(f"Loop Error: {e}")
self.status_msg = f"Error: {str(e)[:50]}"
live.update(self.generate_dashboard())
await asyncio.sleep(5) await asyncio.sleep(5)
if __name__ == "__main__": if __name__ == "__main__":