From d5ec01ffd9e5166dc9ca2c18f0d1fb15f73212fc Mon Sep 17 00:00:00 2001
From: Nawaz Haider <nawazhaider60@gmail.com>
Date: Thu, 1 Jan 2026 22:30:32 +0600
Subject: [PATCH] Refactor trading logic to use MAX_TRADES constant for trade
 limit

---
 config.py | 1 +
 main.py   | 5 +++--
 2 files changed, 4 insertions(+), 2 deletions(-)

diff --git a/config.py b/config.py
index ed5867f..65bc530 100644
--- a/config.py
+++ b/config.py
@@ -8,3 +8,4 @@ PROFIT_MARGIN = 0.02
 TRADING_BPS_THRESHOLD = 50
 MARKET_SESSION_SECONDS = 900
 TIMEZONE = "US/Eastern"
+MAX_TRADES = 1
diff --git a/main.py b/main.py
index 7b9a481..746c9f3 100644
--- a/main.py
+++ b/main.py
@@ -11,6 +11,7 @@ from utils.clob_client_and_order import (
     cache_tocken_trading_infos,
 )
 from utils.cpu_affinity import set_cpu_affinity
+from config import MAX_TRADES
 
 
 gc.disable()
@@ -25,7 +26,6 @@ requests.options = session.options
 
 
 async def main():
-    max_trades = 1
     trades = 0
 
     logger = setup_logging()
@@ -59,6 +59,7 @@ async def main():
             logger.info("Trading session ended. Starting new session.")
             gc.collect()
             await asyncio.sleep(10)
+            trades = 0
             up_token, down_token, market_slug = await fetch_tokens()
             book = OrderBook(up_token, down_token, market_slug)
             asyncio.create_task(cache_tocken_trading_infos(client, book))
@@ -77,7 +78,7 @@ async def main():
         down_ask_price = 1 - up_bid_price
         down_bid_price = 1 - up_ask_price
 
-        if trades < max_trades:
+        if trades < MAX_TRADES:
             trading_side = book.last_signal
 
             if trading_side == SIGNALES.UP: