From 51c69b35baf87f3e74656ee5330f588c1fe108c4 Mon Sep 17 00:00:00 2001 From: doron zarhi Date: Sun, 5 Dec 2021 11:29:51 +0200 Subject: [PATCH] cli: allow passing of udid using PYMOBILEDEVICE3_UDID env var --- pymobiledevice3/cli/cli_common.py | 7 ++++++- 1 file changed, 6 insertions(+), 1 deletion(-) diff --git a/pymobiledevice3/cli/cli_common.py b/pymobiledevice3/cli/cli_common.py index 867e4c1d4..e920e2cbd 100644 --- a/pymobiledevice3/cli/cli_common.py +++ b/pymobiledevice3/cli/cli_common.py @@ -35,11 +35,16 @@ def set_verbosity(ctx, param, value): coloredlogs.set_level(logging.INFO - (value * 10)) +UDID_ENV_VAR = 'PYMOBILEDEVICE3_UDID' + + class Command(click.Command): def __init__(self, *args, **kwargs): super().__init__(*args, **kwargs) self.params[:0] = [ - click.Option(('lockdown', '--udid'), callback=self.udid), + click.Option(('lockdown', '--udid'), envvar=UDID_ENV_VAR, callback=self.udid, + help=f'Device unique identifier. You may pass {UDID_ENV_VAR} environment variable to pass this' + f' option as well'), click.Option(('verbosity', '-v', '--verbose'), count=True, callback=set_verbosity, expose_value=False), ]