cli: add --verbose flag

doronz88 · Oct 31, 2021 · 7e45cc3 · 7e45cc3
1 parent 16c1f4b
commit 7e45cc3
Show file tree

Hide file tree

Showing 2 changed files with 8 additions and 1 deletion.
diff --git a/pymobiledevice3/cli/cli_common.py b/pymobiledevice3/cli/cli_common.py
@@ -1,5 +1,6 @@
 import datetime
 import json
+import logging
 import os
 import uuid
 
@@ -29,11 +30,16 @@ def print_json(buf, colored=True, default=default_json_encoder):
         print(formatted_json)
 
 
+def set_verbosity(ctx, param, value):
+    logging.getLogger().setLevel(logging.INFO - (value * 10))
+
+
 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(('verbosity', '-v', '--verbose'), count=True, callback=set_verbosity, expose_value=False),
         ]
 
     @staticmethod

diff --git a/pymobiledevice3/cli/restore.py b/pymobiledevice3/cli/restore.py
@@ -6,7 +6,7 @@
 import IPython
 import click
 from pygments import highlight, lexers, formatters
-from pymobiledevice3.cli.cli_common import print_json
+from pymobiledevice3.cli.cli_common import print_json, set_verbosity
 from pymobiledevice3.exceptions import IncorrectModeError
 from pymobiledevice3.irecv import IRecv
 from pymobiledevice3.lockdown import list_devices, LockdownClient
@@ -26,6 +26,7 @@ def __init__(self, *args, **kwargs):
         super().__init__(*args, **kwargs)
         self.params[:0] = [
             click.Option(('device', '--ecid'), type=click.INT, callback=self.device),
+            click.Option(('verbosity', '-v', '--verbose'), count=True, callback=set_verbosity, expose_value=False),
         ]
 
     @staticmethod