Mirrors
/
stable-diffusion-webui
mirror of https://github.com/AUTOMATIC1111/stable-diffusion-webui.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

compact get_info_from_repo_path

This commit is contained in:
w-e-w 2024-07-09 01:02:11 +09:00
parent 3f6dcda3e5
commit 4debd4d3ef
1 changed files with 10 additions and 17 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

27
modules/sysinfo.py
View File

@ -5,6 +5,7 @@ import subprocess
import platform import platform
import hashlib import hashlib
import re import re
from pathlib import Path
from modules import paths_internal, timer, shared, extensions, errors, launch_utils from modules import paths_internal, timer, shared, extensions, errors, launch_utils
@ -162,20 +163,19 @@ def get_torch_sysinfo():
def run_git(path, *args): def run_git(path, *args):
try: try:
if os.path.isdir(os.path.join(path, '.git')): return subprocess.check_output([launch_utils.git, '-C', path, *args], shell=False, encoding='utf8').strip()
return subprocess.check_output([launch_utils.git, '-C', path, *args], shell=False, encoding='utf8').strip()
return None
except Exception as e: except Exception as e:
return str(e) return str(e)
def get_info_from_repo_path(path): def get_info_from_repo_path(path: Path):
is_repo = (path / '.git').is_dir()
return { return {
'name': os.path.basename(path), 'name': path.name,
'path': path, 'path': str(path),
'commit': run_git(path, 'rev-parse', 'HEAD'), 'commit': run_git(path, 'rev-parse', 'HEAD') if is_repo else None,
'branch': run_git(path, 'branch', '--show-current'), 'branch': run_git(path, 'branch', '--show-current') if is_repo else None,
'remote': run_git(path, 'remote', 'get-url', 'origin') 'remote': run_git(path, 'remote', 'get-url', 'origin') if is_repo else None,
} }
@ -192,14 +192,7 @@ def get_extensions(*, enabled, fallback_disabled_extensions=None):
} }
return [to_json(x) for x in extensions.extensions if not x.is_builtin and x.enabled == enabled] return [to_json(x) for x in extensions.extensions if not x.is_builtin and x.enabled == enabled]
else: else:
extensions_list = [] return [get_info_from_repo_path(d) for d in Path(paths_internal.extensions_dir).iterdir() if d.is_dir() and enabled != (str(d.name) in fallback_disabled_extensions)]
for extension_dirname in sorted(os.listdir(paths_internal.extensions_dir)):
path = os.path.join(paths_internal.extensions_dir, extension_dirname)
if enabled == (extension_dirname in fallback_disabled_extensions):
continue
if os.path.isdir(path):
extensions_list.append(get_info_from_repo_path(path))
return extensions_list
except Exception as e: except Exception as e:
return str(e) return str(e)