mirror of
https://git.mirrors.martin98.com/https://github.com/Ultimaker/Cura
synced 2025-04-20 04:39:38 +08:00
101 lines
4.5 KiB
Python
101 lines
4.5 KiB
Python
# Copyright (c) 2018 Ultimaker B.V.
|
|
# Cura is released under the terms of the LGPLv3 or higher.
|
|
|
|
import os
|
|
from PyQt5.QtCore import QUrl
|
|
from PyQt5.QtGui import QDesktopServices
|
|
|
|
from typing import Set
|
|
|
|
from UM.Extension import Extension
|
|
from UM.Application import Application
|
|
from UM.Logger import Logger
|
|
from UM.PluginRegistry import PluginRegistry
|
|
from UM.Qt.QtApplication import QtApplication
|
|
from UM.i18n import i18nCatalog
|
|
from UM.Settings.ContainerRegistry import ContainerRegistry
|
|
|
|
from cura.Settings.GlobalStack import GlobalStack
|
|
|
|
from .FirmwareUpdateCheckerJob import FirmwareUpdateCheckerJob
|
|
from .FirmwareUpdateCheckerLookup import FirmwareUpdateCheckerLookup, getSettingsKeyForMachine
|
|
from .FirmwareUpdateCheckerMessage import FirmwareUpdateCheckerMessage
|
|
|
|
i18n_catalog = i18nCatalog("cura")
|
|
|
|
|
|
## This Extension checks for new versions of the firmware based on the latest checked version number.
|
|
# The plugin is currently only usable for applications maintained by Ultimaker. But it should be relatively easy
|
|
# to change it to work for other applications.
|
|
class FirmwareUpdateChecker(Extension):
|
|
|
|
def __init__(self) -> None:
|
|
super().__init__()
|
|
|
|
# Listen to a Signal that indicates a change in the list of printers, just if the user has enabled the
|
|
# "check for updates" option
|
|
Application.getInstance().getPreferences().addPreference("info/automatic_update_check", True)
|
|
if Application.getInstance().getPreferences().getValue("info/automatic_update_check"):
|
|
ContainerRegistry.getInstance().containerAdded.connect(self._onContainerAdded)
|
|
|
|
# Partly initialize after creation, since we need our own path from the plugin-manager.
|
|
self._download_url = None
|
|
self._check_job = None
|
|
self._checked_printer_names = set() # type: Set[str]
|
|
self._lookups = None
|
|
QtApplication.pluginsLoaded.connect(self._onPluginsLoaded)
|
|
|
|
## Callback for the message that is spawned when there is a new version.
|
|
def _onActionTriggered(self, message, action):
|
|
if action == FirmwareUpdateCheckerMessage.STR_ACTION_DOWNLOAD:
|
|
machine_id = message.getMachineId()
|
|
download_url = self._lookups.getRedirectUserFor(machine_id)
|
|
if download_url is not None:
|
|
if QDesktopServices.openUrl(QUrl(download_url)):
|
|
Logger.log("i", "Redirected browser to {0} to show newly available firmware.".format(download_url))
|
|
else:
|
|
Logger.log("e", "Can't reach URL: {0}".format(download_url))
|
|
else:
|
|
Logger.log("e", "Can't find URL for {0}".format(machine_id))
|
|
|
|
def _onContainerAdded(self, container):
|
|
# Only take care when a new GlobalStack was added
|
|
if isinstance(container, GlobalStack):
|
|
self.checkFirmwareVersion(container, True)
|
|
|
|
def _onJobFinished(self, *args, **kwargs):
|
|
self._check_job = None
|
|
|
|
def _onPluginsLoaded(self):
|
|
if self._lookups is not None:
|
|
return
|
|
|
|
self._lookups = FirmwareUpdateCheckerLookup(os.path.join(PluginRegistry.getInstance().getPluginPath(
|
|
"FirmwareUpdateChecker"), "resources/machines.json"))
|
|
|
|
# Initialize the Preference called `latest_checked_firmware` that stores the last version
|
|
# checked for each printer.
|
|
for machine_id in self._lookups.getMachineIds():
|
|
Application.getInstance().getPreferences().addPreference(getSettingsKeyForMachine(machine_id), "")
|
|
|
|
## Connect with software.ultimaker.com, load latest.version and check version info.
|
|
# If the version info is different from the current version, spawn a message to
|
|
# allow the user to download it.
|
|
#
|
|
# \param silent type(boolean) Suppresses messages other than "new version found" messages.
|
|
# This is used when checking for a new firmware version at startup.
|
|
def checkFirmwareVersion(self, container = None, silent = False):
|
|
if self._lookups is None:
|
|
self._onPluginsLoaded()
|
|
|
|
container_name = container.definition.getName()
|
|
if container_name in self._checked_printer_names:
|
|
return
|
|
self._checked_printer_names.add(container_name)
|
|
|
|
self._check_job = FirmwareUpdateCheckerJob(container = container, silent = silent,
|
|
lookups = self._lookups,
|
|
callback = self._onActionTriggered)
|
|
self._check_job.start()
|
|
self._check_job.finished.connect(self._onJobFinished)
|