Browse Source

add menu action to show status of print in dialog

Thomas Buck 3 years ago
parent
commit
a898374987
1 changed files with 58 additions and 17 deletions
  1. 58
    17
      octotray

+ 58
- 17
octotray View File

1
 #!/usr/bin/env python3
1
 #!/usr/bin/env python3
2
 
2
 
3
 # OctoTray Linux Qt System Tray OctoPrint client
3
 # OctoTray Linux Qt System Tray OctoPrint client
4
+#
4
 # depends on:
5
 # depends on:
5
 # - python-pyqt5
6
 # - python-pyqt5
6
 # - curl
7
 # - curl
26
     iconName = "octotray_icon.png"
27
     iconName = "octotray_icon.png"
27
 
28
 
28
     # 0=host, 1=key
29
     # 0=host, 1=key
29
-    # (2=method, 3=menu, 4=on-action, 5=off-action, 6=web-action)
30
+    # (2=method, 3=menu, 4...=actions)
30
     printers = [
31
     printers = [
31
         [ "PRINTER_HOST_HERE", "PRINTER_API_KEY_HERE" ]
32
         [ "PRINTER_HOST_HERE", "PRINTER_API_KEY_HERE" ]
32
     ]
33
     ]
66
             p.append(action)
67
             p.append(action)
67
             menu.addAction(action)
68
             menu.addAction(action)
68
 
69
 
70
+            action = QAction("Get Status")
71
+            action.triggered.connect(lambda chk, x=p: self.printerStatusAction(x))
72
+            p.append(action)
73
+            menu.addAction(action)
74
+
69
             action = QAction("Open Web UI")
75
             action = QAction("Open Web UI")
70
             action.triggered.connect(lambda chk, x=p: self.printerWebAction(x))
76
             action.triggered.connect(lambda chk, x=p: self.printerWebAction(x))
71
             p.append(action)
77
             p.append(action)
99
     def openBrowser(self, url):
105
     def openBrowser(self, url):
100
         os.system("xdg-open http://" + url)
106
         os.system("xdg-open http://" + url)
101
 
107
 
108
+    def showDialog(self, title, text1, text2, question, warning):
109
+        msg = QMessageBox()
110
+
111
+        if warning:
112
+            msg.setIcon(QMessageBox.Warning)
113
+        else:
114
+            msg.setIcon(QMessageBox.Information)
115
+
116
+        msg.setWindowTitle(title)
117
+        msg.setText(text1)
118
+
119
+        if text2 is not None:
120
+            msg.setInformativeText(text2)
121
+
122
+        if question:
123
+            msg.setStandardButtons(QMessageBox.Yes | QMessageBox.No)
124
+        else:
125
+            msg.setStandardButtons(QMessageBox.Ok)
126
+
127
+        retval = msg.exec_()
128
+        if retval == QMessageBox.Yes:
129
+            return True
130
+        return False
131
+
102
     def sendRequest(self, host, headers, path, content = None):
132
     def sendRequest(self, host, headers, path, content = None):
103
         cmdline = 'curl -s'
133
         cmdline = 'curl -s'
104
         for h in headers:
134
         for h in headers:
121
         headers = [ "X-Api-Key: " + key ]
151
         headers = [ "X-Api-Key: " + key ]
122
         return self.sendRequest(host, headers, path)
152
         return self.sendRequest(host, headers, path)
123
 
153
 
124
-    def getStatus(self, host, key):
154
+    def getState(self, host, key):
125
         r = self.sendGetRequest(host, key, "job")
155
         r = self.sendGetRequest(host, key, "job")
126
         try:
156
         try:
127
             rd = json.loads(r)
157
             rd = json.loads(r)
131
             pass
161
             pass
132
         return "Unknown"
162
         return "Unknown"
133
 
163
 
164
+    def getProgress(self, host, key):
165
+        r = self.sendGetRequest(host, key, "job")
166
+        try:
167
+            rd = json.loads(r)
168
+            if "progress" in rd:
169
+                return rd["progress"]
170
+        except json.JSONDecodeError:
171
+            pass
172
+        return "Unknown"
173
+
134
     def getName(self, host, key):
174
     def getName(self, host, key):
135
         r = self.sendGetRequest(host, key, "printerprofiles")
175
         r = self.sendGetRequest(host, key, "printerprofiles")
136
         try:
176
         try:
191
         self.setPrinter(item[0], item[1], item[2], True)
231
         self.setPrinter(item[0], item[1], item[2], True)
192
 
232
 
193
     def printerOffAction(self, item):
233
     def printerOffAction(self, item):
194
-        state = self.getStatus(item[0], item[1])
234
+        state = self.getState(item[0], item[1])
195
         if state in self.statesWithWarning:
235
         if state in self.statesWithWarning:
196
-            if self.showdialog() == True:
236
+            if self.showDialog("OctoTray Warning", "The printer seems to be running currently!", "Do you really want to turn it off?", True, True) == True:
197
                 self.setPrinter(item[0], item[1], item[2], False)
237
                 self.setPrinter(item[0], item[1], item[2], False)
198
         else:
238
         else:
199
             self.setPrinter(item[0], item[1], item[2], False)
239
             self.setPrinter(item[0], item[1], item[2], False)
201
     def printerWebAction(self, item):
241
     def printerWebAction(self, item):
202
         self.openBrowser(item[0])
242
         self.openBrowser(item[0])
203
 
243
 
204
-    def showdialog(self):
205
-        msg = QMessageBox()
206
-        msg.setIcon(QMessageBox.Warning)
207
-
208
-        msg.setWindowTitle("OctoTray Warning")
209
-        msg.setText("The printer seems to be running currently!")
210
-        msg.setInformativeText("Do you really want to turn it off?")
211
-        msg.setStandardButtons(QMessageBox.Yes | QMessageBox.No)
212
-
213
-        retval = msg.exec_()
214
-        if retval == QMessageBox.Yes:
215
-            return True
216
-        return False
244
+    def printerStatusAction(self, item):
245
+        progress = self.getProgress(item[0], item[1])
246
+        s = ""
247
+        warning = False
248
+        if ("completion" in progress) and ("printTime" in progress) and ("printTimeLeft" in progress) and (progress["completion"] != None) and (progress["printTime"] != None) and (progress["printTimeLeft"] != None):
249
+            s = "%.1f%% Completion\n" % progress["completion"]
250
+            s += "Printing since " + time.strftime("%H:%M:%S", time.gmtime(progress["printTime"])) + "\n"
251
+            s += time.strftime("%H:%M:%S", time.gmtime(progress["printTimeLeft"])) + " left\n"
252
+        elif ("completion" in progress) and ("printTime" in progress) and ("printTimeLeft" in progress):
253
+            s = "No job is currently running"
254
+        else:
255
+            s = "Could not read printer status!"
256
+            warning = True
257
+        self.showDialog("OctoTray Status", s, None, False, warning)
217
 
258
 
218
 if __name__ == "__main__":
259
 if __name__ == "__main__":
219
     tray = OctoTray()
260
     tray = OctoTray()

Loading…
Cancel
Save