mirror of
https://git.mirrors.martin98.com/https://github.com/Ultimaker/Cura
synced 2025-06-30 10:25:10 +08:00
2304aeaceb
Did this so that models can be updated more often fixing bugs where when a material is set as "not favorite" in the favorites section, its updated accordingly in its "normal" section. It's also the ground work for saving material section expansion to preferences. Contributes to CURA-5378
43 lines
1.3 KiB
Python
43 lines
1.3 KiB
Python
# Copyright (c) 2018 Ultimaker B.V.
|
|
# Cura is released under the terms of the LGPLv3 or higher.
|
|
|
|
from UM.Logger import Logger
|
|
from cura.Machines.Models.BaseMaterialsModel import BaseMaterialsModel
|
|
|
|
class FavoriteMaterialsModel(BaseMaterialsModel):
|
|
|
|
def __init__(self, parent = None):
|
|
super().__init__(parent)
|
|
self._update()
|
|
|
|
def _update(self):
|
|
|
|
# Perform standard check and reset if the check fails
|
|
if not self._canUpdate():
|
|
self.setItems([])
|
|
return
|
|
|
|
# Get updated list of favorites
|
|
self._favorite_ids = self._material_manager.getFavorites()
|
|
|
|
item_list = []
|
|
|
|
for root_material_id, container_node in self._available_materials.items():
|
|
metadata = container_node.metadata
|
|
|
|
# Do not include the materials from a to-be-removed package
|
|
if bool(metadata.get("removed", False)):
|
|
continue
|
|
|
|
# Only add results for favorite materials
|
|
if root_material_id not in self._favorite_ids:
|
|
continue
|
|
|
|
item = self._createMaterialItem(root_material_id, container_node)
|
|
item_list.append(item)
|
|
|
|
# Sort the item list alphabetically by name
|
|
item_list = sorted(item_list, key = lambda d: d["brand"].upper())
|
|
|
|
self.setItems(item_list)
|