WIP: Rename function to getVariantNode()

This commit is contained in:
Lipu Fei 2018-02-14 21:48:04 +01:00
parent df1d2f137b
commit 4af176dbf3
3 changed files with 5 additions and 6 deletions

View file

@ -72,7 +72,7 @@ class VariantManager:
# Gets the variant InstanceContainer with the given information.
# Almost the same as getVariantMetadata() except that this returns an InstanceContainer if present.
#
def getVariant(self, machine_type_name: str, variant_name: str,
def getVariantNode(self, machine_type_name: str, variant_name: str,
variant_type: Optional[str] = None) -> Optional["InstanceContainer"]:
return self._machine_to_variant_dict_map[machine_type_name].get(variant_name)

View file

@ -44,7 +44,7 @@ class CuraStackBuilder:
if parseBool(machine_definition.getMetaDataEntry("has_variants", False)):
variant_name = machine_definition.getMetaDataEntry("preferred_variant_name")
if variant_name:
variant_node = variant_manager.getVariant(definition_id, variant_name)
variant_node = variant_manager.getVariantNode(definition_id, variant_name)
# Sanity check. If you see this error, the related definition files should be fixed.
if variant_node is None:
raise RuntimeError("Cannot find variant with definition [%s] and variant name [%s]" % (definition_id, variant_name))

View file

@ -592,10 +592,9 @@ class XmlMaterialProfile(InstanceContainer):
from cura.Machines.VariantManager import VariantType
variant_manager = CuraApplication.getInstance().getVariantManager()
variant_node = variant_manager.getVariant(machine_id, VariantType.BUILD_PLATE, buildplate_id)
variant_node = variant_manager.getVariantNode(machine_id, buildplate_id)
if not variant_node:
continue
variant_metadata = variant_node.metadata
# TODO: check if build plate variant exists
@ -623,7 +622,7 @@ class XmlMaterialProfile(InstanceContainer):
continue
variant_manager = CuraApplication.getInstance().getVariantManager()
variant_node = variant_manager.getVariant(machine_id, hotend_name)
variant_node = variant_manager.getVariantNode(machine_id, hotend_name)
if not variant_node:
continue