Compare commits
No commits in common. "87b379d2c54e4bea5f12c4714f8a04f5f338dd39" and "25f3c087c4cce38566e3f915e6f1654a2d871065" have entirely different histories.
87b379d2c5
...
25f3c087c4
17 changed files with 168 additions and 1104 deletions
20
WORKSPACE
20
WORKSPACE
|
@ -62,18 +62,18 @@ load("@arrdem_source_pypi//:requirements.bzl", "install_deps")
|
||||||
# Call it to define repos for your requirements.
|
# Call it to define repos for your requirements.
|
||||||
install_deps()
|
install_deps()
|
||||||
|
|
||||||
# git_repository(
|
git_repository(
|
||||||
# name = "rules_zapp",
|
name = "rules_zapp",
|
||||||
# remote = "https://git.arrdem.com/arrdem/rules_zapp.git",
|
remote = "https://git.arrdem.com/arrdem/rules_zapp.git",
|
||||||
# commit = "72f82e0ace184fe862f1b19c4f71c3bc36cf335b",
|
commit = "72f82e0ace184fe862f1b19c4f71c3bc36cf335b",
|
||||||
# # tag = "0.1.2",
|
# tag = "0.1.2",
|
||||||
# )
|
|
||||||
|
|
||||||
local_repository(
|
|
||||||
name = "rules_zapp",
|
|
||||||
path = "/home/arrdem/Documents/hobby/programming/lang/python/rules_zapp",
|
|
||||||
)
|
)
|
||||||
|
|
||||||
|
# local_repository(
|
||||||
|
# name = "rules_zapp",
|
||||||
|
# path = "/home/arrdem/Documents/hobby/programming/lang/python/rules_zapp",
|
||||||
|
# )
|
||||||
|
|
||||||
####################################################################################################
|
####################################################################################################
|
||||||
# Docker support
|
# Docker support
|
||||||
####################################################################################################
|
####################################################################################################
|
||||||
|
|
|
@ -1,3 +0,0 @@
|
||||||
py_project(
|
|
||||||
name = "gcode-interpreter",
|
|
||||||
)
|
|
|
@ -1,15 +0,0 @@
|
||||||
# Gcode Interpreter
|
|
||||||
|
|
||||||
Extracted [from
|
|
||||||
octoprint](https://raw.githubusercontent.com/OctoPrint/OctoPrint/master/src/octoprint/util/gcodeInterpreter.py), this
|
|
||||||
package provides static analysis (ahem. abstract interpretation) of GCODE scripts for 3d printers to provide key data
|
|
||||||
such as the bounding box through which the tool(s) move, estimated net movement time and the net amount of material
|
|
||||||
extruded.
|
|
||||||
|
|
||||||
## License
|
|
||||||
|
|
||||||
This artifact is licensed under the GNU Affero General Public License http://www.gnu.org/licenses/agpl.html
|
|
||||||
|
|
||||||
Copyright © Reid McKenzie <me@arrdem.com>
|
|
||||||
Copyright © Gina Häußge <osd@foosel.net>
|
|
||||||
Copyright © David Braam
|
|
|
@ -1,868 +0,0 @@
|
||||||
__author__ = "Gina Häußge <osd@foosel.net> based on work by David Braam"
|
|
||||||
__license__ = "GNU Affero General Public License http://www.gnu.org/licenses/agpl.html"
|
|
||||||
__copyright__ = "Copyright (C) 2013 David Braam, Gina Häußge - Released under terms of the AGPLv3 License"
|
|
||||||
|
|
||||||
|
|
||||||
import base64
|
|
||||||
import codecs
|
|
||||||
import io
|
|
||||||
import logging
|
|
||||||
import math
|
|
||||||
import os
|
|
||||||
import re
|
|
||||||
import zlib
|
|
||||||
|
|
||||||
|
|
||||||
class Vector3D:
|
|
||||||
"""
|
|
||||||
3D vector value
|
|
||||||
|
|
||||||
Supports addition, subtraction and multiplication with a scalar value (float, int) as well as calculating the
|
|
||||||
length of the vector.
|
|
||||||
|
|
||||||
Examples:
|
|
||||||
|
|
||||||
>>> a = Vector3D(1.0, 1.0, 1.0)
|
|
||||||
>>> b = Vector3D(4.0, 4.0, 4.0)
|
|
||||||
>>> a + b == Vector3D(5.0, 5.0, 5.0)
|
|
||||||
True
|
|
||||||
>>> b - a == Vector3D(3.0, 3.0, 3.0)
|
|
||||||
True
|
|
||||||
>>> abs(a - b) == Vector3D(3.0, 3.0, 3.0)
|
|
||||||
True
|
|
||||||
>>> a * 2 == Vector3D(2.0, 2.0, 2.0)
|
|
||||||
True
|
|
||||||
>>> a * 2 == 2 * a
|
|
||||||
True
|
|
||||||
>>> a.length == math.sqrt(a.x ** 2 + a.y ** 2 + a.z ** 2)
|
|
||||||
True
|
|
||||||
>>> copied_a = Vector3D(a)
|
|
||||||
>>> a == copied_a
|
|
||||||
True
|
|
||||||
>>> copied_a.x == a.x and copied_a.y == a.y and copied_a.z == a.z
|
|
||||||
True
|
|
||||||
"""
|
|
||||||
|
|
||||||
def __init__(self, *args):
|
|
||||||
if len(args) == 3:
|
|
||||||
(self.x, self.y, self.z) = args
|
|
||||||
|
|
||||||
elif len(args) == 1:
|
|
||||||
# copy constructor
|
|
||||||
other = args[0]
|
|
||||||
if not isinstance(other, Vector3D):
|
|
||||||
raise ValueError("Object to copy must be a Vector3D instance")
|
|
||||||
|
|
||||||
self.x = other.x
|
|
||||||
self.y = other.y
|
|
||||||
self.z = other.z
|
|
||||||
|
|
||||||
@property
|
|
||||||
def length(self):
|
|
||||||
return math.sqrt(self.x * self.x + self.y * self.y + self.z * self.z)
|
|
||||||
|
|
||||||
def __add__(self, other):
|
|
||||||
try:
|
|
||||||
if len(other) == 3:
|
|
||||||
return Vector3D(self.x + other[0], self.y + other[1], self.z + other[2])
|
|
||||||
except TypeError:
|
|
||||||
# doesn't look like a 3-tuple
|
|
||||||
pass
|
|
||||||
|
|
||||||
try:
|
|
||||||
return Vector3D(self.x + other.x, self.y + other.y, self.z + other.z)
|
|
||||||
except AttributeError:
|
|
||||||
# also doesn't look like a Vector3D
|
|
||||||
pass
|
|
||||||
|
|
||||||
raise TypeError(
|
|
||||||
"other must be a Vector3D instance or a list or tuple of length 3"
|
|
||||||
)
|
|
||||||
|
|
||||||
def __sub__(self, other):
|
|
||||||
try:
|
|
||||||
if len(other) == 3:
|
|
||||||
return Vector3D(self.x - other[0], self.y - other[1], self.z - other[2])
|
|
||||||
except TypeError:
|
|
||||||
# doesn't look like a 3-tuple
|
|
||||||
pass
|
|
||||||
|
|
||||||
try:
|
|
||||||
return Vector3D(self.x - other.x, self.y - other.y, self.z - other.z)
|
|
||||||
except AttributeError:
|
|
||||||
# also doesn't look like a Vector3D
|
|
||||||
pass
|
|
||||||
|
|
||||||
raise TypeError(
|
|
||||||
"other must be a Vector3D instance or a list or tuple of length 3"
|
|
||||||
)
|
|
||||||
|
|
||||||
def __mul__(self, other):
|
|
||||||
try:
|
|
||||||
return Vector3D(self.x * other, self.y * other, self.z * other)
|
|
||||||
except TypeError:
|
|
||||||
# doesn't look like a scalar
|
|
||||||
pass
|
|
||||||
|
|
||||||
raise ValueError("other must be a float or int value")
|
|
||||||
|
|
||||||
def __rmul__(self, other):
|
|
||||||
return self.__mul__(other)
|
|
||||||
|
|
||||||
def __abs__(self):
|
|
||||||
return Vector3D(abs(self.x), abs(self.y), abs(self.z))
|
|
||||||
|
|
||||||
def __eq__(self, other):
|
|
||||||
if not isinstance(other, Vector3D):
|
|
||||||
return False
|
|
||||||
return self.x == other.x and self.y == other.y and self.z == other.z
|
|
||||||
|
|
||||||
def __str__(self):
|
|
||||||
return "Vector3D(x={}, y={}, z={}, length={})".format(
|
|
||||||
self.x, self.y, self.z, self.length
|
|
||||||
)
|
|
||||||
|
|
||||||
|
|
||||||
class MinMax3D:
|
|
||||||
"""
|
|
||||||
Tracks minimum and maximum of recorded values
|
|
||||||
|
|
||||||
Examples:
|
|
||||||
|
|
||||||
>>> minmax = MinMax3D()
|
|
||||||
>>> minmax.record(Vector3D(2.0, 2.0, 2.0))
|
|
||||||
>>> minmax.min.x == 2.0 == minmax.max.x and minmax.min.y == 2.0 == minmax.max.y and minmax.min.z == 2.0 == minmax.max.z
|
|
||||||
True
|
|
||||||
>>> minmax.record(Vector3D(1.0, 2.0, 3.0))
|
|
||||||
>>> minmax.min.x == 1.0 and minmax.min.y == 2.0 and minmax.min.z == 2.0
|
|
||||||
True
|
|
||||||
>>> minmax.max.x == 2.0 and minmax.max.y == 2.0 and minmax.max.z == 3.0
|
|
||||||
True
|
|
||||||
>>> minmax.size == Vector3D(1.0, 0.0, 1.0)
|
|
||||||
True
|
|
||||||
>>> empty = MinMax3D()
|
|
||||||
>>> empty.size == Vector3D(0.0, 0.0, 0.0)
|
|
||||||
True
|
|
||||||
>>> weird = MinMax3D(min_z=-1.0)
|
|
||||||
>>> weird.record(Vector3D(2.0, 2.0, 2.0))
|
|
||||||
>>> weird.record(Vector3D(1.0, 2.0, 3.0))
|
|
||||||
>>> weird.min.z == -1.0
|
|
||||||
True
|
|
||||||
>>> weird.size == Vector3D(1.0, 0.0, 4.0)
|
|
||||||
True
|
|
||||||
"""
|
|
||||||
|
|
||||||
def __init__(
|
|
||||||
self,
|
|
||||||
min_x=None,
|
|
||||||
min_y=None,
|
|
||||||
min_z=None,
|
|
||||||
max_x=None,
|
|
||||||
max_y=None,
|
|
||||||
max_z=None,
|
|
||||||
):
|
|
||||||
min_x = min_x if min_x is not None else float("inf")
|
|
||||||
min_y = min_y if min_y is not None else float("inf")
|
|
||||||
min_z = min_z if min_z is not None else float("inf")
|
|
||||||
max_x = max_x if max_x is not None else -float("inf")
|
|
||||||
max_y = max_y if max_y is not None else -float("inf")
|
|
||||||
max_z = max_z if max_z is not None else -float("inf")
|
|
||||||
|
|
||||||
self.min = Vector3D(min_x, min_y, min_z)
|
|
||||||
self.max = Vector3D(max_x, max_y, max_z)
|
|
||||||
|
|
||||||
def record(self, coordinate):
|
|
||||||
"""
|
|
||||||
Records the coordinate, storing the min and max values.
|
|
||||||
|
|
||||||
The input vector components must not be None.
|
|
||||||
"""
|
|
||||||
self.min.x = min(self.min.x, coordinate.x)
|
|
||||||
self.min.y = min(self.min.y, coordinate.y)
|
|
||||||
self.min.z = min(self.min.z, coordinate.z)
|
|
||||||
self.max.x = max(self.max.x, coordinate.x)
|
|
||||||
self.max.y = max(self.max.y, coordinate.y)
|
|
||||||
self.max.z = max(self.max.z, coordinate.z)
|
|
||||||
|
|
||||||
@property
|
|
||||||
def size(self):
|
|
||||||
result = Vector3D()
|
|
||||||
for c in "xyz":
|
|
||||||
min = getattr(self.min, c)
|
|
||||||
max = getattr(self.max, c)
|
|
||||||
value = abs(max - min) if max >= min else 0.0
|
|
||||||
setattr(result, c, value)
|
|
||||||
return result
|
|
||||||
|
|
||||||
@property
|
|
||||||
def dimensions(self):
|
|
||||||
size = self.size
|
|
||||||
return {"width": size.x, "depth": size.y, "height": size.z}
|
|
||||||
|
|
||||||
@property
|
|
||||||
def area(self):
|
|
||||||
return {
|
|
||||||
"minX": None if math.isinf(self.min.x) else self.min.x,
|
|
||||||
"minY": None if math.isinf(self.min.y) else self.min.y,
|
|
||||||
"minZ": None if math.isinf(self.min.z) else self.min.z,
|
|
||||||
"maxX": None if math.isinf(self.max.x) else self.max.x,
|
|
||||||
"maxY": None if math.isinf(self.max.y) else self.max.y,
|
|
||||||
"maxZ": None if math.isinf(self.max.z) else self.max.z,
|
|
||||||
}
|
|
||||||
|
|
||||||
|
|
||||||
class AnalysisAborted(Exception):
|
|
||||||
def __init__(self, reenqueue=True, *args, **kwargs):
|
|
||||||
self.reenqueue = reenqueue
|
|
||||||
Exception.__init__(self, *args, **kwargs)
|
|
||||||
|
|
||||||
|
|
||||||
regex_command = re.compile(
|
|
||||||
r"^\s*((?P<codeGM>[GM]\d+)(\.(?P<subcode>\d+))?|(?P<codeT>T)(?P<tool>\d+))"
|
|
||||||
)
|
|
||||||
"""Regex for a GCODE command."""
|
|
||||||
|
|
||||||
|
|
||||||
class gcode:
|
|
||||||
def __init__(self, incl_layers=False, progress_callback=None):
|
|
||||||
self._logger = logging.getLogger(__name__)
|
|
||||||
self.extrusionAmount = [0]
|
|
||||||
self.extrusionVolume = [0]
|
|
||||||
self.totalMoveTimeMinute = 0
|
|
||||||
self.filename = None
|
|
||||||
self._abort = False
|
|
||||||
self._reenqueue = True
|
|
||||||
self._filamentDiameter = 0
|
|
||||||
self._print_minMax = MinMax3D()
|
|
||||||
self._travel_minMax = MinMax3D()
|
|
||||||
self._progress_callback = progress_callback
|
|
||||||
|
|
||||||
self._incl_layers = incl_layers
|
|
||||||
self._layers = []
|
|
||||||
self._current_layer = None
|
|
||||||
|
|
||||||
def _track_layer(self, pos, arc=None):
|
|
||||||
if not self._incl_layers:
|
|
||||||
return
|
|
||||||
|
|
||||||
if self._current_layer is None or self._current_layer["z"] != pos.z:
|
|
||||||
self._current_layer = {"z": pos.z, "minmax": MinMax3D(), "commands": 1}
|
|
||||||
self._layers.append(self._current_layer)
|
|
||||||
|
|
||||||
elif self._current_layer:
|
|
||||||
self._current_layer["minmax"].record(pos)
|
|
||||||
if arc is not None:
|
|
||||||
self._addArcMinMax(
|
|
||||||
self._current_layer["minmax"],
|
|
||||||
arc["startAngle"],
|
|
||||||
arc["endAngle"],
|
|
||||||
arc["center"],
|
|
||||||
arc["radius"],
|
|
||||||
)
|
|
||||||
|
|
||||||
def _track_command(self):
|
|
||||||
if self._current_layer:
|
|
||||||
self._current_layer["commands"] += 1
|
|
||||||
|
|
||||||
@property
|
|
||||||
def dimensions(self):
|
|
||||||
return self._print_minMax.dimensions
|
|
||||||
|
|
||||||
@property
|
|
||||||
def travel_dimensions(self):
|
|
||||||
return self._travel_minMax.dimensions
|
|
||||||
|
|
||||||
@property
|
|
||||||
def printing_area(self):
|
|
||||||
return self._print_minMax.area
|
|
||||||
|
|
||||||
@property
|
|
||||||
def travel_area(self):
|
|
||||||
return self._travel_minMax.area
|
|
||||||
|
|
||||||
@property
|
|
||||||
def layers(self):
|
|
||||||
return [
|
|
||||||
{
|
|
||||||
"num": num + 1,
|
|
||||||
"z": layer["z"],
|
|
||||||
"commands": layer["commands"],
|
|
||||||
"bounds": {
|
|
||||||
"minX": layer["minmax"].min.x,
|
|
||||||
"maxX": layer["minmax"].max.x,
|
|
||||||
"minY": layer["minmax"].min.y,
|
|
||||||
"maxY": layer["minmax"].max.y,
|
|
||||||
},
|
|
||||||
}
|
|
||||||
for num, layer in enumerate(self._layers)
|
|
||||||
]
|
|
||||||
|
|
||||||
def load(
|
|
||||||
self,
|
|
||||||
filename,
|
|
||||||
throttle=None,
|
|
||||||
speedx=6000,
|
|
||||||
speedy=6000,
|
|
||||||
offsets=None,
|
|
||||||
max_extruders=10,
|
|
||||||
g90_extruder=False,
|
|
||||||
bed_z=0.0,
|
|
||||||
):
|
|
||||||
self._print_minMax.min.z = self._travel_minMax.min.z = bed_z
|
|
||||||
if os.path.isfile(filename):
|
|
||||||
self.filename = filename
|
|
||||||
self._fileSize = os.stat(filename).st_size
|
|
||||||
|
|
||||||
with codecs.open(filename, encoding="utf-8", errors="replace") as f:
|
|
||||||
self._load(
|
|
||||||
f,
|
|
||||||
throttle=throttle,
|
|
||||||
speedx=speedx,
|
|
||||||
speedy=speedy,
|
|
||||||
offsets=offsets,
|
|
||||||
max_extruders=max_extruders,
|
|
||||||
g90_extruder=g90_extruder,
|
|
||||||
)
|
|
||||||
|
|
||||||
def abort(self, reenqueue=True):
|
|
||||||
self._abort = True
|
|
||||||
self._reenqueue = reenqueue
|
|
||||||
|
|
||||||
def _load(
|
|
||||||
self,
|
|
||||||
gcodeFile,
|
|
||||||
throttle=None,
|
|
||||||
speedx=6000,
|
|
||||||
speedy=6000,
|
|
||||||
offsets=None,
|
|
||||||
max_extruders=10,
|
|
||||||
g90_extruder=False,
|
|
||||||
):
|
|
||||||
lineNo = 0
|
|
||||||
readBytes = 0
|
|
||||||
pos = Vector3D(0.0, 0.0, 0.0)
|
|
||||||
currentE = [0.0]
|
|
||||||
totalExtrusion = [0.0]
|
|
||||||
maxExtrusion = [0.0]
|
|
||||||
currentExtruder = 0
|
|
||||||
totalMoveTimeMinute = 0.0
|
|
||||||
relativeE = False
|
|
||||||
relativeMode = False
|
|
||||||
duplicationMode = False
|
|
||||||
scale = 1.0
|
|
||||||
fwretractTime = 0
|
|
||||||
fwretractDist = 0
|
|
||||||
fwrecoverTime = 0
|
|
||||||
feedrate = min(speedx, speedy)
|
|
||||||
if feedrate == 0:
|
|
||||||
# some somewhat sane default if axes speeds are insane...
|
|
||||||
feedrate = 2000
|
|
||||||
|
|
||||||
if offsets is None or not isinstance(offsets, (list, tuple)):
|
|
||||||
offsets = []
|
|
||||||
if len(offsets) < max_extruders:
|
|
||||||
offsets += [(0, 0)] * (max_extruders - len(offsets))
|
|
||||||
|
|
||||||
for line in gcodeFile:
|
|
||||||
if self._abort:
|
|
||||||
raise AnalysisAborted(reenqueue=self._reenqueue)
|
|
||||||
lineNo += 1
|
|
||||||
readBytes += len(line.encode("utf-8"))
|
|
||||||
|
|
||||||
if isinstance(gcodeFile, (io.IOBase, codecs.StreamReaderWriter)):
|
|
||||||
percentage = readBytes / self._fileSize
|
|
||||||
elif isinstance(gcodeFile, (list)):
|
|
||||||
percentage = lineNo / len(gcodeFile)
|
|
||||||
else:
|
|
||||||
percentage = None
|
|
||||||
|
|
||||||
try:
|
|
||||||
if (
|
|
||||||
self._progress_callback is not None
|
|
||||||
and (lineNo % 1000 == 0)
|
|
||||||
and percentage is not None
|
|
||||||
):
|
|
||||||
self._progress_callback(percentage)
|
|
||||||
except Exception as exc:
|
|
||||||
self._logger.debug(
|
|
||||||
"Progress callback %r error: %s", self._progress_callback, exc
|
|
||||||
)
|
|
||||||
|
|
||||||
if ";" in line:
|
|
||||||
comment = line[line.find(";") + 1 :].strip()
|
|
||||||
if comment.startswith("filament_diameter"):
|
|
||||||
# Slic3r
|
|
||||||
filamentValue = comment.split("=", 1)[1].strip()
|
|
||||||
try:
|
|
||||||
self._filamentDiameter = float(filamentValue)
|
|
||||||
except ValueError:
|
|
||||||
try:
|
|
||||||
self._filamentDiameter = float(
|
|
||||||
filamentValue.split(",")[0].strip()
|
|
||||||
)
|
|
||||||
except ValueError:
|
|
||||||
self._filamentDiameter = 0.0
|
|
||||||
elif comment.startswith("CURA_PROFILE_STRING") or comment.startswith(
|
|
||||||
"CURA_OCTO_PROFILE_STRING"
|
|
||||||
):
|
|
||||||
# Cura 15.04.* & OctoPrint Cura plugin
|
|
||||||
if comment.startswith("CURA_PROFILE_STRING"):
|
|
||||||
prefix = "CURA_PROFILE_STRING:"
|
|
||||||
else:
|
|
||||||
prefix = "CURA_OCTO_PROFILE_STRING:"
|
|
||||||
|
|
||||||
curaOptions = self._parseCuraProfileString(comment, prefix)
|
|
||||||
if "filament_diameter" in curaOptions:
|
|
||||||
try:
|
|
||||||
self._filamentDiameter = float(
|
|
||||||
curaOptions["filament_diameter"]
|
|
||||||
)
|
|
||||||
except ValueError:
|
|
||||||
self._filamentDiameter = 0.0
|
|
||||||
elif comment.startswith("filamentDiameter,"):
|
|
||||||
# Simplify3D
|
|
||||||
filamentValue = comment.split(",", 1)[1].strip()
|
|
||||||
try:
|
|
||||||
self._filamentDiameter = float(filamentValue)
|
|
||||||
except ValueError:
|
|
||||||
self._filamentDiameter = 0.0
|
|
||||||
line = line[0 : line.find(";")]
|
|
||||||
|
|
||||||
match = regex_command.search(line)
|
|
||||||
gcode = tool = None
|
|
||||||
if match:
|
|
||||||
values = match.groupdict()
|
|
||||||
if "codeGM" in values and values["codeGM"]:
|
|
||||||
gcode = values["codeGM"]
|
|
||||||
elif "codeT" in values and values["codeT"]:
|
|
||||||
gcode = values["codeT"]
|
|
||||||
tool = int(values["tool"])
|
|
||||||
|
|
||||||
# G codes
|
|
||||||
if gcode in ("G0", "G1", "G00", "G01"): # Move
|
|
||||||
x = getCodeFloat(line, "X")
|
|
||||||
y = getCodeFloat(line, "Y")
|
|
||||||
z = getCodeFloat(line, "Z")
|
|
||||||
e = getCodeFloat(line, "E")
|
|
||||||
f = getCodeFloat(line, "F")
|
|
||||||
|
|
||||||
if x is not None or y is not None or z is not None:
|
|
||||||
# this is a move
|
|
||||||
move = True
|
|
||||||
else:
|
|
||||||
# print head stays on position
|
|
||||||
move = False
|
|
||||||
|
|
||||||
oldPos = pos
|
|
||||||
|
|
||||||
# Use new coordinates if provided. If not provided, use prior coordinates (minus tool offset)
|
|
||||||
# in absolute and 0.0 in relative mode.
|
|
||||||
newPos = Vector3D(
|
|
||||||
x * scale if x is not None else (0.0 if relativeMode else pos.x),
|
|
||||||
y * scale if y is not None else (0.0 if relativeMode else pos.y),
|
|
||||||
z * scale if z is not None else (0.0 if relativeMode else pos.z),
|
|
||||||
)
|
|
||||||
|
|
||||||
if relativeMode:
|
|
||||||
# Relative mode: add to current position
|
|
||||||
pos += newPos
|
|
||||||
else:
|
|
||||||
# Absolute mode: apply tool offsets
|
|
||||||
pos = newPos
|
|
||||||
|
|
||||||
if f is not None and f != 0:
|
|
||||||
feedrate = f
|
|
||||||
|
|
||||||
if e is not None:
|
|
||||||
if relativeMode or relativeE:
|
|
||||||
# e is already relative, nothing to do
|
|
||||||
pass
|
|
||||||
else:
|
|
||||||
e -= currentE[currentExtruder]
|
|
||||||
|
|
||||||
totalExtrusion[currentExtruder] += e
|
|
||||||
currentE[currentExtruder] += e
|
|
||||||
maxExtrusion[currentExtruder] = max(
|
|
||||||
maxExtrusion[currentExtruder], totalExtrusion[currentExtruder]
|
|
||||||
)
|
|
||||||
|
|
||||||
if currentExtruder == 0 and len(currentE) > 1 and duplicationMode:
|
|
||||||
# Copy first extruder length to other extruders
|
|
||||||
for i in range(1, len(currentE)):
|
|
||||||
totalExtrusion[i] += e
|
|
||||||
currentE[i] += e
|
|
||||||
maxExtrusion[i] = max(maxExtrusion[i], totalExtrusion[i])
|
|
||||||
else:
|
|
||||||
e = 0
|
|
||||||
|
|
||||||
# If move, calculate new min/max coordinates
|
|
||||||
if move:
|
|
||||||
self._travel_minMax.record(oldPos)
|
|
||||||
self._travel_minMax.record(pos)
|
|
||||||
if e > 0:
|
|
||||||
# store as print move if extrusion is > 0
|
|
||||||
self._print_minMax.record(oldPos)
|
|
||||||
self._print_minMax.record(pos)
|
|
||||||
|
|
||||||
# move time in x, y, z, will be 0 if no movement happened
|
|
||||||
moveTimeXYZ = abs((oldPos - pos).length / feedrate)
|
|
||||||
|
|
||||||
# time needed for extruding, will be 0 if no extrusion happened
|
|
||||||
extrudeTime = abs(e / feedrate)
|
|
||||||
|
|
||||||
# time to add is maximum of both
|
|
||||||
totalMoveTimeMinute += max(moveTimeXYZ, extrudeTime)
|
|
||||||
|
|
||||||
# process layers if there's extrusion
|
|
||||||
if e:
|
|
||||||
self._track_layer(pos)
|
|
||||||
|
|
||||||
if gcode in ("G2", "G3", "G02", "G03"): # Arc Move
|
|
||||||
x = getCodeFloat(line, "X")
|
|
||||||
y = getCodeFloat(line, "Y")
|
|
||||||
z = getCodeFloat(line, "Z")
|
|
||||||
e = getCodeFloat(line, "E")
|
|
||||||
i = getCodeFloat(line, "I")
|
|
||||||
j = getCodeFloat(line, "J")
|
|
||||||
r = getCodeFloat(line, "R")
|
|
||||||
f = getCodeFloat(line, "F")
|
|
||||||
|
|
||||||
# this is a move or print head stays on position
|
|
||||||
move = (
|
|
||||||
x is not None
|
|
||||||
or y is not None
|
|
||||||
or z is not None
|
|
||||||
or i is not None
|
|
||||||
or j is not None
|
|
||||||
or r is not None
|
|
||||||
)
|
|
||||||
|
|
||||||
oldPos = pos
|
|
||||||
|
|
||||||
# Use new coordinates if provided. If not provided, use prior coordinates (minus tool offset)
|
|
||||||
# in absolute and 0.0 in relative mode.
|
|
||||||
newPos = Vector3D(
|
|
||||||
x * scale if x is not None else (0.0 if relativeMode else pos.x),
|
|
||||||
y * scale if y is not None else (0.0 if relativeMode else pos.y),
|
|
||||||
z * scale if z is not None else (0.0 if relativeMode else pos.z),
|
|
||||||
)
|
|
||||||
|
|
||||||
if relativeMode:
|
|
||||||
# Relative mode: add to current position
|
|
||||||
pos += newPos
|
|
||||||
else:
|
|
||||||
# Absolute mode: apply tool offsets
|
|
||||||
pos = newPos
|
|
||||||
|
|
||||||
if f is not None and f != 0:
|
|
||||||
feedrate = f
|
|
||||||
|
|
||||||
# get radius and offset
|
|
||||||
i = 0 if i is None else i
|
|
||||||
j = 0 if j is None else j
|
|
||||||
r = math.sqrt(i * i + j * j) if r is None else r
|
|
||||||
|
|
||||||
# calculate angles
|
|
||||||
centerArc = Vector3D(oldPos.x + i, oldPos.y + j, oldPos.z)
|
|
||||||
startAngle = math.atan2(oldPos.y - centerArc.y, oldPos.x - centerArc.x)
|
|
||||||
endAngle = math.atan2(pos.y - centerArc.y, pos.x - centerArc.x)
|
|
||||||
arcAngle = endAngle - startAngle
|
|
||||||
|
|
||||||
if gcode in ("G2", "G02"):
|
|
||||||
startAngle, endAngle = endAngle, startAngle
|
|
||||||
arcAngle = -arcAngle
|
|
||||||
if startAngle < 0:
|
|
||||||
startAngle += math.pi * 2
|
|
||||||
if endAngle < 0:
|
|
||||||
endAngle += math.pi * 2
|
|
||||||
if arcAngle < 0:
|
|
||||||
arcAngle += math.pi * 2
|
|
||||||
|
|
||||||
# from now on we only think in counter-clockwise direction
|
|
||||||
|
|
||||||
if e is not None:
|
|
||||||
if relativeMode or relativeE:
|
|
||||||
# e is already relative, nothing to do
|
|
||||||
pass
|
|
||||||
else:
|
|
||||||
e -= currentE[currentExtruder]
|
|
||||||
|
|
||||||
totalExtrusion[currentExtruder] += e
|
|
||||||
currentE[currentExtruder] += e
|
|
||||||
maxExtrusion[currentExtruder] = max(
|
|
||||||
maxExtrusion[currentExtruder], totalExtrusion[currentExtruder]
|
|
||||||
)
|
|
||||||
|
|
||||||
if currentExtruder == 0 and len(currentE) > 1 and duplicationMode:
|
|
||||||
# Copy first extruder length to other extruders
|
|
||||||
for i in range(1, len(currentE)):
|
|
||||||
totalExtrusion[i] += e
|
|
||||||
currentE[i] += e
|
|
||||||
maxExtrusion[i] = max(maxExtrusion[i], totalExtrusion[i])
|
|
||||||
else:
|
|
||||||
e = 0
|
|
||||||
|
|
||||||
# If move, calculate new min/max coordinates
|
|
||||||
if move:
|
|
||||||
self._travel_minMax.record(oldPos)
|
|
||||||
self._travel_minMax.record(pos)
|
|
||||||
self._addArcMinMax(
|
|
||||||
self._travel_minMax, startAngle, endAngle, centerArc, r
|
|
||||||
)
|
|
||||||
if e > 0:
|
|
||||||
# store as print move if extrusion is > 0
|
|
||||||
self._print_minMax.record(oldPos)
|
|
||||||
self._print_minMax.record(pos)
|
|
||||||
self._addArcMinMax(
|
|
||||||
self._print_minMax, startAngle, endAngle, centerArc, r
|
|
||||||
)
|
|
||||||
|
|
||||||
# calculate 3d arc length
|
|
||||||
arcLengthXYZ = math.sqrt((oldPos.z - pos.z) ** 2 + (arcAngle * r) ** 2)
|
|
||||||
|
|
||||||
# move time in x, y, z, will be 0 if no movement happened
|
|
||||||
moveTimeXYZ = abs(arcLengthXYZ / feedrate)
|
|
||||||
|
|
||||||
# time needed for extruding, will be 0 if no extrusion happened
|
|
||||||
extrudeTime = abs(e / feedrate)
|
|
||||||
|
|
||||||
# time to add is maximum of both
|
|
||||||
totalMoveTimeMinute += max(moveTimeXYZ, extrudeTime)
|
|
||||||
|
|
||||||
# process layers if there's extrusion
|
|
||||||
if e:
|
|
||||||
self._track_layer(
|
|
||||||
pos,
|
|
||||||
{
|
|
||||||
"startAngle": startAngle,
|
|
||||||
"endAngle": endAngle,
|
|
||||||
"center": centerArc,
|
|
||||||
"radius": r,
|
|
||||||
},
|
|
||||||
)
|
|
||||||
|
|
||||||
elif gcode == "G4": # Delay
|
|
||||||
S = getCodeFloat(line, "S")
|
|
||||||
if S is not None:
|
|
||||||
totalMoveTimeMinute += S / 60
|
|
||||||
P = getCodeFloat(line, "P")
|
|
||||||
if P is not None:
|
|
||||||
totalMoveTimeMinute += P / 60 / 1000
|
|
||||||
elif gcode == "G10": # Firmware retract
|
|
||||||
totalMoveTimeMinute += fwretractTime
|
|
||||||
elif gcode == "G11": # Firmware retract recover
|
|
||||||
totalMoveTimeMinute += fwrecoverTime
|
|
||||||
elif gcode == "G20": # Units are inches
|
|
||||||
scale = 25.4
|
|
||||||
elif gcode == "G21": # Units are mm
|
|
||||||
scale = 1.0
|
|
||||||
elif gcode == "G28": # Home
|
|
||||||
x = getCodeFloat(line, "X")
|
|
||||||
y = getCodeFloat(line, "Y")
|
|
||||||
z = getCodeFloat(line, "Z")
|
|
||||||
origin = Vector3D(0.0, 0.0, 0.0)
|
|
||||||
if x is None and y is None and z is None:
|
|
||||||
pos = origin
|
|
||||||
else:
|
|
||||||
pos = Vector3D(pos)
|
|
||||||
if x is not None:
|
|
||||||
pos.x = origin.x
|
|
||||||
if y is not None:
|
|
||||||
pos.y = origin.y
|
|
||||||
if z is not None:
|
|
||||||
pos.z = origin.z
|
|
||||||
elif gcode == "G90": # Absolute position
|
|
||||||
relativeMode = False
|
|
||||||
if g90_extruder:
|
|
||||||
relativeE = False
|
|
||||||
elif gcode == "G91": # Relative position
|
|
||||||
relativeMode = True
|
|
||||||
if g90_extruder:
|
|
||||||
relativeE = True
|
|
||||||
elif gcode == "G92":
|
|
||||||
x = getCodeFloat(line, "X")
|
|
||||||
y = getCodeFloat(line, "Y")
|
|
||||||
z = getCodeFloat(line, "Z")
|
|
||||||
e = getCodeFloat(line, "E")
|
|
||||||
|
|
||||||
if e is None and x is None and y is None and z is None:
|
|
||||||
# no parameters, set all axis to 0
|
|
||||||
currentE[currentExtruder] = 0.0
|
|
||||||
pos.x = 0.0
|
|
||||||
pos.y = 0.0
|
|
||||||
pos.z = 0.0
|
|
||||||
else:
|
|
||||||
# some parameters set, only set provided axes
|
|
||||||
if e is not None:
|
|
||||||
currentE[currentExtruder] = e
|
|
||||||
if x is not None:
|
|
||||||
pos.x = x
|
|
||||||
if y is not None:
|
|
||||||
pos.y = y
|
|
||||||
if z is not None:
|
|
||||||
pos.z = z
|
|
||||||
# M codes
|
|
||||||
elif gcode == "M82": # Absolute E
|
|
||||||
relativeE = False
|
|
||||||
elif gcode == "M83": # Relative E
|
|
||||||
relativeE = True
|
|
||||||
elif gcode in ("M207", "M208"): # Firmware retract settings
|
|
||||||
s = getCodeFloat(line, "S")
|
|
||||||
f = getCodeFloat(line, "F")
|
|
||||||
if s is not None and f is not None:
|
|
||||||
if gcode == "M207":
|
|
||||||
# Ensure division is valid
|
|
||||||
if f > 0:
|
|
||||||
fwretractTime = s / f
|
|
||||||
else:
|
|
||||||
fwretractTime = 0
|
|
||||||
fwretractDist = s
|
|
||||||
else:
|
|
||||||
if f > 0:
|
|
||||||
fwrecoverTime = (fwretractDist + s) / f
|
|
||||||
else:
|
|
||||||
fwrecoverTime = 0
|
|
||||||
elif gcode == "M605": # Duplication/Mirroring mode
|
|
||||||
s = getCodeInt(line, "S")
|
|
||||||
if s in [2, 4, 5, 6]:
|
|
||||||
# Duplication / Mirroring mode selected. Printer firmware copies extrusion commands
|
|
||||||
# from first extruder to all other extruders
|
|
||||||
duplicationMode = True
|
|
||||||
else:
|
|
||||||
duplicationMode = False
|
|
||||||
|
|
||||||
# T codes
|
|
||||||
elif tool is not None:
|
|
||||||
if tool > max_extruders:
|
|
||||||
self._logger.warning(
|
|
||||||
"GCODE tried to select tool %d, that looks wrong, ignoring for GCODE analysis"
|
|
||||||
% tool
|
|
||||||
)
|
|
||||||
elif tool == currentExtruder:
|
|
||||||
pass
|
|
||||||
else:
|
|
||||||
pos.x -= (
|
|
||||||
offsets[currentExtruder][0]
|
|
||||||
if currentExtruder < len(offsets)
|
|
||||||
else 0
|
|
||||||
)
|
|
||||||
pos.y -= (
|
|
||||||
offsets[currentExtruder][1]
|
|
||||||
if currentExtruder < len(offsets)
|
|
||||||
else 0
|
|
||||||
)
|
|
||||||
|
|
||||||
currentExtruder = tool
|
|
||||||
|
|
||||||
pos.x += (
|
|
||||||
offsets[currentExtruder][0]
|
|
||||||
if currentExtruder < len(offsets)
|
|
||||||
else 0
|
|
||||||
)
|
|
||||||
pos.y += (
|
|
||||||
offsets[currentExtruder][1]
|
|
||||||
if currentExtruder < len(offsets)
|
|
||||||
else 0
|
|
||||||
)
|
|
||||||
|
|
||||||
if len(currentE) <= currentExtruder:
|
|
||||||
for _ in range(len(currentE), currentExtruder + 1):
|
|
||||||
currentE.append(0.0)
|
|
||||||
if len(maxExtrusion) <= currentExtruder:
|
|
||||||
for _ in range(len(maxExtrusion), currentExtruder + 1):
|
|
||||||
maxExtrusion.append(0.0)
|
|
||||||
if len(totalExtrusion) <= currentExtruder:
|
|
||||||
for _ in range(len(totalExtrusion), currentExtruder + 1):
|
|
||||||
totalExtrusion.append(0.0)
|
|
||||||
|
|
||||||
if gcode or tool:
|
|
||||||
self._track_command()
|
|
||||||
|
|
||||||
if throttle is not None:
|
|
||||||
throttle(lineNo, readBytes)
|
|
||||||
if self._progress_callback is not None:
|
|
||||||
self._progress_callback(100.0)
|
|
||||||
|
|
||||||
self.extrusionAmount = maxExtrusion
|
|
||||||
self.extrusionVolume = [0] * len(maxExtrusion)
|
|
||||||
for i in range(len(maxExtrusion)):
|
|
||||||
radius = self._filamentDiameter / 2
|
|
||||||
self.extrusionVolume[i] = (
|
|
||||||
self.extrusionAmount[i] * (math.pi * radius * radius)
|
|
||||||
) / 1000
|
|
||||||
self.totalMoveTimeMinute = totalMoveTimeMinute
|
|
||||||
|
|
||||||
def _parseCuraProfileString(self, comment, prefix):
|
|
||||||
return {
|
|
||||||
key: value
|
|
||||||
for (key, value) in map(
|
|
||||||
lambda x: x.split(b"=", 1),
|
|
||||||
zlib.decompress(base64.b64decode(comment[len(prefix) :])).split(b"\b"),
|
|
||||||
)
|
|
||||||
}
|
|
||||||
|
|
||||||
def _intersectsAngle(self, start, end, angle):
|
|
||||||
if end < start and angle == 0:
|
|
||||||
# angle crosses 0 degrees
|
|
||||||
return True
|
|
||||||
else:
|
|
||||||
return start <= angle <= end
|
|
||||||
|
|
||||||
def _addArcMinMax(self, minmax, startAngle, endAngle, centerArc, radius):
|
|
||||||
startDeg = math.degrees(startAngle)
|
|
||||||
endDeg = math.degrees(endAngle)
|
|
||||||
|
|
||||||
if self._intersectsAngle(startDeg, endDeg, 0):
|
|
||||||
# arc crosses positive x
|
|
||||||
minmax.max.x = max(minmax.max.x, centerArc.x + radius)
|
|
||||||
if self._intersectsAngle(startDeg, endDeg, 90):
|
|
||||||
# arc crosses positive y
|
|
||||||
minmax.max.y = max(minmax.max.y, centerArc.y + radius)
|
|
||||||
if self._intersectsAngle(startDeg, endDeg, 180):
|
|
||||||
# arc crosses negative x
|
|
||||||
minmax.min.x = min(minmax.min.x, centerArc.x - radius)
|
|
||||||
if self._intersectsAngle(startDeg, endDeg, 270):
|
|
||||||
# arc crosses negative y
|
|
||||||
minmax.min.y = min(minmax.min.y, centerArc.y - radius)
|
|
||||||
|
|
||||||
def get_result(self):
|
|
||||||
result = {
|
|
||||||
"total_time": self.totalMoveTimeMinute,
|
|
||||||
"extrusion_length": self.extrusionAmount,
|
|
||||||
"extrusion_volume": self.extrusionVolume,
|
|
||||||
"dimensions": self.dimensions,
|
|
||||||
"printing_area": self.printing_area,
|
|
||||||
"travel_dimensions": self.travel_dimensions,
|
|
||||||
"travel_area": self.travel_area,
|
|
||||||
}
|
|
||||||
if self._incl_layers:
|
|
||||||
result["layers"] = self.layers
|
|
||||||
|
|
||||||
return result
|
|
||||||
|
|
||||||
|
|
||||||
def getCodeInt(line, code):
|
|
||||||
return getCode(line, code, int)
|
|
||||||
|
|
||||||
|
|
||||||
def getCodeFloat(line, code):
|
|
||||||
return getCode(line, code, float)
|
|
||||||
|
|
||||||
|
|
||||||
def getCode(line, code, c):
|
|
||||||
n = line.find(code) + 1
|
|
||||||
if n < 1:
|
|
||||||
return None
|
|
||||||
m = line.find(" ", n)
|
|
||||||
try:
|
|
||||||
if m < 0:
|
|
||||||
result = c(line[n:])
|
|
||||||
else:
|
|
||||||
result = c(line[n:m])
|
|
||||||
except ValueError:
|
|
||||||
return None
|
|
||||||
|
|
||||||
if math.isnan(result) or math.isinf(result):
|
|
||||||
return None
|
|
||||||
|
|
||||||
return result
|
|
|
@ -17,10 +17,9 @@ from tentacles.blueprints import (
|
||||||
printer_ui,
|
printer_ui,
|
||||||
user_ui,
|
user_ui,
|
||||||
)
|
)
|
||||||
from tentacles.db import Db
|
|
||||||
from tentacles.globals import _ctx, Ctx, ctx
|
from tentacles.globals import _ctx, Ctx, ctx
|
||||||
from tentacles.workers import *
|
from tentacles.store import Store
|
||||||
from tentacles.workers import assign_jobs, Worker
|
from tentacles.workers import Worker
|
||||||
|
|
||||||
|
|
||||||
@click.group()
|
@click.group()
|
||||||
|
@ -29,7 +28,7 @@ def cli():
|
||||||
|
|
||||||
|
|
||||||
def db_factory(app):
|
def db_factory(app):
|
||||||
store = Db(app.config.get("db", {}).get("uri"))
|
store = Store(app.config.get("db", {}).get("uri"))
|
||||||
store.connect()
|
store.connect()
|
||||||
return store
|
return store
|
||||||
|
|
||||||
|
@ -39,11 +38,10 @@ def custom_ctx(app, wsgi_app):
|
||||||
store = db_factory(app)
|
store = db_factory(app)
|
||||||
token = _ctx.set(Ctx(store))
|
token = _ctx.set(Ctx(store))
|
||||||
try:
|
try:
|
||||||
with store.savepoint():
|
return wsgi_app(environ, start_response)
|
||||||
return wsgi_app(environ, start_response)
|
|
||||||
finally:
|
finally:
|
||||||
store.close()
|
|
||||||
_ctx.reset(token)
|
_ctx.reset(token)
|
||||||
|
store.close()
|
||||||
|
|
||||||
return helper
|
return helper
|
||||||
|
|
||||||
|
@ -58,21 +56,21 @@ def user_session():
|
||||||
if (
|
if (
|
||||||
(
|
(
|
||||||
(session_id := request.cookies.get("sid", ""))
|
(session_id := request.cookies.get("sid", ""))
|
||||||
and (row := ctx.db.try_key(kid=session_id))
|
and (uid := ctx.db.try_key(session_id))
|
||||||
)
|
)
|
||||||
or (
|
or (
|
||||||
request.authorization
|
request.authorization
|
||||||
and request.authorization.token
|
and request.authorization.token
|
||||||
and (row := ctx.db.try_key(kid=request.authorization.token))
|
and (uid := ctx.db.try_key(request.authorization.token))
|
||||||
)
|
)
|
||||||
or (
|
or (
|
||||||
(api_key := request.headers.get("x-api-key"))
|
(api_key := request.headers.get("x-api-key"))
|
||||||
and (row := ctx.db.try_key(kid=api_key))
|
and (uid := ctx.db.try_key(api_key))
|
||||||
)
|
)
|
||||||
):
|
):
|
||||||
ctx.sid = row.id
|
ctx.sid = session_id
|
||||||
ctx.uid = row.user_id
|
ctx.uid = uid
|
||||||
user = ctx.db.fetch_user(row.user_id)
|
user = ctx.db.fetch_user(uid)
|
||||||
ctx.gid = user.group_id
|
ctx.gid = user.group_id
|
||||||
ctx.username = user.name
|
ctx.username = user.name
|
||||||
ctx.is_admin = user.group_id == 0
|
ctx.is_admin = user.group_id == 0
|
||||||
|
@ -103,17 +101,14 @@ def make_app():
|
||||||
@cli.command()
|
@cli.command()
|
||||||
@click.option("--hostname", "hostname", type=str, default="0.0.0.0")
|
@click.option("--hostname", "hostname", type=str, default="0.0.0.0")
|
||||||
@click.option("--port", "port", type=int, default=8080)
|
@click.option("--port", "port", type=int, default=8080)
|
||||||
@click.option("--trace/--no-trace", "trace", default=False)
|
|
||||||
@click.option("--config", type=Path)
|
@click.option("--config", type=Path)
|
||||||
def serve(hostname: str, port: int, config: Path, trace: bool):
|
def serve(hostname: str, port: int, config: Path):
|
||||||
logging.basicConfig(
|
logging.basicConfig(
|
||||||
format="%(asctime)s %(threadName)s - %(name)s - %(levelname)s - %(message)s",
|
format="%(asctime)s %(relativeCreated)6d %(threadName)s - %(name)s - %(levelname)s - %(message)s",
|
||||||
level=logging.INFO,
|
level=logging.INFO,
|
||||||
)
|
)
|
||||||
|
|
||||||
logging.getLogger("tentacles").setLevel(logging.DEBUG)
|
logging.getLogger("tentacles").setLevel(logging.DEBUG)
|
||||||
if trace:
|
|
||||||
logging.getLogger("tentacles.db").setLevel(logging.DEBUG - 1)
|
|
||||||
|
|
||||||
app = make_app()
|
app = make_app()
|
||||||
|
|
||||||
|
@ -138,14 +133,8 @@ def serve(hostname: str, port: int, config: Path, trace: bool):
|
||||||
server.shutdown_timeout = 1
|
server.shutdown_timeout = 1
|
||||||
server.subscribe()
|
server.subscribe()
|
||||||
|
|
||||||
# Spawn the worker thread(s)
|
# Spawn the worker thread
|
||||||
|
Worker(cherrypy.engine, app, db_factory, frequency=5).start()
|
||||||
Worker(cherrypy.engine, app, db_factory, poll_printers, frequency=5).start()
|
|
||||||
Worker(cherrypy.engine, app, db_factory, assign_jobs, frequency=5).start()
|
|
||||||
Worker(cherrypy.engine, app, db_factory, push_jobs, frequency=5).start()
|
|
||||||
Worker(cherrypy.engine, app, db_factory, revoke_jobs, frequency=5).start()
|
|
||||||
Worker(cherrypy.engine, app, db_factory, pull_jobs, frequency=5).start()
|
|
||||||
Worker(cherrypy.engine, app, db_factory, send_emails, frequency=5).start()
|
|
||||||
|
|
||||||
# Run the server
|
# Run the server
|
||||||
cherrypy.engine.start()
|
cherrypy.engine.start()
|
||||||
|
|
|
@ -98,12 +98,10 @@ def create_file(location: Optional[str] = None):
|
||||||
return {"error": "file exists already"}, 409
|
return {"error": "file exists already"}, 409
|
||||||
|
|
||||||
file.save(sanitized_path)
|
file.save(sanitized_path)
|
||||||
fid = ctx.db.create_file(
|
fid = ctx.db.create_file(ctx.uid, file.filename, sanitized_path)
|
||||||
uid=ctx.uid, filename=file.filename, path=sanitized_path
|
|
||||||
)
|
|
||||||
|
|
||||||
if request.form.get("print", "").lower() == "true":
|
if request.form.get("print", "").lower() == "true":
|
||||||
ctx.db.create_job(uid=ctx.uid, fid=fid)
|
ctx.db.create_job(ctx.uid, fid)
|
||||||
|
|
||||||
return {"status": "ok"}, 202
|
return {"status": "ok"}, 202
|
||||||
|
|
||||||
|
@ -121,7 +119,7 @@ def get_files():
|
||||||
"owner": ctx.uid,
|
"owner": ctx.uid,
|
||||||
"upload_date": f.upload_date,
|
"upload_date": f.upload_date,
|
||||||
}
|
}
|
||||||
for f in ctx.db.list_files(uid=ctx.uid)
|
for f in ctx.db.list_files(ctx.uid)
|
||||||
]
|
]
|
||||||
}, 200
|
}, 200
|
||||||
|
|
||||||
|
@ -152,7 +150,7 @@ def get_jobs():
|
||||||
"finished_at": j.finished_at,
|
"finished_at": j.finished_at,
|
||||||
"printer_id": j.printer_id,
|
"printer_id": j.printer_id,
|
||||||
}
|
}
|
||||||
for j in ctx.db.list_jobs(uid=ctx.uid)
|
for j in ctx.db.list_jobs()
|
||||||
]
|
]
|
||||||
}, 200
|
}, 200
|
||||||
|
|
||||||
|
|
|
@ -42,10 +42,9 @@ def manipulate_files():
|
||||||
return render_template("files.html.j2"), code
|
return render_template("files.html.j2"), code
|
||||||
|
|
||||||
case "delete":
|
case "delete":
|
||||||
file = ctx.db.fetch_file(uid=ctx.uid, fid=int(request.form.get("file_id")))
|
file = ctx.db.fetch_file(ctx.uid, int(request.form.get("file_id")))
|
||||||
if any(
|
if any(
|
||||||
job.finished_at is None
|
job.finished_at is None for job in ctx.db.list_jobs_by_file(file.id)
|
||||||
for job in ctx.db.list_jobs_by_file(uid=ctx.uid, fid=file.id)
|
|
||||||
):
|
):
|
||||||
flash("File is in use", category="error")
|
flash("File is in use", category="error")
|
||||||
return render_template("files.html.j2"), 400
|
return render_template("files.html.j2"), 400
|
||||||
|
@ -53,7 +52,7 @@ def manipulate_files():
|
||||||
if os.path.exists(file.path):
|
if os.path.exists(file.path):
|
||||||
os.unlink(file.path)
|
os.unlink(file.path)
|
||||||
|
|
||||||
ctx.db.delete_file(uid=ctx.uid, fid=file.id)
|
ctx.db.delete_file(ctx.uid, file.id)
|
||||||
flash("File deleted", category="info")
|
flash("File deleted", category="info")
|
||||||
|
|
||||||
case _:
|
case _:
|
||||||
|
|
|
@ -29,24 +29,22 @@ def list_jobs():
|
||||||
def manipulate_jobs():
|
def manipulate_jobs():
|
||||||
match request.form.get("action"):
|
match request.form.get("action"):
|
||||||
case "enqueue":
|
case "enqueue":
|
||||||
ctx.db.create_job(uid=ctx.uid, fid=int(request.form.get("file_id")))
|
ctx.db.create_job(ctx.uid, int(request.form.get("file_id")))
|
||||||
flash("Job created!", category="info")
|
flash("Job created!", category="info")
|
||||||
|
|
||||||
case "duplicate":
|
case "duplicate":
|
||||||
if job := ctx.db.fetch_job(
|
if job := ctx.db.fetch_job(ctx.uid, int(request.form.get("job_id"))):
|
||||||
uid=ctx.uid, jid=int(request.form.get("job_id"))
|
ctx.db.create_job(ctx.uid, job.file_id)
|
||||||
):
|
|
||||||
ctx.db.create_job(uid=ctx.uid, fid=job.file_id)
|
|
||||||
flash("Job created!", category="info")
|
flash("Job created!", category="info")
|
||||||
else:
|
else:
|
||||||
flash("Could not duplicate", category="error")
|
flash("Could not duplicate", category="error")
|
||||||
|
|
||||||
case "cancel":
|
case "cancel":
|
||||||
ctx.db.cancel_job(uid=ctx.uid, jid=int(request.form.get("job_id")))
|
ctx.db.cancel_job(ctx.uid, int(request.form.get("job_id")))
|
||||||
flash("Cancellation reqested", category="info")
|
flash("Cancellation reqested", category="info")
|
||||||
|
|
||||||
case "delete":
|
case "delete":
|
||||||
ctx.db.delete_job(uid=ctx.uid, jid=int(request.form.get("job_id")))
|
ctx.db.delete_job(ctx.uid, int(request.form.get("job_id")))
|
||||||
flash("Job deleted", category="info")
|
flash("Job deleted", category="info")
|
||||||
|
|
||||||
case _:
|
case _:
|
||||||
|
|
|
@ -36,10 +36,9 @@ def add_printer():
|
||||||
assert request.form["url"]
|
assert request.form["url"]
|
||||||
assert request.form["api_key"]
|
assert request.form["api_key"]
|
||||||
ctx.db.try_create_printer(
|
ctx.db.try_create_printer(
|
||||||
name=request.form["name"],
|
request.form["name"],
|
||||||
url=request.form["url"],
|
request.form["url"],
|
||||||
api_key=request.form["api_key"],
|
request.form["api_key"],
|
||||||
sid=0, # Disconnected
|
|
||||||
)
|
)
|
||||||
flash("Printer created")
|
flash("Printer created")
|
||||||
return redirect("/printers")
|
return redirect("/printers")
|
||||||
|
|
|
@ -42,13 +42,13 @@ def get_login():
|
||||||
|
|
||||||
@BLUEPRINT.route("/user/login", methods=["POST"])
|
@BLUEPRINT.route("/user/login", methods=["POST"])
|
||||||
def post_login():
|
def post_login():
|
||||||
if row := ctx.db.try_login(
|
if sid := ctx.db.try_login(
|
||||||
username=(username := request.form["username"]),
|
username := request.form["username"],
|
||||||
password=salt(request.form["password"]),
|
salt(request.form["password"]),
|
||||||
ttl=timedelta(days=1),
|
timedelta(days=1),
|
||||||
):
|
):
|
||||||
resp = redirect("/")
|
resp = redirect("/")
|
||||||
resp.set_cookie("sid", row.id)
|
resp.set_cookie("sid", sid)
|
||||||
flash(f"Welcome, {username}", category="success")
|
flash(f"Welcome, {username}", category="success")
|
||||||
return resp
|
return resp
|
||||||
|
|
||||||
|
@ -72,7 +72,7 @@ def post_register():
|
||||||
username = request.form["username"]
|
username = request.form["username"]
|
||||||
email = request.form["email"]
|
email = request.form["email"]
|
||||||
group_id = 1 # Normal users
|
group_id = 1 # Normal users
|
||||||
status_id = -3 # Unverified
|
status_id = -2 # Unverified
|
||||||
|
|
||||||
for user_config in current_app.config.get("users", []):
|
for user_config in current_app.config.get("users", []):
|
||||||
if user_config["email"] == email:
|
if user_config["email"] == email:
|
||||||
|
@ -85,17 +85,13 @@ def post_register():
|
||||||
break
|
break
|
||||||
|
|
||||||
if user := ctx.db.try_create_user(
|
if user := ctx.db.try_create_user(
|
||||||
username=username,
|
username, email, salt(request.form["password"]), group_id, status_id
|
||||||
email=email,
|
|
||||||
password=salt(request.form["password"]),
|
|
||||||
gid=group_id,
|
|
||||||
sid=status_id,
|
|
||||||
):
|
):
|
||||||
if user.status_id == -3:
|
if user.status_id == -2:
|
||||||
ctx.db.create_email(
|
ctx.db.create_email(
|
||||||
uid=user.id,
|
user.id,
|
||||||
subject="Tentacles email confirmation",
|
"Tentacles email confirmation",
|
||||||
body=render_template(
|
render_template(
|
||||||
"verification_email.html.j2",
|
"verification_email.html.j2",
|
||||||
username=user.name,
|
username=user.name,
|
||||||
token_id=user.verification_token,
|
token_id=user.verification_token,
|
||||||
|
@ -107,10 +103,6 @@ def post_register():
|
||||||
"Please check your email for a verification request",
|
"Please check your email for a verification request",
|
||||||
category="success",
|
category="success",
|
||||||
)
|
)
|
||||||
|
|
||||||
elif user.status_id == 1:
|
|
||||||
flash("Welcome, please log in", category="success")
|
|
||||||
|
|
||||||
return render_template("register.html.j2")
|
return render_template("register.html.j2")
|
||||||
|
|
||||||
except Exception as e:
|
except Exception as e:
|
||||||
|
@ -123,7 +115,7 @@ def post_register():
|
||||||
@BLUEPRINT.route("/user/logout")
|
@BLUEPRINT.route("/user/logout")
|
||||||
def logout():
|
def logout():
|
||||||
# Invalidate the user's authorization
|
# Invalidate the user's authorization
|
||||||
ctx.db.delete_key(uid=ctx.uid, kid=ctx.sid)
|
ctx.db.delete_key(ctx.uid, ctx.sid)
|
||||||
resp = redirect("/")
|
resp = redirect("/")
|
||||||
resp.set_cookie("sid", "")
|
resp.set_cookie("sid", "")
|
||||||
return resp
|
return resp
|
||||||
|
@ -140,7 +132,7 @@ def get_settings():
|
||||||
@BLUEPRINT.route("/user", methods=["POST"])
|
@BLUEPRINT.route("/user", methods=["POST"])
|
||||||
def post_settings():
|
def post_settings():
|
||||||
if request.form["action"] == "add":
|
if request.form["action"] == "add":
|
||||||
ttl_spec = request.form.get("ttl", "")
|
ttl_spec = request.form.get("ttl")
|
||||||
if ttl_spec == "forever":
|
if ttl_spec == "forever":
|
||||||
ttl = None
|
ttl = None
|
||||||
elif m := re.fullmatch(r"(\d+)d", ttl_spec):
|
elif m := re.fullmatch(r"(\d+)d", ttl_spec):
|
||||||
|
@ -149,9 +141,7 @@ def post_settings():
|
||||||
flash("Bad request", category="error")
|
flash("Bad request", category="error")
|
||||||
return render_template("user.html.j2"), 400
|
return render_template("user.html.j2"), 400
|
||||||
|
|
||||||
ctx.db.create_key(
|
ctx.db.create_key(ctx.sid, ttl, request.form.get("name"))
|
||||||
uid=ctx.uid, ttl=ttl, name=request.form.get("name", "anonymous")
|
|
||||||
)
|
|
||||||
flash("Key created", category="success")
|
flash("Key created", category="success")
|
||||||
|
|
||||||
elif request.form["action"] == "revoke":
|
elif request.form["action"] == "revoke":
|
||||||
|
|
|
@ -3,13 +3,13 @@
|
||||||
from contextvars import ContextVar
|
from contextvars import ContextVar
|
||||||
|
|
||||||
from attrs import define
|
from attrs import define
|
||||||
from tentacles.db import Db
|
from tentacles.store import Store
|
||||||
from werkzeug.local import LocalProxy
|
from werkzeug.local import LocalProxy
|
||||||
|
|
||||||
|
|
||||||
@define
|
@define
|
||||||
class Ctx:
|
class Ctx:
|
||||||
db: Db
|
db: Store
|
||||||
uid: int = None
|
uid: int = None
|
||||||
gid: int = None
|
gid: int = None
|
||||||
sid: str = None
|
sid: str = None
|
||||||
|
|
|
@ -93,6 +93,7 @@ CREATE TABLE IF NOT EXISTS jobs (
|
||||||
id INTEGER PRIMARY KEY AUTOINCREMENT
|
id INTEGER PRIMARY KEY AUTOINCREMENT
|
||||||
, user_id INTEGER NOT NULL
|
, user_id INTEGER NOT NULL
|
||||||
, file_id INTEGER NOT NULL
|
, file_id INTEGER NOT NULL
|
||||||
|
, priority INTEGER CHECK(priority IS NOT NULL AND 0 <= priority)
|
||||||
, started_at TEXT
|
, started_at TEXT
|
||||||
, cancelled_at TEXT
|
, cancelled_at TEXT
|
||||||
, finished_at TEXT
|
, finished_at TEXT
|
||||||
|
@ -197,8 +198,7 @@ INSERT INTO user_keys (
|
||||||
)
|
)
|
||||||
VALUES (:uid, :name, :expiration)
|
VALUES (:uid, :name, :expiration)
|
||||||
RETURNING
|
RETURNING
|
||||||
id
|
id
|
||||||
, user_id
|
|
||||||
;
|
;
|
||||||
|
|
||||||
-- name: try-login^
|
-- name: try-login^
|
||||||
|
@ -232,8 +232,7 @@ WHERE
|
||||||
|
|
||||||
-- name: try-key^
|
-- name: try-key^
|
||||||
SELECT
|
SELECT
|
||||||
id
|
user_id
|
||||||
, user_id
|
|
||||||
FROM user_keys
|
FROM user_keys
|
||||||
WHERE
|
WHERE
|
||||||
(expiration IS NULL OR unixepoch(expiration) > unixepoch('now'))
|
(expiration IS NULL OR unixepoch(expiration) > unixepoch('now'))
|
||||||
|
@ -266,7 +265,7 @@ INSERT INTO printers (
|
||||||
, api_key
|
, api_key
|
||||||
, status_id
|
, status_id
|
||||||
)
|
)
|
||||||
VALUES (:name, :url, :api_key, :sid)
|
VALUES (:name, :url, :api_key, :status_id)
|
||||||
RETURNING
|
RETURNING
|
||||||
id
|
id
|
||||||
;
|
;
|
||||||
|
@ -311,10 +310,10 @@ WHERE
|
||||||
-- name: update-printer-status!
|
-- name: update-printer-status!
|
||||||
UPDATE printers
|
UPDATE printers
|
||||||
SET
|
SET
|
||||||
status_id = (SELECT id FROM printer_statuses WHERE name = :status or id = :status)
|
status_id = (SELECT id FROM printer_statuses WHERE name = :status)
|
||||||
, last_poll_date = datetime('now')
|
, last_poll_date = datetime('now')
|
||||||
WHERE
|
WHERE
|
||||||
id = :pid
|
id = :uid
|
||||||
;
|
;
|
||||||
|
|
||||||
----------------------------------------------------------------------------------------------------
|
----------------------------------------------------------------------------------------------------
|
||||||
|
@ -365,10 +364,16 @@ WHERE
|
||||||
INSERT INTO jobs (
|
INSERT INTO jobs (
|
||||||
user_id
|
user_id
|
||||||
, file_id
|
, file_id
|
||||||
|
, priority
|
||||||
)
|
)
|
||||||
VALUES (
|
VALUES (
|
||||||
:uid
|
:uid
|
||||||
, :fid
|
, :fid,
|
||||||
|
, (
|
||||||
|
SELECT priority + :priority
|
||||||
|
FROM users
|
||||||
|
WHERE uid = :uid
|
||||||
|
)
|
||||||
)
|
)
|
||||||
RETURNING
|
RETURNING
|
||||||
id
|
id
|
||||||
|
@ -379,7 +384,7 @@ SELECT
|
||||||
FROM jobs
|
FROM jobs
|
||||||
WHERE
|
WHERE
|
||||||
user_id = :uid
|
user_id = :uid
|
||||||
AND id = :jid
|
AND id = :fid
|
||||||
;
|
;
|
||||||
|
|
||||||
-- name: list-jobs
|
-- name: list-jobs
|
||||||
|
@ -396,7 +401,6 @@ SELECT
|
||||||
FROM jobs
|
FROM jobs
|
||||||
WHERE
|
WHERE
|
||||||
file_id = :fid
|
file_id = :fid
|
||||||
, uid = :uid
|
|
||||||
;
|
;
|
||||||
|
|
||||||
-- name: list-job-queue
|
-- name: list-job-queue
|
||||||
|
@ -406,9 +410,11 @@ FROM jobs
|
||||||
WHERE
|
WHERE
|
||||||
finished_at IS NULL
|
finished_at IS NULL
|
||||||
AND (:uid IS NULL OR user_id = :uid)
|
AND (:uid IS NULL OR user_id = :uid)
|
||||||
|
ORDER BY
|
||||||
|
priority DESC
|
||||||
;
|
;
|
||||||
|
|
||||||
-- name: poll-job-queue^
|
-- name: poll-job-queue
|
||||||
SELECT
|
SELECT
|
||||||
*
|
*
|
||||||
FROM jobs
|
FROM jobs
|
||||||
|
@ -416,6 +422,8 @@ WHERE
|
||||||
started_at IS NULL
|
started_at IS NULL
|
||||||
AND finished_at IS NULL
|
AND finished_at IS NULL
|
||||||
AND printer_id IS NULL
|
AND printer_id IS NULL
|
||||||
|
ORDER BY
|
||||||
|
priority DESC
|
||||||
LIMIT 1
|
LIMIT 1
|
||||||
;
|
;
|
||||||
|
|
||||||
|
|
|
@ -8,7 +8,6 @@ from importlib.resources import files
|
||||||
from inspect import signature
|
from inspect import signature
|
||||||
import logging
|
import logging
|
||||||
import sqlite3
|
import sqlite3
|
||||||
from time import sleep
|
|
||||||
from types import GeneratorType, new_class
|
from types import GeneratorType, new_class
|
||||||
from typing import Optional
|
from typing import Optional
|
||||||
|
|
||||||
|
@ -28,7 +27,7 @@ def qfn(name, f):
|
||||||
# Force lazy values for convenience
|
# Force lazy values for convenience
|
||||||
if isinstance(res, GeneratorType):
|
if isinstance(res, GeneratorType):
|
||||||
res = list(res)
|
res = list(res)
|
||||||
log.log(logging.DEBUG - 1, "%s (%r) -> %r", name, kwargs, res)
|
print("%s -> %r" % (name, res))
|
||||||
return res
|
return res
|
||||||
|
|
||||||
_helper.__name__ = f.__name__
|
_helper.__name__ = f.__name__
|
||||||
|
@ -65,7 +64,7 @@ class LoginError(StoreError):
|
||||||
pass
|
pass
|
||||||
|
|
||||||
|
|
||||||
class Db(Queries):
|
class Store(Queries):
|
||||||
def __init__(self, path):
|
def __init__(self, path):
|
||||||
self._path = path
|
self._path = path
|
||||||
self._conn: sqlite3.Connection = None
|
self._conn: sqlite3.Connection = None
|
||||||
|
@ -107,24 +106,9 @@ class Db(Queries):
|
||||||
try:
|
try:
|
||||||
self.begin()
|
self.begin()
|
||||||
yield self
|
yield self
|
||||||
exc = None
|
self.commit()
|
||||||
for attempt in range(5):
|
|
||||||
try:
|
|
||||||
self.commit()
|
|
||||||
break
|
|
||||||
except sqlite3.OperationalError as e:
|
|
||||||
exc = e
|
|
||||||
if e.sqlite_errorcode == 6:
|
|
||||||
sleep(0.1 * attempt)
|
|
||||||
continue
|
|
||||||
else:
|
|
||||||
raise e
|
|
||||||
else:
|
|
||||||
raise exc
|
|
||||||
|
|
||||||
except sqlite3.Error:
|
except sqlite3.Error:
|
||||||
self.rollback()
|
self.rollback()
|
||||||
log.exception("Forced to roll back!")
|
|
||||||
|
|
||||||
return _helper()
|
return _helper()
|
||||||
|
|
||||||
|
@ -138,11 +122,9 @@ class Db(Queries):
|
||||||
################################################################################
|
################################################################################
|
||||||
# Wrappers for doing Python type mapping
|
# Wrappers for doing Python type mapping
|
||||||
|
|
||||||
def create_key(self, *, uid: int, name: str, ttl: Optional[timedelta]):
|
def create_key(self, *, uid: int, name: str, ttl: timedelta):
|
||||||
return super().create_key(
|
return super().create_key(
|
||||||
uid=uid,
|
uid=uid, name=name, expiration=(datetime.now() + ttl).isoformat()
|
||||||
name=name,
|
|
||||||
expiration=((datetime.now() + ttl).isoformat() if ttl else None),
|
|
||||||
)
|
)
|
||||||
|
|
||||||
def try_login(
|
def try_login(
|
||||||
|
@ -189,6 +171,3 @@ class Db(Queries):
|
||||||
"""
|
"""
|
||||||
|
|
||||||
super().refresh_key(kid=kid, expiration=(datetime.now() + ttl).isoformat())
|
super().refresh_key(kid=kid, expiration=(datetime.now() + ttl).isoformat())
|
||||||
|
|
||||||
def finish_job(self, *, jid: int, state: str, message: Optional[str] = None):
|
|
||||||
super().finish_job(jid=jid, state=state, message=message)
|
|
|
@ -35,7 +35,7 @@
|
||||||
{% macro job_state(job) %}
|
{% macro job_state(job) %}
|
||||||
{{ 'queued' if (not job.finished_at and not job.printer_id and not job.cancelled_at) else
|
{{ 'queued' if (not job.finished_at and not job.printer_id and not job.cancelled_at) else
|
||||||
'running' if (not job.finished_at and job.printer_id and not job.cancelled_at) else
|
'running' if (not job.finished_at and job.printer_id and not job.cancelled_at) else
|
||||||
'cancelling' if (not job.finished_at and job.cancelled_at) else
|
'cancelling' if (not job.finished_at and job.printer_id and job.cancelled_at) else
|
||||||
job.state }}
|
job.state }}
|
||||||
{% endmacro %}
|
{% endmacro %}
|
||||||
|
|
||||||
|
|
|
@ -8,10 +8,11 @@ Mostly related to monitoring and managing Printer state.
|
||||||
"""
|
"""
|
||||||
|
|
||||||
from contextlib import closing
|
from contextlib import closing
|
||||||
|
from datetime import datetime, timedelta
|
||||||
import logging
|
import logging
|
||||||
from pathlib import Path
|
from pathlib import Path
|
||||||
from threading import Event
|
from threading import Event
|
||||||
from typing import Callable
|
from time import sleep
|
||||||
from urllib import parse as urlparse
|
from urllib import parse as urlparse
|
||||||
|
|
||||||
from cherrypy.process.plugins import Monitor
|
from cherrypy.process.plugins import Monitor
|
||||||
|
@ -24,7 +25,7 @@ from requests.exceptions import (
|
||||||
HTTPError,
|
HTTPError,
|
||||||
Timeout,
|
Timeout,
|
||||||
)
|
)
|
||||||
from tentacles.db import Db
|
from tentacles.store import Store
|
||||||
|
|
||||||
|
|
||||||
class OctoRest(_OR):
|
class OctoRest(_OR):
|
||||||
|
@ -45,16 +46,36 @@ log = logging.getLogger(__name__)
|
||||||
SHUTDOWN = Event()
|
SHUTDOWN = Event()
|
||||||
|
|
||||||
|
|
||||||
def poll_printers(app: App, db: Db) -> None:
|
def corn_job(every: timedelta):
|
||||||
|
def _decorator(f):
|
||||||
|
def _helper(*args, **kwargs):
|
||||||
|
last = None
|
||||||
|
while not SHUTDOWN.is_set():
|
||||||
|
if not last or (datetime.now() - last) > every:
|
||||||
|
log.debug(f"Ticking job {f.__name__}")
|
||||||
|
try:
|
||||||
|
last = datetime.now()
|
||||||
|
f(*args, **kwargs)
|
||||||
|
except Exception:
|
||||||
|
log.exception(f"Error while procesing task {f.__name__}")
|
||||||
|
else:
|
||||||
|
sleep(1)
|
||||||
|
|
||||||
|
return _helper
|
||||||
|
|
||||||
|
return _decorator
|
||||||
|
|
||||||
|
|
||||||
|
def poll_printers(app: App, store: Store) -> None:
|
||||||
"""Poll printers for their status."""
|
"""Poll printers for their status."""
|
||||||
|
|
||||||
for printer in db.list_printers():
|
for printer in store.list_printers():
|
||||||
mapped_job = db.fetch_job_by_printer(pid=printer.id)
|
mapped_job = store.fetch_job_by_printer(printer.id)
|
||||||
|
|
||||||
def _set_status(status: str):
|
def _set_status(status: str):
|
||||||
if printer.status != status:
|
if printer.status != status:
|
||||||
log.info(f"Printer {printer.id} {printer.status} -> {status}")
|
print(f"Printer {printer.id} {printer.status} -> {status}")
|
||||||
db.update_printer_status(pid=printer.id, status=status)
|
store.update_printer_status(printer.id, status)
|
||||||
|
|
||||||
try:
|
try:
|
||||||
client = OctoRest(url=printer.url, apikey=printer.api_key)
|
client = OctoRest(url=printer.url, apikey=printer.api_key)
|
||||||
|
@ -70,7 +91,7 @@ def poll_printers(app: App, db: Db) -> None:
|
||||||
# polling tasks. This violates separation of concerns a bit,
|
# polling tasks. This violates separation of concerns a bit,
|
||||||
# but appears required for correctness.
|
# but appears required for correctness.
|
||||||
if mapped_job:
|
if mapped_job:
|
||||||
db.finish_job(jid=mapped_job.id, state="error")
|
store.finish_job(mapped_job.id, "error")
|
||||||
|
|
||||||
_set_status("error")
|
_set_status("error")
|
||||||
|
|
||||||
|
@ -108,24 +129,24 @@ def poll_printers(app: App, db: Db) -> None:
|
||||||
)
|
)
|
||||||
|
|
||||||
|
|
||||||
def assign_jobs(app: App, db: Db) -> None:
|
def assign_jobs(app: App, store: Store) -> None:
|
||||||
"""Assign jobs to printers. Uploading files and job state management is handled separately."""
|
"""Assign jobs to printers. Uploading files and job state management is handled separately."""
|
||||||
|
|
||||||
for printer in db.list_idle_printers():
|
for printer_id in store.list_idle_printers():
|
||||||
if job := db.poll_job_queue():
|
if job_id := store.poll_job_queue():
|
||||||
db.assign_job(jid=job.id, pid=printer.id)
|
store.assign_job(job_id, printer_id)
|
||||||
log.info(f"Mapped job {job.id} to printer {printer.id}")
|
print(f"Mapped job {job_id} to printer {printer_id}")
|
||||||
|
|
||||||
|
|
||||||
def push_jobs(app: App, db: Db) -> None:
|
def push_jobs(app: App, store: Store) -> None:
|
||||||
"""Ensure that job files are uploaded and started to the assigned printer."""
|
"""Ensure that job files are uploaded and started to the assigned printer."""
|
||||||
|
|
||||||
for job in db.list_mapped_jobs():
|
for job in store.list_mapped_jobs():
|
||||||
printer = db.fetch_printer(pid=job.printer_id)
|
printer = store.fetch_printer(job.printer_id)
|
||||||
file = db.fetch_file(uid=job.user_id, fid=job.file_id)
|
file = store.fetch_file(job.user_id, job.file_id)
|
||||||
if not file:
|
if not file:
|
||||||
log.error(f"Job {job.id} no longer maps to a file")
|
log.error(f"Job {job.id} no longer maps to a file")
|
||||||
db.delete_job(job.user_id, job.id)
|
store.delete_job(job.user_id, job.id)
|
||||||
|
|
||||||
try:
|
try:
|
||||||
client = OctoRest(url=printer.url, apikey=printer.api_key)
|
client = OctoRest(url=printer.url, apikey=printer.api_key)
|
||||||
|
@ -136,15 +157,11 @@ def push_jobs(app: App, db: Db) -> None:
|
||||||
printer_state = {"error": printer_job.get("error")}
|
printer_state = {"error": printer_job.get("error")}
|
||||||
|
|
||||||
if printer_state.get("error"):
|
if printer_state.get("error"):
|
||||||
log.warn(f"Printer {printer.id} is in error, can't push")
|
print(f"Printer {printer.id} is in error, can't push")
|
||||||
continue
|
continue
|
||||||
|
|
||||||
try:
|
try:
|
||||||
if not client.files_info("local", Path(file.path).name):
|
client.upload(file.path)
|
||||||
client.upload(file.path)
|
|
||||||
else:
|
|
||||||
log.info("Don't need to upload the job!")
|
|
||||||
|
|
||||||
except HTTPError as e:
|
except HTTPError as e:
|
||||||
if e.response.status_code == 409:
|
if e.response.status_code == 409:
|
||||||
pass
|
pass
|
||||||
|
@ -153,7 +170,7 @@ def push_jobs(app: App, db: Db) -> None:
|
||||||
|
|
||||||
client.select(Path(file.path).name)
|
client.select(Path(file.path).name)
|
||||||
client.start()
|
client.start()
|
||||||
db.start_job(job.id)
|
store.start_job(job.id)
|
||||||
except TimeoutError:
|
except TimeoutError:
|
||||||
pass
|
pass
|
||||||
|
|
||||||
|
@ -161,19 +178,19 @@ def push_jobs(app: App, db: Db) -> None:
|
||||||
log.exception("Oop")
|
log.exception("Oop")
|
||||||
|
|
||||||
|
|
||||||
def revoke_jobs(app: App, db: Db) -> None:
|
def revoke_jobs(app: App, store: Store) -> None:
|
||||||
"""Ensure that job files are uploaded and started to the assigned printer.
|
"""Ensure that job files are uploaded and started to the assigned printer.
|
||||||
|
|
||||||
Note that this will ALSO cancel jobs out of the print queue.
|
Note that this will ALSO cancel jobs out of the print queue.
|
||||||
|
|
||||||
"""
|
"""
|
||||||
|
|
||||||
for job in db.list_canceling_jobs():
|
for job in store.list_cancelled_jobs():
|
||||||
if job.printer_id:
|
if job.printer_id:
|
||||||
printer = db.fetch_printer(pid=job.printer_id)
|
printer = store.fetch_printer(job.printer_id)
|
||||||
|
|
||||||
try:
|
try:
|
||||||
log.info(f"Cancelling running job {job.id}")
|
print(f"Cancelling running job {job.id}")
|
||||||
client = OctoRest(url=printer.url, apikey=printer.api_key)
|
client = OctoRest(url=printer.url, apikey=printer.api_key)
|
||||||
try:
|
try:
|
||||||
client.cancel()
|
client.cancel()
|
||||||
|
@ -183,8 +200,8 @@ def revoke_jobs(app: App, db: Db) -> None:
|
||||||
else:
|
else:
|
||||||
raise
|
raise
|
||||||
|
|
||||||
log.info(f"Job {job.id} -> cancelled")
|
print(f"Job {job.id} -> cancelled")
|
||||||
db.finish_job(jid=job.id, state="cancelled")
|
store.finish_job(job.id, "cancelled")
|
||||||
|
|
||||||
except TimeoutError:
|
except TimeoutError:
|
||||||
pass
|
pass
|
||||||
|
@ -193,15 +210,15 @@ def revoke_jobs(app: App, db: Db) -> None:
|
||||||
log.exception("Oop")
|
log.exception("Oop")
|
||||||
|
|
||||||
else:
|
else:
|
||||||
log.info(f"Unmapped job {job.id} became cancelled")
|
print(f"Unmapped job {job.id} became cancelled")
|
||||||
db.finish_job(jid=job.id, state="cancelled")
|
store.finish_job(job.id, "cancelled")
|
||||||
|
|
||||||
|
|
||||||
def pull_jobs(app: App, db: Db) -> None:
|
def pull_jobs(app: App, store: Store) -> None:
|
||||||
"""Poll the state of mapped printers to control jobs."""
|
"""Poll the state of mapped printers to control jobs."""
|
||||||
|
|
||||||
for job in db.list_running_jobs():
|
for job in store.list_running_jobs():
|
||||||
printer = db.fetch_printer(pid=job.printer_id)
|
printer = store.fetch_printer(job.printer_id)
|
||||||
try:
|
try:
|
||||||
client = OctoRest(url=printer.url, apikey=printer.api_key)
|
client = OctoRest(url=printer.url, apikey=printer.api_key)
|
||||||
job_state = client.job_info()
|
job_state = client.job_info()
|
||||||
|
@ -214,19 +231,19 @@ def pull_jobs(app: App, db: Db) -> None:
|
||||||
pass
|
pass
|
||||||
|
|
||||||
elif job_state.get("progress", {}).get("completion", 0.0) == 100.0:
|
elif job_state.get("progress", {}).get("completion", 0.0) == 100.0:
|
||||||
log.info(f"Job {job.id} has succeeded")
|
print(f"Job {job.id} has succeeded")
|
||||||
db.finish_job(jid=job.id, state="success")
|
store.finish_job(job.id, "success")
|
||||||
|
|
||||||
elif printer_state.get("error"):
|
elif printer_state.get("error"):
|
||||||
log.warn(f"Job {job.id} has failed")
|
print(f"Job {job.id} has failed")
|
||||||
db.finish_job(jid=job.id, state="failed")
|
store.finish_job(job.id, "failed")
|
||||||
|
|
||||||
elif printer_state.get("cancelling"):
|
elif printer_state.get("cancelling"):
|
||||||
log.info(f"Job {job.id} has been acknowledged as cancelled")
|
print(f"Job {job.id} has been acknowledged as cancelled")
|
||||||
db.finish_job(jid=job.id, state="cancelled")
|
store.finish_job(job.id, "cancelled")
|
||||||
|
|
||||||
else:
|
else:
|
||||||
log.warn(
|
print(
|
||||||
f"Job {job.id} is in a weird state {job_state.get('progress')!r} {printer_state!r}"
|
f"Job {job.id} is in a weird state {job_state.get('progress')!r} {printer_state!r}"
|
||||||
)
|
)
|
||||||
|
|
||||||
|
@ -237,60 +254,35 @@ def pull_jobs(app: App, db: Db) -> None:
|
||||||
log.exception("Oop")
|
log.exception("Oop")
|
||||||
|
|
||||||
|
|
||||||
def send_emails(app, db: Db):
|
def send_emails(app, store: Store):
|
||||||
with closing(
|
with closing(
|
||||||
FastMailSMTP(
|
FastMailSMTP(
|
||||||
app.config.get("fastmail", {}).get("username"),
|
app.config.get("fastmail", {}).get("username"),
|
||||||
app.config.get("fastmail", {}).get("key"),
|
app.config.get("fastmail", {}).get("key"),
|
||||||
)
|
)
|
||||||
) as fm:
|
) as fm:
|
||||||
for message in db.poll_email_queue():
|
for message in store.poll_spool():
|
||||||
fm.send_message(
|
fm.send_message(
|
||||||
from_addr="root@tirefireind.us",
|
from_addr="root@tirefireind.us",
|
||||||
to_addrs=[message.to],
|
to_addrs=[message.to],
|
||||||
subject=message.subject,
|
subject=message.subject,
|
||||||
msg=message.body,
|
msg=message.body,
|
||||||
)
|
)
|
||||||
db.send_email(message.id)
|
store.send_email(message.id)
|
||||||
|
|
||||||
|
|
||||||
def once(f):
|
|
||||||
val = uninitialized = object()
|
|
||||||
|
|
||||||
def _helper(*args, **kwargs):
|
|
||||||
nonlocal val
|
|
||||||
if val is uninitialized:
|
|
||||||
val = f(*args, **kwargs)
|
|
||||||
return val
|
|
||||||
|
|
||||||
return _helper
|
|
||||||
|
|
||||||
|
|
||||||
def toil(*fs):
|
|
||||||
def _helper(*args, **kwargs):
|
|
||||||
for f in fs:
|
|
||||||
f(*args, **kwargs)
|
|
||||||
|
|
||||||
_helper.__name__ = "toil"
|
|
||||||
return _helper
|
|
||||||
|
|
||||||
|
|
||||||
class Worker(Monitor):
|
class Worker(Monitor):
|
||||||
def __init__(
|
def __init__(self, bus, app, db_factory, **kwargs):
|
||||||
self,
|
|
||||||
bus,
|
|
||||||
app: App,
|
|
||||||
db_factory: Callable[[App], Db],
|
|
||||||
callback: Callable[[App, Db], None],
|
|
||||||
**kwargs,
|
|
||||||
):
|
|
||||||
self._app = app
|
self._app = app
|
||||||
self._db_factory = db_factory
|
self._db_factory = db_factory
|
||||||
self._callback = callback
|
super().__init__(bus, self.callback, **kwargs)
|
||||||
super().__init__(
|
|
||||||
bus, self.callback, **kwargs, name=f"Async {callback.__name__}"
|
|
||||||
)
|
|
||||||
|
|
||||||
def callback(self):
|
def callback(self):
|
||||||
with closing(self._db_factory(self._app)) as db:
|
log.debug("Tick")
|
||||||
self._callback(self._app, db)
|
with self._app.app_context(), closing(self._db_factory(self._app)) as store:
|
||||||
|
poll_printers(self._app, store)
|
||||||
|
assign_jobs(self._app, store)
|
||||||
|
push_jobs(self._app, store)
|
||||||
|
revoke_jobs(self._app, store)
|
||||||
|
pull_jobs(self._app, store)
|
||||||
|
send_emails(self._app, store)
|
||||||
|
|
|
@ -3,12 +3,12 @@
|
||||||
from datetime import timedelta
|
from datetime import timedelta
|
||||||
|
|
||||||
import pytest
|
import pytest
|
||||||
from tentacles.db import Db
|
import tentacles.store as s
|
||||||
|
|
||||||
|
|
||||||
@pytest.yield_fixture
|
@pytest.yield_fixture
|
||||||
def db():
|
def store():
|
||||||
conn = Db(":memory:")
|
conn = s.Store(":memory:")
|
||||||
conn.connect()
|
conn.connect()
|
||||||
yield conn
|
yield conn
|
||||||
conn.close()
|
conn.close()
|
||||||
|
@ -25,9 +25,9 @@ def password_testy():
|
||||||
|
|
||||||
|
|
||||||
@pytest.fixture
|
@pytest.fixture
|
||||||
def uid_testy(db: Db, username_testy, password_testy):
|
def uid_testy(store: s.Store, username_testy, password_testy):
|
||||||
with db.savepoint():
|
with store.savepoint():
|
||||||
return db.try_create_user(
|
return store.try_create_user(
|
||||||
username=username_testy,
|
username=username_testy,
|
||||||
email=username_testy,
|
email=username_testy,
|
||||||
password=password_testy,
|
password=password_testy,
|
||||||
|
@ -41,10 +41,8 @@ def login_ttl():
|
||||||
|
|
||||||
|
|
||||||
@pytest.fixture
|
@pytest.fixture
|
||||||
def sid_testy(db: Db, uid_testy, username_testy, password_testy, login_ttl):
|
def sid_testy(store, uid_testy, username_testy, password_testy, login_ttl):
|
||||||
with db.savepoint():
|
with store.savepoint():
|
||||||
res = db.try_login(
|
return store.try_login(
|
||||||
username=username_testy, password=password_testy, ttl=login_ttl
|
username=username_testy, password=password_testy, ttl=login_ttl
|
||||||
)
|
).id
|
||||||
assert res.user_id == uid_testy
|
|
||||||
return res.id
|
|
||||||
|
|
|
@ -1,13 +1,13 @@
|
||||||
#!/usr/bin/env python3
|
#!/usr/bin/env python3
|
||||||
|
|
||||||
from tentacles.db import Db
|
from tentacles.store import Store
|
||||||
|
|
||||||
|
|
||||||
def test_db_initializes(store: Db):
|
def test_store_initializes(store: Store):
|
||||||
assert isinstance(store, Db)
|
assert isinstance(store, Store)
|
||||||
|
|
||||||
|
|
||||||
def test_db_savepoint(store: Db):
|
def test_store_savepoint(store: Store):
|
||||||
obj = store.savepoint()
|
obj = store.savepoint()
|
||||||
|
|
||||||
assert hasattr(obj, "__enter__")
|
assert hasattr(obj, "__enter__")
|
||||||
|
|
Loading…
Reference in a new issue