Cura/plugins/DigitalLibrary/src/DigitalFactoryProjectModel.py

# 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.Logger import Logger
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()