fallback Extensions info

This commit is contained in:
w-e-w 2024-07-08 20:31:50 +09:00
parent dd4f798b97
commit 27d96fa608
1 changed files with 42 additions and 14 deletions

View File

@ -99,6 +99,7 @@ def get_packages():
def get_dict(): def get_dict():
config = get_config()
res = { res = {
"Platform": platform.platform(), "Platform": platform.platform(),
"Python": platform.python_version(), "Python": platform.python_version(),
@ -114,10 +115,10 @@ def get_dict():
"Exceptions": errors.get_exceptions(), "Exceptions": errors.get_exceptions(),
"CPU": get_cpu_info(), "CPU": get_cpu_info(),
"RAM": get_ram_info(), "RAM": get_ram_info(),
"Extensions": get_extensions(enabled=True), "Extensions": get_extensions(enabled=True, fallback_disabled_extensions=config.get('disabled_extensions', [])),
"Inactive extensions": get_extensions(enabled=False), "Inactive extensions": get_extensions(enabled=False, fallback_disabled_extensions=config.get('disabled_extensions', [])),
"Environment": get_environment(), "Environment": get_environment(),
"Config": get_config(), "Config": config,
"Startup": timer.startup_record, "Startup": timer.startup_record,
"Packages": get_packages(), "Packages": get_packages(),
} }
@ -159,9 +160,28 @@ def get_torch_sysinfo():
return str(e) return str(e)
def get_extensions(*, enabled): 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 None
except Exception as e:
return str(e)
def get_info_from_repo_path(path):
return {
'name': os.path.basename(path),
'path': path,
'version': run_git(path, 'rev-parse', 'HEAD'),
'branch': run_git(path, 'branch', '--show-current'),
'remote': run_git(path, 'remote', 'get-url', 'origin')
}
def get_extensions(*, enabled, fallback_disabled_extensions=None):
try:
if extensions.extensions:
def to_json(x: extensions.Extension): def to_json(x: extensions.Extension):
return { return {
"name": x.name, "name": x.name,
@ -170,8 +190,16 @@ def get_extensions(*, enabled):
"branch": x.branch, "branch": x.branch,
"remote": x.remote, "remote": x.remote,
} }
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:
extensions_list = []
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)