UltimakerCloudScope.py 1.2 KB

123456789101112131415161718192021222324252627282930313233343536
  1. # Copyright (c) 2021 Ultimaker B.V.
  2. # Cura is released under the terms of the LGPLv3 or higher.
  3. from PyQt6.QtNetwork import QNetworkRequest
  4. from UM.Logger import Logger
  5. from UM.TaskManagement.HttpRequestScope import DefaultUserAgentScope
  6. from typing import TYPE_CHECKING
  7. if TYPE_CHECKING:
  8. from cura.CuraApplication import CuraApplication
  9. from cura.API.Account import Account
  10. class UltimakerCloudScope(DefaultUserAgentScope):
  11. """
  12. Add an Authorization header to the request for Ultimaker Cloud Api requests, if available.
  13. Also add the user agent headers (see DefaultUserAgentScope).
  14. """
  15. def __init__(self, application: "CuraApplication"):
  16. super().__init__(application)
  17. api = application.getCuraAPI()
  18. self._account: Account = api.account
  19. def requestHook(self, request: QNetworkRequest):
  20. super().requestHook(request)
  21. token = self._account.accessToken
  22. if not self._account.isLoggedIn or token is None:
  23. Logger.debug("User is not logged in for Cloud API request to {url}".format(url = request.url().toDisplayString()))
  24. return
  25. header_dict = {
  26. "Authorization": "Bearer {}".format(token)
  27. }
  28. self.addHeaders(request, header_dict)