mirror of
https://github.com/Ultimaker/Cura.git
synced 2025-07-08 23:46:22 -06:00
111 lines
No EOL
3.7 KiB
Python
111 lines
No EOL
3.7 KiB
Python
from UM.OutputDevice.OutputDevice import OutputDevice
|
|
from UM.OutputDevice import OutputDeviceError
|
|
import threading
|
|
import json
|
|
import time
|
|
import base64
|
|
import requests
|
|
|
|
from . import HttpUploadDataStream
|
|
from UM.i18n import i18nCatalog
|
|
from UM.Signal import Signal, SignalEmitter
|
|
from UM.Application import Application
|
|
from UM.Logger import Logger
|
|
i18n_catalog = i18nCatalog("cura")
|
|
|
|
class WifiConnection(OutputDevice, SignalEmitter):
|
|
def __init__(self, key, address, info):
|
|
super().__init__(key)
|
|
self._address = address
|
|
self._key = key
|
|
self._info = info
|
|
self._http_lock = threading.Lock()
|
|
self._http_connection = None
|
|
self._file = None
|
|
self._do_update = True
|
|
self._thread = None
|
|
|
|
self._json_printer_state = None
|
|
|
|
self._is_connected = False
|
|
|
|
self._api_version = "1"
|
|
self._api_prefix = "/api/v" + self._api_version + "/"
|
|
self.setName(key)
|
|
self.setShortDescription(i18n_catalog.i18nc("@action:button", "Print with WIFI"))
|
|
self.setDescription(i18n_catalog.i18nc("@info:tooltip", "Print with WIFI"))
|
|
self.setIconName("print")
|
|
|
|
connectionStateChanged = Signal()
|
|
|
|
def getKey(self):
|
|
return self._key
|
|
|
|
def isConnected(self):
|
|
return self._is_connected
|
|
|
|
## Set the connection state of this connection.
|
|
# Although we use a restfull API, we do poll the api to check if the machine is still responding.
|
|
def setConnectionState(self, state):
|
|
if state != self._is_connected:
|
|
Logger.log("i", "setting connection state of %s to %s " %(self._address, state))
|
|
self._is_connected = state
|
|
self.connectionStateChanged.emit(self._address)
|
|
else:
|
|
self._is_connected = state
|
|
|
|
def _update(self):
|
|
while self._thread:
|
|
self.setConnectionState(True)
|
|
try:
|
|
reply = self._httpGet("printer")
|
|
if reply.status_code == 200:
|
|
self._json_printer_state = reply.json()
|
|
if not self._is_connected:
|
|
self.setConnectionState(True)
|
|
else:
|
|
self.setConnectionState(False)
|
|
except:
|
|
self.setConnectionState(False)
|
|
time.sleep(1)
|
|
|
|
def close(self):
|
|
self._do_update = False
|
|
self._is_connected = False
|
|
|
|
def requestWrite(self, node, file_name = None):
|
|
self._file = getattr( Application.getInstance().getController().getScene(), "gcode_list")
|
|
self.startPrint()
|
|
|
|
## Start the polling thread.
|
|
def connect(self):
|
|
if self._thread is None:
|
|
self._do_update = True
|
|
self._thread = threading.Thread(target = self._update)
|
|
self._thread.daemon = True
|
|
self._thread.start()
|
|
|
|
def getCameraImage(self):
|
|
pass #Do Nothing
|
|
|
|
def startPrint(self):
|
|
try:
|
|
result = self._httpPost("print_job", self._file)
|
|
except Exception as e:
|
|
Logger.log("e" , "An exception occured in wifi connection: %s" % str(e))
|
|
|
|
def _httpGet(self, path):
|
|
return requests.get("http://" + self._address + self._api_prefix + path)
|
|
|
|
def _httpPost(self, path, file_data):
|
|
with self._http_lock:
|
|
files_dict = {}
|
|
if isinstance(file_data, list): # in case a list with strings is sent
|
|
single_string_file_data = ""
|
|
for line in file_data:
|
|
single_string_file_data += line
|
|
files_dict = {"file":("test.gcode", single_string_file_data)}
|
|
else:
|
|
files_dict = {"file":("test.gcode", file_data)}
|
|
|
|
return requests.post("http://" + self._address + self._api_prefix + path, files = files_dict) |