NetworkClusterPrinterOutputDevice.py 31 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680681682683684685686687688689690691692693694695696697698699700701702703704705706707708709710711712713714715716717718719720721722
  1. import datetime
  2. import getpass
  3. import gzip
  4. import json
  5. import os
  6. import os.path
  7. import time
  8. from enum import Enum
  9. from PyQt5.QtNetwork import QNetworkRequest, QHttpPart, QHttpMultiPart
  10. from PyQt5.QtCore import QUrl, QByteArray, pyqtSlot, pyqtProperty, QCoreApplication, QTimer, pyqtSignal, QObject
  11. from PyQt5.QtGui import QDesktopServices
  12. from PyQt5.QtNetwork import QNetworkAccessManager, QNetworkReply
  13. from PyQt5.QtQml import QQmlComponent, QQmlContext
  14. from UM.Application import Application
  15. from UM.Decorators import override
  16. from UM.Logger import Logger
  17. from UM.Message import Message
  18. from UM.OutputDevice import OutputDeviceError
  19. from UM.i18n import i18nCatalog
  20. from UM.Qt.Duration import Duration, DurationFormat
  21. from UM.PluginRegistry import PluginRegistry
  22. from . import NetworkPrinterOutputDevice
  23. i18n_catalog = i18nCatalog("cura")
  24. class OutputStage(Enum):
  25. ready = 0
  26. uploading = 2
  27. class NetworkClusterPrinterOutputDevice(NetworkPrinterOutputDevice.NetworkPrinterOutputDevice):
  28. printJobsChanged = pyqtSignal()
  29. printersChanged = pyqtSignal()
  30. selectedPrinterChanged = pyqtSignal()
  31. def __init__(self, key, address, properties, api_prefix):
  32. super().__init__(key, address, properties, api_prefix)
  33. # Store the address of the master.
  34. self._master_address = address
  35. name_property = properties.get(b"name", b"")
  36. if name_property:
  37. name = name_property.decode("utf-8")
  38. else:
  39. name = key
  40. self._authentication_state = NetworkPrinterOutputDevice.AuthState.Authenticated # The printer is always authenticated
  41. self.setName(name)
  42. description = i18n_catalog.i18nc("@action:button Preceded by 'Ready to'.", "Print over network")
  43. self.setShortDescription(description)
  44. self.setDescription(description)
  45. self._stage = OutputStage.ready
  46. host_override = os.environ.get("CLUSTER_OVERRIDE_HOST", "")
  47. if host_override:
  48. Logger.log(
  49. "w",
  50. "Environment variable CLUSTER_OVERRIDE_HOST is set to [%s], cluster hosts are now set to this host",
  51. host_override)
  52. self._host = "http://" + host_override
  53. else:
  54. self._host = "http://" + address
  55. # is the same as in NetworkPrinterOutputDevicePlugin
  56. self._cluster_api_version = "1"
  57. self._cluster_api_prefix = "/cluster-api/v" + self._cluster_api_version + "/"
  58. self._api_base_uri = self._host + self._cluster_api_prefix
  59. self._file_name = None
  60. self._progress_message = None
  61. self._request = None
  62. self._reply = None
  63. # The main reason to keep the 'multipart' form data on the object
  64. # is to prevent the Python GC from claiming it too early.
  65. self._multipart = None
  66. self._print_view = None
  67. self._request_job = []
  68. self._monitor_view_qml_path = os.path.join(os.path.dirname(os.path.abspath(__file__)), "ClusterMonitorItem.qml")
  69. self._control_view_qml_path = os.path.join(os.path.dirname(os.path.abspath(__file__)), "ClusterControlItem.qml")
  70. self._print_jobs = []
  71. self._print_job_by_printer_uuid = {}
  72. self._print_job_by_uuid = {} # Print jobs by their own uuid
  73. self._printers = []
  74. self._printers_dict = {} # by unique_name
  75. self._connected_printers_type_count = []
  76. self._automatic_printer = {"unique_name": "", "friendly_name": "Automatic"} # empty unique_name IS automatic selection
  77. self._selected_printer = self._automatic_printer
  78. self._cluster_status_update_timer = QTimer()
  79. self._cluster_status_update_timer.setInterval(5000)
  80. self._cluster_status_update_timer.setSingleShot(False)
  81. self._cluster_status_update_timer.timeout.connect(self._requestClusterStatus)
  82. self._can_pause = True
  83. self._can_abort = True
  84. self._can_pre_heat_bed = False
  85. self._cluster_size = int(properties.get(b"cluster_size", 0))
  86. self._cleanupRequest()
  87. #These are texts that are to be translated for future features.
  88. temporary_translation = i18n_catalog.i18n("This printer is not set up to host a group of connected Ultimaker 3 printers.")
  89. temporary_translation2 = i18n_catalog.i18nc("Count is number of printers.", "This printer is the host for a group of {count} connected Ultimaker 3 printers.").format(count = 3)
  90. temporary_translation3 = i18n_catalog.i18n("{printer_name} has finished printing '{job_name}'. Please collect the print and confirm clearing the build plate.") #When finished.
  91. temporary_translation4 = i18n_catalog.i18n("{printer_name} is reserved to print '{job_name}'. Please change the printer's configuration to match the job, for it to start printing.") #When configuration changed.
  92. ## No authentication, so requestAuthentication should do exactly nothing
  93. @pyqtSlot()
  94. def requestAuthentication(self, message_id = None, action_id = "Retry"):
  95. pass # Cura Connect doesn't do any authorization
  96. def setAuthenticationState(self, auth_state):
  97. self._authentication_state = NetworkPrinterOutputDevice.AuthState.Authenticated # The printer is always authenticated
  98. def _verifyAuthentication(self):
  99. pass
  100. def _checkAuthentication(self):
  101. Logger.log("d", "_checkAuthentication Cura Connect - nothing to be done")
  102. @pyqtProperty(QObject, notify=selectedPrinterChanged)
  103. def controlItem(self):
  104. # TODO: Probably not the nicest way to do this. This needs to be done better at some point in time.
  105. if not self._control_component:
  106. self._createControlViewFromQML()
  107. name = self._selected_printer.get("friendly_name")
  108. if name == self._automatic_printer.get("friendly_name") or name == "":
  109. return self._control_item
  110. # Let cura use the default.
  111. return None
  112. @pyqtSlot(int, result = str)
  113. def getTimeCompleted(self, time_remaining):
  114. current_time = time.time()
  115. datetime_completed = datetime.datetime.fromtimestamp(current_time + time_remaining)
  116. return "{hour:02d}:{minute:02d}".format(hour = datetime_completed.hour, minute = datetime_completed.minute)
  117. @pyqtSlot(int, result = str)
  118. def getDateCompleted(self, time_remaining):
  119. current_time = time.time()
  120. datetime_completed = datetime.datetime.fromtimestamp(current_time + time_remaining)
  121. return (datetime_completed.strftime("%a %b ") + "{day}".format(day=datetime_completed.day)).upper()
  122. @pyqtProperty(int, constant = True)
  123. def clusterSize(self):
  124. return self._cluster_size
  125. @pyqtProperty(str, notify=selectedPrinterChanged)
  126. def name(self):
  127. # Show the name of the selected printer.
  128. # This is not the nicest way to do this, but changes to the Cura UI are required otherwise.
  129. name = self._selected_printer.get("friendly_name")
  130. if name != self._automatic_printer.get("friendly_name"):
  131. return name
  132. # Return name of cluster master.
  133. return self._properties.get(b"name", b"").decode("utf-8")
  134. def connect(self):
  135. super().connect()
  136. self._cluster_status_update_timer.start()
  137. def close(self):
  138. super().close()
  139. self._cluster_status_update_timer.stop()
  140. def _setJobState(self, job_state):
  141. if not self._selected_printer:
  142. return
  143. selected_printer_uuid = self._printers_dict[self._selected_printer["unique_name"]]["uuid"]
  144. if selected_printer_uuid not in self._print_job_by_printer_uuid:
  145. return
  146. print_job_uuid = self._print_job_by_printer_uuid[selected_printer_uuid]["uuid"]
  147. url = QUrl(self._api_base_uri + "print_jobs/" + print_job_uuid + "/action")
  148. put_request = QNetworkRequest(url)
  149. put_request.setHeader(QNetworkRequest.ContentTypeHeader, "application/json")
  150. data = '{"action": "' + job_state + '"}'
  151. self._manager.put(put_request, data.encode())
  152. def _requestClusterStatus(self):
  153. # TODO: Handle timeout. We probably want to know if the cluster is still reachable or not.
  154. url = QUrl(self._api_base_uri + "printers/")
  155. printers_request = QNetworkRequest(url)
  156. self._addUserAgentHeader(printers_request)
  157. self._manager.get(printers_request)
  158. # See _finishedPrintersRequest()
  159. if self._printers: # if printers is not empty
  160. url = QUrl(self._api_base_uri + "print_jobs/")
  161. print_jobs_request = QNetworkRequest(url)
  162. self._addUserAgentHeader(print_jobs_request)
  163. self._manager.get(print_jobs_request)
  164. # See _finishedPrintJobsRequest()
  165. def _finishedPrintJobsRequest(self, reply):
  166. try:
  167. json_data = json.loads(bytes(reply.readAll()).decode("utf-8"))
  168. except json.decoder.JSONDecodeError:
  169. Logger.log("w", "Received an invalid print job state message: Not valid JSON.")
  170. return
  171. self.setPrintJobs(json_data)
  172. def _finishedPrintersRequest(self, reply):
  173. try:
  174. json_data = json.loads(bytes(reply.readAll()).decode("utf-8"))
  175. except json.decoder.JSONDecodeError:
  176. Logger.log("w", "Received an invalid print job state message: Not valid JSON.")
  177. return
  178. self.setPrinters(json_data)
  179. def materialHotendChangedMessage(self, callback):
  180. pass # Do nothing.
  181. def _startCameraStream(self):
  182. ## Request new image
  183. url = QUrl("http://" + self._printers_dict[self._selected_printer["unique_name"]]["ip_address"] + ":8080/?action=stream")
  184. self._image_request = QNetworkRequest(url)
  185. self._addUserAgentHeader(self._image_request)
  186. self._image_reply = self._manager.get(self._image_request)
  187. self._image_reply.downloadProgress.connect(self._onStreamDownloadProgress)
  188. def spawnPrintView(self):
  189. if self._print_view is None:
  190. path = QUrl.fromLocalFile(os.path.join(self._plugin_path, "PrintWindow.qml"))
  191. component = QQmlComponent(Application.getInstance()._engine, path)
  192. self._print_context = QQmlContext(Application.getInstance()._engine.rootContext())
  193. self._print_context.setContextProperty("OutputDevice", self)
  194. self._print_view = component.create(self._print_context)
  195. if component.isError():
  196. Logger.log("e", " Errors creating component: \n%s", "\n".join(
  197. [e.toString() for e in component.errors()]))
  198. if self._print_view is not None:
  199. self._print_view.show()
  200. ## Store job info, show Print view for settings
  201. def requestWrite(self, nodes, file_name=None, filter_by_machine=False, file_handler=None, **kwargs):
  202. self._selected_printer = self._automatic_printer # reset to default option
  203. self._request_job = [nodes, file_name, filter_by_machine, file_handler, kwargs]
  204. if self._stage != OutputStage.ready:
  205. if self._error_message:
  206. self._error_message.hide()
  207. self._error_message = Message(
  208. i18n_catalog.i18nc("@info:status",
  209. "Sending new jobs (temporarily) blocked, still sending the previous print job."))
  210. self._error_message.show()
  211. return
  212. if len(self._printers) > 1:
  213. self.spawnPrintView() # Ask user how to print it.
  214. elif len(self._printers) == 1:
  215. # If there is only one printer, don't bother asking.
  216. self.selectAutomaticPrinter()
  217. self.sendPrintJob()
  218. else:
  219. # Cluster has no printers, warn the user of this.
  220. if self._error_message:
  221. self._error_message.hide()
  222. self._error_message = Message(
  223. i18n_catalog.i18nc("@info:status",
  224. "Unable to send new print job: this 3D printer is not (yet) set up to host a group of connected Ultimaker 3 printers."))
  225. self._error_message.show()
  226. ## Actually send the print job, called from the dialog
  227. # :param: require_printer_name: name of printer, or ""
  228. @pyqtSlot()
  229. def sendPrintJob(self):
  230. nodes, file_name, filter_by_machine, file_handler, kwargs = self._request_job
  231. require_printer_name = self._selected_printer["unique_name"]
  232. self._send_gcode_start = time.time()
  233. Logger.log("d", "Sending print job [%s] to host..." % file_name)
  234. if self._stage != OutputStage.ready:
  235. Logger.log("d", "Unable to send print job as the state is %s", self._stage)
  236. raise OutputDeviceError.DeviceBusyError()
  237. self._stage = OutputStage.uploading
  238. self._file_name = "%s.gcode.gz" % file_name
  239. self._showProgressMessage()
  240. new_request = self._buildSendPrintJobHttpRequest(require_printer_name)
  241. if new_request is None or self._stage != OutputStage.uploading:
  242. return
  243. self._request = new_request
  244. self._reply = self._manager.post(self._request, self._multipart)
  245. self._reply.uploadProgress.connect(self._onUploadProgress)
  246. # See _finishedPostPrintJobRequest()
  247. def _buildSendPrintJobHttpRequest(self, require_printer_name):
  248. api_url = QUrl(self._api_base_uri + "print_jobs/")
  249. request = QNetworkRequest(api_url)
  250. # Create multipart request and add the g-code.
  251. self._multipart = QHttpMultiPart(QHttpMultiPart.FormDataType)
  252. # Add gcode
  253. part = QHttpPart()
  254. part.setHeader(QNetworkRequest.ContentDispositionHeader,
  255. 'form-data; name="file"; filename="%s"' % self._file_name)
  256. gcode = getattr(Application.getInstance().getController().getScene(), "gcode_list")
  257. compressed_gcode = self._compressGcode(gcode)
  258. if compressed_gcode is None:
  259. return None # User aborted print, so stop trying.
  260. part.setBody(compressed_gcode)
  261. self._multipart.append(part)
  262. # require_printer_name "" means automatic
  263. if require_printer_name:
  264. self._multipart.append(self.__createKeyValueHttpPart("require_printer_name", require_printer_name))
  265. user_name = self.__get_username()
  266. if user_name is None:
  267. user_name = "unknown"
  268. self._multipart.append(self.__createKeyValueHttpPart("owner", user_name))
  269. self._addUserAgentHeader(request)
  270. return request
  271. def _compressGcode(self, gcode):
  272. self._compressing_print = True
  273. batched_line = ""
  274. max_chars_per_line = int(1024 * 1024 / 4) # 1 / 4 MB
  275. byte_array_file_data = b""
  276. def _compressDataAndNotifyQt(data_to_append):
  277. compressed_data = gzip.compress(data_to_append.encode("utf-8"))
  278. self._progress_message.setProgress(-1) # Tickle the message so that it's clear that it's still being used.
  279. QCoreApplication.processEvents() # Ensure that the GUI does not freeze.
  280. # Pretend that this is a response, as zipping might take a bit of time.
  281. self._last_response_time = time.time()
  282. return compressed_data
  283. if gcode is None:
  284. Logger.log("e", "Unable to find sliced gcode, returning empty.")
  285. return byte_array_file_data
  286. for line in gcode:
  287. if not self._compressing_print:
  288. self._progress_message.hide()
  289. return None # Stop trying to zip, abort was called.
  290. batched_line += line
  291. # if the gcode was read from a gcode file, self._gcode will be a list of all lines in that file.
  292. # Compressing line by line in this case is extremely slow, so we need to batch them.
  293. if len(batched_line) < max_chars_per_line:
  294. continue
  295. byte_array_file_data += _compressDataAndNotifyQt(batched_line)
  296. batched_line = ""
  297. # Also compress the leftovers.
  298. if batched_line:
  299. byte_array_file_data += _compressDataAndNotifyQt(batched_line)
  300. return byte_array_file_data
  301. def __createKeyValueHttpPart(self, key, value):
  302. metadata_part = QHttpPart()
  303. metadata_part.setHeader(QNetworkRequest.ContentTypeHeader, 'text/plain')
  304. metadata_part.setHeader(QNetworkRequest.ContentDispositionHeader, 'form-data; name="%s"' % (key))
  305. metadata_part.setBody(bytearray(value, "utf8"))
  306. return metadata_part
  307. def __get_username(self):
  308. try:
  309. return getpass.getuser()
  310. except:
  311. Logger.log("d", "Could not get the system user name, returning 'unknown' instead.")
  312. return None
  313. def _finishedPrintJobPostRequest(self, reply):
  314. self._stage = OutputStage.ready
  315. if self._progress_message:
  316. self._progress_message.hide()
  317. self._progress_message = None
  318. self.writeFinished.emit(self)
  319. if reply.error():
  320. self._showRequestFailedMessage(reply)
  321. self.writeError.emit(self)
  322. else:
  323. self._showRequestSucceededMessage()
  324. self.writeSuccess.emit(self)
  325. self._cleanupRequest()
  326. def _showRequestFailedMessage(self, reply):
  327. if reply is not None:
  328. Logger.log("w", "Unable to send print job to group {cluster_name}: {error_string} ({error})".format(
  329. cluster_name = self.getName(),
  330. error_string = str(reply.errorString()),
  331. error = str(reply.error())))
  332. error_message_template = i18n_catalog.i18nc("@info:status", "Unable to send print job to group {cluster_name}.")
  333. message = Message(text=error_message_template.format(
  334. cluster_name = self.getName()))
  335. message.show()
  336. def _showRequestSucceededMessage(self):
  337. confirmation_message_template = i18n_catalog.i18nc(
  338. "@info:status",
  339. "Sent {file_name} to group {cluster_name}."
  340. )
  341. file_name = os.path.basename(self._file_name).split(".")[0]
  342. message_text = confirmation_message_template.format(cluster_name = self.getName(), file_name = file_name)
  343. message = Message(text=message_text)
  344. button_text = i18n_catalog.i18nc("@action:button", "Show print jobs")
  345. button_tooltip = i18n_catalog.i18nc("@info:tooltip", "Opens the print jobs interface in your browser.")
  346. message.addAction("open_browser", button_text, "globe", button_tooltip)
  347. message.actionTriggered.connect(self._onMessageActionTriggered)
  348. message.show()
  349. def setPrintJobs(self, print_jobs):
  350. #TODO: hack, last seen messes up the check, so drop it.
  351. for job in print_jobs:
  352. del job["last_seen"]
  353. # Strip any extensions
  354. job["name"] = self._removeGcodeExtension(job["name"])
  355. if self._print_jobs != print_jobs:
  356. old_print_jobs = self._print_jobs
  357. self._print_jobs = print_jobs
  358. self._notifyFinishedPrintJobs(old_print_jobs, print_jobs)
  359. self._notifyConfigurationChangeRequired(old_print_jobs, print_jobs)
  360. # Yes, this is a hacky way of doing it, but it's quick and the API doesn't give the print job per printer
  361. # for some reason. ugh.
  362. self._print_job_by_printer_uuid = {}
  363. self._print_job_by_uuid = {}
  364. for print_job in print_jobs:
  365. if "printer_uuid" in print_job and print_job["printer_uuid"] is not None:
  366. self._print_job_by_printer_uuid[print_job["printer_uuid"]] = print_job
  367. self._print_job_by_uuid[print_job["uuid"]] = print_job
  368. self.printJobsChanged.emit()
  369. def _removeGcodeExtension(self, name):
  370. parts = name.split(".")
  371. if parts[-1].upper() == "GZ":
  372. parts = parts[:-1]
  373. if parts[-1].upper() == "GCODE":
  374. parts = parts[:-1]
  375. return ".".join(parts)
  376. def _notifyFinishedPrintJobs(self, old_print_jobs, new_print_jobs):
  377. """Notify the user when any of their print jobs have just completed.
  378. Arguments:
  379. old_print_jobs -- the previous list of print job status information as returned by the cluster REST API.
  380. new_print_jobs -- the current list of print job status information as returned by the cluster REST API.
  381. """
  382. if old_print_jobs is None:
  383. return
  384. username = self.__get_username()
  385. if username is None:
  386. return
  387. our_old_print_jobs = self.__filterOurPrintJobs(old_print_jobs)
  388. our_old_not_finished_print_jobs = [pj for pj in our_old_print_jobs if pj["status"] != "wait_cleanup"]
  389. our_new_print_jobs = self.__filterOurPrintJobs(new_print_jobs)
  390. our_new_finished_print_jobs = [pj for pj in our_new_print_jobs if pj["status"] == "wait_cleanup"]
  391. old_not_finished_print_job_uuids = set([pj["uuid"] for pj in our_old_not_finished_print_jobs])
  392. for print_job in our_new_finished_print_jobs:
  393. if print_job["uuid"] in old_not_finished_print_job_uuids:
  394. printer_name = self.__getPrinterNameFromUuid(print_job["printer_uuid"])
  395. if printer_name is None:
  396. printer_name = i18n_catalog.i18nc("@label", "Unknown")
  397. message_text = (i18n_catalog.i18nc("@info:status",
  398. "Printer '{printer_name}' has finished printing '{job_name}'.")
  399. .format(printer_name=printer_name, job_name=print_job["name"]))
  400. message = Message(text=message_text, title=i18n_catalog.i18nc("@info:status", "Print finished"))
  401. Application.getInstance().showMessage(message)
  402. Application.getInstance().showToastMessage(
  403. i18n_catalog.i18nc("@info:status", "Print finished"),
  404. message_text)
  405. def __filterOurPrintJobs(self, print_jobs):
  406. username = self.__get_username()
  407. return [print_job for print_job in print_jobs if print_job["owner"] == username]
  408. def _notifyConfigurationChangeRequired(self, old_print_jobs, new_print_jobs):
  409. if old_print_jobs is None:
  410. return
  411. old_change_required_print_jobs = self.__filterConfigChangePrintJobs(self.__filterOurPrintJobs(old_print_jobs))
  412. new_change_required_print_jobs = self.__filterConfigChangePrintJobs(self.__filterOurPrintJobs(new_print_jobs))
  413. old_change_required_print_job_uuids = set([pj["uuid"] for pj in old_change_required_print_jobs])
  414. for print_job in new_change_required_print_jobs:
  415. if print_job["uuid"] not in old_change_required_print_job_uuids:
  416. printer_name = self.__getPrinterNameFromUuid(print_job["assigned_to"])
  417. if printer_name is None:
  418. # don't report on yet unknown printers
  419. continue
  420. message_text = (i18n_catalog.i18n("{printer_name} is reserved to print '{job_name}'. Please change the printer's configuration to match the job, for it to start printing.")
  421. .format(printer_name=printer_name, job_name=print_job["name"]))
  422. message = Message(text=message_text, title=i18n_catalog.i18nc("@label:status", "Action required"))
  423. Application.getInstance().showMessage(message)
  424. Application.getInstance().showToastMessage(
  425. i18n_catalog.i18nc("@label:status", "Action required"),
  426. message_text)
  427. def __filterConfigChangePrintJobs(self, print_jobs):
  428. return filter(self.__isConfigurationChangeRequiredPrintJob, print_jobs)
  429. def __isConfigurationChangeRequiredPrintJob(self, print_job):
  430. if print_job["status"] == "queued":
  431. changes_required = print_job.get("configuration_changes_required", [])
  432. return len(changes_required) != 0
  433. return False
  434. def __getPrinterNameFromUuid(self, printer_uuid):
  435. for printer in self._printers:
  436. if printer["uuid"] == printer_uuid:
  437. return printer["friendly_name"]
  438. return None
  439. def setPrinters(self, printers):
  440. if self._printers != printers:
  441. self._connected_printers_type_count = []
  442. printers_count = {}
  443. self._printers = printers
  444. self._printers_dict = dict((p["unique_name"], p) for p in printers) # for easy lookup by unique_name
  445. for printer in printers:
  446. variant = printer["machine_variant"]
  447. if variant in printers_count:
  448. printers_count[variant] += 1
  449. else:
  450. printers_count[variant] = 1
  451. for type in printers_count:
  452. self._connected_printers_type_count.append({"machine_type": type, "count": printers_count[type]})
  453. self.printersChanged.emit()
  454. @pyqtProperty("QVariantList", notify=printersChanged)
  455. def connectedPrintersTypeCount(self):
  456. return self._connected_printers_type_count
  457. @pyqtProperty("QVariantList", notify=printersChanged)
  458. def connectedPrinters(self):
  459. return self._printers
  460. @pyqtProperty(int, notify=printJobsChanged)
  461. def numJobsPrinting(self):
  462. num_jobs_printing = 0
  463. for job in self._print_jobs:
  464. if job["status"] in ["printing", "wait_cleanup", "sent_to_printer", "pre_print", "post_print"]:
  465. num_jobs_printing += 1
  466. return num_jobs_printing
  467. @pyqtProperty(int, notify=printJobsChanged)
  468. def numJobsQueued(self):
  469. num_jobs_queued = 0
  470. for job in self._print_jobs:
  471. if job["status"] == "queued":
  472. num_jobs_queued += 1
  473. return num_jobs_queued
  474. @pyqtProperty("QVariantMap", notify=printJobsChanged)
  475. def printJobsByUUID(self):
  476. return self._print_job_by_uuid
  477. @pyqtProperty("QVariantMap", notify=printJobsChanged)
  478. def printJobsByPrinterUUID(self):
  479. return self._print_job_by_printer_uuid
  480. @pyqtProperty("QVariantList", notify=printJobsChanged)
  481. def printJobs(self):
  482. return self._print_jobs
  483. @pyqtProperty("QVariantList", notify=printersChanged)
  484. def printers(self):
  485. return [self._automatic_printer, ] + self._printers
  486. @pyqtSlot(str, str)
  487. def selectPrinter(self, unique_name, friendly_name):
  488. self.stopCamera()
  489. self._selected_printer = {"unique_name": unique_name, "friendly_name": friendly_name}
  490. Logger.log("d", "Selected printer: %s %s", friendly_name, unique_name)
  491. # TODO: Probably not the nicest way to do this. This needs to be done better at some point in time.
  492. if unique_name == "":
  493. self._address = self._master_address
  494. else:
  495. self._address = self._printers_dict[self._selected_printer["unique_name"]]["ip_address"]
  496. self.selectedPrinterChanged.emit()
  497. def _updateJobState(self, job_state):
  498. name = self._selected_printer.get("friendly_name")
  499. if name == "" or name == "Automatic":
  500. # TODO: This is now a bit hacked; If no printer is selected, don't show job state.
  501. if self._job_state != "":
  502. self._job_state = ""
  503. self.jobStateChanged.emit()
  504. else:
  505. if self._job_state != job_state:
  506. self._job_state = job_state
  507. self.jobStateChanged.emit()
  508. @pyqtSlot()
  509. def selectAutomaticPrinter(self):
  510. self.stopCamera()
  511. self._selected_printer = self._automatic_printer
  512. self.selectedPrinterChanged.emit()
  513. @pyqtProperty("QVariant", notify=selectedPrinterChanged)
  514. def selectedPrinterName(self):
  515. return self._selected_printer.get("unique_name", "")
  516. def getPrintJobsUrl(self):
  517. return self._host + "/print_jobs"
  518. def getPrintersUrl(self):
  519. return self._host + "/printers"
  520. def _showProgressMessage(self):
  521. progress_message_template = i18n_catalog.i18nc("@info:progress",
  522. "Sending <filename>{file_name}</filename> to group {cluster_name}")
  523. file_name = os.path.basename(self._file_name).split(".")[0]
  524. self._progress_message = Message(progress_message_template.format(file_name = file_name, cluster_name = self.getName()), 0, False, -1)
  525. self._progress_message.addAction("Abort", i18n_catalog.i18nc("@action:button", "Cancel"), None, "")
  526. self._progress_message.actionTriggered.connect(self._onMessageActionTriggered)
  527. self._progress_message.show()
  528. def _addUserAgentHeader(self, request):
  529. request.setRawHeader(b"User-agent", b"CuraPrintClusterOutputDevice Plugin")
  530. def _cleanupRequest(self):
  531. self._request = None
  532. self._stage = OutputStage.ready
  533. self._file_name = None
  534. def _onFinished(self, reply):
  535. super()._onFinished(reply)
  536. reply_url = reply.url().toString()
  537. status_code = reply.attribute(QNetworkRequest.HttpStatusCodeAttribute)
  538. if status_code == 500:
  539. Logger.log("w", "Request to {url} returned a 500.".format(url = reply_url))
  540. return
  541. if reply.error() == QNetworkReply.ContentOperationNotPermittedError:
  542. # It was probably "/api/v1/materials" for legacy UM3
  543. return
  544. if reply.error() == QNetworkReply.ContentNotFoundError:
  545. # It was probably "/api/v1/print_job" for legacy UM3
  546. return
  547. if reply.operation() == QNetworkAccessManager.PostOperation:
  548. if self._cluster_api_prefix + "print_jobs" in reply_url:
  549. self._finishedPrintJobPostRequest(reply)
  550. return
  551. # We need to do this check *after* we process the post operation!
  552. # If the sending of g-code is cancelled by the user it will result in an error, but we do need to handle this.
  553. if reply.error() != QNetworkReply.NoError:
  554. Logger.log("e", "After requesting [%s] we got a network error [%s]. Not processing anything...", reply_url, reply.error())
  555. return
  556. elif reply.operation() == QNetworkAccessManager.GetOperation:
  557. if self._cluster_api_prefix + "print_jobs" in reply_url:
  558. self._finishedPrintJobsRequest(reply)
  559. elif self._cluster_api_prefix + "printers" in reply_url:
  560. self._finishedPrintersRequest(reply)
  561. @pyqtSlot()
  562. def openPrintJobControlPanel(self):
  563. Logger.log("d", "Opening print job control panel...")
  564. QDesktopServices.openUrl(QUrl(self.getPrintJobsUrl()))
  565. @pyqtSlot()
  566. def openPrinterControlPanel(self):
  567. Logger.log("d", "Opening printer control panel...")
  568. QDesktopServices.openUrl(QUrl(self.getPrintersUrl()))
  569. def _onMessageActionTriggered(self, message, action):
  570. if action == "open_browser":
  571. QDesktopServices.openUrl(QUrl(self.getPrintJobsUrl()))
  572. if action == "Abort":
  573. Logger.log("d", "User aborted sending print to remote.")
  574. self._progress_message.hide()
  575. self._compressing_print = False
  576. if self._reply:
  577. self._reply.abort()
  578. self._stage = OutputStage.ready
  579. Application.getInstance().showPrintMonitor.emit(False)
  580. @pyqtSlot(int, result=str)
  581. def formatDuration(self, seconds):
  582. return Duration(seconds).getDisplayString(DurationFormat.Format.Short)
  583. ## For cluster below
  584. def _get_plugin_directory_name(self):
  585. current_file_absolute_path = os.path.realpath(__file__)
  586. directory_path = os.path.dirname(current_file_absolute_path)
  587. _, directory_name = os.path.split(directory_path)
  588. return directory_name
  589. @property
  590. def _plugin_path(self):
  591. return PluginRegistry.getInstance().getPluginPath(self._get_plugin_directory_name())