Update trade_manager.py

trade_manager.py

@@ -1,4 +1,4 @@
-# trade_manager.py (Updated to V6.7 - 1M Indicator & Order Book Trigger)
+# trade_manager.py (Updated to V6.8 - Added Post-Close Cycle Trigger)
 import asyncio
 import json
 import time
@@ -8,15 +8,12 @@ from datetime import datetime, timedelta
 from typing import Dict, Any, List
 from collections import deque, defaultdict 
 
-# 🔴 --- START OF CHANGE (V6.7) --- 🔴
-# (إضافة pandas لتحليل مؤشرات 1-دقيقة)
 import pandas as pd
 try:
     import pandas_ta as ta
 except ImportError:
     print("⚠️ مكتبة pandas_ta غير موجودة، مؤشرات الحارس (Sentry 1m) ستفشل.")
     ta = None
-# 🔴 --- END OF CHANGE --- 🔴
 
 try:
     import ccxt.async_support as ccxtasync
@@ -47,16 +44,11 @@ class TacticalData:
         self.last_kucoin_trade_id = None
         self.last_confirmation_trade_ids = defaultdict(lambda: None)
         
-        # 🔴 --- START OF CHANGE (V6.7) --- 🔴
-        # (حذف: self.one_min_rsi)
-        # (إضافة: بيانات ومؤشرات 1-دقيقة)
-        self.ohlcv_1m = deque(maxlen=100) # (لتخزين 100 شمعة 1-دقيقة)
-        self.indicators_1m = {} # (لتخزين EMA 9, EMA 21, MACD Hist)
+        self.ohlcv_1m = deque(maxlen=100) 
+        self.indicators_1m = {} 
         self.last_1m_candle_timestamp = None
-        # 🔴 --- END OF CHANGE --- 🔴
 
     def add_trade(self, trade):
-        """إضافة صفقة KuCoin (الأساسية)"""
         trade_id = trade.get('id')
         if trade_id and trade_id == self.last_kucoin_trade_id:
             return 
@@ -77,7 +69,6 @@ class TacticalData:
         except Exception: pass
 
     def add_confirmation_trade(self, exchange_id: str, trade: Dict):
-        """(جديد) إضافة صفقة تأكيد (Bybit, OKX, etc.)"""
         trade_id = trade.get('id')
         if trade_id and trade_id == self.last_confirmation_trade_ids[exchange_id]:
             return 
@@ -103,43 +94,34 @@ class TacticalData:
             return {"bids_depth": bids_depth, "asks_depth": asks_depth}
         except Exception: return {"bids_depth": 0, "asks_depth": 0}
 
-    # 🔴 --- START OF CHANGE (V6.7) --- 🔴
-    # (حذف دالة get_1m_rsi التقريبية)
-
     def add_1m_ohlcv(self, ohlcv_data: List):
-        """(جديد V6.7) إضافة شموع 1-دقيقة وحساب المؤشرات"""
         if not ohlcv_data:
             return
         
-        # (إضافة الشموع الجديدة فقط)
         new_candles_added = False
         for candle in ohlcv_data:
             timestamp = candle[0]
             if timestamp and timestamp != self.last_1m_candle_timestamp:
                 if self.ohlcv_1m and timestamp < self.ohlcv_1m[-1][0]:
-                    continue # (تجاهل الشموع القديمة إذا حدث تداخل)
+                    continue 
                 
                 self.ohlcv_1m.append(candle)
                 self.last_1m_candle_timestamp = timestamp
                 new_candles_added = True
         
-        # (حساب المؤشرات فقط إذا تغيرت البيانات)
-        if new_candles_added and len(self.ohlcv_1m) >= 26: # (26 كافية لـ EMA 21 و MACD)
+        if new_candles_added and len(self.ohlcv_1m) >= 26: 
             self._analyze_1m_indicators()
 
     def _analyze_1m_indicators(self):
-        """(جديد V6.7) حساب مؤشرات 1-دقيقة الحقيقية"""
         if ta is None or len(self.ohlcv_1m) < 26:
             self.indicators_1m = {}
             return
             
         try:
-            # (تحويل deque إلى DataFrame للمعالجة)
             df = pd.DataFrame(list(self.ohlcv_1m), columns=['timestamp', 'open', 'high', 'low', 'close', 'volume'])
             df[['open', 'high', 'low', 'close', 'volume']] = df[['open', 'high', 'low', 'close', 'volume']].astype(float)
             close = df['close']
 
-            # (حساب المؤشرات المطلوبة)
             ema_9 = ta.ema(close, length=9)
             ema_21 = ta.ema(close, length=21)
             macd_data = ta.macd(close, fast=12, slow=26, signal=9)
@@ -157,29 +139,25 @@ class TacticalData:
                 self.indicators_1m = {}
                 
         except Exception as e:
-            # print(f"⚠️ [Sentry] خطأ في حساب مؤشرات 1m لـ {self.symbol}: {e}")
             self.indicators_1m = {}
-    # 🔴 --- END OF CHANGE --- 🔴
 
     def get_tactical_snapshot(self):
-        """(محدث V6.7) لإرجاع مؤشرات 1-دقيقة الحقيقية"""
         agg_cvd = sum(self.confirmation_cvd.values())
         
-        # 🔴 --- (تغيير V6.7) --- 🔴
         return {
             "cvd_kucoin": self.cvd,
             "cvd_confirmation_sources": dict(self.confirmation_cvd),
             "cvd_confirmation_aggregate": agg_cvd,
             "large_trades_count_5m": len([t for t in self.large_trades if t.get('timestamp') and (time.time() - t['timestamp']/1000) < 300]),
-            # (حذف rsi_1m_approx)
-            "indicators_1m": self.indicators_1m, # (إضافة المؤشرات الجديدة)
+            "indicators_1m": self.indicators_1m,
             "ob_analysis": self.analyze_order_book()
         }
-        # 🔴 --- نهاية التغيير --- 🔴
 
 
 class TradeManager:
-    def __init__(self, r2_service, learning_hub=None, data_manager=None, state_manager=None):
+    # 🔴 --- START OF CHANGE (V6.8) --- 🔴
+    def __init__(self, r2_service, learning_hub=None, data_manager=None, state_manager=None, callback_on_close=None):
+    # 🔴 --- END OF CHANGE --- 🔴
         if not CCXT_ASYNC_AVAILABLE:
             raise RuntimeError("مكتبة 'ccxt.async_support' غير متاحة.")
 
@@ -187,6 +165,12 @@ class TradeManager:
         self.learning_hub = learning_hub
         self.data_manager = data_manager
         self.state_manager = state_manager
+        
+        # 🔴 --- START OF CHANGE (V6.8) --- 🔴
+        # (حفظ دالة رد النداء)
+        self.callback_on_close = callback_on_close
+        # 🔴 --- END OF CHANGE --- 🔴
+        
         self.is_running = False
         self.sentry_watchlist = {}
         self.sentry_tasks = {}
@@ -354,8 +338,6 @@ class TradeManager:
                     await asyncio.sleep(10)
                     continue
 
-                # 🔴 --- START OF CHANGE (V6.7) --- 🔴
-                # (جلب 3 أنواع بيانات بالتوازي)
                 tasks = {
                     'ob': asyncio.create_task(self.kucoin_rest.fetch_order_book(symbol, limit=20)),
                     'trades': asyncio.create_task(self.kucoin_rest.fetch_trades(symbol, since=int((time.time() - 60) * 1000), limit=50)),
@@ -365,23 +347,19 @@ class TradeManager:
                 await asyncio.wait(tasks.values(), return_when=asyncio.ALL_COMPLETED)
                 
                 if symbol not in self.tactical_data_cache:
-                    continue # (ربما تم إيقاف المراقبة أثناء الجلب)
+                    continue 
 
-                # 1. معالجة دفتر الطلبات
                 if not tasks['ob'].exception():
                     self.tactical_data_cache[symbol].set_order_book(tasks['ob'].result())
 
-                # 2. معالجة آخر الصفقات
                 if not tasks['trades'].exception():
                     trades = tasks['trades'].result()
                     trades.sort(key=lambda x: x['timestamp'])
                     for trade in trades:
                         self.tactical_data_cache[symbol].add_trade(trade)
                 
-                # 3. معالجة شموع 1-دقيقة
                 if not tasks['ohlcv_1m'].exception():
                     self.tactical_data_cache[symbol].add_1m_ohlcv(tasks['ohlcv_1m'].result())
-                # 🔴 --- END OF CHANGE --- 🔴
 
                 await asyncio.sleep(self.polling_interval)
 
@@ -472,8 +450,6 @@ class TradeManager:
                         is_still_on_watchlist = symbol in self.sentry_watchlist
 
                     if is_still_on_watchlist:
-                        # 🔴 --- (تغيير V6.7) --- 🔴
-                        # (snapshot يحتوي الآن على مؤشرات 1-دقيقة الحقيقية)
                         trigger = self._check_entry_trigger(symbol, strategy_hint, snapshot)
                         if trigger:
                             print(f"✅ [Sentry] زناد دخول تكتيكي لـ {symbol} (استراتيجية: {strategy_hint})")
@@ -489,11 +465,9 @@ class TradeManager:
                 raise 
             except Exception as e: print(f"❌ [Sentry] خطأ في حلقة التحليل التكتيكي لـ {symbol}: {e}"); traceback.print_exc()
 
-    # 🔴 --- START OF CHANGE (V6.7) --- 🔴
     def _check_entry_trigger(self, symbol: str, strategy_hint: str, data: Dict) -> bool:
         """(محدث V6.7) زناد ثلاثي: CVD + دفتر الطلبات + مؤشرات 1-دقيقة"""
         
-        # --- جلب البيانات ---
         cvd_kucoin = data.get('cvd_kucoin', 0)
         
         ob_analysis = data.get('ob_analysis', {})
@@ -504,38 +478,24 @@ class TradeManager:
         ema_9_1m = indicators_1m.get('ema_9')
         ema_21_1m = indicators_1m.get('ema_21')
         
-        # (للاستراتيجيات الأخرى)
         large_trades = data.get('large_trades_count_5m', 0)
 
-        # --- منطق الزناد ---
         if strategy_hint in ['breakout_momentum', 'trend_following']:
             
-            # (الشرط 0: يجب أن تتوفر بيانات 1-دقيقة)
             if ema_9_1m is None or ema_21_1m is None:
                 return False
                 
-            # (الشرط 1: زخم الصفقات - CVD)
             cvd_check = (cvd_kucoin > 0)
-            
-            # (الشرط 2: زخم دفتر الطلبات - OB Depth)
             ob_check = (bids_depth > asks_depth)
-            
-            # (الشرط 3: زخم الاتجاه - 1m EMAs)
             ema_check = (ema_9_1m > ema_21_1m)
 
             if cvd_check and ob_check and ema_check:
                 print(f"   [Trigger] {symbol} (Momentum): CVD+, OB+, 1m_EMA+. الدخول!")
                 return True
-            # (إذا فشل، لا نطبع شيئاً لتقليل التشويش)
                  
         elif strategy_hint == 'mean_reversion':
-            # (لم نطور منطق 1-دقيقة للانعكاس بعد، نستخدم المنطق القديم)
-            # (ملاحظة: مؤشر rsi_1m_approx لم يعد موجوداً، لذا هذا الزناد معطل مؤقتاً)
-            # rsi = data.get('rsi_1m_approx', 50) # (محذوف)
-            # if (rsi < 35): 
-            #     print(f"   [Trigger] {symbol} Reversion: RSI={rsi:.1f}")
-            #     return True
-            pass # (يبقى معطلاً حتى نضيف مؤشرات انعكاس 1-دقيقة)
+            # (ملاحظة: هذا الزناد معطل مؤقتاً لعدم وجود مؤشرات انعكاس 1-دقيقة)
+            pass 
                 
         elif strategy_hint == 'volume_spike':
              if (large_trades > 0): 
@@ -543,7 +503,6 @@ class TradeManager:
                  return True
                  
         return False
-    # 🔴 --- END OF CHANGE --- 🔴
 
     def _check_exit_trigger(self, trade: Dict, data: Dict, tactical_data: TacticalData) -> str:
         """(محدث V6.6) يراقب وقف الخسارة وجني الأرباح باستخدام (Bid) و (Last Trade Price)"""
@@ -763,6 +722,15 @@ class TradeManager:
                 print(f"🧠 [LearningHub] تشغيل التعلم (Reflector+Stats) لـ {symbol}...")
                 await self.learning_hub.analyze_trade_and_learn(trade_to_close, reason)
             else: print("⚠️ [Sentry] LearningHub غير متاح، تم تخطي التعلم.")
+            
+            # 🔴 --- START OF CHANGE (V6.8) --- 🔴
+            # (تشغيل الدورة الجديدة بعد الإغلاق)
+            if self.callback_on_close:
+                print("🔄 [Executor] Trade closed. Scheduling immediate Explorer cycle...")
+                # (استخدام create_task لتجنب الجمود (Deadlock))
+                asyncio.create_task(self.callback_on_close())
+            # 🔴 --- END OF CHANGE --- 🔴
+            
             print(f"✅ [Executor] تم إغلاق الصفقة (الوهمية) {symbol} - السبب: {reason} - PnL: {pnl_percent:+.2f}%")
             return True
         except Exception as e: print(f"❌ [Executor] فشل فادح أثناء إغلاق الصفقة (الوهمية) {symbol}: {e}"); traceback.print_exc(); raise
@@ -839,4 +807,4 @@ class TradeManager:
         except Exception as e: print(f"❌ Failed to get trade by symbol {symbol}: {e}"); return None
 
 
-print(f"✅ Trade Manager loaded - V6.7 (1M Indicator & OB Trigger) (ccxt.async_support: {CCXT_ASYNC_AVAILABLE})")
+print(f"✅ Trade Manager loaded - V6.8 (1M Trigger + Post-Close Cycle) (ccxt.async_support: {CCXT_ASYNC_AVAILABLE})")