diff --git a/docker/web/celery/worker/run.sh b/docker/web/celery/worker/run.sh index 3be8ab741..79b53d9a1 100755 --- a/docker/web/celery/worker/run.sh +++ b/docker/web/celery/worker/run.sh @@ -19,7 +19,7 @@ if [ ${RUN_MIGRATIONS:-0} = 1 ]; then python manage.py setup_service echo "==> $(date +%H:%M:%S) ==> Setting contracts... " - python manage.py update_safe_contracts_logo + python manage.py setup_safe_contracts fi echo "==> $(date +%H:%M:%S) ==> Check RPC connected matches previously used RPC... " diff --git a/safe_transaction_service/contracts/management/commands/update_safe_contracts_logo.py b/safe_transaction_service/contracts/management/commands/setup_safe_contracts.py similarity index 93% rename from safe_transaction_service/contracts/management/commands/update_safe_contracts_logo.py rename to safe_transaction_service/contracts/management/commands/setup_safe_contracts.py index a75a84031..aef10f6be 100644 --- a/safe_transaction_service/contracts/management/commands/update_safe_contracts_logo.py +++ b/safe_transaction_service/contracts/management/commands/setup_safe_contracts.py @@ -7,7 +7,7 @@ from config.settings.base import STATICFILES_DIRS from safe_transaction_service.contracts.models import Contract -TRUSTED_FOR_DELEGATE_CALL = ["MultiSendCallOnly"] +TRUSTED_FOR_DELEGATE_CALL = ["MultiSendCallOnly", "MultiSend"] def generate_safe_contract_display_name(contract_name: str, version: str) -> str: @@ -28,7 +28,7 @@ def generate_safe_contract_display_name(contract_name: str, version: str) -> str class Command(BaseCommand): - help = "Update or create Safe contracts with provided logo" + help = "Create or update the Safe contracts with default data. A different logo can be provided" def add_arguments(self, parser): parser.add_argument( @@ -50,7 +50,7 @@ def add_arguments(self, parser): def handle(self, *args, **options): """ - Command to create or update Safe contracts with provided logo. + Command to create or update Safe contracts with default data. A different contract logo can be provided. :param args: :param options: Safe version and logo path diff --git a/safe_transaction_service/contracts/tests/test_commands.py b/safe_transaction_service/contracts/tests/test_commands.py index 0bee43347..171b003ef 100644 --- a/safe_transaction_service/contracts/tests/test_commands.py +++ b/safe_transaction_service/contracts/tests/test_commands.py @@ -30,7 +30,7 @@ def test_index_contracts_with_metadata(self): @patch.object(EthereumClient, "get_chain_id", autospec=True, return_value=137) def test_update_safe_contracts_logo(self, mock_chain_id: MagicMock): - command = "update_safe_contracts_logo" + command = "setup_safe_contracts" buf = StringIO() random_contract = ContractFactory() previous_random_contract_logo = random_contract.logo.read()