olas-operate-middleware 0.9.0__py3-none-any.whl → 0.10.1__py3-none-any.whl

operate/services/service.py

@@ -63,8 +63,8 @@ from autonomy.deploy.generators.docker_compose.base import DockerComposeGenerato
 from autonomy.deploy.generators.kubernetes.base import KubernetesGenerator
 from docker import from_env
 
-from operate.constants import CONFIG_JSON, DEPLOYMENT_DIR, DEPLOYMENT_JSON, ZERO_ADDRESS
-from operate.keys import Keys
+from operate.constants import CONFIG_JSON, DEPLOYMENT_DIR, DEPLOYMENT_JSON
+from operate.keys import KeysManager
 from operate.operate_http.exceptions import NotAllowed
 from operate.operate_types import (
     Chain,
@@ -92,10 +92,10 @@ from operate.utils.ssl import create_ssl_certificate
 SAFE_CONTRACT_ADDRESS = "safe_contract_address"
 ALL_PARTICIPANTS = "all_participants"
 CONSENSUS_THRESHOLD = "consensus_threshold"
-SERVICE_CONFIG_VERSION = 7
+SERVICE_CONFIG_VERSION = 8
 SERVICE_CONFIG_PREFIX = "sc-"
 
-NON_EXISTENT_MULTISIG = "0xm"
+NON_EXISTENT_MULTISIG = None
 NON_EXISTENT_TOKEN = -1
 
 DEFAULT_TRADER_ENV_VARS = {
@@ -446,7 +446,7 @@ class Deployment(LocalResource):
         builder = ServiceBuilder.from_dir(
             path=service.package_absolute_path,
             keys_file=self.path / DEFAULT_KEYS_FILE,
-            number_of_agents=len(service.keys),
+            number_of_agents=len(service.agent_addresses),
         )
         builder.deplopyment_type = KubernetesGenerator.deployment_type
         (
@@ -489,12 +489,8 @@ class Deployment(LocalResource):
         keys_file.write_text(
             json.dumps(
                 [
-                    {
-                        "address": key.address,
-                        "private_key": key.private_key,
-                        "ledger": key.ledger.name.lower(),
-                    }
-                    for key in service.keys
+                    KeysManager().get(address).json
+                    for address in service.agent_addresses
                 ],
                 indent=4,
             ),
@@ -504,7 +500,7 @@ class Deployment(LocalResource):
             builder = ServiceBuilder.from_dir(
                 path=service.package_absolute_path,
                 keys_file=keys_file,
-                number_of_agents=len(service.keys),
+                number_of_agents=len(service.agent_addresses),
             )
             builder.deplopyment_type = DockerComposeGenerator.deployment_type
             builder.try_update_abci_connection_params()
@@ -616,12 +612,8 @@ class Deployment(LocalResource):
         keys_file.write_text(
             json.dumps(
                 [
-                    {
-                        "address": key.address,
-                        "private_key": key.private_key,
-                        "ledger": key.ledger.name.lower(),
-                    }
-                    for key in service.keys
+                    KeysManager().get(address).json
+                    for address in service.agent_addresses
                 ],
                 indent=4,
             ),
@@ -631,7 +623,7 @@ class Deployment(LocalResource):
             builder = ServiceBuilder.from_dir(
                 path=service.package_absolute_path,
                 keys_file=keys_file,
-                number_of_agents=len(service.keys),
+                number_of_agents=len(service.agent_addresses),
             )
             builder.deplopyment_type = HostDeploymentGenerator.deployment_type
             builder.try_update_abci_connection_params()
@@ -779,7 +771,7 @@ class Service(LocalResource):
     service_config_id: str
     hash: str
     hash_history: t.Dict[int, str]
-    keys: Keys
+    agent_addresses: t.List[str]
     home_chain: str
     chain_configs: ChainConfigs
     description: str
@@ -796,7 +788,7 @@ class Service(LocalResource):
     _file = CONFIG_JSON
 
     @staticmethod
-    def _determine_agent_id(service_name: str) -> int:
+    def determine_agent_id(service_name: str) -> int:
         """Determine the appropriate agent ID based on service name."""
         service_name_lower = service_name.lower()
         if "mech" in service_name_lower:
@@ -807,208 +799,6 @@ class Service(LocalResource):
             return AGENT_TYPE_IDS["modius"]
         return AGENT_TYPE_IDS["trader"]
 
-    @classmethod
-    def migrate_format(cls, path: Path) -> bool:  # pylint: disable=too-many-statements
-        """Migrate the JSON file format if needed."""
-
-        if not path.is_dir():
-            return False
-
-        if not path.name.startswith(SERVICE_CONFIG_PREFIX) and not path.name.startswith(
-            "bafybei"
-        ):
-            return False
-
-        if path.name.startswith("bafybei"):
-            backup_name = f"backup_{int(time.time())}_{path.name}"
-            backup_path = path.parent / backup_name
-            shutil.copytree(path, backup_path)
-            deployment_path = backup_path / "deployment"
-            if deployment_path.is_dir():
-                shutil.rmtree(deployment_path)
-
-        with open(path / Service._file, "r", encoding="utf-8") as file:
-            data = json.load(file)
-
-        version = data.get("version", 0)
-        if version > SERVICE_CONFIG_VERSION:
-            raise RuntimeError(
-                f"Service configuration in {path} has version {version}, which means it was created with a newer version of olas-operate-middleware. Only configuration versions <= {SERVICE_CONFIG_VERSION} are supported by this version of olas-operate-middleware."
-            )
-
-        # Complete missing env vars for trader
-        if "trader" in data["name"].lower():
-            data.setdefault("env_variables", {})
-
-            for key, value in DEFAULT_TRADER_ENV_VARS.items():
-                if key not in data["env_variables"]:
-                    data["env_variables"][key] = value
-
-            with open(path / Service._file, "w", encoding="utf-8") as file:
-                json.dump(data, file, indent=2)
-
-        if version == SERVICE_CONFIG_VERSION:
-            return False
-
-        # Migration steps for older versions
-        if version == 0:
-            new_data = {
-                "version": 2,
-                "hash": data.get("hash"),
-                "keys": data.get("keys"),
-                "home_chain_id": "100",  # This is the default value for version 2 - do not change, will be corrected below
-                "chain_configs": {
-                    "100": {  # This is the default value for version 2 - do not change, will be corrected below
-                        "ledger_config": {
-                            "rpc": data.get("ledger_config", {}).get("rpc"),
-                            "type": data.get("ledger_config", {}).get("type"),
-                            "chain": data.get("ledger_config", {}).get("chain"),
-                        },
-                        "chain_data": {
-                            "instances": data.get("chain_data", {}).get(
-                                "instances", []
-                            ),
-                            "token": data.get("chain_data", {}).get("token"),
-                            "multisig": data.get("chain_data", {}).get("multisig"),
-                            "staked": data.get("chain_data", {}).get("staked", False),
-                            "on_chain_state": data.get("chain_data", {}).get(
-                                "on_chain_state", 3
-                            ),
-                            "user_params": {
-                                "staking_program_id": "pearl_alpha",
-                                "nft": data.get("chain_data", {})
-                                .get("user_params", {})
-                                .get("nft"),
-                                "threshold": data.get("chain_data", {})
-                                .get("user_params", {})
-                                .get("threshold"),
-                                "use_staking": data.get("chain_data", {})
-                                .get("user_params", {})
-                                .get("use_staking"),
-                                "cost_of_bond": data.get("chain_data", {})
-                                .get("user_params", {})
-                                .get("cost_of_bond"),
-                                "fund_requirements": data.get("chain_data", {})
-                                .get("user_params", {})
-                                .get("fund_requirements", {}),
-                                "agent_id": data.get("chain_data", {})
-                                .get("user_params", {})
-                                .get("agent_id", "14"),
-                            },
-                        },
-                    }
-                },
-                "service_path": data.get("service_path", ""),
-                "name": data.get("name", ""),
-            }
-            data = new_data
-
-        if version < 4:
-            # Add missing fields introduced in later versions, if necessary.
-            for _, chain_data in data.get("chain_configs", {}).items():
-                chain_data.setdefault("chain_data", {}).setdefault(
-                    "user_params", {}
-                ).setdefault("use_mech_marketplace", False)
-                service_name = data.get("name", "")
-                agent_id = cls._determine_agent_id(service_name)
-                chain_data.setdefault("chain_data", {}).setdefault("user_params", {})[
-                    "agent_id"
-                ] = agent_id
-
-            data["description"] = data.setdefault("description", data.get("name"))
-            data["hash_history"] = data.setdefault(
-                "hash_history", {int(time.time()): data["hash"]}
-            )
-
-            if "service_config_id" not in data:
-                service_config_id = Service.get_new_service_config_id(path)
-                new_path = path.parent / service_config_id
-                data["service_config_id"] = service_config_id
-                path = path.rename(new_path)
-
-            old_to_new_ledgers = ["ethereum", "solana"]
-            for key_data in data["keys"]:
-                key_data["ledger"] = old_to_new_ledgers[key_data["ledger"]]
-
-            old_to_new_chains = [
-                "ethereum",
-                "goerli",
-                "gnosis",
-                "solana",
-                "optimism",
-                "base",
-                "mode",
-            ]
-            new_chain_configs = {}
-            for chain_id, chain_data in data["chain_configs"].items():
-                chain_data["ledger_config"]["chain"] = old_to_new_chains[
-                    chain_data["ledger_config"]["chain"]
-                ]
-                del chain_data["ledger_config"]["type"]
-                new_chain_configs[Chain.from_id(int(chain_id)).value] = chain_data  # type: ignore
-
-            data["chain_configs"] = new_chain_configs
-            data["home_chain"] = data.setdefault("home_chain", Chain.from_id(int(data.get("home_chain_id", "100"))).value)  # type: ignore
-            del data["home_chain_id"]
-
-            if "env_variables" not in data:
-                if data["name"] == "valory/trader_pearl":
-                    data["env_variables"] = DEFAULT_TRADER_ENV_VARS
-                else:
-                    data["env_variables"] = {}
-
-        if version < 5:
-            new_chain_configs = {}
-            for chain, chain_data in data["chain_configs"].items():
-                fund_requirements = chain_data["chain_data"]["user_params"][
-                    "fund_requirements"
-                ]
-                if ZERO_ADDRESS not in fund_requirements:
-                    chain_data["chain_data"]["user_params"]["fund_requirements"] = {
-                        ZERO_ADDRESS: fund_requirements
-                    }
-
-                new_chain_configs[chain] = chain_data  # type: ignore
-            data["chain_configs"] = new_chain_configs
-
-        if version < 7:
-            if data["home_chain"] == "optimistic":
-                data["home_chain"] = Chain.OPTIMISM.value
-
-            if "optimistic" in data["chain_configs"]:
-                data["chain_configs"]["optimism"] = data["chain_configs"].pop(
-                    "optimistic"
-                )
-
-            for _, chain_config in data["chain_configs"].items():
-                if chain_config["ledger_config"]["chain"] == "optimistic":
-                    chain_config["ledger_config"]["chain"] = Chain.OPTIMISM.value
-
-        data["version"] = SERVICE_CONFIG_VERSION
-
-        # Redownload service path
-        if "service_path" in data:
-            package_absolute_path = path / Path(data["service_path"]).name
-            data.pop("service_path")
-        else:
-            package_absolute_path = path / data["package_path"]
-
-        if package_absolute_path.exists() and package_absolute_path.is_dir():
-            shutil.rmtree(package_absolute_path)
-
-        package_absolute_path = Path(
-            IPFSTool().download(
-                hash_id=data["hash"],
-                target_dir=path,
-            )
-        )
-        data["package_path"] = str(package_absolute_path.name)
-
-        with open(path / Service._file, "w", encoding="utf-8") as file:
-            json.dump(data, file, indent=2)
-
-        return True
-
     @classmethod
     def load(cls, path: Path) -> "Service":
         """Load a service"""
@@ -1063,7 +853,7 @@ class Service(LocalResource):
 
     @staticmethod
     def new(  # pylint: disable=too-many-locals
-        keys: Keys,
+        agent_addresses: t.List[str],
         service_template: ServiceTemplate,
         storage: Path,
     ) -> "Service":
@@ -1105,7 +895,7 @@ class Service(LocalResource):
             name=service_template["name"],
             description=service_template["description"],
             hash=service_template["hash"],
-            keys=keys,
+            agent_addresses=agent_addresses,
             home_chain=service_template["home_chain"],
             hash_history={current_timestamp: service_template["hash"]},
             chain_configs=chain_configs,

operate/utils/__init__.py

@@ -19,10 +19,54 @@
 
 """Helper utilities."""
 
+import functools
 import shutil
 import time
 import typing as t
 from pathlib import Path
+from threading import Lock
+
+
+class SingletonMeta(type):
+    """A metaclass for creating thread-safe singleton classes."""
+
+    _instances: t.Dict[t.Type, t.Any] = {}
+    _lock: Lock = Lock()
+    _class_locks: t.Dict[t.Type, Lock] = {}
+
+    def __new__(
+        cls, name: str, bases: t.Tuple[type, ...], dct: t.Dict[str, t.Any]
+    ) -> t.Type:
+        """Create a new class with thread-safe methods."""
+        # Wrap all callable methods (except special methods) with thread safety
+        for key, value in list(dct.items()):
+            if callable(value) and not key.startswith("__"):
+                dct[key] = cls._make_thread_safe(value)
+
+        new_class = super().__new__(cls, name, bases, dct)
+        cls._class_locks[new_class] = Lock()
+        return new_class
+
+    @staticmethod
+    def _make_thread_safe(func: t.Callable) -> t.Callable:
+        """Wrap a function to make it thread-safe."""
+
+        @functools.wraps(func)
+        def wrapper(self: t.Any, *args: t.Any, **kwargs: t.Any) -> t.Any:
+            class_lock = SingletonMeta._class_locks.get(type(self))
+            if class_lock:
+                with class_lock:
+                    return func(self, *args, **kwargs)
+            return func(self, *args, **kwargs)
+
+        return wrapper
+
+    def __call__(cls, *args: t.Any, **kwargs: t.Any) -> t.Any:
+        """Override the __call__ method to control instance creation."""
+        with cls._lock:
+            if cls not in cls._instances:
+                cls._instances[cls] = super().__call__(*args, **kwargs)
+        return cls._instances[cls]
 
 
 def create_backup(path: Path) -> Path:

operate/utils/gnosis.py

@@ -42,7 +42,7 @@ from operate.constants import (
 from operate.operate_types import Chain
 
 
-logger = setup_logger(name="operate.manager")
+logger = setup_logger(name="operate.utils.gnosis")
 MAX_UINT256 = 2**256 - 1
 SENTINEL_OWNERS = "0x0000000000000000000000000000000000000001"
 
@@ -490,7 +490,7 @@ def drain_eoa(
     crypto: Crypto,
     withdrawal_address: str,
     chain_id: int,
-) -> str:
+) -> t.Optional[str]:
     """Drain all the native tokens from the crypto wallet."""
     tx_helper = TxSettler(
         ledger_api=ledger_api,
@@ -516,7 +516,7 @@ def drain_eoa(
         )
         tx = ledger_api.update_with_gas_estimate(
             transaction=tx,
-            raise_on_try=True,
+            raise_on_try=False,
         )
 
         chain_fee = tx["gas"] * tx["maxFeePerGas"]
@@ -530,7 +530,6 @@ def drain_eoa(
 
         tx["value"] = ledger_api.get_balance(crypto.address) - chain_fee
         if tx["value"] <= 0:
-            logger.warning(f"No balance to drain from wallet: {crypto.address}")
             raise ChainInteractionError(
                 f"No balance to drain from wallet: {crypto.address}"
             )
@@ -542,14 +541,25 @@ def drain_eoa(
         return tx
 
     setattr(tx_helper, "build", _build_tx)  # noqa: B010
-    tx_receipt = tx_helper.transact(
-        method=lambda: {},
-        contract="",
-        kwargs={},
-        dry_run=False,
-    )
-    tx_hash = tx_receipt.get("transactionHash", "").hex()
-    return tx_hash
+    try:
+        tx_receipt = tx_helper.transact(
+            method=lambda: {},
+            contract="",
+            kwargs={},
+            dry_run=False,
+        )
+    except ChainInteractionError as e:
+        if "No balance to drain from wallet" in str(e):
+            logger.warning(f"Failed to drain wallet {crypto.address} with error: {e}.")
+            return None
+
+        raise e
+
+    tx_hash = tx_receipt.get("transactionHash", None)
+    if tx_hash is not None:
+        return tx_hash.hex()
+
+    return None
 
 
 def get_asset_balance(

operate/wallet/master.py

@@ -29,7 +29,6 @@ from pathlib import Path
 
 from aea.crypto.base import Crypto, LedgerApi
 from aea.crypto.registries import make_ledger_api
-from aea.helpers.logging import setup_logger
 from aea_ledger_ethereum import DEFAULT_GAS_PRICE_STRATEGIES, EIP1559, GWEI, to_wei
 from aea_ledger_ethereum.ethereum import EthereumApi, EthereumCrypto
 from autonomy.chain.base import registry_contracts
@@ -516,6 +515,9 @@ class EthereumMasterWallet(MasterWallet):
                 asset_address=asset,
                 address=self.safes[chain] if from_safe else self.crypto.address,
             )
+            if balance <= 0:
+                continue
+
             self.transfer_asset(
                 to=withdrawal_address,
                 amount=balance,
@@ -808,6 +810,17 @@ class EthereumMasterWallet(MasterWallet):
                 safes[chain] = address
         data["safes"] = safes
 
+        if "optimistic" in data.get("safes", {}):
+            data["safes"]["optimism"] = data["safes"].pop("optimistic")
+            migrated = True
+
+        if "optimistic" in data.get("safe_chains"):
+            data["safe_chains"] = [
+                "optimism" if chain == "optimistic" else chain
+                for chain in data["safe_chains"]
+            ]
+            migrated = True
+
         with open(wallet_path, "w", encoding="utf-8") as file:
             json.dump(data, file, indent=2)
 
@@ -825,13 +838,13 @@ class MasterWalletManager:
     def __init__(
         self,
         path: Path,
+        logger: logging.Logger,
         password: t.Optional[str] = None,
-        logger: t.Optional[logging.Logger] = None,
     ) -> None:
         """Initialize master wallet manager."""
         self.path = path
+        self.logger = logger
         self._password = password
-        self.logger = logger or setup_logger(name="operate.master_wallet_manager")
 
     @property
     def json(self) -> t.List[t.Dict]:
@@ -927,20 +940,3 @@ class MasterWalletManager:
             if not self.exists(ledger_type=ledger_type):
                 continue
             yield LEDGER_TYPE_TO_WALLET_CLASS[ledger_type].load(path=self.path)
-
-    def migrate_wallet_configs(self) -> None:
-        """Migrate old wallet config formats to new ones, if applies."""
-
-        print(self.path)
-
-        for ledger_type in LedgerType:
-            if not self.exists(ledger_type=ledger_type):
-                continue
-
-            wallet_class = LEDGER_TYPE_TO_WALLET_CLASS.get(ledger_type)
-            if wallet_class is None:
-                continue
-
-            migrated = wallet_class.migrate_format(path=self.path)
-            if migrated:
-                self.logger.info(f"Wallet {wallet_class} has been migrated.")