From 041954f573ee9fc32e1442673225e4e54d215dc0 Mon Sep 17 00:00:00 2001 From: Dalena Date: Thu, 14 Nov 2024 10:30:45 -0600 Subject: [PATCH] fix: use convert to check for address --- silverback/_cli.py | 10 ++++++---- 1 file changed, 6 insertions(+), 4 deletions(-) diff --git a/silverback/_cli.py b/silverback/_cli.py index a7e85db1..62e79094 100644 --- a/silverback/_cli.py +++ b/silverback/_cli.py @@ -6,7 +6,7 @@ import click import yaml # type: ignore[import-untyped] -from ape import Contract +from ape import Contract, convert from ape.cli import ( AccountAliasPromptChoice, ConnectedProviderCommand, @@ -15,8 +15,9 @@ ape_cli_context, network_option, ) -from ape.exceptions import Abort, ApeException +from ape.exceptions import Abort, ApeException, ConversionError from ape.logging import LogLevel +from ape.types import AddressType from silverback._click_ext import ( SectionedHelpGroup, @@ -550,13 +551,14 @@ def create_payment_stream( accepted_tokens = platform.get_accepted_tokens(network.chain_id) if token: try: + convert(token, AddressType) token_symbol = Contract(token).symbol() token = accepted_tokens.get(token_symbol) - except ValueError: + except ConversionError: token = accepted_tokens.get(token) if token is None: - raise click.UsageError("Token not found in accepted tokens.") + raise click.UsageError(f"Token not found in {accepted_tokens}.") if not token: token = accepted_tokens.get(