38 lines
1.0 KiB
Python
38 lines
1.0 KiB
Python
import os
|
|
import shutil
|
|
|
|
|
|
__all__ = ["ToolNotFound", "tool_env_var", "has_tool", "require_tool"]
|
|
|
|
|
|
class ToolNotFound(Exception):
|
|
pass
|
|
|
|
|
|
def tool_env_var(name):
|
|
return name.upper().replace("-", "_")
|
|
|
|
|
|
def _get_tool(name):
|
|
return os.environ.get(tool_env_var(name), name)
|
|
|
|
|
|
def has_tool(name):
|
|
return shutil.which(_get_tool(name)) is not None
|
|
|
|
|
|
def require_tool(name):
|
|
env_var = tool_env_var(name)
|
|
path = _get_tool(name)
|
|
if shutil.which(path) is None:
|
|
if env_var in os.environ:
|
|
raise ToolNotFound("Could not find required tool {} in {} as "
|
|
"specified via the {} environment variable".
|
|
format(name, path, env_var))
|
|
else:
|
|
raise ToolNotFound("Could not find required tool {} in PATH. Place "
|
|
"it directly in PATH or specify path explicitly "
|
|
"via the {} environment variable".
|
|
format(name, env_var))
|
|
return path
|