Update trade_manager.py

trade_manager.py

@@ -18,6 +18,18 @@ class TradeManager:
 
     async def open_trade(self, symbol, decision, current_price):
         try:
+            # ✅ الإصلاح: التحقق من نوع الصفقة ومنع صفقات SHORT
+            trade_type = decision.get("trade_type", "LONG") # الافتراضي هو LONG
+            if trade_type == "SHORT":
+                print(f"⚠️ تم رفض فتح صفقة SHORT لـ {symbol}. النظام مصمم لـ SPOT فقط.")
+                await self.r2_service.save_system_logs_async({
+                    "trade_open_rejected": True,
+                    "reason": "SHORT trade not allowed for SPOT system",
+                    "symbol": symbol,
+                    "llm_decision": decision
+                })
+                return None # لا تفتح الصفقة
+
             portfolio_state = await self.r2_service.get_portfolio_state_async()
             available_capital = portfolio_state.get("current_capital_usd", 0)
 
@@ -43,7 +55,7 @@ class TradeManager:
                 "status": "OPEN",
                 "stop_loss": decision.get("stop_loss"),
                 "take_profit": decision.get("take_profit"),
-                "trade_type": decision.get("trade_type"),
+                "trade_type": "LONG", # ✅ الإصلاح: فرض LONG دائماً لـ SPOT
                 "position_size_usd": available_capital,
                 "expected_target_minutes": expected_target_minutes,
                 "expected_target_time": expected_target_time,
@@ -65,11 +77,11 @@ class TradeManager:
                 "symbol": symbol,
                 "position_size": available_capital,
                 "expected_minutes": expected_target_minutes,
-                "trade_type": decision.get("trade_type", "LONG"),
+                "trade_type": "LONG", # تم فرض LONG
                 "strategy": strategy
             })
 
-            print(f"✅ تم فتح صفقة جديدة لـ {symbol} باستراتيجية {strategy}")
+            print(f"✅ تم فتح صفقة جديدة (LONG) لـ {symbol} باستراتيجية {strategy}")
             return new_trade
 
         except Exception as e:
@@ -91,7 +103,8 @@ class TradeManager:
             
             entry_price = trade_to_close['entry_price']
             position_size = trade_to_close['position_size_usd']
-            trade_type = trade_to_close.get('trade_type', 'LONG')
+            # ✅ الإصلاح: بما أن النظام SPOT فقط، نفترض دائماً LONG لحساب PnL
+            trade_type = "LONG" # trade_to_close.get('trade_type', 'LONG')
             strategy = trade_to_close.get('strategy', 'unknown')
             
             pnl = 0.0
@@ -99,12 +112,13 @@ class TradeManager:
             
             if entry_price and entry_price > 0 and close_price and close_price > 0:
                 try:
-                    if trade_type == 'LONG':
-                        pnl_percent = ((close_price - entry_price) / entry_price) * 100
-                        pnl = position_size * (pnl_percent / 100)
-                    elif trade_type == 'SHORT':
-                        pnl_percent = ((entry_price - close_price) / entry_price) * 100
-                        pnl = position_size * (pnl_percent / 100)
+                    # ✅ الإصلاح: استخدام معادلة LONG فقط
+                    # if trade_type == 'LONG':
+                    pnl_percent = ((close_price - entry_price) / entry_price) * 100
+                    pnl = position_size * (pnl_percent / 100)
+                    # elif trade_type == 'SHORT': # <-- إزالة هذا القسم
+                    #     pnl_percent = ((entry_price - close_price) / entry_price) * 100
+                    #     pnl = position_size * (pnl_percent / 100)
                           
                 except (TypeError, ZeroDivisionError) as calc_error:
                     pnl = 0.0
@@ -153,7 +167,7 @@ class TradeManager:
                 "new_capital": new_capital,
                 "strategy": strategy,
                 "position_size": position_size,
-                "trade_type": trade_type,
+                "trade_type": trade_type, # سيظل LONG دائماً
                 "reason": reason
             })
 
@@ -193,6 +207,7 @@ class TradeManager:
             trade_to_update['strategy'] = original_strategy
             trade_to_update['decision_data'] = re_analysis_decision
             trade_to_update['is_monitored'] = True
+            trade_to_update['trade_type'] = "LONG" # ✅ الإصلاح: التأكد من بقائها LONG
 
             open_trades = await self.r2_service.get_open_trades_async()
             for i, trade in enumerate(open_trades):
@@ -304,7 +319,6 @@ class TradeManager:
                     
                 # الحصول على السعر الحالي مع مهلة زمنية
                 try:
-                    # ✅ الإصلاح: استخدام await بشكل صحيح مع دالة get_latest_price_async
                     current_price = await asyncio.wait_for(
                         self.data_manager.get_latest_price_async(symbol), 
                         timeout=10
@@ -324,43 +338,55 @@ class TradeManager:
                 entry_price = trade['entry_price']
                 stop_loss = trade.get('stop_loss')
                 take_profit = trade.get('take_profit')
+                # ✅ الإصلاح: نفترض LONG دائماً عند التحقق من الإغلاق لـ SPOT
+                trade_type = "LONG"
                 should_close, close_reason = False, ""
                 
-                # التحقق من شروط الإغلاق
+                # التحقق من شروط الإغلاق (مع افتراض LONG)
                 if stop_loss and current_price <= stop_loss:
-                    should_close, close_reason = True, f"وصول وقف الخسارة: {current_price} <= {stop_loss}"
+                    should_close, close_reason = True, f"وصول وقف الخسارة (LONG): {current_price} <= {stop_loss}"
                 elif take_profit and current_price >= take_profit:
-                    should_close, close_reason = True, f"وصول جني الأرباح: {current_price} >= {take_profit}"
+                    should_close, close_reason = True, f"وصول جني الأرباح (LONG): {current_price} >= {take_profit}"
                 
-                # تحديث وقف الخسارة الديناميكي
+                # تحديث وقف الخسارة الديناميكي (Trailing Stop Loss) لصفقات LONG
                 if not should_close and current_price > entry_price:
-                    dynamic_stop = current_price * 0.98
-                    if dynamic_stop > (stop_loss or 0):
-                        trade['stop_loss'] = dynamic_stop
+                    # مثال بسيط: تحديد وقف خسارة عند 2% تحت السعر الحالي إذا كان أعلى من وقف الخسارة الأصلي
+                    potential_new_stop = current_price * 0.98
+                    current_stop_loss = trade.get('stop_loss', 0) or 0 # التعامل مع None
+                    if potential_new_stop > current_stop_loss:
+                        trade['stop_loss'] = potential_new_stop
+                        print(f"📈 {symbol}: تم تحديث وقف الخسارة الديناميكي إلى {potential_new_stop:.6f}")
+                        # ملاحظة: هذا التحديث محلي للمهمة، قد تحتاج لحفظه في R2 إذا أردت استمراره
                 
                 # إغلاق الصفقة إذا لزم الأمر
                 if should_close:
                     if self.r2_service.acquire_lock():
                         try:
-                            await self.immediate_close_trade(symbol, current_price, close_reason)
+                            # قبل الإغلاق، تأكد من جلب أحدث نسخة من الصفقة من R2
+                            latest_trade_data = await self.r2_service.get_trade_by_symbol_async(symbol)
+                            if latest_trade_data and latest_trade_data['status'] == 'OPEN':
+                                await self.immediate_close_trade(symbol, current_price, close_reason)
+                            else:
+                                print(f"⚠️ الصفقة {symbol} لم تعد مفتوحة أو غير موجودة، تم إلغاء الإغلاق.")
                         except Exception as close_error:
                             print(f"❌ فشل الإغلاق التلقائي لـ {symbol}: {close_error}")
                         finally:
                             self.r2_service.release_lock()
                     
-                    break
+                    break # الخروج من حلقة المراقبة بعد محاولة الإغلاق
                 
                 # إعادة تعيين عداد الأخطاء عند النجاح
                 if symbol in self.monitoring_errors:
                     self.monitoring_errors[symbol] = 0
                     
                 # التحقق من وقت المراقبة الطويل
-                monitoring_duration = time.time() - start_time
+                monitoring_duration = time.time() - self.monitoring_tasks[symbol]['start_time'] # استخدام وقت البدء الأصلي
                 if monitoring_duration > max_monitoring_time:
-                    print(f"🕒 انتهى وقت مراقبة الصفقة {symbol}")
+                    print(f"🕒 انتهى وقت مراقبة الصفقة {symbol} ({monitoring_duration:.0f}s > {max_monitoring_time}s)")
+                    # يمكنك إضافة منطق هنا لإغلاق الصفقة إذا طالت مدتها، أو تركها لإعادة التحليل
                     break
                     
-                await asyncio.sleep(15)
+                await asyncio.sleep(15) # انتظار 15 ثانية بين كل فحص
                 
             except Exception as error:
                 error_count = self._increment_monitoring_error(symbol)
@@ -374,11 +400,22 @@ class TradeManager:
                         "error_count": error_count,
                         "error": str(error)
                     })
-                    break
+                    # اختياري: محاولة إغلاق الصفقة كإجراء أخير
+                    # current_price_fallback = await self.data_manager.get_latest_price_async(symbol)
+                    # if current_price_fallback and self.r2_service.acquire_lock():
+                    #    try: await self.immediate_close_trade(symbol, current_price_fallback, "Forced close due to monitoring errors")
+                    #    finally: self.r2_service.release_lock()
+                    break # الخروج من الحلقة بسبب الأخطاء
                     
-                await asyncio.sleep(30)
+                await asyncio.sleep(30) # انتظار أطول بعد الخطأ
 
         print(f"🛑 توقيف مراقبة الصفقة: {symbol}")
+        # التأكد من إزالة المهمة عند الخروج من الحلقة
+        if symbol in self.monitoring_tasks:
+            del self.monitoring_tasks[symbol]
+        if symbol in self.monitoring_errors:
+            del self.monitoring_errors[symbol]
+
 
     def _increment_monitoring_error(self, symbol):
         if symbol not in self.monitoring_errors: