mirror of
https://github.com/alicevision/Meshroom.git
synced 2025-04-28 17:57:16 +02:00
f88e7d2a0a
If a compatibility issue is found and the assertion is raised, a message is now displayed to indicate which template (with its path) and which node (with its node type) caused the issue.
58 lines
2.3 KiB
Python
58 lines
2.3 KiB
Python
#!/usr/bin/env python
|
|
# coding:utf-8
|
|
|
|
from meshroom.core.graph import Graph
|
|
from meshroom.core import pipelineTemplates, Version
|
|
from meshroom.core.node import CompatibilityIssue, CompatibilityNode
|
|
|
|
import json
|
|
import meshroom
|
|
|
|
|
|
def test_templateVersions():
|
|
"""
|
|
This test checks that there is no compatibility issue with the nodes saved in the template files.
|
|
It fails when an upgrade of a templates is needed. Any template can still be opened even if its
|
|
nodes are not up-to-date, as they will be automatically upgraded.
|
|
"""
|
|
|
|
assert len(pipelineTemplates) >= 1
|
|
|
|
for _, path in pipelineTemplates.items():
|
|
with open(path) as jsonFile:
|
|
fileData = json.load(jsonFile)
|
|
|
|
graphData = fileData.get(Graph.IO.Keys.Graph, fileData)
|
|
|
|
assert isinstance(graphData, dict)
|
|
|
|
header = fileData.get(Graph.IO.Keys.Header, {})
|
|
assert header.get("template", False)
|
|
nodesVersions = header.get(Graph.IO.Keys.NodesVersions, {})
|
|
|
|
for _, nodeData in graphData.items():
|
|
nodeType = nodeData["nodeType"]
|
|
assert nodeType in meshroom.core.nodesDesc
|
|
|
|
nodeDesc = meshroom.core.nodesDesc[nodeType]
|
|
currentNodeVersion = meshroom.core.nodeVersion(nodeDesc)
|
|
|
|
inputs = nodeData.get("inputs", {})
|
|
internalInputs = nodeData.get("internalInputs", {})
|
|
version = nodesVersions.get(nodeType, None)
|
|
|
|
compatibilityIssue = None
|
|
|
|
if version and currentNodeVersion and Version(version).major != Version(currentNodeVersion).major:
|
|
compatibilityIssue = CompatibilityIssue.VersionConflict
|
|
else:
|
|
for attrName, value in inputs.items():
|
|
if not CompatibilityNode.attributeDescFromName(nodeDesc.inputs, attrName, value):
|
|
compatibilityIssue = CompatibilityIssue.DescriptionConflict
|
|
break
|
|
for attrName, value in internalInputs.items():
|
|
if not CompatibilityNode.attributeDescFromName(nodeDesc.internalInputs, attrName, value):
|
|
compatibilityIssue = CompatibilityIssue.DescriptionConflict
|
|
break
|
|
|
|
assert compatibilityIssue is None, "{} in {} for node {}".format(compatibilityIssue, path, nodeType)
|