feat:add pancake address

bnb-chain · Nov 2, 2023 · 7a02264 · 7a02264
1 parent a3fb218
commit 7a02264
Show file tree

Hide file tree

Showing 3 changed files with 10 additions and 2 deletions.
diff --git a/config/settings/base.py b/config/settings/base.py
@@ -555,3 +555,5 @@
         "api_key": {"type": "apiKey", "in": "header", "name": "Authorization"}
     },
 }
+
+EXTERNAL_ADDRESS_LIST = env("EXTERNAL_ADDRESS_LIST", default="").split(",")
diff --git a/safe_transaction_service/history/services/safe_service.py b/safe_transaction_service/history/services/safe_service.py
@@ -2,6 +2,7 @@
 from dataclasses import dataclass, replace
 from datetime import datetime
 from typing import Optional, Tuple, Union
+from django.conf import settings
 
 from eth_typing import ChecksumAddress
 from web3 import Web3
@@ -165,6 +166,9 @@ def get_safe_info_from_blockchain(self, safe_address: ChecksumAddress) -> SafeIn
             safe = Safe(safe_address, self.ethereum_client)
             safe_info = safe.retrieve_all_info()
             # Return same master copy information than the db method
+            if safe_address in settings.EXTERNAL_ADDRESS_LIST:
+                return safe_info
+
             return replace(
                 safe_info,
                 version=SafeMasterCopy.objects.get_version_for_address(
@@ -175,7 +179,7 @@ def get_safe_info_from_blockchain(self, safe_address: ChecksumAddress) -> SafeIn
             raise NodeConnectionException from exc
         except CannotRetrieveSafeInfoException as exc:
             raise CannotGetSafeInfoFromBlockchain(safe_address) from exc
-
+            
     def get_safe_info_from_db(self, safe_address: ChecksumAddress) -> SafeInfo:
         try:
             return SafeLastStatus.objects.get_or_generate(safe_address).get_safe_info()

diff --git a/safe_transaction_service/history/views.py b/safe_transaction_service/history/views.py
@@ -1160,7 +1160,9 @@ def get(self, request, address, *args, **kwargs):
                 },
             )
 
-        if not SafeContract.objects.filter(address=address).exists():
+        logger.debug("param:%s", address)
+        logger.debug(settings.EXTERNAL_ADDRESS_LIST)
+        if (not SafeContract.objects.filter(address=address).exists()) and (address not in settings.EXTERNAL_ADDRESS_LIST):
             return Response(status=status.HTTP_404_NOT_FOUND)
 
         try: