print_queue.py 31 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680681682683684685686687688689690691692693694695696697698699700701702703704705706707708709710711712713714715716717718719720721722723724725726727728729730731732733734735736737738739740741742743744745746747748749750751752753754755756757758759760761762763764765766767768769770771772773774775776777778779780781782783784785786
  1. """API routes for print queue management."""
  2. import json
  3. import logging
  4. import zipfile
  5. from datetime import datetime
  6. from pathlib import Path
  7. import defusedxml.ElementTree as ET
  8. from fastapi import APIRouter, Depends, HTTPException, Query
  9. from sqlalchemy import func, select
  10. from sqlalchemy.ext.asyncio import AsyncSession
  11. from sqlalchemy.orm import selectinload
  12. from backend.app.core.auth import RequirePermissionIfAuthEnabled, require_ownership_permission
  13. from backend.app.core.config import settings
  14. from backend.app.core.database import get_db
  15. from backend.app.core.permissions import Permission
  16. from backend.app.models.archive import PrintArchive
  17. from backend.app.models.library import LibraryFile
  18. from backend.app.models.print_queue import PrintQueueItem
  19. from backend.app.models.printer import Printer
  20. from backend.app.models.user import User
  21. from backend.app.schemas.print_queue import (
  22. PrintQueueBulkUpdate,
  23. PrintQueueBulkUpdateResponse,
  24. PrintQueueItemCreate,
  25. PrintQueueItemResponse,
  26. PrintQueueItemUpdate,
  27. PrintQueueReorder,
  28. )
  29. from backend.app.services.notification_service import notification_service
  30. from backend.app.utils.printer_models import normalize_printer_model, normalize_printer_model_id
  31. from backend.app.utils.threemf_tools import extract_filament_usage_from_3mf
  32. logger = logging.getLogger(__name__)
  33. router = APIRouter(prefix="/queue", tags=["queue"])
  34. def _extract_filament_types_from_3mf(file_path: Path, plate_id: int | None = None) -> list[str]:
  35. """Extract unique filament types from a 3MF file.
  36. Args:
  37. file_path: Path to the 3MF file
  38. plate_id: Optional plate index to filter for (for multi-plate files)
  39. Returns:
  40. List of unique filament types (e.g., ["PLA", "PETG"])
  41. """
  42. types: set[str] = set()
  43. try:
  44. with zipfile.ZipFile(file_path, "r") as zf:
  45. if "Metadata/slice_info.config" not in zf.namelist():
  46. return []
  47. content = zf.read("Metadata/slice_info.config").decode()
  48. root = ET.fromstring(content)
  49. if plate_id is not None:
  50. # Find the plate element with matching index
  51. for plate_elem in root.findall(".//plate"):
  52. plate_index = None
  53. for meta in plate_elem.findall("metadata"):
  54. if meta.get("key") == "index":
  55. try:
  56. plate_index = int(meta.get("value", "0"))
  57. except ValueError:
  58. pass # Skip plate with unparseable index
  59. break
  60. if plate_index == plate_id:
  61. for filament_elem in plate_elem.findall("filament"):
  62. filament_type = filament_elem.get("type", "")
  63. used_g = filament_elem.get("used_g", "0")
  64. try:
  65. used_grams = float(used_g)
  66. except (ValueError, TypeError):
  67. used_grams = 0
  68. if used_grams > 0 and filament_type:
  69. types.add(filament_type)
  70. break
  71. else:
  72. # No plate_id specified - extract all filaments with used_g > 0
  73. for filament_elem in root.findall(".//filament"):
  74. filament_type = filament_elem.get("type", "")
  75. used_g = filament_elem.get("used_g", "0")
  76. try:
  77. used_grams = float(used_g)
  78. except (ValueError, TypeError):
  79. used_grams = 0
  80. if used_grams > 0 and filament_type:
  81. types.add(filament_type)
  82. except Exception as e:
  83. logger.warning("Failed to extract filament types from %s: %s", file_path, e)
  84. return sorted(types)
  85. def _extract_print_time_from_3mf(file_path: Path, plate_id: int | None = None) -> int | None:
  86. """Extract print time (prediction) from a 3MF file.
  87. Args:
  88. file_path: Path to the 3MF file
  89. plate_id: Optional plate index to filter for (for multi-plate files)
  90. Returns:
  91. Print time in seconds, or None if not found
  92. """
  93. try:
  94. with zipfile.ZipFile(file_path, "r") as zf:
  95. if "Metadata/slice_info.config" not in zf.namelist():
  96. return None
  97. content = zf.read("Metadata/slice_info.config").decode()
  98. root = ET.fromstring(content)
  99. if plate_id is not None:
  100. for plate_elem in root.findall(".//plate"):
  101. plate_index = None
  102. for meta in plate_elem.findall("metadata"):
  103. if meta.get("key") == "index":
  104. try:
  105. plate_index = int(meta.get("value", "0"))
  106. except ValueError:
  107. pass # Skip plate with unparseable index
  108. break
  109. if plate_index == plate_id:
  110. for meta in plate_elem.findall("metadata"):
  111. if meta.get("key") == "prediction":
  112. try:
  113. return int(meta.get("value", "0"))
  114. except ValueError:
  115. return None
  116. break
  117. else:
  118. plate_elem = root.find(".//plate")
  119. if plate_elem is not None:
  120. for meta in plate_elem.findall("metadata"):
  121. if meta.get("key") == "prediction":
  122. try:
  123. return int(meta.get("value", "0"))
  124. except ValueError:
  125. return None
  126. except Exception as e:
  127. logger.warning("Failed to extract print time from %s: %s", file_path, e)
  128. return None
  129. def _enrich_response(item: PrintQueueItem) -> PrintQueueItemResponse:
  130. """Add nested archive/printer/library_file info to response."""
  131. # Parse ams_mapping from JSON string BEFORE model_validate
  132. ams_mapping_parsed = None
  133. if item.ams_mapping:
  134. try:
  135. ams_mapping_parsed = json.loads(item.ams_mapping)
  136. except json.JSONDecodeError:
  137. ams_mapping_parsed = None
  138. # Parse required_filament_types from JSON string
  139. required_filament_types_parsed = None
  140. if item.required_filament_types:
  141. try:
  142. required_filament_types_parsed = json.loads(item.required_filament_types)
  143. except json.JSONDecodeError:
  144. required_filament_types_parsed = None
  145. # Create response with parsed ams_mapping
  146. item_dict = {
  147. "id": item.id,
  148. "printer_id": item.printer_id,
  149. "target_model": item.target_model,
  150. "target_location": item.target_location,
  151. "required_filament_types": required_filament_types_parsed,
  152. "waiting_reason": item.waiting_reason,
  153. "archive_id": item.archive_id,
  154. "library_file_id": item.library_file_id,
  155. "position": item.position,
  156. "scheduled_time": item.scheduled_time,
  157. "require_previous_success": item.require_previous_success,
  158. "auto_off_after": item.auto_off_after,
  159. "manual_start": item.manual_start,
  160. "ams_mapping": ams_mapping_parsed,
  161. "plate_id": item.plate_id,
  162. "bed_levelling": item.bed_levelling,
  163. "flow_cali": item.flow_cali,
  164. "vibration_cali": item.vibration_cali,
  165. "layer_inspect": item.layer_inspect,
  166. "timelapse": item.timelapse,
  167. "use_ams": item.use_ams,
  168. "status": item.status,
  169. "started_at": item.started_at,
  170. "completed_at": item.completed_at,
  171. "error_message": item.error_message,
  172. "created_at": item.created_at,
  173. # User tracking (Issue #206)
  174. "created_by_id": item.created_by_id,
  175. "created_by_username": item.created_by.username if item.created_by else None,
  176. }
  177. response = PrintQueueItemResponse(**item_dict)
  178. if item.archive:
  179. response.archive_name = item.archive.print_name or item.archive.filename
  180. response.archive_thumbnail = item.archive.thumbnail_path
  181. response.print_time_seconds = item.archive.print_time_seconds
  182. response.filament_used_grams = item.archive.filament_used_grams
  183. if item.plate_id:
  184. archive_path = settings.base_dir / item.archive.file_path
  185. if archive_path.exists():
  186. plate_time = _extract_print_time_from_3mf(archive_path, item.plate_id)
  187. plate_weight = sum(f["used_g"] for f in extract_filament_usage_from_3mf(archive_path, item.plate_id))
  188. if plate_time is not None:
  189. response.print_time_seconds = plate_time
  190. if plate_weight > 0:
  191. response.filament_used_grams = plate_weight
  192. if item.library_file:
  193. response.library_file_name = (
  194. item.library_file.file_metadata.get("print_name") if item.library_file.file_metadata else None
  195. )
  196. if not response.library_file_name:
  197. response.library_file_name = item.library_file.filename
  198. response.library_file_thumbnail = item.library_file.thumbnail_path
  199. # Get print time from library file metadata if no archive
  200. if not item.archive and item.library_file.file_metadata:
  201. response.print_time_seconds = item.library_file.file_metadata.get("print_time_seconds")
  202. response.filament_used_grams = item.library_file.file_metadata.get("filament_used_grams")
  203. if item.plate_id:
  204. lib_path = Path(item.library_file.file_path)
  205. library_file_path = lib_path if lib_path.is_absolute() else settings.base_dir / item.library_file.file_path
  206. if library_file_path.exists():
  207. plate_time = _extract_print_time_from_3mf(library_file_path, item.plate_id)
  208. plate_weight = sum(
  209. f["used_g"] for f in extract_filament_usage_from_3mf(library_file_path, item.plate_id)
  210. )
  211. if plate_time is not None:
  212. response.print_time_seconds = plate_time
  213. if plate_weight > 0:
  214. response.filament_used_grams = plate_weight
  215. if item.printer:
  216. response.printer_name = item.printer.name
  217. return response
  218. @router.get("/", response_model=list[PrintQueueItemResponse])
  219. async def list_queue(
  220. printer_id: int | None = Query(None, description="Filter by printer (-1 for unassigned)"),
  221. status: str | None = Query(None, description="Filter by status"),
  222. db: AsyncSession = Depends(get_db),
  223. _: User | None = RequirePermissionIfAuthEnabled(Permission.QUEUE_READ),
  224. ):
  225. """List all queue items, optionally filtered by printer or status."""
  226. query = (
  227. select(PrintQueueItem)
  228. .options(
  229. selectinload(PrintQueueItem.archive),
  230. selectinload(PrintQueueItem.printer),
  231. selectinload(PrintQueueItem.library_file),
  232. selectinload(PrintQueueItem.created_by),
  233. )
  234. .order_by(PrintQueueItem.printer_id.nulls_first(), PrintQueueItem.position)
  235. )
  236. if printer_id is not None:
  237. if printer_id == -1:
  238. # Special value: filter for unassigned items
  239. query = query.where(PrintQueueItem.printer_id.is_(None))
  240. else:
  241. query = query.where(PrintQueueItem.printer_id == printer_id)
  242. if status:
  243. query = query.where(PrintQueueItem.status == status)
  244. result = await db.execute(query)
  245. items = result.scalars().all()
  246. return [_enrich_response(item) for item in items]
  247. @router.post("/", response_model=PrintQueueItemResponse)
  248. async def add_to_queue(
  249. data: PrintQueueItemCreate,
  250. db: AsyncSession = Depends(get_db),
  251. current_user: User | None = RequirePermissionIfAuthEnabled(Permission.QUEUE_CREATE),
  252. ):
  253. """Add an item to the print queue."""
  254. # Normalize target_model (e.g., "Bambu Lab X1E" / "C13" -> "X1E")
  255. target_model_norm = None
  256. if data.target_model:
  257. target_model_norm = (
  258. normalize_printer_model(data.target_model)
  259. or normalize_printer_model_id(data.target_model)
  260. or data.target_model
  261. )
  262. # Validate that either archive_id or library_file_id is provided
  263. if not data.archive_id and not data.library_file_id:
  264. raise HTTPException(400, "Either archive_id or library_file_id must be provided")
  265. # Cannot specify both printer_id and target_model
  266. if data.printer_id and target_model_norm:
  267. raise HTTPException(400, "Cannot specify both printer_id and target_model")
  268. # Validate printer exists (if assigned)
  269. if data.printer_id is not None:
  270. result = await db.execute(select(Printer).where(Printer.id == data.printer_id))
  271. if not result.scalar_one_or_none():
  272. raise HTTPException(400, "Printer not found")
  273. # Validate target_model has active printers
  274. if target_model_norm:
  275. result = await db.execute(
  276. select(Printer).where(Printer.model == target_model_norm).where(Printer.is_active == True) # noqa: E712
  277. )
  278. if not result.scalars().first():
  279. raise HTTPException(400, f"No active printers for model: {target_model_norm}")
  280. # Validate archive exists (if provided) and get it for filament extraction
  281. archive = None
  282. if data.archive_id:
  283. result = await db.execute(select(PrintArchive).where(PrintArchive.id == data.archive_id))
  284. archive = result.scalar_one_or_none()
  285. if not archive:
  286. raise HTTPException(400, "Archive not found")
  287. # Validate library file exists (if provided) and get it for filament extraction
  288. library_file = None
  289. if data.library_file_id:
  290. result = await db.execute(select(LibraryFile).where(LibraryFile.id == data.library_file_id))
  291. library_file = result.scalar_one_or_none()
  292. if not library_file:
  293. raise HTTPException(400, "Library file not found")
  294. # Extract filament types for model-based assignment (used by scheduler for validation)
  295. required_filament_types = None
  296. if target_model_norm:
  297. # Get file path from archive or library file
  298. file_path = None
  299. if archive:
  300. file_path = settings.base_dir / archive.file_path
  301. elif library_file:
  302. lib_path = Path(library_file.file_path)
  303. file_path = lib_path if lib_path.is_absolute() else settings.base_dir / library_file.file_path
  304. if file_path and file_path.exists():
  305. filament_types = _extract_filament_types_from_3mf(file_path, data.plate_id)
  306. if filament_types:
  307. required_filament_types = json.dumps(filament_types)
  308. logger.info("Extracted filament types for model-based queue: %s", filament_types)
  309. # Get next position for this printer (or for unassigned/model-based items)
  310. if data.printer_id is not None:
  311. result = await db.execute(
  312. select(func.max(PrintQueueItem.position))
  313. .where(PrintQueueItem.printer_id == data.printer_id)
  314. .where(PrintQueueItem.status == "pending")
  315. )
  316. else:
  317. # For unassigned/model-based items, get max position across all unassigned
  318. result = await db.execute(
  319. select(func.max(PrintQueueItem.position))
  320. .where(PrintQueueItem.printer_id.is_(None))
  321. .where(PrintQueueItem.status == "pending")
  322. )
  323. max_pos = result.scalar() or 0
  324. item = PrintQueueItem(
  325. printer_id=data.printer_id,
  326. target_model=target_model_norm,
  327. target_location=data.target_location,
  328. required_filament_types=required_filament_types,
  329. archive_id=data.archive_id,
  330. library_file_id=data.library_file_id,
  331. scheduled_time=data.scheduled_time,
  332. require_previous_success=data.require_previous_success,
  333. auto_off_after=data.auto_off_after,
  334. manual_start=data.manual_start,
  335. ams_mapping=json.dumps(data.ams_mapping) if data.ams_mapping else None,
  336. plate_id=data.plate_id,
  337. bed_levelling=data.bed_levelling,
  338. flow_cali=data.flow_cali,
  339. vibration_cali=data.vibration_cali,
  340. layer_inspect=data.layer_inspect,
  341. timelapse=data.timelapse,
  342. use_ams=data.use_ams,
  343. position=max_pos + 1,
  344. status="pending",
  345. created_by_id=current_user.id if current_user else None,
  346. )
  347. db.add(item)
  348. await db.commit()
  349. await db.refresh(item)
  350. # Load relationships for response
  351. await db.refresh(item, ["archive", "printer", "library_file", "created_by"])
  352. source_name = f"archive {data.archive_id}" if data.archive_id else f"library file {data.library_file_id}"
  353. target_desc = data.printer_id or (f"model {target_model_norm}" if target_model_norm else "unassigned")
  354. logger.info("Added %s to queue for %s", source_name, target_desc)
  355. # MQTT relay - publish queue job added
  356. try:
  357. from backend.app.services.mqtt_relay import mqtt_relay
  358. await mqtt_relay.on_queue_job_added(
  359. job_id=item.id,
  360. filename=item.archive.filename if item.archive else "",
  361. printer_id=item.printer_id,
  362. printer_name=item.printer.name if item.printer else None,
  363. )
  364. except Exception:
  365. pass # Don't fail queue add if MQTT fails
  366. # Send notification for job added
  367. try:
  368. job_name = (
  369. item.archive.filename
  370. if item.archive
  371. else item.library_file.filename
  372. if item.library_file
  373. else f"Job #{item.id}"
  374. )
  375. job_name = job_name.replace(".gcode.3mf", "").replace(".3mf", "")
  376. target = (
  377. item.printer.name if item.printer else (f"Any {item.target_model}" if target_model_norm else "Unassigned")
  378. )
  379. await notification_service.on_queue_job_added(
  380. job_name=job_name,
  381. target=target,
  382. db=db,
  383. printer_id=item.printer_id,
  384. printer_name=item.printer.name if item.printer else None,
  385. )
  386. except Exception:
  387. pass # Don't fail queue add if notification fails
  388. return _enrich_response(item)
  389. @router.patch("/bulk", response_model=PrintQueueBulkUpdateResponse)
  390. async def bulk_update_queue_items(
  391. data: PrintQueueBulkUpdate,
  392. db: AsyncSession = Depends(get_db),
  393. auth_result: tuple[User | None, bool] = Depends(
  394. require_ownership_permission(
  395. Permission.QUEUE_UPDATE_ALL,
  396. Permission.QUEUE_UPDATE_OWN,
  397. )
  398. ),
  399. ):
  400. """Bulk update multiple queue items with the same values.
  401. Only pending items can be updated. Non-pending items are skipped.
  402. Items not owned by the user are also skipped (unless user has *_all permission).
  403. """
  404. user, can_modify_all = auth_result
  405. if not data.item_ids:
  406. raise HTTPException(400, "No item IDs provided")
  407. # Get fields to update (exclude item_ids and unset fields)
  408. update_data = data.model_dump(exclude={"item_ids"}, exclude_unset=True)
  409. if not update_data:
  410. raise HTTPException(400, "No fields to update")
  411. # Validate printer_id if being changed
  412. if "printer_id" in update_data and update_data["printer_id"] is not None:
  413. result = await db.execute(select(Printer).where(Printer.id == update_data["printer_id"]))
  414. if not result.scalar_one_or_none():
  415. raise HTTPException(400, "Printer not found")
  416. # Fetch all items
  417. result = await db.execute(select(PrintQueueItem).where(PrintQueueItem.id.in_(data.item_ids)))
  418. items = result.scalars().all()
  419. updated_count = 0
  420. skipped_count = 0
  421. for item in items:
  422. if item.status != "pending":
  423. skipped_count += 1
  424. continue
  425. # Ownership check
  426. if not can_modify_all and item.created_by_id != user.id:
  427. skipped_count += 1
  428. continue
  429. for field, value in update_data.items():
  430. setattr(item, field, value)
  431. updated_count += 1
  432. await db.commit()
  433. logger.info("Bulk updated %s queue items, skipped %s", updated_count, skipped_count)
  434. return PrintQueueBulkUpdateResponse(
  435. updated_count=updated_count,
  436. skipped_count=skipped_count,
  437. message=f"Updated {updated_count} items"
  438. + (f", skipped {skipped_count} non-pending/not-owned" if skipped_count else ""),
  439. )
  440. @router.get("/{item_id}", response_model=PrintQueueItemResponse)
  441. async def get_queue_item(
  442. item_id: int,
  443. db: AsyncSession = Depends(get_db),
  444. _: User | None = RequirePermissionIfAuthEnabled(Permission.QUEUE_READ),
  445. ):
  446. """Get a specific queue item."""
  447. result = await db.execute(
  448. select(PrintQueueItem)
  449. .options(
  450. selectinload(PrintQueueItem.archive),
  451. selectinload(PrintQueueItem.printer),
  452. selectinload(PrintQueueItem.library_file),
  453. selectinload(PrintQueueItem.created_by),
  454. )
  455. .where(PrintQueueItem.id == item_id)
  456. )
  457. item = result.scalar_one_or_none()
  458. if not item:
  459. raise HTTPException(404, "Queue item not found")
  460. return _enrich_response(item)
  461. @router.patch("/{item_id}", response_model=PrintQueueItemResponse)
  462. async def update_queue_item(
  463. item_id: int,
  464. data: PrintQueueItemUpdate,
  465. db: AsyncSession = Depends(get_db),
  466. auth_result: tuple[User | None, bool] = Depends(
  467. require_ownership_permission(
  468. Permission.QUEUE_UPDATE_ALL,
  469. Permission.QUEUE_UPDATE_OWN,
  470. )
  471. ),
  472. ):
  473. """Update a queue item."""
  474. user, can_modify_all = auth_result
  475. result = await db.execute(select(PrintQueueItem).where(PrintQueueItem.id == item_id))
  476. item = result.scalar_one_or_none()
  477. if not item:
  478. raise HTTPException(404, "Queue item not found")
  479. # Ownership check
  480. if not can_modify_all:
  481. if item.created_by_id != user.id:
  482. raise HTTPException(403, "You can only update your own queue items")
  483. if item.status != "pending":
  484. raise HTTPException(400, "Can only update pending items")
  485. update_data = data.model_dump(exclude_unset=True)
  486. # Normalize target_model if being updated
  487. if "target_model" in update_data and update_data["target_model"]:
  488. update_data["target_model"] = (
  489. normalize_printer_model(update_data["target_model"])
  490. or normalize_printer_model_id(update_data["target_model"])
  491. or update_data["target_model"]
  492. )
  493. # Cannot specify both printer_id and target_model
  494. new_printer_id = update_data.get("printer_id", item.printer_id)
  495. new_target_model = update_data.get("target_model", item.target_model)
  496. if new_printer_id and new_target_model:
  497. raise HTTPException(400, "Cannot specify both printer_id and target_model")
  498. # Validate new printer_id if being changed (and not None)
  499. if "printer_id" in update_data and update_data["printer_id"] is not None:
  500. result = await db.execute(select(Printer).where(Printer.id == update_data["printer_id"]))
  501. if not result.scalar_one_or_none():
  502. raise HTTPException(400, "Printer not found")
  503. # Validate target_model has active printers
  504. if "target_model" in update_data and update_data["target_model"]:
  505. result = await db.execute(
  506. select(Printer).where(Printer.model == update_data["target_model"]).where(Printer.is_active == True) # noqa: E712
  507. )
  508. if not result.scalars().first():
  509. raise HTTPException(400, f"No active printers for model: {update_data['target_model']}")
  510. # Serialize ams_mapping to JSON for TEXT column storage
  511. if "ams_mapping" in update_data:
  512. update_data["ams_mapping"] = json.dumps(update_data["ams_mapping"]) if update_data["ams_mapping"] else None
  513. for field, value in update_data.items():
  514. setattr(item, field, value)
  515. await db.commit()
  516. await db.refresh(item, ["archive", "printer", "library_file", "created_by"])
  517. logger.info("Updated queue item %s", item_id)
  518. return _enrich_response(item)
  519. @router.delete("/{item_id}")
  520. async def delete_queue_item(
  521. item_id: int,
  522. db: AsyncSession = Depends(get_db),
  523. auth_result: tuple[User | None, bool] = Depends(
  524. require_ownership_permission(
  525. Permission.QUEUE_DELETE_ALL,
  526. Permission.QUEUE_DELETE_OWN,
  527. )
  528. ),
  529. ):
  530. """Remove an item from the queue."""
  531. user, can_modify_all = auth_result
  532. result = await db.execute(select(PrintQueueItem).where(PrintQueueItem.id == item_id))
  533. item = result.scalar_one_or_none()
  534. if not item:
  535. raise HTTPException(404, "Queue item not found")
  536. # Ownership check
  537. if not can_modify_all:
  538. if item.created_by_id != user.id:
  539. raise HTTPException(403, "You can only delete your own queue items")
  540. if item.status == "printing":
  541. raise HTTPException(400, "Cannot delete item that is currently printing")
  542. await db.delete(item)
  543. await db.commit()
  544. logger.info("Deleted queue item %s", item_id)
  545. return {"message": "Queue item deleted"}
  546. @router.post("/reorder")
  547. async def reorder_queue(
  548. data: PrintQueueReorder,
  549. db: AsyncSession = Depends(get_db),
  550. _: User | None = RequirePermissionIfAuthEnabled(Permission.QUEUE_UPDATE_ALL),
  551. ):
  552. """Bulk update positions for queue items."""
  553. for reorder_item in data.items:
  554. result = await db.execute(select(PrintQueueItem).where(PrintQueueItem.id == reorder_item.id))
  555. item = result.scalar_one_or_none()
  556. if item and item.status == "pending":
  557. item.position = reorder_item.position
  558. await db.commit()
  559. logger.info("Reordered %s queue items", len(data.items))
  560. return {"message": f"Reordered {len(data.items)} items"}
  561. @router.post("/{item_id}/cancel")
  562. async def cancel_queue_item(
  563. item_id: int,
  564. db: AsyncSession = Depends(get_db),
  565. auth_result: tuple[User | None, bool] = Depends(
  566. require_ownership_permission(
  567. Permission.QUEUE_UPDATE_ALL,
  568. Permission.QUEUE_UPDATE_OWN,
  569. )
  570. ),
  571. ):
  572. """Cancel a pending queue item."""
  573. user, can_modify_all = auth_result
  574. result = await db.execute(select(PrintQueueItem).where(PrintQueueItem.id == item_id))
  575. item = result.scalar_one_or_none()
  576. if not item:
  577. raise HTTPException(404, "Queue item not found")
  578. # Ownership check
  579. if not can_modify_all:
  580. if item.created_by_id != user.id:
  581. raise HTTPException(403, "You can only cancel your own queue items")
  582. if item.status not in ("pending",):
  583. raise HTTPException(400, f"Cannot cancel item with status '{item.status}'")
  584. item.status = "cancelled"
  585. item.completed_at = datetime.now()
  586. await db.commit()
  587. logger.info("Cancelled queue item %s", item_id)
  588. return {"message": "Queue item cancelled"}
  589. @router.post("/{item_id}/stop")
  590. async def stop_queue_item(
  591. item_id: int,
  592. db: AsyncSession = Depends(get_db),
  593. _: User | None = RequirePermissionIfAuthEnabled(Permission.QUEUE_UPDATE_ALL),
  594. ):
  595. """Stop an actively printing queue item."""
  596. import asyncio
  597. from backend.app.models.smart_plug import SmartPlug
  598. from backend.app.services.printer_manager import printer_manager
  599. from backend.app.services.tasmota import tasmota_service
  600. result = await db.execute(select(PrintQueueItem).where(PrintQueueItem.id == item_id))
  601. item = result.scalar_one_or_none()
  602. if not item:
  603. raise HTTPException(404, "Queue item not found")
  604. if item.status != "printing":
  605. raise HTTPException(400, f"Can only stop items that are printing, current status: '{item.status}'")
  606. # Capture values we need for background task
  607. printer_id = item.printer_id
  608. auto_off_after = item.auto_off_after
  609. # Try to send stop command to printer
  610. stop_sent = False
  611. try:
  612. stop_sent = printer_manager.stop_print(printer_id)
  613. if not stop_sent:
  614. logger.warning("stop_print returned False for printer %s - printer may not be connected", printer_id)
  615. except Exception as e:
  616. logger.error("Error sending stop command for queue item %s: %s", item_id, e)
  617. # Update queue item status regardless - if printer is off, print is already stopped
  618. item.status = "cancelled"
  619. item.completed_at = datetime.now()
  620. item.error_message = "Stopped by user" if stop_sent else "Stopped by user (printer was offline)"
  621. await db.commit()
  622. # Get smart plug info if auto-off is enabled
  623. plug_ip = None
  624. if auto_off_after:
  625. result = await db.execute(select(SmartPlug).where(SmartPlug.printer_id == printer_id))
  626. plug = result.scalar_one_or_none()
  627. if plug and plug.enabled:
  628. plug_ip = plug.ip_address
  629. logger.info("Stopped printing queue item %s (stop command sent: %s)", item_id, stop_sent)
  630. # Schedule background task for cooldown + power off
  631. if plug_ip:
  632. async def cooldown_and_poweroff():
  633. logger.info("Auto-off: Waiting for printer %s to cool down before power off...", printer_id)
  634. await printer_manager.wait_for_cooldown(printer_id, target_temp=50.0, timeout=600)
  635. # Re-fetch plug since we're in a new async context
  636. from backend.app.core.database import async_session
  637. async with async_session() as new_db:
  638. result = await new_db.execute(select(SmartPlug).where(SmartPlug.printer_id == printer_id))
  639. plug = result.scalar_one_or_none()
  640. if plug and plug.enabled:
  641. logger.info("Auto-off: Powering off printer %s", printer_id)
  642. await tasmota_service.turn_off(plug)
  643. asyncio.create_task(cooldown_and_poweroff())
  644. return {"message": "Print stopped" if stop_sent else "Queue item cancelled (printer was offline)"}
  645. @router.post("/{item_id}/start")
  646. async def start_queue_item(
  647. item_id: int,
  648. db: AsyncSession = Depends(get_db),
  649. _: User | None = RequirePermissionIfAuthEnabled(Permission.QUEUE_UPDATE_OWN),
  650. ):
  651. """Manually start a staged (manual_start) queue item.
  652. This clears the manual_start flag so the scheduler will pick it up,
  653. or starts immediately if the printer is ready.
  654. """
  655. result = await db.execute(
  656. select(PrintQueueItem)
  657. .options(selectinload(PrintQueueItem.archive), selectinload(PrintQueueItem.printer))
  658. .where(PrintQueueItem.id == item_id)
  659. )
  660. item = result.scalar_one_or_none()
  661. if not item:
  662. raise HTTPException(404, "Queue item not found")
  663. if item.status != "pending":
  664. raise HTTPException(400, f"Can only start pending items, current status: '{item.status}'")
  665. # Clear manual_start flag so scheduler picks it up
  666. item.manual_start = False
  667. await db.commit()
  668. await db.refresh(item, ["archive", "printer", "library_file", "created_by"])
  669. logger.info("Manually started queue item %s (cleared manual_start flag)", item_id)
  670. return _enrich_response(item)