cli: allow passing of udid using PYMOBILEDEVICE3_UDID env var

doronz88 · Dec 5, 2021 · 51c69b3 · 51c69b3
1 parent 701c96a
commit 51c69b3
Showing 1 changed file with 6 additions and 1 deletion.
diff --git 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),
         ]