BIOSUtilities/biosutilities/common/externals.py
Plato Mavropoulos 35564f31b7 BIOSUtilities v24.11.10
Re-written public attributes of Apple EFI ID
Improved memory consumption for all utilities
Adjusted README with consolidated requirements
2024-11-11 01:18:34 +02:00

89 lines
2.4 KiB
Python

#!/usr/bin/env python3 -B
# coding=utf-8
"""
Copyright (C) 2022-2024 Plato Mavropoulos
"""
import os
import re
import shutil
import sys
from importlib.abc import Loader
from importlib.machinery import ModuleSpec
from importlib.util import module_from_spec, spec_from_file_location
from types import ModuleType
from typing import Type
from biosutilities.common.paths import is_dir, is_file_read, project_root
from biosutilities.common.texts import to_string
def get_external_path(cmd: str | list | tuple) -> str:
""" Get external dependency path (PATH environment variable or "external" directory) """
external_root: str = os.path.join(project_root(), 'external')
external_path: str | None = external_root if is_dir(in_path=external_root) else None
for command in cmd if isinstance(cmd, (list, tuple)) else [to_string(in_object=cmd)]:
command_path: str | None = shutil.which(command, path=external_path)
if command_path and is_file_read(in_path=command_path):
return command_path
raise OSError(f'{to_string(in_object=cmd, sep_char=", ")} requirement could not be found!')
def big_script_tool() -> Type | None:
""" Get Intel BIOS Guard Script Tool class """
try:
bgst: str = get_external_path(cmd='big_script_tool')
bgst_spec: ModuleSpec | None = spec_from_file_location('big_script_tool', re.sub(r'\.PY$', '.py', bgst))
if bgst_spec and isinstance(bgst_spec.loader, Loader):
bgst_module: ModuleType | None = module_from_spec(bgst_spec)
if bgst_module:
sys.modules['big_script_tool'] = bgst_module
bgst_spec.loader.exec_module(bgst_module)
return getattr(bgst_module, 'BigScript')
except OSError:
pass
return None
def comextract_path() -> str:
""" Get ToshibaComExtractor path """
return get_external_path(cmd=['comextract', 'ComExtract'])
def szip_path() -> str:
""" Get 7-Zip path """
return get_external_path(cmd=['7zzs', '7zz', '7z'])
def tiano_path() -> str:
""" Get TianoCompress path """
return get_external_path(cmd=['TianoCompress', 'tianocompress'])
def uefifind_path() -> str:
""" Get UEFIFind path """
return get_external_path(cmd=['uefifind', 'UEFIFind'])
def uefiextract_path() -> str:
""" Get UEFIExtract path """
return get_external_path(cmd=['uefiextract', 'UEFIExtract'])