3pp-mirror/Meshroom
1
0
Fork 0
mirror of https://github.com/alicevision/Meshroom.git synced 2025-07-13 14:57:22 +02:00
Code Issues Projects Releases Packages Wiki Activity

[core] plugins: Add class for DirTreeProcessEnv

Browse source 
This class handles processes without a specific environment as well
as those using Conda or a VirtualEnv.
This commit is contained in:
Candice Bentéjac 2025-06-10 17:16:55 +02:00
parent 210d1fba4e
commit dc0059132f
1 changed files with 31 additions and 0 deletions
Download patch file Download diff file Expand all files Collapse all files

31
meshroom/core/plugins.py
View file

@ -11,6 +11,7 @@ from pathlib import Path
from meshroom.common import BaseObject
from meshroom.core import desc
from meshroom.core.desc.node import _MESHROOM_ROOT
def validateNodeDesc(nodeDesc: desc.Node) -> list:
"""
@ -82,6 +83,36 @@ class ProcessEnv(BaseObject):
return ""
class DirTreeProcessEnv(ProcessEnv):
"""
"""
def __init__(self, folder: str, envType: ProcessEnvType = ProcessEnvType.DEFAULT, uri: str = ""):
super().__init__(folder, envType)
if envType == ProcessEnvType.REZ:
raise RuntimeError("Wrong process environment type.")
if not uri and envType != ProcessEnvType.DEFAULT:
raise RuntimeError("URI should be provided for the process environment.")
self.uri = uri
self.binPaths: list = [str(Path(folder, "bin"))]
self.libPaths: list = [str(Path(folder, "lib")), str(Path(folder, "lib64"))]
self.pythonPaths: list = [str(Path(folder))] + self.binPaths
def getEnvDict(self) -> dict:
env = os.environ.copy()
env["PYTHONPATH"] = f"{_MESHROOM_ROOT}{os.pathsep}{os.pathsep.join(self.pythonPaths)}{os.pathsep}{os.getenv('PYTHONPATH', '')}"
env["LD_LIBRARY_PATH"] = f"{os.pathsep.join(self.libPaths)}{os.pathsep}{os.getenv('LD_LIBRARY_PATH', '')}"
env["PATH"] = f"{os.pathsep.join(self.binPaths)}{os.pathsep}{os.getenv('PATH', '')}"
return env
def getCommandPrefix(self) -> str:
if self._processEnvType == ProcessEnvType.CONDA:
return f"conda run -n {self.uri} "
if self._processEnvType == ProcessEnvType.VIRTUALENV:
return f"{self.uri}/bin/python "
return super().getCommandPrefix()
class NodePluginStatus(Enum):
"""