Linux PyQt tray application to control OctoPrint instances
Ви не можете вибрати більше 25 тем Теми мають розпочинатися з літери або цифри, можуть містити дефіси (-) і не повинні перевищувати 35 символів.

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680681682683684685686687688689690691692693694695696697698699700701702703704705706707708709710711712713714715716717718719720721722723724725726727728729730731732733734735736737738739740741742743744745746747748749750751752753754755756757758759760761762763764765766767768769770771772773774775776777778779780781782783784785786787788789790791792793794795796797798799800801802803804805806807808809810811812813814815816817818819820821822823824825826827828829830831832833834835836837838839840841842843844845846847848849850851852853854855856857
  1. #!/usr/bin/env python3
  2. # OctoTray Linux Qt System Tray OctoPrint client
  3. #
  4. # depends on:
  5. # - python-pyqt5
  6. #
  7. # see also:
  8. # https://doc.qt.io/qt-5/qtwidgets-widgets-imageviewer-example.html
  9. # https://stackoverflow.com/a/22618496
  10. import json
  11. import sys
  12. import os
  13. import time
  14. import string
  15. import urllib.parse
  16. import urllib.request
  17. from os import path
  18. from PyQt5 import QtWidgets, QtGui, QtCore, QtNetwork
  19. from PyQt5.QtWidgets import QSystemTrayIcon, QAction, QMenu, QMessageBox, QWidget, QLabel, QVBoxLayout, QHBoxLayout, QDesktopWidget, QSizePolicy, QSlider, QLayout, QTableWidget, QTableWidgetItem, QPushButton
  20. from PyQt5.QtGui import QIcon, QPixmap, QImageReader, QDesktopServices
  21. from PyQt5.QtCore import QCoreApplication, QSettings, QUrl, QTimer, QSize, Qt, QSettings
  22. class SettingsWindow(QWidget):
  23. columns = [ "Hostname", "API Key", "Tool Preheat", "Bed Preheat" ]
  24. presets = [ "octopi.local", "000000000_API_KEY_HERE_000000000", "0", "0" ]
  25. def __init__(self, parent, *args, **kwargs):
  26. super(SettingsWindow, self).__init__(*args, **kwargs)
  27. self.parent = parent
  28. self.setWindowTitle(parent.name + " Settings")
  29. self.setWindowIcon(parent.icon)
  30. box = QVBoxLayout()
  31. self.setLayout(box)
  32. helpText = "Usage:\n"
  33. helpText += "1st Column: Printer Hostname or IP address\n"
  34. helpText += "2nd Column: OctoPrint API Key (32 char hexadecimal)\n"
  35. helpText += "3rd Column: Tool Preheat Temperature (0 to disable)\n"
  36. helpText += "4th Column: Bed Preheat Temperature (0 to disable)"
  37. self.helpText = QLabel(helpText)
  38. box.addWidget(self.helpText, 0)
  39. box.setAlignment(self.helpText, Qt.AlignHCenter)
  40. buttons = QHBoxLayout()
  41. box.addLayout(buttons, 0)
  42. self.add = QPushButton("&Add Printer")
  43. self.add.clicked.connect(self.addPrinter)
  44. buttons.addWidget(self.add)
  45. self.remove = QPushButton("&Remove Printer")
  46. self.remove.clicked.connect(self.removePrinter)
  47. buttons.addWidget(self.remove)
  48. printers = self.parent.readSettings()
  49. self.rows = len(printers)
  50. self.table = QTableWidget(self.rows, len(self.columns))
  51. box.addWidget(self.table, 1)
  52. for i in range(0, self.rows):
  53. p = printers[i]
  54. for j in range(0, len(self.columns)):
  55. text = p[j]
  56. if (j >= 2) and (j <= 3) and (text == None):
  57. text = "0"
  58. item = QTableWidgetItem(text)
  59. self.table.setItem(i, j, item)
  60. buttons2 = QHBoxLayout()
  61. box.addLayout(buttons2, 0)
  62. self.up = QPushButton("Move &Up")
  63. self.up.clicked.connect(self.moveUp)
  64. buttons2.addWidget(self.up)
  65. self.down = QPushButton("Move &Down")
  66. self.down.clicked.connect(self.moveDown)
  67. buttons2.addWidget(self.down)
  68. self.table.setHorizontalHeaderLabels(self.columns)
  69. self.table.setSizeAdjustPolicy(QtWidgets.QAbstractScrollArea.AdjustToContents)
  70. self.table.resizeColumnsToContents()
  71. if self.rows <= 0:
  72. self.addPrinter()
  73. def tableToList(self):
  74. printers = []
  75. for i in range(0, self.rows):
  76. p = []
  77. for j in range(0, len(self.columns)):
  78. text = self.table.item(i, j).text()
  79. if (j >= 2) and (j <= 3) and (text == "0"):
  80. text = None
  81. p.append(text)
  82. printers.append(p)
  83. return printers
  84. def settingsValid(self, printers):
  85. for p in printers:
  86. # p[0] needs to be valid hostname or IP
  87. # TODO
  88. # p[1] needs to be valid API key (hexadecimal, 32 chars)
  89. if (len(p[1]) != 32) or not all(c in string.hexdigits for c in p[1]):
  90. return (False, "API Key not 32-digit hexadecimal")
  91. # p[2] and p[3] need to be integer temperatures (0...999)
  92. for s in [ p[2], p[3] ]:
  93. if s == None:
  94. s = "0"
  95. if (len(s) < 1) or (len(s) > 3) or not all(c in string.digits for c in s):
  96. return (False, "Temperature not a number from 0...999")
  97. return (True, "")
  98. def closeEvent(self, event):
  99. oldPrinters = [item[0:len(self.columns)] for item in self.parent.printers]
  100. newPrinters = self.tableToList()
  101. valid, errorText = self.settingsValid(newPrinters)
  102. if valid == False:
  103. r = self.parent.showDialog(self.parent.name + " Settings Invalid", errorText + "!", "Do you want to edit it again?", True, True, False)
  104. if r == True:
  105. event.ignore()
  106. return
  107. else:
  108. self.parent.removeSettingsWindow()
  109. return
  110. if oldPrinters != newPrinters:
  111. r = self.parent.showDialog(self.parent.name + " Settings Changed", "Do you want to save the new list of printers?", "This will restart the application!", True, False, False)
  112. if r == True:
  113. self.parent.writeSettings(newPrinters)
  114. self.parent.restartApp()
  115. self.parent.removeSettingsWindow()
  116. def addPrinter(self):
  117. self.rows += 1
  118. self.table.setRowCount(self.rows)
  119. for i in range(0, len(self.columns)):
  120. item = QTableWidgetItem(self.presets[i])
  121. self.table.setItem(self.rows - 1, i, item)
  122. self.table.resizeColumnsToContents()
  123. def removePrinter(self):
  124. r = self.table.currentRow()
  125. if (r >= 0) and (r < self.rows):
  126. self.rows -= 1
  127. self.table.removeRow(r)
  128. def moveUp(self):
  129. i = self.table.currentRow()
  130. if i <= 0:
  131. return
  132. host = self.table.item(i, 0).text()
  133. key = self.table.item(i, 1).text()
  134. self.table.item(i, 0).setText(self.table.item(i - 1, 0).text())
  135. self.table.item(i, 1).setText(self.table.item(i - 1, 1).text())
  136. self.table.item(i - 1, 0).setText(host)
  137. self.table.item(i - 1, 1).setText(key)
  138. def moveDown(self):
  139. i = self.table.currentRow()
  140. if i >= (self.rows - 1):
  141. return
  142. host = self.table.item(i, 0).text()
  143. key = self.table.item(i, 1).text()
  144. self.table.item(i, 0).setText(self.table.item(i + 1, 0).text())
  145. self.table.item(i, 1).setText(self.table.item(i + 1, 1).text())
  146. self.table.item(i + 1, 0).setText(host)
  147. self.table.item(i + 1, 1).setText(key)
  148. class AspectRatioPixmapLabel(QLabel):
  149. def __init__(self, *args, **kwargs):
  150. super(AspectRatioPixmapLabel, self).__init__(*args, **kwargs)
  151. self.setMinimumSize(1, 1)
  152. self.setScaledContents(False)
  153. self.pix = QPixmap(0, 0)
  154. def setPixmap(self, p):
  155. self.pix = p
  156. super(AspectRatioPixmapLabel, self).setPixmap(self.scaledPixmap())
  157. def heightForWidth(self, width):
  158. if self.pix.isNull():
  159. return self.height()
  160. else:
  161. return (self.pix.height() * width) / self.pix.width()
  162. def sizeHint(self):
  163. w = self.width()
  164. return QSize(int(w), int(self.heightForWidth(w)))
  165. def scaledPixmap(self):
  166. return self.pix.scaled(self.size(), Qt.KeepAspectRatio, Qt.SmoothTransformation)
  167. def resizeEvent(self, e):
  168. if not self.pix.isNull():
  169. super(AspectRatioPixmapLabel, self).setPixmap(self.scaledPixmap())
  170. class CamWindow(QWidget):
  171. reloadDelayDefault = 1000 # in ms
  172. statusDelay = 5 * 1000 # in ms
  173. reloadOn = True
  174. sliderFactor = 100
  175. def __init__(self, parent, printer, *args, **kwargs):
  176. super(CamWindow, self).__init__(*args, **kwargs)
  177. self.app = parent.app
  178. self.manager = parent.manager
  179. self.manager.finished.connect(self.handleResponse)
  180. self.parent = parent
  181. self.printer = printer
  182. self.host = self.printer[0]
  183. self.url = "http://" + self.host + ":8080/?action=snapshot"
  184. self.setWindowTitle(parent.name + " Webcam Stream")
  185. self.setWindowIcon(parent.icon)
  186. box = QVBoxLayout()
  187. self.setLayout(box)
  188. label = QLabel(self.url)
  189. box.addWidget(label, 0)
  190. box.setAlignment(label, Qt.AlignHCenter)
  191. self.img = AspectRatioPixmapLabel()
  192. self.img.setPixmap(QPixmap(640, 480))
  193. box.addWidget(self.img, 1)
  194. slide = QHBoxLayout()
  195. box.addLayout(slide, 0)
  196. self.slider = QSlider(Qt.Horizontal)
  197. self.slider.setMinimum(int(0 / self.sliderFactor))
  198. self.slider.setMaximum(int(2000 / self.sliderFactor))
  199. self.slider.setTickInterval(int(100 / self.sliderFactor))
  200. self.slider.setPageStep(int(100 / self.sliderFactor))
  201. self.slider.setSingleStep(int(100 / self.sliderFactor))
  202. self.slider.setTickPosition(QSlider.TicksBelow)
  203. self.slider.setValue(int(self.reloadDelayDefault / self.sliderFactor))
  204. self.slider.valueChanged.connect(self.sliderChanged)
  205. slide.addWidget(self.slider, 1)
  206. self.slideLabel = QLabel(str(self.reloadDelayDefault) + "ms")
  207. slide.addWidget(self.slideLabel, 0)
  208. self.statusLabel = QLabel("Status: unavailable")
  209. box.addWidget(self.statusLabel, 0)
  210. box.setAlignment(label, Qt.AlignHCenter)
  211. self.loadImage()
  212. self.loadStatus()
  213. def getHost(self):
  214. return self.host
  215. def sliderChanged(self):
  216. self.slideLabel.setText(str(self.slider.value() * self.sliderFactor) + "ms")
  217. def closeEvent(self, event):
  218. self.reloadOn = False
  219. self.url = ""
  220. self.parent.removeWebcamWindow(self)
  221. def scheduleLoadImage(self):
  222. if self.reloadOn:
  223. QTimer.singleShot(self.slider.value() * self.sliderFactor, self.loadImage)
  224. def scheduleLoadStatus(self):
  225. if self.reloadOn:
  226. QTimer.singleShot(self.statusDelay, self.loadStatus)
  227. def loadImage(self):
  228. url = QUrl(self.url)
  229. request = QtNetwork.QNetworkRequest(url)
  230. self.manager.get(request)
  231. def loadStatus(self):
  232. s = "Status: "
  233. t = self.parent.getTemperatureString(self.host, self.printer[1])
  234. if len(t) > 0:
  235. s += t
  236. else:
  237. s += "Unknown"
  238. progress = self.parent.getProgress(self.host, self.printer[1])
  239. if ("completion" in progress) and ("printTime" in progress) and ("printTimeLeft" in progress) and (progress["completion"] != None) and (progress["printTime"] != None) and (progress["printTimeLeft"] != None):
  240. s += " - %.1f%%" % progress["completion"]
  241. s += " - runtime "
  242. s += time.strftime("%H:%M:%S", time.gmtime(progress["printTime"]))
  243. s += " - "
  244. s += time.strftime("%H:%M:%S", time.gmtime(progress["printTimeLeft"])) + " left"
  245. self.statusLabel.setText(s)
  246. self.scheduleLoadStatus()
  247. def handleResponse(self, reply):
  248. if reply.url().url() == self.url:
  249. if reply.error() == QtNetwork.QNetworkReply.NoError:
  250. reader = QImageReader(reply)
  251. reader.setAutoTransform(True)
  252. image = reader.read()
  253. if image != None:
  254. if image.colorSpace().isValid():
  255. image.convertToColorSpace(QColorSpace.SRgb)
  256. self.img.setPixmap(QPixmap.fromImage(image))
  257. self.scheduleLoadImage()
  258. else:
  259. print("Error decoding image: " + reader.errorString())
  260. else:
  261. print("Error loading image: " + reply.errorString())
  262. class OctoTray():
  263. name = "OctoTray"
  264. vendor = "xythobuz"
  265. version = "0.3"
  266. iconName = "octotray_icon.png"
  267. iconPaths = [
  268. path.abspath(path.dirname(__file__)),
  269. "data",
  270. "/usr/share/pixmaps",
  271. ".",
  272. "..",
  273. "../data"
  274. ]
  275. networkTimeout = 2.0 # in s
  276. # list of lists, inner lists contain printer data:
  277. # first elements as in SettingsWindow.columns
  278. # 0=host 1=key 2=tool-preheat 3=bed-preheat
  279. # rest used for system-commands, menu, actions
  280. printers = []
  281. statesWithWarning = [
  282. "Printing", "Pausing", "Paused"
  283. ]
  284. camWindows = []
  285. settingsWindow = None
  286. def __init__(self, app):
  287. QCoreApplication.setApplicationName(self.name)
  288. self.app = app
  289. if not QSystemTrayIcon.isSystemTrayAvailable():
  290. self.showDialog("OctoTray Error", "System Tray is not available on this platform!", "", False, False, True)
  291. sys.exit(0)
  292. self.manager = QtNetwork.QNetworkAccessManager()
  293. self.menu = QMenu()
  294. self.printers = self.readSettings()
  295. unknownCount = 0
  296. for p in self.printers:
  297. method = self.getMethod(p[0], p[1])
  298. print("Printer " + p[0] + " has method " + method)
  299. if method == "unknown":
  300. unknownCount += 1
  301. action = QAction(p[0])
  302. action.setEnabled(False)
  303. p.append(action)
  304. self.menu.addAction(action)
  305. continue
  306. commands = self.getSystemCommands(p[0], p[1])
  307. p.append(commands)
  308. menu = QMenu(self.getName(p[0], p[1]))
  309. p.append(menu)
  310. self.menu.addMenu(menu)
  311. if method == "psucontrol":
  312. action = QAction("Turn On PSU")
  313. action.triggered.connect(lambda chk, x=p: self.printerOnAction(x))
  314. p.append(action)
  315. menu.addAction(action)
  316. action = QAction("Turn Off PSU")
  317. action.triggered.connect(lambda chk, x=p: self.printerOffAction(x))
  318. p.append(action)
  319. menu.addAction(action)
  320. for i in range(0, len(commands)):
  321. action = QAction(commands[i].title())
  322. action.triggered.connect(lambda chk, x=p, y=i: self.printerSystemCommandAction(x, y))
  323. p.append(action)
  324. menu.addAction(action)
  325. if (p[2] != None) or (p[3] != None):
  326. menu.addSeparator()
  327. if p[2] != None:
  328. action = QAction("Preheat Tool")
  329. action.triggered.connect(lambda chk, x=p: self.printerHeatTool(x))
  330. p.append(action)
  331. menu.addAction(action)
  332. if p[3] != None:
  333. action = QAction("Preheat Bed")
  334. action.triggered.connect(lambda chk, x=p: self.printerHeatBed(x))
  335. p.append(action)
  336. menu.addAction(action)
  337. if (p[2] != None) or (p[3] != None):
  338. action = QAction("Cooldown")
  339. action.triggered.connect(lambda chk, x=p: self.printerCooldown(x))
  340. p.append(action)
  341. menu.addAction(action)
  342. menu.addSeparator()
  343. action = QAction("Get Status")
  344. action.triggered.connect(lambda chk, x=p: self.printerStatusAction(x))
  345. p.append(action)
  346. menu.addAction(action)
  347. action = QAction("Show Webcam")
  348. action.triggered.connect(lambda chk, x=p: self.printerWebcamAction(x))
  349. p.append(action)
  350. menu.addAction(action)
  351. action = QAction("Open Web UI")
  352. action.triggered.connect(lambda chk, x=p: self.printerWebAction(x))
  353. p.append(action)
  354. menu.addAction(action)
  355. self.menu.addSeparator()
  356. self.settingsAction = QAction("&Settings")
  357. self.settingsAction.triggered.connect(self.showSettingsAction)
  358. self.menu.addAction(self.settingsAction)
  359. self.refreshAction = QAction("&Refresh")
  360. self.refreshAction.triggered.connect(self.restartApp)
  361. self.menu.addAction(self.refreshAction)
  362. self.quitAction = QAction("&Quit")
  363. self.quitAction.triggered.connect(self.exit)
  364. self.menu.addAction(self.quitAction)
  365. self.iconPathName = None
  366. for p in self.iconPaths:
  367. if os.path.isfile(path.join(p, self.iconName)):
  368. self.iconPathName = path.join(p, self.iconName)
  369. break
  370. if self.iconPathName == None:
  371. self.showDialog("OctoTray Error", "Icon file has not been found!", "", False, False, True)
  372. sys.exit(0)
  373. self.icon = QIcon()
  374. self.pic = QPixmap(32, 32)
  375. self.pic.load(self.iconPathName)
  376. self.icon = QIcon(self.pic)
  377. self.trayIcon = QSystemTrayIcon(self.icon)
  378. self.trayIcon.setToolTip(self.name + " " + self.version)
  379. self.trayIcon.setContextMenu(self.menu)
  380. self.trayIcon.setVisible(True)
  381. def readSettings(self):
  382. settings = QSettings(self.vendor, self.name)
  383. printers = []
  384. l = settings.beginReadArray("printers")
  385. for i in range(0, l):
  386. settings.setArrayIndex(i)
  387. p = []
  388. p.append(settings.value("host"))
  389. p.append(settings.value("key"))
  390. p.append(settings.value("tool_preheat"))
  391. p.append(settings.value("bed_preheat"))
  392. printers.append(p)
  393. settings.endArray()
  394. return printers
  395. def writeSettings(self, printers):
  396. settings = QSettings(self.vendor, self.name)
  397. settings.remove("printers")
  398. settings.beginWriteArray("printers")
  399. for i in range(0, len(printers)):
  400. p = printers[i]
  401. settings.setArrayIndex(i)
  402. settings.setValue("host", p[0])
  403. settings.setValue("key", p[1])
  404. settings.setValue("tool_preheat", p[2])
  405. settings.setValue("bed_preheat", p[3])
  406. settings.endArray()
  407. del settings
  408. def openBrowser(self, url):
  409. QDesktopServices.openUrl(QUrl("http://" + url))
  410. def showDialog(self, title, text1, text2 = "", question = False, warning = False, error = False):
  411. msg = QMessageBox()
  412. if error:
  413. msg.setIcon(QMessageBox.Critical)
  414. elif warning:
  415. msg.setIcon(QMessageBox.Warning)
  416. elif question:
  417. msg.setIcon(QMessageBox.Question)
  418. else:
  419. msg.setIcon(QMessageBox.Information)
  420. msg.setWindowTitle(title)
  421. msg.setText(text1)
  422. if text2 is not None:
  423. msg.setInformativeText(text2)
  424. if question:
  425. msg.setStandardButtons(QMessageBox.Yes | QMessageBox.No)
  426. else:
  427. msg.setStandardButtons(QMessageBox.Ok)
  428. retval = msg.exec_()
  429. if retval == QMessageBox.Yes:
  430. return True
  431. else:
  432. return False
  433. def sendRequest(self, host, headers, path, content = None):
  434. url = "http://" + host + "/api/" + path
  435. if content == None:
  436. request = urllib.request.Request(url, None, headers)
  437. else:
  438. data = content.encode('ascii')
  439. request = urllib.request.Request(url, data, headers)
  440. try:
  441. with urllib.request.urlopen(request, None, self.networkTimeout) as response:
  442. text = response.read()
  443. return text
  444. except (urllib.error.URLError, urllib.error.HTTPError) as error:
  445. print("Error requesting URL \"" + url + "\": \"" + str(error) + "\"")
  446. return "error"
  447. except socket.timeout:
  448. print("Timeout waiting for response to \"" + url + "\"")
  449. return "timeout"
  450. def sendPostRequest(self, host, key, path, content):
  451. headers = {
  452. "Content-Type": "application/json",
  453. "X-Api-Key": key
  454. }
  455. return self.sendRequest(host, headers, path, content)
  456. def sendGetRequest(self, host, key, path):
  457. headers = {
  458. "X-Api-Key": key
  459. }
  460. return self.sendRequest(host, headers, path)
  461. def getTemperatureIsSafe(self, host, key):
  462. r = self.sendGetRequest(host, key, "printer")
  463. try:
  464. rd = json.loads(r)
  465. if "temperature" in rd:
  466. if ("tool0" in rd["temperature"]) and ("actual" in rd["temperature"]["tool0"]):
  467. if rd["temperature"]["tool0"]["actual"] > 50.0:
  468. return False
  469. if ("tool1" in rd["temperature"]) and ("actual" in rd["temperature"]["tool1"]):
  470. if rd["temperature"]["tool1"]["actual"] > 50.0:
  471. return False
  472. except json.JSONDecodeError:
  473. pass
  474. return True
  475. def getTemperatureString(self, host, key):
  476. r = self.sendGetRequest(host, key, "printer")
  477. s = ""
  478. try:
  479. rd = json.loads(r)
  480. if ("state" in rd) and ("text" in rd["state"]):
  481. s += rd["state"]["text"]
  482. if "temperature" in rd:
  483. s += " - "
  484. if "temperature" in rd:
  485. if "bed" in rd["temperature"]:
  486. if "actual" in rd["temperature"]["bed"]:
  487. s += "B"
  488. s += "%.1f" % rd["temperature"]["bed"]["actual"]
  489. if "target" in rd["temperature"]["bed"]:
  490. s += "/"
  491. s += "%.1f" % rd["temperature"]["bed"]["target"]
  492. s += " "
  493. if "tool0" in rd["temperature"]:
  494. if "actual" in rd["temperature"]["tool0"]:
  495. s += "T"
  496. s += "%.1f" % rd["temperature"]["tool0"]["actual"]
  497. if "target" in rd["temperature"]["tool0"]:
  498. s += "/"
  499. s += "%.1f" % rd["temperature"]["tool0"]["target"]
  500. s += " "
  501. if "tool1" in rd["temperature"]:
  502. if "actual" in rd["temperature"]["tool1"]:
  503. s += "T"
  504. s += "%.1f" % rd["temperature"]["tool1"]["actual"]
  505. if "target" in rd["temperature"]["tool1"]:
  506. s += "/"
  507. s += "%.1f" % rd["temperature"]["tool1"]["target"]
  508. s += " "
  509. except json.JSONDecodeError:
  510. pass
  511. return s.strip()
  512. def getState(self, host, key):
  513. r = self.sendGetRequest(host, key, "job")
  514. try:
  515. rd = json.loads(r)
  516. if "state" in rd:
  517. return rd["state"]
  518. except json.JSONDecodeError:
  519. pass
  520. return "Unknown"
  521. def getProgress(self, host, key):
  522. r = self.sendGetRequest(host, key, "job")
  523. try:
  524. rd = json.loads(r)
  525. if "progress" in rd:
  526. return rd["progress"]
  527. except json.JSONDecodeError:
  528. pass
  529. return "Unknown"
  530. def getName(self, host, key):
  531. r = self.sendGetRequest(host, key, "printerprofiles")
  532. try:
  533. rd = json.loads(r)
  534. if "profiles" in rd:
  535. p = next(iter(rd["profiles"]))
  536. if "name" in rd["profiles"][p]:
  537. return rd["profiles"][p]["name"]
  538. except json.JSONDecodeError:
  539. pass
  540. return host
  541. def getMethod(self, host, key):
  542. r = self.sendGetRequest(host, key, "plugin/psucontrol")
  543. if r == "timeout":
  544. return "unknown"
  545. try:
  546. rd = json.loads(r)
  547. if "isPSUOn" in rd:
  548. return "psucontrol"
  549. except json.JSONDecodeError:
  550. pass
  551. r = self.sendGetRequest(host, key, "system/commands/custom")
  552. if r == "timeout":
  553. return "unknown"
  554. try:
  555. rd = json.loads(r)
  556. for c in rd:
  557. if "action" in c:
  558. # we have some custom commands and no psucontrol
  559. # so lets try to use that instead of skipping
  560. # the printer completely with 'unknown'
  561. return "system"
  562. except json.JSONDecodeError:
  563. pass
  564. return "unknown"
  565. def getSystemCommands(self, host, key):
  566. l = []
  567. r = self.sendGetRequest(host, key, "system/commands/custom")
  568. try:
  569. rd = json.loads(r)
  570. if len(rd) > 0:
  571. print("system commands available for " + host + ":")
  572. for c in rd:
  573. if "action" in c:
  574. print(" - " + c["action"])
  575. l.append(c["action"])
  576. except json.JSONDecodeError:
  577. pass
  578. return l
  579. def setPSUControl(self, host, key, state):
  580. cmd = "turnPSUOff"
  581. if state:
  582. cmd = "turnPSUOn"
  583. return self.sendPostRequest(host, key, "plugin/psucontrol", '{ "command":"' + cmd + '" }')
  584. def setSystemCommand(self, host, key, cmd):
  585. cmd = urllib.parse.quote(cmd)
  586. return self.sendPostRequest(host, key, "system/commands/custom/" + cmd, '')
  587. def exit(self):
  588. QCoreApplication.quit()
  589. def printerSystemCommandAction(self, item, index):
  590. if "off" in item[2][index].lower():
  591. state = self.getState(item[0], item[1])
  592. if state in self.statesWithWarning:
  593. if self.showDialog("OctoTray Warning", "The printer seems to be running currently!", "Do you really want to run '" + item[2][index] + "'?", True, True) == False:
  594. return
  595. safe = self.getTemperatureIsSafe(item[0], item[1])
  596. if safe == False:
  597. if self.showDialog("OctoTray Warning", "The printer seems to still be hot!", "Do you really want to turn it off?", True, True) == False:
  598. return
  599. self.setSystemCommand(item[0], item[1], item[2][index])
  600. def printerOnAction(self, item):
  601. self.setPSUControl(item[0], item[1], True)
  602. def printerOffAction(self, item):
  603. state = self.getState(item[0], item[1])
  604. if state in self.statesWithWarning:
  605. if self.showDialog("OctoTray Warning", "The printer seems to be running currently!", "Do you really want to turn it off?", True, True) == False:
  606. return
  607. safe = self.getTemperatureIsSafe(item[0], item[1])
  608. if safe == False:
  609. if self.showDialog("OctoTray Warning", "The printer seems to still be hot!", "Do you really want to turn it off?", True, True) == False:
  610. return
  611. self.setPSUControl(item[0], item[1], False)
  612. def printerWebAction(self, item):
  613. self.openBrowser(item[0])
  614. def printerStatusAction(self, item):
  615. progress = self.getProgress(item[0], item[1])
  616. s = item[0] + "\n"
  617. warning = False
  618. if ("completion" in progress) and ("printTime" in progress) and ("printTimeLeft" in progress) and (progress["completion"] != None) and (progress["printTime"] != None) and (progress["printTimeLeft"] != None):
  619. s += "%.1f%% Completion\n" % progress["completion"]
  620. s += "Printing since " + time.strftime("%H:%M:%S", time.gmtime(progress["printTime"])) + "\n"
  621. s += time.strftime("%H:%M:%S", time.gmtime(progress["printTimeLeft"])) + " left"
  622. elif ("completion" in progress) and ("printTime" in progress) and ("printTimeLeft" in progress):
  623. s += "No job is currently running"
  624. else:
  625. s += "Could not read printer status!"
  626. warning = True
  627. t = self.getTemperatureString(item[0], item[1])
  628. if len(t) > 0:
  629. s += "\n" + t
  630. self.showDialog("OctoTray Status", s, None, False, warning)
  631. def setTemperature(self, host, key, what, temp):
  632. path = "printer/bed"
  633. s = "{\"command\": \"target\", \"target\": " + temp + "}"
  634. if "tool" in what:
  635. path = "printer/tool"
  636. s = "{\"command\": \"target\", \"targets\": {\"" + what + "\": " + temp + "}}"
  637. if temp == None:
  638. temp = 0
  639. self.sendPostRequest(host, key, path, s)
  640. def printerHeatTool(self, p):
  641. self.setTemperature(p[0], p[1], "tool0", p[2])
  642. def printerHeatBed(self, p):
  643. self.setTemperature(p[0], p[1], "bed", p[3])
  644. def printerCooldown(self, p):
  645. state = self.getState(p[0], p[1])
  646. if state in self.statesWithWarning:
  647. if self.showDialog("OctoTray Warning", "The printer seems to be running currently!", "Do you really want to turn it off?", True, True) == False:
  648. return
  649. self.setTemperature(p[0], p[1], "tool0", 0)
  650. self.setTemperature(p[0], p[1], "bed", 0)
  651. def printerWebcamAction(self, item):
  652. for cw in self.camWindows:
  653. if cw.getHost() == item[0]:
  654. cw.show()
  655. cw.activateWindow()
  656. return
  657. window = CamWindow(self, item)
  658. self.camWindows.append(window)
  659. window.show()
  660. window.activateWindow()
  661. screenGeometry = QDesktopWidget().screenGeometry()
  662. x = (screenGeometry.width() - window.width()) / 2
  663. y = (screenGeometry.height() - window.height()) / 2
  664. x += screenGeometry.x()
  665. y += screenGeometry.y()
  666. window.setGeometry(int(x), int(y), int(window.width()), int(window.height()))
  667. def removeWebcamWindow(self, window):
  668. self.camWindows.remove(window)
  669. def showSettingsAction(self):
  670. if self.settingsWindow != None:
  671. self.settingsWindow.show()
  672. self.settingsWindow.activateWindow()
  673. return
  674. self.settingsWindow = SettingsWindow(self)
  675. self.settingsWindow.show()
  676. self.settingsWindow.activateWindow()
  677. screenGeometry = QDesktopWidget().screenGeometry()
  678. x = (screenGeometry.width() - self.settingsWindow.width()) / 2
  679. y = (screenGeometry.height() - self.settingsWindow.height()) / 2
  680. x += screenGeometry.x()
  681. y += screenGeometry.y()
  682. self.settingsWindow.setGeometry(int(x), int(y), int(self.settingsWindow.width()), int(self.settingsWindow.height()) + 50)
  683. def removeSettingsWindow(self):
  684. self.settingsWindow = None
  685. def restartApp(self):
  686. QCoreApplication.exit(42)
  687. def closeAll(self):
  688. for cw in self.camWindows:
  689. cw.close()
  690. if self.settingsWindow != None:
  691. self.settingsWindow.close()
  692. self.trayIcon.setVisible(False)
  693. if __name__ == "__main__":
  694. app = QtWidgets.QApplication(sys.argv)
  695. tray = OctoTray(app)
  696. rc = app.exec_()
  697. while rc == 42:
  698. tray.closeAll()
  699. tray = OctoTray(app)
  700. rc = app.exec_()
  701. sys.exit(rc)