def INITIALIZE():
from PyFlow.UI.Tool import REGISTER_TOOL
from PyFlow.UI.Widgets.InputWidgets import REGISTER_UI_INPUT_WIDGET_PIN_FACTORY
from PyFlow.UI.Canvas.UINodeBase import REGISTER_UI_NODE_FACTORY
from PyFlow.UI.Canvas.UIPinBase import REGISTER_UI_PIN_FACTORY
# TODO: Check for duplicated package names
for importer, modname, ispkg in pkgutil.iter_modules(Packages.__path__):
if ispkg:
mod = importer.find_module(modname).load_module(modname)
package = getattr(mod, modname)()
__PACKAGES[modname] = package
for name, package in __PACKAGES.items():
packageName = package.__class__.__name__
for node in package.GetNodeClasses().values():
node._packageName = packageName
for pin in package.GetPinClasses().values():
pin._packageName = packageName
uiPinsFactory = package.UIPinsFactory()
REGISTER_UI_PIN_FACTORY(packageName, uiPinsFactory)
uiPinInputWidgetsFactory = package.PinsInputWidgetFactory()
REGISTER_UI_INPUT_WIDGET_PIN_FACTORY(packageName, uiPinInputWidgetsFactory)
uiNodesFactory = package.UINodesFactory()
REGISTER_UI_NODE_FACTORY(packageName, uiNodesFactory)
for toolClass in package.GetToolClasses().values():
REGISTER_TOOL(packageName, toolClass)
def INITIALIZE(additionalPackageLocations=[]):
from PyFlow.UI.Tool import REGISTER_TOOL
from PyFlow.UI.Widgets.InputWidgets import REGISTER_UI_INPUT_WIDGET_PIN_FACTORY
from PyFlow.UI.Canvas.UINodeBase import REGISTER_UI_NODE_FACTORY
from PyFlow.UI.Canvas.UIPinBase import REGISTER_UI_PIN_FACTORY
from PyFlow import ConfigManager
packagePaths = Packages.__path__
# check for additional package locations
if "PYFLOW_PACKAGES_PATHS" in os.environ:
delim = ';'
pathsString = os.environ["PYFLOW_PACKAGES_PATHS"]
# remove delimeters from right
pathsString = pathsString.rstrip(delim)
for packagesRoot in pathsString.split(delim):
if os.path.exists(packagesRoot):
packagePaths.append(packagesRoot)
packagePaths.extend(additionalPackageLocations)
for importer, modname, ispkg in pkgutil.iter_modules(packagePaths):
if ispkg:
mod = importer.find_module(modname).load_module(modname)
package = getattr(mod, modname)()
__PACKAGES[modname] = package
registeredInternalPinDataTypes = set()
for name, package in __PACKAGES.items():
packageName = package.__class__.__name__
for node in package.GetNodeClasses().values():
node._packageName = packageName
for pin in package.GetPinClasses().values():
pin._packageName = packageName
if pin.IsValuePin():
internalType = pin.internalDataStructure()
if internalType in registeredInternalPinDataTypes:
raise Exception(
"Pin with {0} internal data type alredy been registered"
.format(internalType))
registeredInternalPinDataTypes.add(internalType)
uiPinsFactory = package.UIPinsFactory()
if uiPinsFactory is not None:
REGISTER_UI_PIN_FACTORY(packageName, uiPinsFactory)
uiPinInputWidgetsFactory = package.PinsInputWidgetFactory()
if uiPinInputWidgetsFactory is not None:
REGISTER_UI_INPUT_WIDGET_PIN_FACTORY(packageName,
uiPinInputWidgetsFactory)
uiNodesFactory = package.UINodesFactory()
if uiNodesFactory is not None:
REGISTER_UI_NODE_FACTORY(packageName, uiNodesFactory)
for toolClass in package.GetToolClasses().values():
REGISTER_TOOL(packageName, toolClass)
getHashableDataTypes()
def INITIALIZE(additionalPackageLocations=[], software=""):
sys.path.append(
str((Path(__file__).parent / Path('Packages')).resolve().absolute()))
from PyFlow.UI.Tool import REGISTER_TOOL
from PyFlow.UI.Widgets.InputWidgets import REGISTER_UI_INPUT_WIDGET_PIN_FACTORY
from PyFlow.UI.Canvas.UINodeBase import REGISTER_UI_NODE_FACTORY
from PyFlow.UI.Canvas.UIPinBase import REGISTER_UI_PIN_FACTORY
from PyFlow import ConfigManager
from Qt.QtWidgets import QMessageBox
packagePaths = Packages.__path__
def ensurePackagePath(inPath):
for subFolder in os.listdir(inPath):
subFolderPath = os.path.join(inPath, subFolder)
if os.path.isdir(subFolderPath):
if "PyFlow" in os.listdir(subFolderPath):
subFolderPath = os.path.join(subFolderPath, "PyFlow",
"Packages")
if os.path.exists(subFolderPath):
return subFolderPath
return inPath
def recursePackagePaths(inPath):
paths = []
for subFolder in os.listdir(inPath):
subFolderPath = os.path.join(inPath, subFolder)
if os.path.isdir(subFolderPath):
if "PyFlow" in os.listdir(subFolderPath):
subFolderPath = os.path.join(subFolderPath, "PyFlow",
"Packages")
if os.path.exists(subFolderPath):
paths.append(subFolderPath)
return paths
# check for additional package locations
if "PYFLOW_PACKAGES_PATHS" in os.environ:
delim = ';'
pathsString = os.environ["PYFLOW_PACKAGES_PATHS"]
# remove delimeters from right
pathsString = pathsString.rstrip(delim)
for packagesRoot in pathsString.split(delim):
if os.path.exists(packagesRoot):
paths = recursePackagePaths(packagesRoot)
packagePaths.extend(paths)
for packagePathId in range(len(additionalPackageLocations)):
packagePath = additionalPackageLocations[packagePathId]
packagePath = ensurePackagePath(packagePath)
additionalPackageLocations[packagePathId] = packagePath
packagePaths.extend(additionalPackageLocations)
for importer, modname, ispkg in pkgutil.iter_modules(packagePaths):
try:
if ispkg:
mod = importer.find_module(modname).load_module(modname)
package = getattr(mod, modname)()
__PACKAGES[modname] = package
__PACKAGE_PATHS[modname] = os.path.normpath(mod.__path__[0])
except Exception as e:
QMessageBox.critical(
None, str("Fatal error"),
"Error On Module %s :\n%s" % (modname, str(e)))
continue
registeredInternalPinDataTypes = set()
for name, package in __PACKAGES.items():
packageName = package.__class__.__name__
for node in package.GetNodeClasses().values():
node._packageName = packageName
for pin in package.GetPinClasses().values():
pin._packageName = packageName
if pin.IsValuePin():
internalType = pin.internalDataStructure()
if internalType in registeredInternalPinDataTypes:
raise Exception(
"Pin with {0} internal data type already been registered"
.format(internalType))
registeredInternalPinDataTypes.add(internalType)
uiPinsFactory = package.UIPinsFactory()
if uiPinsFactory is not None:
REGISTER_UI_PIN_FACTORY(packageName, uiPinsFactory)
uiPinInputWidgetsFactory = package.PinsInputWidgetFactory()
if uiPinInputWidgetsFactory is not None:
REGISTER_UI_INPUT_WIDGET_PIN_FACTORY(packageName,
uiPinInputWidgetsFactory)
uiNodesFactory = package.UINodesFactory()
if uiNodesFactory is not None:
REGISTER_UI_NODE_FACTORY(packageName, uiNodesFactory)
for toolClass in package.GetToolClasses().values():
supportedSoftwares = toolClass.supportedSoftwares()
if "any" not in supportedSoftwares:
if software not in supportedSoftwares:
continue
REGISTER_TOOL(packageName, toolClass)
getHashableDataTypes()