Make timeout configurable for cli (#564)

This commit is contained in:
Teemu R 2023-12-04 16:44:27 +01:00 committed by GitHub
parent bfd1d6ae0a
commit 347cbfe3bd
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23

View File

@ -131,6 +131,13 @@ def json_formatter_cb(result, **kwargs):
@click.option( @click.option(
"--json", default=False, is_flag=True, help="Output raw device response as JSON." "--json", default=False, is_flag=True, help="Output raw device response as JSON."
) )
@click.option(
"--timeout",
envvar="KASA_TIMEOUT",
default=5,
required=False,
help="Timeout for device communications.",
)
@click.option( @click.option(
"--discovery-timeout", "--discovery-timeout",
envvar="KASA_DISCOVERY_TIMEOUT", envvar="KASA_DISCOVERY_TIMEOUT",
@ -163,6 +170,7 @@ async def cli(
debug, debug,
type, type,
json, json,
timeout,
discovery_timeout, discovery_timeout,
username, username,
password, password,
@ -234,7 +242,7 @@ async def cli(
if type is not None: if type is not None:
device_type = DeviceType.from_value(type) device_type = DeviceType.from_value(type)
dev = await SmartDevice.connect( dev = await SmartDevice.connect(
host, credentials=credentials, device_type=device_type host, credentials=credentials, device_type=device_type, timeout=timeout
) )
else: else:
echo("No --type defined, discovering..") echo("No --type defined, discovering..")