New program-level -q flag, that must be placed before the command wor…

…d. Currently used by 'header', 'listkeys', and 'metadata' commands

readme.md

@@ -2,13 +2,14 @@
 
 This is a light-weight utility program for [safetensors files](https://github.com/huggingface/safetensors "safetensors files") written in Python only (no major external dependencies). Currently it can do the following:
 
-    
+
     Usage: safetensors_util.py [OPTIONS] COMMAND [ARGS]...
-    
+
     Options:
-      --version  Show the version and exit.
-      --help     Show this message and exit.
-
+      --version    Show the version and exit.
+      -q, --quiet  quiet mode, don't print informational stuff
+      --help       Show this message and exit.
+
     Commands:
       checklora   see if input file is a SD 1.x LoRA file
       extracthdr  extract file header and save to output file

safetensors_util.py

@@ -13,16 +13,19 @@
                                   help="overwrite existing files")
 fix_ued_flag=click.option("-pm","--parse-more",default=False,is_flag=True, show_default=True,
                           help="when printing metadata, unescaped doublequotes to make text more readable" )
+quiet_flag=click.option("-q","--quiet",default=False,is_flag=True, show_default=True,
+                        help="Quiet mode, don't print informational stuff" )
 
 @click.group()
-
 @click.version_option(version=3)
+@quiet_flag
 
 @click.pass_context
-def cli(ctx):
+def cli(ctx,quiet:bool):
     # ensure that ctx.obj exists and is a dict (in case `cli()` is called
     # by means other than the `if` block below)
     ctx.ensure_object(dict)
+    ctx.obj['quiet'] = quiet
 
 @cli.command(name="header",short_help="print file header")
 @readonly_input_file

safetensors_worker.py

@@ -55,7 +55,7 @@ def WriteMetadataToHeader(cmdLine:dict,in_st_file:str,in_json_file:str,output_fi
     return i
 
 def PrintHeader(cmdLine:dict,input_file:str) -> int:
-    s=SafeTensorsFile.open_file(input_file)
+    s=SafeTensorsFile.open_file(input_file,cmdLine['quiet'])
     js=s.get_header()
     if js is None: return -1
 
@@ -106,7 +106,7 @@ def _ParseMore(d:dict):
             _ParseMore(value)
 
 def PrintMetadata(cmdLine:dict,input_file:str) -> int:
-    s=SafeTensorsFile.open_file(input_file)
+    s=SafeTensorsFile.open_file(input_file,cmdLine['quiet'])
     js=s.get_header()
     if js is None: return -1
 
@@ -121,7 +121,7 @@ def PrintMetadata(cmdLine:dict,input_file:str) -> int:
     return 0
 
 def HeaderKeysToLists(cmdLine:dict,input_file:str) -> int:
-    s=SafeTensorsFile.open_file(input_file,quiet=True)
+    s=SafeTensorsFile.open_file(input_file,cmdLine['quiet'])
     js=s.get_header()
     if js is None: return -1