Update trade_manager.py

trade_manager.py

@@ -1,4 +1,4 @@
-# trade_manager.py (Updated to V5.8 - Added Asyncio Lock for Watchlist)
+# trade_manager.py (Updated to V5.9 - Added trend_following to Triggers)
 import asyncio
 import json
 import time
@@ -133,9 +133,7 @@ class TradeManager:
         self.sentry_tasks = {}
         self.tactical_data_cache = {}
         
-        # 🔴 --- START OF CHANGE (FIX RACE CONDITION) --- 🔴
-        self.sentry_lock = asyncio.Lock() # إضافة قفل لحماية Watchlist
-        # 🔴 --- END OF CHANGE --- 🔴
+        self.sentry_lock = asyncio.Lock() # (إصلاح V5.8)
         
         self.kucoin_rest = None # (المنصة الأساسية)
         self.confirmation_exchanges = {} # (المنصات الثانوية للتأكيد)
@@ -189,8 +187,7 @@ class TradeManager:
             for ex in self.confirmation_exchanges.values(): await ex.close()
             raise
 
-    # 🔴 --- START OF CHANGE (FIX RACE CONDITION) --- 🔴
-    async def start_sentry_and_monitoring_loops(self):
+    async def start_sentry_and_monitoring_loops(self): # (محدث V5.8)
         """الحلقة الرئيسية للحارس (Sentry) ومراقب الخروج (Exit Monitor)."""
         self.is_running = True
         print(f"✅ [Sentry] بدء حلقات المراقبة التكتيكية (Layer 2 - API Polling)...")
@@ -236,7 +233,6 @@ class TradeManager:
                 
             except Exception as error:
                 print(f"❌ [Sentry] خطأ في الحلقة الرئيسية: {error}"); traceback.print_exc(); await asyncio.sleep(60)
-    # 🔴 --- END OF CHANGE --- 🔴
 
     async def stop_sentry_loops(self):
         """(محدث) إيقاف جميع مهام المراقبة وإغلاق جميع اتصالات REST"""
@@ -259,13 +255,11 @@ class TradeManager:
             print("✅ [Sentry] تم إغلاق اتصالات التداول (REST).")
         except Exception as e: print(f"⚠️ [Sentry] خطأ أثناء إغلاق الاتصالات: {e}")
 
-    # 🔴 --- START OF CHANGE (FIX RACE CONDITION) --- 🔴
-    async def update_sentry_watchlist(self, candidates: List[Dict]):
+    async def update_sentry_watchlist(self, candidates: List[Dict]): # (محدث V5.8)
         """تحديث قائمة المراقبة التي يستخدمها الحارس (Sentry)."""
         async with self.sentry_lock: # (استخدام القفل قبل الكتابة)
             self.sentry_watchlist = {c['symbol']: c for c in candidates}
             print(f"ℹ️ [Sentry] تم تحديث Watchlist. عدد المرشحين: {len(self.sentry_watchlist)}")
-    # 🔴 --- END OF CHANGE --- 🔴
 
     def get_sentry_status(self):
         """(محدث) لواجهة برمجة التطبيقات /system-status"""
@@ -399,7 +393,7 @@ class TradeManager:
     
 
     async def _run_tactical_analysis_loop(self, symbol: str, strategy_hint: str):
-        """(محدث) (دماغ الحارس) يشغل التحليل التكتيكي كل ثانية."""
+        """(محدث V5.8) (دماغ الحارس) يشغل التحليل التكتيكي كل ثانية."""
         while self.is_running:
             await asyncio.sleep(1) # (التحليل السريع كل ثانية)
             try:
@@ -419,9 +413,7 @@ class TradeManager:
                         current_price = tactical_data.order_book['bids'][0][0] if tactical_data.order_book and tactical_data.order_book.get('bids') and len(tactical_data.order_book['bids']) > 0 else None
                         if current_price: await self.immediate_close_trade(symbol, current_price, f"Tactical Exit: {exit_reason}")
                 else:
-                    # (منطق الدخول - محدث)
-                    # (التحقق من قائمة المراقبة يتم الآن في الحلقة الرئيسية)
-                    # (لكننا نحتاج إلى التحقق مرة أخرى هنا لضمان عدم حدوث سباق حالات)
+                    # (منطق الدخول - محدث V5.8)
                     async with self.sentry_lock:
                         is_still_on_watchlist = symbol in self.sentry_watchlist
 
@@ -442,8 +434,9 @@ class TradeManager:
                 raise # تمرير الإلغاء
             except Exception as e: print(f"❌ [Sentry] خطأ في حلقة التحليل التكتيكي لـ {symbol}: {e}"); traceback.print_exc()
 
+    # 🔴 --- START OF CHANGE (LOGIC FIXED V5.9) --- 🔴
     def _check_entry_trigger(self, symbol: str, strategy_hint: str, data: Dict) -> bool:
-        """(محدث V5.7) إصلاح منطق الزناد للسماح باستراتيجيات الانعكاس."""
+        """(محدث V5.9) إضافة trend_following إلى منطق الزناد."""
         
         rsi = data.get('rsi_1m_approx', 50)
         cvd_kucoin = data.get('cvd_kucoin', 0)
@@ -452,9 +445,10 @@ class TradeManager:
         cvd_conf_agg = data.get('cvd_confirmation_aggregate', 0)
         cvd_conf_sources_count = len(data.get('cvd_confirmation_sources', {}))
         
-        # (تم حذف شرط CVD العام من هنا)
+        # (تم حذف شرط CVD العام من هنا - إصلاح V5.7)
 
-        if strategy_hint == 'breakout_momentum':
+        # (إصلاح V5.9: دمج trend_following مع breakout_momentum)
+        if strategy_hint in ['breakout_momentum', 'trend_following']:
             # استراتيجية الزخم *تتطلب* تدفقاً إيجابياً
             if cvd_kucoin <= 0:
                 return False # (لا يوجد زخم إذا كان CVD سلبياً)
@@ -463,18 +457,16 @@ class TradeManager:
             if (rsi > 55):
                  # (الفيتو المخفf: نوقف فقط إذا كانت المنصات الأخرى تبيع ضدنا بقوة)
                  if cvd_conf_sources_count > 0 and cvd_conf_agg < (cvd_kucoin * -0.5):
-                     print(f"   [Trigger Hold] {symbol} Breakout: KuCoin CVD ({cvd_kucoin:.0f}) إيجابي، لكن منصات التأكيد ({cvd_conf_agg:.0f}) تبيع بقوة.")
+                     print(f"   [Trigger Hold] {symbol} Breakout/Trend: KuCoin CVD ({cvd_kucoin:.0f}) إيجابي، لكن منصات التأكيد ({cvd_conf_agg:.0f}) تبيع بقوة.")
                      return False
                  
-                 print(f"   [Trigger] {symbol} Breakout: K_CVD={cvd_kucoin:.0f}, C_CVD_Agg={cvd_conf_agg:.0f}, RSI={rsi:.1f}"); 
+                 print(f"   [Trigger] {symbol} Breakout/Trend: K_CVD={cvd_kucoin:.0f}, C_CVD_Agg={cvd_conf_agg:.0f}, RSI={rsi:.1f}"); 
                  return True
                  
         elif strategy_hint == 'mean_reversion':
             # استراتيجية الانعكاس *لا* تتطلب CVD إيجابي
             # (الشروط المخففة: RSI < 35)
             if (rsi < 35): 
-                # (اختياري: يمكننا إضافة شرط أن يكون CVD قد بدأ في التحسن،
-                # ولكن في الوقت الحالي، RSI < 35 هو زناد كافٍ)
                 print(f"   [Trigger] {symbol} Reversion: RSI={rsi:.1f}, K_CVD={cvd_kucoin:.0f} (CVD check ignored)")
                 return True
                 
@@ -486,15 +478,18 @@ class TradeManager:
                  return True
                  
         return False
-    # 🔴 --- END OF LOGIC FIX (V5.7) --- 🔴
+    # 🔴 --- END OF CHANGE (LOGIC FIXED V5.9) --- 🔴
 
     def _check_exit_trigger(self, trade: Dict, data: Dict) -> str: # (لا تغيير هنا)
         """يحدد ما إذا كان يجب الخروج من صفقة مفتوحة تكتيكياً."""
         rsi = data.get('rsi_1m_approx', 50); cvd_kucoin = data.get('cvd_kucoin', 0); symbol = trade['symbol']
-        if trade.get('strategy') == 'breakout_momentum':
-            if rsi < 40 and cvd_kucoin < 0: return "Momentum reversal (RSI < 40 + CVD Negative)"
+        
+        # (تحديث منطق الخروج ليشمل trend_following)
+        if trade.get('strategy') in ['breakout_momentum', 'trend_following']:
+            if rsi < 40 and cvd_kucoin < 0: return "Momentum/Trend reversal (RSI < 40 + CVD Negative)"
         if trade.get('strategy') == 'mean_reversion':
             if rsi > 75: return "Mean Reversion Target Hit (RSI > 75)"
+            
         if symbol in self.tactical_data_cache and self.tactical_data_cache[symbol].order_book:
             ob = self.tactical_data_cache[symbol].order_book
             if ob and ob.get('bids') and len(ob['bids']) > 0: # (تحقق إضافي)
@@ -627,7 +622,6 @@ class TradeManager:
             await self.r2_service.save_open_trades_async(trades_to_keep)
             
             # (لا حاجة لإزالة المهمة يدوياً، الحلقة الرئيسية ستفعل ذلك)
-            # if symbol in self.sentry_tasks: print(f"ℹ️ [Sentry] الصفقة {symbol} أغلقت، ستتوقف المراقبة.")
             
             await self.r2_service.save_system_logs_async({
                 "trade_closed": True, "symbol": symbol, "pnl_usd": pnl, "pnl_percent": pnl_percent,
@@ -715,4 +709,4 @@ class TradeManager:
         except Exception as e: print(f"❌ Failed to get trade by symbol {symbol}: {e}"); return None
 
 
-print(f"✅ Trade Manager loaded - V5.8 (Added Watchlist Lock) (ccxt.async_support: {CCXT_ASYNC_AVAILABLE})")
+print(f"✅ Trade Manager loaded - V5.9 (Fixed trend_following Trigger) (ccxt.async_support: {CCXT_ASYNC_AVAILABLE})")