123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263 |
- # Copyright (c) 2021 Ultimaker B.V.
- # Cura is released under the terms of the LGPLv3 or higher.
- from typing import List, Optional
- from PyQt6.QtCore import Qt, pyqtSignal
- from UM.Qt.ListModel import ListModel
- from .DigitalFactoryProjectResponse import DigitalFactoryProjectResponse
- PROJECT_UPDATED_AT_DATETIME_FORMAT = "%d-%m-%Y"
- class DigitalFactoryProjectModel(ListModel):
- DisplayNameRole = Qt.ItemDataRole.UserRole + 1
- LibraryProjectIdRole = Qt.ItemDataRole.UserRole + 2
- DescriptionRole = Qt.ItemDataRole.UserRole + 3
- ThumbnailUrlRole = Qt.ItemDataRole.UserRole + 5
- UsernameRole = Qt.ItemDataRole.UserRole + 6
- LastUpdatedRole = Qt.ItemDataRole.UserRole + 7
- dfProjectModelChanged = pyqtSignal()
- def __init__(self, parent = None) -> None:
- super().__init__(parent)
- self.addRoleName(self.DisplayNameRole, "displayName")
- self.addRoleName(self.LibraryProjectIdRole, "libraryProjectId")
- self.addRoleName(self.DescriptionRole, "description")
- self.addRoleName(self.ThumbnailUrlRole, "thumbnailUrl")
- self.addRoleName(self.UsernameRole, "username")
- self.addRoleName(self.LastUpdatedRole, "lastUpdated")
- self._projects = [] # type: List[DigitalFactoryProjectResponse]
- def setProjects(self, df_projects: List[DigitalFactoryProjectResponse]) -> None:
- if self._projects == df_projects:
- return
- self._items.clear()
- self._projects = df_projects
- # self.sortProjectsBy("display_name")
- self._update(df_projects)
- def extendProjects(self, df_projects: List[DigitalFactoryProjectResponse]) -> None:
- if not df_projects:
- return
- self._projects.extend(df_projects)
- # self.sortProjectsBy("display_name")
- self._update(df_projects)
- def clearProjects(self) -> None:
- self.clear()
- self._projects.clear()
- self.dfProjectModelChanged.emit()
- def _update(self, df_projects: List[DigitalFactoryProjectResponse]) -> None:
- for project in df_projects:
- self.appendItem({
- "displayName" : project.display_name,
- "libraryProjectId" : project.library_project_id,
- "description": project.description,
- "thumbnailUrl": project.thumbnail_url,
- "username": project.username,
- "lastUpdated": project.last_updated.strftime(PROJECT_UPDATED_AT_DATETIME_FORMAT) if project.last_updated else "",
- })
- self.dfProjectModelChanged.emit()
|