86 lines
3.3 KiB
Python
86 lines
3.3 KiB
Python
from huey import SqliteHuey
|
|
import subprocess
|
|
import os
|
|
from .models import db, PrintFile, SystemConfig
|
|
|
|
huey = SqliteHuey(filename='huey_queue.db')
|
|
|
|
import configparser
|
|
|
|
@huey.task()
|
|
def slice_stl_task(file_id, stl_filepath, quality_preset=None):
|
|
# This is run by the Huey worker
|
|
# We need to create an app context to interact with the database
|
|
from app import create_app
|
|
app = create_app()
|
|
with app.app_context():
|
|
print_file = PrintFile.query.get(file_id)
|
|
if not print_file:
|
|
return
|
|
|
|
# Cache variables and commit slicing status
|
|
gcode_filename = print_file.filename.rsplit('.', 1)[0] + '.gcode'
|
|
gcode_filepath = os.path.join(app.config['UPLOAD_FOLDER'], gcode_filename)
|
|
print_file.status = 'slicing'
|
|
db.session.commit()
|
|
|
|
# Remove DB session to avoid locking the sqlite db during long slicing operations
|
|
db.session.remove()
|
|
|
|
try:
|
|
# Create Cura engine options
|
|
# use our local minimal configurations detached from the entire Cura framework
|
|
print_config_path = os.path.abspath(os.path.join(app.root_path, '..', 'print_config'))
|
|
printers_path = os.path.join(print_config_path, 'printers')
|
|
extruders_path = os.path.join(print_config_path, 'extruders')
|
|
materials_path = os.path.join(print_config_path, 'materials')
|
|
presets_path = os.path.join(print_config_path, 'presets')
|
|
env = os.environ.copy()
|
|
env["CURA_ENGINE_SEARCH_PATH"] = f"{printers_path}:{extruders_path}:{materials_path}:{presets_path}"
|
|
|
|
command = [
|
|
"CuraEngine", "slice",
|
|
"-j", os.path.join(printers_path, "creality_ender3v3se.def.json")
|
|
]
|
|
|
|
# Apply quality presets if any
|
|
if quality_preset:
|
|
config = configparser.ConfigParser()
|
|
preset_path = os.path.join(presets_path, 'creality', 'base', quality_preset)
|
|
if os.path.exists(preset_path):
|
|
config.read(preset_path)
|
|
if config.has_section('values'):
|
|
for key, val in config.items('values'):
|
|
command.extend(['-s', f"{key}={val}"])
|
|
|
|
command.extend([
|
|
"-l", stl_filepath,
|
|
"-o", gcode_filepath
|
|
])
|
|
|
|
app.logger.info(f"Running command: {' '.join(command)}")
|
|
process = subprocess.Popen(command, stdout=subprocess.PIPE, stderr=subprocess.PIPE, env=env)
|
|
stdout, stderr = process.communicate()
|
|
|
|
# Re-fetch print_file and update status
|
|
print_file = PrintFile.query.get(file_id)
|
|
if not print_file:
|
|
return
|
|
|
|
if process.returncode == 0:
|
|
print_file.status = 'sliced'
|
|
else:
|
|
print_file.status = 'failed'
|
|
app.logger.error(f"CuraEngine Error: {stderr.decode()}")
|
|
|
|
except Exception as e:
|
|
# Re-fetch in case of exception
|
|
print_file = PrintFile.query.get(file_id)
|
|
if print_file:
|
|
print_file.status = 'failed'
|
|
app.logger.error(f"Subprocess Exception: {e}")
|
|
|
|
db.session.commit()
|
|
db.session.remove()
|
|
|