基础的切片和质量控制

2026-04-10 13:58:18 +08:00
commit 975f06eb46
3302 changed files with 650758 additions and 0 deletions
--- a/app/tasks.py
+++ b/app/tasks.py
@@ -0,0 +1,85 @@
+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()
+