Update whale_news_data.py

whale_news_data.py

@@ -32,15 +32,17 @@ class EnhancedWhaleMonitor:
         self._validate_api_keys()
         
         self.whale_threshold_usd = 50000
-        self.contracts_db = contracts_db or {}
+        
+        # تحسين تخزين العقود والشبكات
+        self.contracts_db = {}
+        self._initialize_contracts_db(contracts_db or {})
+        
         self.data_manager = None
         self.r2_service = r2_service
         
         self.address_labels = {}
         self._initialize_comprehensive_exchange_addresses()
         
-        self.contract_cache = {}
-        self.symbol_networks = {}
         self.token_price_cache = {}
         self.token_decimals_cache = {}
         
@@ -115,6 +117,38 @@ class EnhancedWhaleMonitor:
         if self.r2_service:
             asyncio.create_task(self._load_contracts_from_r2())
 
+    def _initialize_contracts_db(self, initial_contracts):
+        """تهيئة قاعدة بيانات العقود مع تحسين تخزين الشبكات"""
+        print("🔄 تهيئة قاعدة بيانات العقود...")
+        
+        for symbol, contract_data in initial_contracts.items():
+            if isinstance(contract_data, dict):
+                # إذا كانت البيانات بالفعل في الصيغة الجديدة
+                self.contracts_db[symbol] = contract_data
+            else:
+                # إذا كانت البيانات في الصيغة القديمة، نحولها للصيغة الجديدة
+                self.contracts_db[symbol] = {
+                    'address': contract_data,
+                    'network': self._detect_network_from_address(contract_data)
+                }
+        
+        print(f"✅ تم تحميل {len(self.contracts_db)} عقد في قاعدة البيانات")
+
+    def _detect_network_from_address(self, address):
+        """اكتشاف الشبكة من عنوان العقد"""
+        if not isinstance(address, str):
+            return 'ethereum'  # قيمة افتراضية
+            
+        address_lower = address.lower()
+        
+        # اكتشاف الشبكة بناءً على تنسيق العنوان
+        if address_lower.startswith('0x') and len(address_lower) == 42:
+            return 'ethereum'  # معظم عناوين EVM تبدأ بـ 0x
+        elif len(address_lower) == 44 and not address_lower.startswith('0x'):
+            return 'solana'   # عناوين Solana عادة 44 حرف
+        else:
+            return 'ethereum'  # افتراضي
+
     def _validate_api_keys(self):
         """التحقق من صحة مفاتيح API المطلوبة فقط"""
         print("🔍 التحقق من صحة مفاتيح API...")
@@ -313,8 +347,22 @@ class EnhancedWhaleMonitor:
             response = self.r2_service.s3_client.get_object(Bucket="trading", Key=key)
             contracts_data = json.loads(response['Body'].read())
             
-            self.contracts_db.update(contracts_data)
-            print(f"✅ تم تحميل {len(contracts_data)} عقد من R2")
+            # تحسين تحميل البيانات مع اكتشاف الشبكات
+            loaded_count = 0
+            for symbol, contract_data in contracts_data.items():
+                if isinstance(contract_data, dict):
+                    # البيانات بالفعل في الصيغة الجديدة
+                    self.contracts_db[symbol] = contract_data
+                    loaded_count += 1
+                else:
+                    # تحويل البيانات القديمة للصيغة الجديدة
+                    self.contracts_db[symbol] = {
+                        'address': contract_data,
+                        'network': self._detect_network_from_address(contract_data)
+                    }
+                    loaded_count += 1
+            
+            print(f"✅ تم تحميل {loaded_count} عقد من R2")
             
         except Exception as e:
             print(f"⚠️ لم يتم العثور على قاعدة بيانات العقود في R2: {e}")
@@ -336,13 +384,23 @@ class EnhancedWhaleMonitor:
             
             # 1. البحث عن عقد العملة إذا لم يتم توفيره
             if not contract_address:
-                contract_address = await self._find_contract_address_enhanced(symbol)
-                if not contract_address:
+                contract_info = await self._find_contract_address_enhanced(symbol)
+                if not contract_info:
                     print(f"❌ لم يتم العثور على عقد للعملة {symbol}")
                     return self._create_no_contract_response(symbol)
+                
+                contract_address = contract_info['address']
+                network = contract_info['network']
+            else:
+                # إذا تم توفير العنوان، نحتاج لاكتشاف الشبكة
+                contract_info = await self._find_contract_address_enhanced(symbol)
+                if contract_info and 'network' in contract_info:
+                    network = contract_info['network']
+                else:
+                    # اكتشاف الشبكة من العنوان إذا لم نجد معلومات
+                    network = self._detect_network_from_address(contract_address)
+                    print(f"🌐 تم اكتشاف الشبكة تلقائياً: {network}")
             
-            # 2. تحديد الشبكة المناسبة فقط
-            network = self.symbol_networks.get(symbol.lower())
             if not network:
                 print(f"❌ لم يتم تحديد شبكة للعملة {symbol}")
                 return self._create_no_contract_response(symbol)
@@ -944,7 +1002,9 @@ class EnhancedWhaleMonitor:
                 'ATOM': 'cosmos',
                 'UNI': 'uniswap',
                 'AAVE': 'aave',
-                'KCS': 'kucoin-shares'
+                'KCS': 'kucoin-shares',
+                'MIRA': 'mirai-2',
+                'XL1': 'xl1'
             }
             
             coingecko_id = symbol_mapping.get(base_symbol, base_symbol.lower())
@@ -1078,34 +1138,41 @@ class EnhancedWhaleMonitor:
         
         print(f"🔍 البحث عن عقد للعملة: {symbol} (الرمز الأساسي: {base_symbol})")
         
-        # 1. البحث في الكاش أولاً
-        if symbol_lower in self.contract_cache:
-            print(f"   ✅ وجد في الكاش: {self.contract_cache[symbol_lower]}")
-            return self.contract_cache[symbol_lower]
-        
-        # 2. البحث في قاعدة البيانات المحلية
-        for key, address in self.contracts_db.items():
-            if symbol_lower in key.lower():
-                print(f"   ✅ وجد في قاعدة البيانات المحلية: {address}")
-                self.contract_cache[symbol_lower] = address
-                return address
+        # 1. البحث في قاعدة البيانات المحلية
+        if symbol_lower in self.contracts_db:
+            contract_info = self.contracts_db[symbol_lower]
+            print(f"   ✅ وجد في قاعدة البيانات المحلية: {contract_info}")
+            
+            # إذا كانت البيانات في الصيغة القديمة، نحولها للصيغة الجديدة
+            if isinstance(contract_info, str):
+                network = self._detect_network_from_address(contract_info)
+                contract_info = {
+                    'address': contract_info,
+                    'network': network
+                }
+                self.contracts_db[symbol_lower] = contract_info
+                print(f"   🔄 تم تحويل البيانات للصيغة الجديدة: {contract_info}")
+            
+            return contract_info
         
-        # 3. البحث في CoinGecko مع تحديد الشبكة
+        # 2. البحث في CoinGecko مع تحديد الشبكة
         print(f"   🔍 البحث في CoinGecko عن {base_symbol}...")
         coingecko_result = await self._find_contract_via_coingecko(base_symbol)
         
         if coingecko_result:
             address, network = coingecko_result
-            self.contract_cache[symbol_lower] = address
-            self.symbol_networks[symbol_lower] = network
-            self.contracts_db[symbol_lower] = address
+            contract_info = {
+                'address': address,
+                'network': network
+            }
+            self.contracts_db[symbol_lower] = contract_info
             
             # حفظ في R2 إذا كان متوفراً
             if self.r2_service:
                 await self._save_contracts_to_r2()
             
             print(f"   ✅ تم العثور على عقد {symbol} عبر CoinGecko على شبكة {network}: {address}")
-            return address
+            return contract_info
         else:
             print(f"   ❌ فشل العثور على عقد لـ {symbol} في CoinGecko")