From 894ba2719f248527d5cb1db2e49bad4169ea58e6 Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?Cl=C3=A9ment=20Walter?= Date: Mon, 31 Jul 2023 10:31:16 +0300 Subject: [PATCH] Add update_default option when deploying an account --- packages/starksheet-cairo/scripts/add_accounts.py | 4 +++- packages/starksheet-cairo/utils/starknet.py | 7 ++++--- 2 files changed, 7 insertions(+), 4 deletions(-) diff --git a/packages/starksheet-cairo/scripts/add_accounts.py b/packages/starksheet-cairo/scripts/add_accounts.py index 3ddcf44..c49c0bc 100644 --- a/packages/starksheet-cairo/scripts/add_accounts.py +++ b/packages/starksheet-cairo/scripts/add_accounts.py @@ -23,7 +23,9 @@ async def main(): # %% Deploy accounts for i in range(count): private_key = random.randbytes(31).hex() - account = await deploy_starknet_account(private_key=private_key, amount=amount) + account = await deploy_starknet_account( + private_key=private_key, amount=amount, update_default=False + ) Path(f"{i}").mkdir(exist_ok=True, parents=True) (Path(f"{i}") / ".env").write_text( f"ACCOUNT_ADDRESS=0x{account.address:x}\nPRIVATE_KEY=0x{private_key}\n" diff --git a/packages/starksheet-cairo/utils/starknet.py b/packages/starksheet-cairo/utils/starknet.py index b645e6a..08f3824 100644 --- a/packages/starksheet-cairo/utils/starknet.py +++ b/packages/starksheet-cairo/utils/starknet.py @@ -286,7 +286,7 @@ def _convert_offset_to_hex(obj): async def deploy_starknet_account( - private_key=None, amount: Union[int, float] = 1 + private_key=None, amount: Union[int, float] = 1, update_default=True ) -> Account: compile_contract( {"contract_name": "OpenzeppelinAccount", "is_account_contract": True} @@ -323,8 +323,9 @@ async def deploy_starknet_account( status = "✅" if status == TransactionStatus.ACCEPTED_ON_L2 else "❌" logger.info(f"{status} Account deployed at address {hex(res.account.address)}") - NETWORK["account_address"] = hex(res.account.address) - NETWORK["private_key"] = hex(key_pair.private_key) + if update_default: + NETWORK["account_address"] = hex(res.account.address) + NETWORK["private_key"] = hex(key_pair.private_key) return res.account