Skip to content

Commit

Permalink
hide --gradio-auth and --api-auth values from /internal/sysinfo report
Browse files Browse the repository at this point in the history
  • Loading branch information
AUTOMATIC1111 committed Aug 27, 2023
1 parent 66d7630 commit d63117a
Showing 1 changed file with 17 additions and 1 deletion.
18 changes: 17 additions & 1 deletion modules/sysinfo.py
Original file line number Diff line number Diff line change
Expand Up @@ -82,7 +82,7 @@ def get_dict():
"Data path": paths_internal.data_path,
"Extensions dir": paths_internal.extensions_dir,
"Checksum": checksum_token,
"Commandline": sys.argv,
"Commandline": get_argv(),
"Torch env info": get_torch_sysinfo(),
"Exceptions": get_exceptions(),
"CPU": {
Expand Down Expand Up @@ -123,6 +123,22 @@ def get_environment():
return {k: os.environ[k] for k in sorted(os.environ) if k in environment_whitelist}


def get_argv():
res = []

for v in sys.argv:
if shared.cmd_opts.gradio_auth and shared.cmd_opts.gradio_auth == v:
res.append("<hidden>")
continue

if shared.cmd_opts.api_auth and shared.cmd_opts.api_auth == v:
res.append("<hidden>")
continue

res.append(v)

return res

re_newline = re.compile(r"\r*\n")


Expand Down

0 comments on commit d63117a

Please sign in to comment.