added debug contracts option

skalenetwork · Oct 15, 2024 · 9507ffa · 9507ffa
1 parent a69b3c3
commit 9507ffa
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 2 deletions.
diff --git a/skale/skale_base.py b/skale/skale_base.py
@@ -52,7 +52,8 @@ def __init__(
             wallet: BaseWallet | None = None,
             state_path: str | None = None,
             ts_diff: int | None = None,
-            provider_timeout: int = 30):
+            provider_timeout: int = 30,
+            debug: bool = False):
         logger.info('Initializing skale.py, endpoint: %s, wallet: %s',
                     endpoint, type(wallet).__name__)
         self._endpoint = endpoint
@@ -65,6 +66,7 @@ def __init__(
         self.instance = self.project.get_instance(alias_or_address)
         self.__contracts: Dict[str, BaseContract[Self]] = {}
         self.__contracts_info: Dict[str, ContractInfo[Self]] = {}
+        self.debug = debug
         self.set_contracts_info()
         if wallet:
             self.wallet = wallet

diff --git a/skale/skale_paymaster.py b/skale/skale_paymaster.py
@@ -38,7 +38,7 @@ def project_name(self) -> str:
     def contracts_info(self) -> List[ContractInfo[SkalePaymaster]]:
         import skale.contracts.paymaster as contracts
         return [
-            ContractInfo('paymaster', 'Paymaster',
+            ContractInfo('paymaster', 'FastForwardPaymaster' if self.debug else 'Paymaster',
                          contracts.Paymaster, ContractTypes.API, False)
 
         ]