Update r2.py

r2.py

@@ -1,4 +1,4 @@
-# r2.py (محدث)
+# r2.py (محدث V10.2 - Whale Learning Storage)
 import os, traceback, json, time
 from datetime import datetime, timedelta
 import asyncio
@@ -11,6 +11,12 @@ R2_SECRET_ACCESS_KEY = os.getenv("R2_SECRET_ACCESS_KEY")
 BUCKET_NAME = "trading"
 INITIAL_CAPITAL = 10.0
 
+# 🔴 --- (جديد V10.2) أسماء ملفات التعلم --- 🔴
+WHALE_LEARNING_PENDING_KEY = "learning_whale_pending_records.json"
+WHALE_LEARNING_COMPLETED_KEY = "learning_whale_completed_records.json"
+WHALE_LEARNING_CONFIG_KEY = "learning_whale_optimal_config.json"
+
+
 class R2Service:
     def __init__(self):
         try:
@@ -317,7 +323,7 @@ class R2Service:
             if updated:
                 await self.save_open_trades_async(open_trades)
                 status = "ENABLED" if is_monitored else "DISABLED"
-                print(f"✅ Real-time monitoring {status} for {symbol}")
+                print(f"✅ Real-time monitoring {STATUS} for {symbol}")
             else:
                 print(f"⚠️ Trade {symbol} not found for monitoring status update")
             return updated
@@ -373,5 +379,117 @@ class R2Service:
         except Exception as e:
             print(f"❌ فشل حفظ سجل تدقيق التحليل في R2: {e}")
 
+    # 🔴 --- START OF CHANGE (V10.2 - Whale Learning Storage) --- 🔴
+    
+    async def _load_json_file_from_r2(self, key: str, default: Any = []) -> Any:
+        """دالة مساعدة لتحميل ملف JSON بأمان."""
+        try:
+            response = self.s3_client.get_object(Bucket=self.BUCKET_NAME, Key=key)
+            return json.loads(response['Body'].read())
+        except ClientError as e:
+            if e.response['Error']['Code'] == 'NoSuchKey':
+                print(f"ℹ️ [R2Service] لم يتم العثور على ملف '{key}'. سيتم استخدام القيمة الافتراضية.")
+                return default
+            else:
+                print(f"❌ [R2Service] خطأ ClientError أثناء تحميل '{key}': {e}")
+                raise
+        except Exception as e:
+            print(f"❌ [R2Service] خطأ عام أثناء تحميل '{key}': {e}")
+            return default
+
+    async def _save_json_file_to_r2(self, key: str, data: Any):
+        """دالة مساعدة لحفظ ملف JSON بأمان."""
+        try:
+            data_json = json.dumps(data, indent=2, ensure_ascii=False).encode('utf-8')
+            self.s3_client.put_object(
+                Bucket=self.BUCKET_NAME, Key=key, Body=data_json, ContentType="application/json"
+            )
+        except Exception as e:
+            print(f"❌ [R2Service] فشل حفظ الملف '{key}' إلى R2: {e}")
+            traceback.print_exc()
+
+    async def save_whale_learning_record_async(self, record: Dict[str, Any]):
+        """
+        (جديد V10.2)
+        يحفظ سجلاً "معلقاً" جديداً في ملف PENDING.
+        """
+        try:
+            pending_records = await self._load_json_file_from_r2(WHALE_LEARNING_PENDING_KEY, default=[])
+            pending_records.append(record)
+            await self._save_json_file_to_r2(WHALE_LEARNING_PENDING_KEY, pending_records)
+            print(f"✅ [R2Service] تم حفظ سجل تعلم الحيتان (PENDING) لـ {record['symbol']}.")
+        except Exception as e:
+            print(f"❌ [R2Service] فشل في save_whale_learning_record_async: {e}")
+
+    async def get_pending_whale_learning_records_async(self) -> List[Dict[str, Any]]:
+        """
+        (جديد V10.2)
+        يجلب جميع السجلات "المعلقة" من ملف PENDING.
+        """
+        try:
+            return await self._load_json_file_from_r2(WHALE_LEARNING_PENDING_KEY, default=[])
+        except Exception as e:
+            print(f"❌ [R2Service] فشل في get_pending_whale_learning_records_async: {e}")
+            return []
+
+    async def update_completed_whale_learning_record_async(self, completed_record: Dict[str, Any]):
+        """
+        (جديد V10.2)
+        1. يحفظ السجل "المكتمل" في ملف COMPLETED.
+        2. يزيل السجل من ملف PENDING.
+        """
+        try:
+            record_id = completed_record.get("record_id")
+            if not record_id:
+                print("❌ [R2Service] لا يمكن تحديث سجل مكتمل بدون record_id.")
+                return
+
+            # 1. الحفظ في ملف المكتمل (يحتفظ بآخر 5000 سجل مكتمل)
+            completed_records = await self._load_json_file_from_r2(WHALE_LEARNING_COMPLETED_KEY, default=[])
+            completed_records.append(completed_record)
+            if len(completed_records) > 5000:
+                completed_records = completed_records[-5000:]
+            await self._save_json_file_to_r2(WHALE_LEARNING_COMPLETED_KEY, completed_records)
+
+            # 2. الإزالة من ملف المعلق
+            pending_records = await self._load_json_file_from_r2(WHALE_LEARNING_PENDING_KEY, default=[])
+            updated_pending_records = [
+                rec for rec in pending_records if rec.get("record_id") != record_id
+            ]
+            await self._save_json_file_to_r2(WHALE_LEARNING_PENDING_KEY, updated_pending_records)
+            
+            print(f"✅ [R2Service] تم نقل سجل تعلم الحيتان (COMPLETED) لـ {completed_record['symbol']} (ID: {record_id}).")
+
+        except Exception as e:
+            print(f"❌ [R2Service] فشل في update_completed_whale_learning_record_async: {e}")
+            
+    async def get_all_completed_whale_records_async(self) -> List[Dict[str, Any]]:
+        """
+        (جديد V10.2)
+        يجلب *جميع* السجلات المكتملة (لتحليل الارتباط).
+        """
+        try:
+            return await self._load_json_file_from_r2(WHALE_LEARNING_COMPLETED_KEY, default=[])
+        except Exception as e:
+            print(f"❌ [R2Service] فشل في get_all_completed_whale_records_async: {e}")
+            return []
+            
+    async def save_whale_learning_config_async(self, config: Dict[str, Any]):
+        """
+        (جديد V10.2)
+        يحفظ ملف الإعدادات (الأوزان) الذي نتج عن التعلم.
+        """
+        await self._save_json_file_to_r2(WHALE_LEARNING_CONFIG_KEY, config)
+        print(f"✅ [R2Service] تم حفظ إعدادات تعلم الحيتان المثلى.")
+        
+    async def load_whale_learning_config_async(self) -> Dict[str, Any]:
+        """
+        (جديد V10.2)
+        يجلب ملف الإعدادات (الأوزان) الذي نتج عن التعلم.
+        """
+        return await self._load_json_file_from_r2(WHALE_LEARNING_CONFIG_KEY, default={})
+
+    # 🔴 --- END OF CHANGE --- 🔴
+
 
 print("✅ Enhanced R2 Service Loaded - Comprehensive Logging System with Candidates Support")