Browse Source

Customizable progress bar

koerschens 3 years ago
parent
commit
957e77b7e9
1 changed files with 10 additions and 4 deletions
  1. 10 4
      python/gpu_status.py

+ 10 - 4
python/gpu_status.py

@@ -45,7 +45,13 @@ parser.add_argument(
     "--columns", 
     type=int,
     default=-1,
-    help="The number of columns the node information is formatted in. By default, this is set by the compactness style (--compact, --ultracompact)."
+    help="The number of columns the node information is formatted in. By default, this is set by the compactness style (default, --compact, --ultracompact). Changing this value might break formatting."
+)
+parser.add_argument(
+    "--pbar", 
+    type=int,
+    default=-1,
+    help="The length of the progress bar. By default, this is set by the compactness style (default, --compact, --ultracompact). Changing this value might break formatting."
 )
 parser.add_argument(
     "-uc", 
@@ -77,7 +83,7 @@ response = requests.get(
 if args.ultracompact:
     renderer = Renderer(
         columns=3 if args.columns == -1 else args.columns,
-        progress_bar_width=20,
+        progress_bar_width=20 if args.pbar == -1 else args.pbar,
         gpu_inner_spacings=False,
         gpu_outer_spacings=False,
         node_names=args.filter,
@@ -88,7 +94,7 @@ if args.ultracompact:
 elif args.compact:
     renderer = Renderer(
         columns=2 if args.columns == -1 else args.columns,
-        progress_bar_width=40,
+        progress_bar_width=40 if args.pbar == -1 else args.pbar,
         gpu_inner_spacings=True,
         gpu_outer_spacings=False,
         node_names=args.filter,
@@ -98,7 +104,7 @@ elif args.compact:
 else:
     renderer = Renderer(
         columns=1 if args.columns == -1 else args.columns,
-        progress_bar_width=50,
+        progress_bar_width=50 if args.pbar == -1 else args.pbar,
         gpu_inner_spacings=True,
         gpu_outer_spacings=True,
         node_names=args.filter,