mirror of
https://github.com/Ultimaker/Cura.git
synced 2025-07-07 15:07:28 -06:00
109 lines
4.4 KiB
Python
109 lines
4.4 KiB
Python
# Copyright (c) 2018 Ultimaker B.V.
|
|
# Cura is released under the terms of the LGPLv3 or higher.
|
|
|
|
import re
|
|
from typing import Dict
|
|
|
|
from PyQt5.QtCore import Qt, pyqtProperty, pyqtSignal
|
|
|
|
from UM.Qt.ListModel import ListModel
|
|
|
|
## Model that holds cura packages. By setting the filter property the instances held by this model can be changed.
|
|
class PackagesModel(ListModel):
|
|
IdRole = Qt.UserRole + 1
|
|
TypeRole = Qt.UserRole + 2
|
|
NameRole = Qt.UserRole + 3
|
|
VersionRole = Qt.UserRole + 4
|
|
AuthorNameRole = Qt.UserRole + 5
|
|
AuthorEmailRole = Qt.UserRole + 6
|
|
DescriptionRole = Qt.UserRole + 7
|
|
IconURLRole = Qt.UserRole + 8
|
|
ImageURLsRole = Qt.UserRole + 9
|
|
DownloadURLRole = Qt.UserRole + 10
|
|
LastUpdatedRole = Qt.UserRole + 11
|
|
|
|
def __init__(self, parent = None):
|
|
super().__init__(parent)
|
|
|
|
self._packages_metadata = None
|
|
|
|
self.addRoleName(PackagesModel.IdRole, "id")
|
|
self.addRoleName(PackagesModel.TypeRole, "type")
|
|
self.addRoleName(PackagesModel.NameRole, "name")
|
|
self.addRoleName(PackagesModel.VersionRole, "version")
|
|
self.addRoleName(PackagesModel.AuthorNameRole, "author_name")
|
|
self.addRoleName(PackagesModel.AuthorEmailRole, "author_email")
|
|
self.addRoleName(PackagesModel.DescriptionRole, "description")
|
|
self.addRoleName(PackagesModel.IconURLRole, "icon_url")
|
|
self.addRoleName(PackagesModel.ImageURLsRole, "image_urls")
|
|
self.addRoleName(PackagesModel.DownloadURLRole, "download_url")
|
|
self.addRoleName(PackagesModel.LastUpdatedRole, "last_updated")
|
|
|
|
# List of filters for queries. The result is the union of the each list of results.
|
|
self._filter = {} # type: Dict[str,str]
|
|
|
|
def setMetadata(self, data):
|
|
self._packages_metadata = data
|
|
self._update()
|
|
|
|
def _update(self):
|
|
items = []
|
|
|
|
for package in self._packages_metadata:
|
|
items.append({
|
|
"id": package["package_id"],
|
|
"type": package["package_type"],
|
|
"name": package["display_name"],
|
|
"version": package["package_version"],
|
|
"author_name": package["author"]["name"],
|
|
"author_email": package["author"]["email"],
|
|
"description": package["description"],
|
|
"icon_url": package["icon_url"] if "icon_url" in package else None,
|
|
"image_urls": package["image_urls"],
|
|
"download_url": package["download_url"],
|
|
"last_updated": package["last_updated"]
|
|
})
|
|
|
|
# Filter on all the key-word arguments.
|
|
for key, value in self._filter.items():
|
|
if "*" in value:
|
|
key_filter = lambda candidate, key = key, value = value: self._matchRegExp(candidate, key, value)
|
|
else:
|
|
key_filter = lambda candidate, key = key, value = value: self._matchString(candidate, key, value)
|
|
items = filter(key_filter, items)
|
|
|
|
# Execute all filters.
|
|
filtered_items = list(items)
|
|
|
|
filtered_items.sort(key = lambda k: k["name"])
|
|
self.setItems(filtered_items)
|
|
|
|
## Set the filter of this model based on a string.
|
|
# \param filter_dict \type{Dict} Dictionary to do the filtering by.
|
|
def setFilter(self, filter_dict: Dict[str, str]) -> None:
|
|
if filter_dict != self._filter:
|
|
self._filter = filter_dict
|
|
self._update()
|
|
|
|
@pyqtProperty("QVariantMap", fset = setFilter, constant = True)
|
|
def filter(self) -> Dict[str, str]:
|
|
return self._filter
|
|
|
|
# Check to see if a container matches with a regular expression
|
|
def _matchRegExp(self, metadata, property_name, value):
|
|
if property_name not in metadata:
|
|
return False
|
|
value = re.escape(value) #Escape for regex patterns.
|
|
value = "^" + value.replace("\\*", ".*") + "$" #Instead of (now escaped) asterisks, match on any string. Also add anchors for a complete match.
|
|
if self._ignore_case:
|
|
value_pattern = re.compile(value, re.IGNORECASE)
|
|
else:
|
|
value_pattern = re.compile(value)
|
|
|
|
return value_pattern.match(str(metadata[property_name]))
|
|
|
|
# Check to see if a container matches with a string
|
|
def _matchString(self, metadata, property_name, value):
|
|
if property_name not in metadata:
|
|
return False
|
|
return value.lower() == str(metadata[property_name]).lower()
|