bambu_mqtt.py 29 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680681682683684685686687688689690691692693694695696697698699700701702703704705706707708709710711712713714715716717718719720721722723724725726727728729730731732733
  1. import json
  2. import ssl
  3. import asyncio
  4. import logging
  5. import time
  6. from collections import deque
  7. from datetime import datetime
  8. from typing import Callable
  9. from dataclasses import dataclass, field
  10. import paho.mqtt.client as mqtt
  11. logger = logging.getLogger(__name__)
  12. @dataclass
  13. class MQTTLogEntry:
  14. """Log entry for MQTT message debugging."""
  15. timestamp: str
  16. topic: str
  17. direction: str # "in" or "out"
  18. payload: dict
  19. @dataclass
  20. class HMSError:
  21. """Health Management System error from printer."""
  22. code: str
  23. module: int
  24. severity: int # 1=fatal, 2=serious, 3=common, 4=info
  25. message: str = ""
  26. @dataclass
  27. class KProfile:
  28. """Pressure advance (K) calibration profile from printer."""
  29. slot_id: int
  30. extruder_id: int
  31. nozzle_id: str
  32. nozzle_diameter: str
  33. filament_id: str
  34. name: str
  35. k_value: str
  36. n_coef: str = "0.000000"
  37. ams_id: int = 0
  38. tray_id: int = -1
  39. setting_id: str | None = None
  40. @dataclass
  41. class PrinterState:
  42. connected: bool = False
  43. state: str = "unknown"
  44. current_print: str | None = None
  45. subtask_name: str | None = None
  46. progress: float = 0.0
  47. remaining_time: int = 0
  48. layer_num: int = 0
  49. total_layers: int = 0
  50. temperatures: dict = field(default_factory=dict)
  51. raw_data: dict = field(default_factory=dict)
  52. gcode_file: str | None = None
  53. subtask_id: str | None = None
  54. hms_errors: list = field(default_factory=list) # List of HMSError
  55. kprofiles: list = field(default_factory=list) # List of KProfile
  56. class BambuMQTTClient:
  57. """MQTT client for Bambu Lab printer communication."""
  58. MQTT_PORT = 8883
  59. def __init__(
  60. self,
  61. ip_address: str,
  62. serial_number: str,
  63. access_code: str,
  64. on_state_change: Callable[[PrinterState], None] | None = None,
  65. on_print_start: Callable[[dict], None] | None = None,
  66. on_print_complete: Callable[[dict], None] | None = None,
  67. on_ams_change: Callable[[list], None] | None = None,
  68. ):
  69. self.ip_address = ip_address
  70. self.serial_number = serial_number
  71. self.access_code = access_code
  72. self.on_state_change = on_state_change
  73. self.on_print_start = on_print_start
  74. self.on_print_complete = on_print_complete
  75. self.on_ams_change = on_ams_change
  76. self.state = PrinterState()
  77. self._client: mqtt.Client | None = None
  78. self._loop: asyncio.AbstractEventLoop | None = None
  79. self._previous_gcode_state: str | None = None
  80. self._previous_gcode_file: str | None = None
  81. self._was_running: bool = False # Track if we've seen RUNNING state for current print
  82. self._completion_triggered: bool = False # Prevent duplicate completion triggers
  83. self._message_log: deque[MQTTLogEntry] = deque(maxlen=100)
  84. self._logging_enabled: bool = False
  85. self._last_message_time: float = 0.0 # Track when we last received a message
  86. self._previous_ams_hash: str | None = None # Track AMS changes
  87. # K-profile command tracking
  88. self._sequence_id: int = 0
  89. self._pending_kprofile_response: asyncio.Event | None = None
  90. self._kprofile_response_data: list | None = None
  91. @property
  92. def topic_subscribe(self) -> str:
  93. return f"device/{self.serial_number}/report"
  94. @property
  95. def topic_publish(self) -> str:
  96. return f"device/{self.serial_number}/request"
  97. def _on_connect(self, client, userdata, flags, rc, properties=None):
  98. if rc == 0:
  99. self.state.connected = True
  100. client.subscribe(self.topic_subscribe)
  101. # Request full status update
  102. self._request_push_all()
  103. # Immediately broadcast connection state change
  104. if self.on_state_change:
  105. self.on_state_change(self.state)
  106. else:
  107. self.state.connected = False
  108. def _on_disconnect(self, client, userdata, disconnect_flags=None, rc=None, properties=None):
  109. # Ignore spurious disconnect callbacks if we've received a message recently
  110. # Paho-mqtt sometimes fires disconnect callbacks while the connection is still active
  111. time_since_last_message = time.time() - self._last_message_time
  112. if time_since_last_message < 30.0 and self._last_message_time > 0:
  113. logger.debug(
  114. f"[{self.serial_number}] Ignoring spurious disconnect (last message {time_since_last_message:.1f}s ago)"
  115. )
  116. return
  117. logger.warning(f"[{self.serial_number}] MQTT disconnected: rc={rc}, flags={disconnect_flags}")
  118. self.state.connected = False
  119. if self.on_state_change:
  120. self.on_state_change(self.state)
  121. def _on_message(self, client, userdata, msg):
  122. try:
  123. payload = json.loads(msg.payload.decode())
  124. # Track last message time - receiving a message proves we're connected
  125. self._last_message_time = time.time()
  126. self.state.connected = True
  127. # Log message if logging is enabled
  128. if self._logging_enabled:
  129. self._message_log.append(MQTTLogEntry(
  130. timestamp=datetime.now().isoformat(),
  131. topic=msg.topic,
  132. direction="in",
  133. payload=payload,
  134. ))
  135. self._process_message(payload)
  136. except json.JSONDecodeError:
  137. pass
  138. def _process_message(self, payload: dict):
  139. """Process incoming MQTT message from printer."""
  140. # Handle top-level AMS data (comes outside of "print" key)
  141. # Wrap in try/except to prevent breaking the MQTT connection
  142. if "ams" in payload:
  143. try:
  144. self._handle_ams_data(payload["ams"])
  145. except Exception as e:
  146. logger.error(f"[{self.serial_number}] Error handling AMS data: {e}")
  147. if "print" in payload:
  148. print_data = payload["print"]
  149. # Log when we see gcode_state changes
  150. if "gcode_state" in print_data:
  151. logger.info(
  152. f"[{self.serial_number}] Received gcode_state: {print_data.get('gcode_state')}, "
  153. f"gcode_file: {print_data.get('gcode_file')}, subtask_name: {print_data.get('subtask_name')}"
  154. )
  155. # Check for K-profile response (extrusion_cali)
  156. if "command" in print_data:
  157. logger.debug(f"[{self.serial_number}] Received command response: {print_data.get('command')}")
  158. if "command" in print_data and print_data.get("command") == "extrusion_cali_get":
  159. self._handle_kprofile_response(print_data)
  160. self._update_state(print_data)
  161. def _handle_ams_data(self, ams_data: list):
  162. """Handle AMS data changes for Spoolman integration.
  163. This is called when we receive top-level AMS data in MQTT messages.
  164. It detects changes and triggers the callback for Spoolman sync.
  165. """
  166. import hashlib
  167. # Store AMS data in raw_data so it's accessible via API
  168. if "ams" not in self.state.raw_data:
  169. self.state.raw_data["ams"] = ams_data
  170. else:
  171. self.state.raw_data["ams"] = ams_data
  172. # Create a hash of relevant AMS data to detect changes
  173. ams_hash_data = []
  174. for ams_unit in ams_data:
  175. for tray in ams_unit.get("tray", []):
  176. # Include fields that matter for filament tracking
  177. ams_hash_data.append(
  178. f"{ams_unit.get('id')}:{tray.get('id')}:"
  179. f"{tray.get('tray_type')}:{tray.get('tag_uid')}:{tray.get('remain')}"
  180. )
  181. ams_hash = hashlib.md5(":".join(ams_hash_data).encode()).hexdigest()
  182. # Only trigger callback if AMS data actually changed
  183. if ams_hash != self._previous_ams_hash:
  184. self._previous_ams_hash = ams_hash
  185. if self.on_ams_change:
  186. logger.info(f"[{self.serial_number}] AMS data changed, triggering sync callback")
  187. self.on_ams_change(ams_data)
  188. def _update_state(self, data: dict):
  189. """Update printer state from message data."""
  190. previous_state = self.state.state
  191. # Update state fields
  192. if "gcode_state" in data:
  193. self.state.state = data["gcode_state"]
  194. if "gcode_file" in data:
  195. self.state.gcode_file = data["gcode_file"]
  196. self.state.current_print = data["gcode_file"]
  197. if "subtask_name" in data:
  198. self.state.subtask_name = data["subtask_name"]
  199. # Prefer subtask_name as current_print if available
  200. if data["subtask_name"]:
  201. self.state.current_print = data["subtask_name"]
  202. if "subtask_id" in data:
  203. self.state.subtask_id = data["subtask_id"]
  204. if "mc_percent" in data:
  205. self.state.progress = float(data["mc_percent"])
  206. if "mc_remaining_time" in data:
  207. self.state.remaining_time = int(data["mc_remaining_time"])
  208. if "layer_num" in data:
  209. self.state.layer_num = int(data["layer_num"])
  210. if "total_layer_num" in data:
  211. self.state.total_layers = int(data["total_layer_num"])
  212. # Temperature data
  213. temps = {}
  214. # Log all temperature-related fields for debugging (only when we have temp data)
  215. temp_fields = {k: v for k, v in data.items() if 'temp' in k.lower() or 'nozzle' in k.lower()}
  216. if temp_fields and not hasattr(self, '_temp_fields_logged'):
  217. logger.info(f"[{self.serial_number}] Temperature fields in MQTT data: {temp_fields}")
  218. self._temp_fields_logged = True
  219. if "bed_temper" in data:
  220. temps["bed"] = float(data["bed_temper"])
  221. if "bed_target_temper" in data:
  222. temps["bed_target"] = float(data["bed_target_temper"])
  223. if "nozzle_temper" in data:
  224. temps["nozzle"] = float(data["nozzle_temper"])
  225. if "nozzle_target_temper" in data:
  226. temps["nozzle_target"] = float(data["nozzle_target_temper"])
  227. # Second nozzle for dual-extruder printers (H2 series)
  228. # Try multiple possible field names used by different firmware versions
  229. if "nozzle_temper_2" in data:
  230. temps["nozzle_2"] = float(data["nozzle_temper_2"])
  231. elif "right_nozzle_temper" in data:
  232. temps["nozzle_2"] = float(data["right_nozzle_temper"])
  233. if "nozzle_target_temper_2" in data:
  234. temps["nozzle_2_target"] = float(data["nozzle_target_temper_2"])
  235. elif "right_nozzle_target_temper" in data:
  236. temps["nozzle_2_target"] = float(data["right_nozzle_target_temper"])
  237. # Also check for left nozzle as primary (some H2 models)
  238. if "left_nozzle_temper" in data and "nozzle" not in temps:
  239. temps["nozzle"] = float(data["left_nozzle_temper"])
  240. if "left_nozzle_target_temper" in data and "nozzle_target" not in temps:
  241. temps["nozzle_target"] = float(data["left_nozzle_target_temper"])
  242. if "chamber_temper" in data:
  243. temps["chamber"] = float(data["chamber_temper"])
  244. if temps:
  245. self.state.temperatures = temps
  246. # Parse HMS (Health Management System) errors
  247. if "hms" in data:
  248. hms_list = data["hms"]
  249. self.state.hms_errors = []
  250. if isinstance(hms_list, list):
  251. for hms in hms_list:
  252. if isinstance(hms, dict):
  253. # HMS format: {"attr": code, "code": full_code}
  254. # The code is a hex string, severity is in bits
  255. code = hms.get("code", hms.get("attr", "0"))
  256. if isinstance(code, int):
  257. code = hex(code)
  258. # Parse severity from code (typically last 4 bits indicate level)
  259. try:
  260. code_int = int(str(code).replace("0x", ""), 16) if code else 0
  261. severity = (code_int >> 16) & 0xF # Extract severity bits
  262. module = (code_int >> 24) & 0xFF # Extract module bits
  263. except (ValueError, TypeError):
  264. severity = 3
  265. module = 0
  266. self.state.hms_errors.append(HMSError(
  267. code=str(code),
  268. module=module,
  269. severity=severity if severity > 0 else 3,
  270. ))
  271. # Preserve AMS data when updating raw_data (AMS comes at top level, not in print)
  272. ams_data = self.state.raw_data.get("ams")
  273. self.state.raw_data = data
  274. if ams_data is not None:
  275. self.state.raw_data["ams"] = ams_data
  276. # Log state transitions for debugging
  277. if "gcode_state" in data:
  278. logger.debug(
  279. f"[{self.serial_number}] gcode_state: {self._previous_gcode_state} -> {self.state.state}, "
  280. f"file: {self.state.gcode_file}, subtask: {self.state.subtask_name}"
  281. )
  282. # Detect print start (state changes TO RUNNING with a file)
  283. current_file = self.state.gcode_file or self.state.current_print
  284. is_new_print = (
  285. self.state.state == "RUNNING"
  286. and self._previous_gcode_state != "RUNNING"
  287. and current_file
  288. )
  289. # Also detect if file changed while running (new print started)
  290. is_file_change = (
  291. self.state.state == "RUNNING"
  292. and current_file
  293. and current_file != self._previous_gcode_file
  294. and self._previous_gcode_file is not None
  295. )
  296. # Track RUNNING state for more robust completion detection
  297. if self.state.state == "RUNNING" and current_file:
  298. if not self._was_running:
  299. logger.info(f"[{self.serial_number}] Now tracking RUNNING state for {current_file}")
  300. self._was_running = True
  301. self._completion_triggered = False
  302. if is_new_print or is_file_change:
  303. # Clear any old HMS errors when a new print starts
  304. self.state.hms_errors = []
  305. # Reset completion tracking for new print
  306. self._was_running = True
  307. self._completion_triggered = False
  308. if (is_new_print or is_file_change) and self.on_print_start:
  309. logger.info(
  310. f"[{self.serial_number}] PRINT START detected - file: {current_file}, "
  311. f"subtask: {self.state.subtask_name}, is_new: {is_new_print}, is_file_change: {is_file_change}"
  312. )
  313. self.on_print_start({
  314. "filename": current_file,
  315. "subtask_name": self.state.subtask_name,
  316. "raw_data": data,
  317. })
  318. # Detect print completion (FINISH = success, FAILED = error, IDLE = aborted)
  319. # Use _was_running flag in addition to _previous_gcode_state for more robust detection
  320. # This handles cases where server restarts during a print
  321. should_trigger_completion = (
  322. self.state.state in ("FINISH", "FAILED")
  323. and not self._completion_triggered
  324. and self.on_print_complete
  325. and (
  326. self._previous_gcode_state == "RUNNING" # Normal transition
  327. or (self._was_running and self._previous_gcode_state != self.state.state) # After server restart
  328. )
  329. )
  330. # For IDLE, only trigger if we just came from RUNNING (explicit abort/cancel)
  331. if (
  332. self.state.state == "IDLE"
  333. and self._previous_gcode_state == "RUNNING"
  334. and not self._completion_triggered
  335. and self.on_print_complete
  336. ):
  337. should_trigger_completion = True
  338. if should_trigger_completion:
  339. if self.state.state == "FINISH":
  340. status = "completed"
  341. elif self.state.state == "FAILED":
  342. status = "failed"
  343. else:
  344. status = "aborted"
  345. logger.info(
  346. f"[{self.serial_number}] PRINT COMPLETE detected - state: {self.state.state}, "
  347. f"status: {status}, file: {self._previous_gcode_file or current_file}, "
  348. f"subtask: {self.state.subtask_name}, was_running: {self._was_running}"
  349. )
  350. self._completion_triggered = True
  351. self._was_running = False
  352. self.on_print_complete({
  353. "status": status,
  354. "filename": self._previous_gcode_file or current_file,
  355. "subtask_name": self.state.subtask_name,
  356. "raw_data": data,
  357. })
  358. self._previous_gcode_state = self.state.state
  359. if current_file:
  360. self._previous_gcode_file = current_file
  361. if self.on_state_change:
  362. self.on_state_change(self.state)
  363. def _request_push_all(self):
  364. """Request full status update from printer."""
  365. if self._client:
  366. message = {"pushing": {"command": "pushall"}}
  367. self._client.publish(self.topic_publish, json.dumps(message))
  368. def connect(self, loop: asyncio.AbstractEventLoop | None = None):
  369. """Connect to the printer MQTT broker.
  370. Args:
  371. loop: The asyncio event loop to use for thread-safe callbacks.
  372. If not provided, will try to get the running loop.
  373. """
  374. self._loop = loop
  375. self._client = mqtt.Client(
  376. callback_api_version=mqtt.CallbackAPIVersion.VERSION2,
  377. client_id=f"bambutrack_{self.serial_number}",
  378. protocol=mqtt.MQTTv311,
  379. )
  380. self._client.username_pw_set("bblp", self.access_code)
  381. self._client.on_connect = self._on_connect
  382. self._client.on_disconnect = self._on_disconnect
  383. self._client.on_message = self._on_message
  384. # TLS setup - Bambu uses self-signed certs
  385. ssl_context = ssl.create_default_context()
  386. ssl_context.check_hostname = False
  387. ssl_context.verify_mode = ssl.CERT_NONE
  388. self._client.tls_set_context(ssl_context)
  389. # Use shorter keepalive (15s) for faster disconnect detection
  390. # Paho considers connection lost after 1.5x keepalive with no response
  391. self._client.connect_async(self.ip_address, self.MQTT_PORT, keepalive=15)
  392. self._client.loop_start()
  393. def start_print(self, filename: str, plate_id: int = 1):
  394. """Start a print job on the printer.
  395. The file should already be uploaded to /cache/ on the printer via FTP.
  396. """
  397. if self._client and self.state.connected:
  398. # Bambu print command format
  399. # Based on: https://github.com/darkorb/bambu-ftp-and-print
  400. command = {
  401. "print": {
  402. "sequence_id": 0,
  403. "command": "project_file",
  404. "param": f"Metadata/plate_{plate_id}.gcode",
  405. "subtask_name": filename,
  406. "url": f"ftp://{filename}",
  407. "timelapse": False,
  408. "bed_leveling": True,
  409. "flow_cali": True,
  410. "vibration_cali": True,
  411. "layer_inspect": False,
  412. "use_ams": True,
  413. }
  414. }
  415. logger.info(f"[{self.serial_number}] Sending print command: {json.dumps(command)}")
  416. self._client.publish(self.topic_publish, json.dumps(command))
  417. return True
  418. return False
  419. def stop_print(self) -> bool:
  420. """Stop the current print job."""
  421. if self._client and self.state.connected:
  422. command = {
  423. "print": {
  424. "command": "stop",
  425. "sequence_id": "0"
  426. }
  427. }
  428. self._client.publish(self.topic_publish, json.dumps(command))
  429. logger.info(f"[{self.serial_number}] Sent stop print command")
  430. return True
  431. return False
  432. def disconnect(self):
  433. """Disconnect from the printer."""
  434. if self._client:
  435. self._client.loop_stop()
  436. self._client.disconnect()
  437. self._client = None
  438. self.state.connected = False
  439. def send_command(self, command: dict):
  440. """Send a command to the printer."""
  441. if self._client and self.state.connected:
  442. # Log outgoing message if logging is enabled
  443. if self._logging_enabled:
  444. self._message_log.append(MQTTLogEntry(
  445. timestamp=datetime.now().isoformat(),
  446. topic=self.topic_publish,
  447. direction="out",
  448. payload=command,
  449. ))
  450. self._client.publish(self.topic_publish, json.dumps(command))
  451. def enable_logging(self, enabled: bool = True):
  452. """Enable or disable MQTT message logging."""
  453. self._logging_enabled = enabled
  454. # Don't clear logs when stopping - user can manually clear with clear_logs()
  455. def get_logs(self) -> list[MQTTLogEntry]:
  456. """Get all logged MQTT messages."""
  457. return list(self._message_log)
  458. def clear_logs(self):
  459. """Clear the message log."""
  460. self._message_log.clear()
  461. @property
  462. def logging_enabled(self) -> bool:
  463. """Check if logging is enabled."""
  464. return self._logging_enabled
  465. def _handle_kprofile_response(self, data: dict):
  466. """Handle K-profile response from printer."""
  467. filaments = data.get("filaments", [])
  468. profiles = []
  469. # Log first profile to see what fields the printer returns
  470. if filaments and isinstance(filaments[0], dict):
  471. logger.debug(f"[{self.serial_number}] Raw K-profile fields: {list(filaments[0].keys())}")
  472. logger.debug(f"[{self.serial_number}] First K-profile: {filaments[0]}")
  473. for i, f in enumerate(filaments):
  474. if isinstance(f, dict):
  475. try:
  476. # cali_idx is the actual slot/calibration index from the printer
  477. cali_idx = f.get("cali_idx", i)
  478. profiles.append(KProfile(
  479. slot_id=cali_idx,
  480. extruder_id=int(f.get("extruder_id", 0)),
  481. nozzle_id=str(f.get("nozzle_id", "")),
  482. nozzle_diameter=str(f.get("nozzle_diameter", "0.4")),
  483. filament_id=str(f.get("filament_id", "")),
  484. name=str(f.get("name", "")),
  485. k_value=str(f.get("k_value", "0.000000")),
  486. n_coef=str(f.get("n_coef", "0.000000")),
  487. ams_id=int(f.get("ams_id", 0)),
  488. tray_id=int(f.get("tray_id", -1)),
  489. setting_id=f.get("setting_id"),
  490. ))
  491. except (ValueError, TypeError) as e:
  492. logger.warning(f"Failed to parse K-profile: {e}")
  493. self.state.kprofiles = profiles
  494. self._kprofile_response_data = profiles
  495. # Signal that we received the response
  496. # Use thread-safe method since MQTT callbacks run in a different thread
  497. if self._pending_kprofile_response:
  498. if self._loop and self._loop.is_running():
  499. self._loop.call_soon_threadsafe(self._pending_kprofile_response.set)
  500. else:
  501. # Fallback for when loop is not available
  502. self._pending_kprofile_response.set()
  503. logger.info(f"[{self.serial_number}] Received {len(profiles)} K-profiles")
  504. async def get_kprofiles(self, nozzle_diameter: str = "0.4", timeout: float = 10.0) -> list[KProfile]:
  505. """Request K-profiles from the printer.
  506. Args:
  507. nozzle_diameter: Filter by nozzle diameter (e.g., "0.4")
  508. timeout: Timeout in seconds to wait for response
  509. Returns:
  510. List of KProfile objects
  511. """
  512. if not self._client or not self.state.connected:
  513. logger.warning(f"[{self.serial_number}] Cannot get K-profiles: not connected")
  514. return []
  515. # Capture current event loop for thread-safe callback
  516. try:
  517. self._loop = asyncio.get_running_loop()
  518. except RuntimeError:
  519. logger.warning(f"[{self.serial_number}] No running event loop")
  520. return []
  521. # Set up response event
  522. self._sequence_id += 1
  523. self._pending_kprofile_response = asyncio.Event()
  524. self._kprofile_response_data = None
  525. # Send the command
  526. command = {
  527. "print": {
  528. "command": "extrusion_cali_get",
  529. "filament_id": "",
  530. "nozzle_diameter": nozzle_diameter,
  531. "sequence_id": str(self._sequence_id),
  532. }
  533. }
  534. logger.info(f"[{self.serial_number}] Requesting K-profiles for nozzle {nozzle_diameter}")
  535. self._client.publish(self.topic_publish, json.dumps(command))
  536. # Wait for response
  537. try:
  538. await asyncio.wait_for(self._pending_kprofile_response.wait(), timeout=timeout)
  539. return self._kprofile_response_data or []
  540. except asyncio.TimeoutError:
  541. logger.warning(f"[{self.serial_number}] Timeout waiting for K-profiles response")
  542. return []
  543. finally:
  544. self._pending_kprofile_response = None
  545. def set_kprofile(
  546. self,
  547. filament_id: str,
  548. name: str,
  549. k_value: str,
  550. nozzle_diameter: str = "0.4",
  551. nozzle_id: str = "HS00-0.4",
  552. extruder_id: int = 0,
  553. setting_id: str | None = None,
  554. slot_id: int = 0,
  555. ) -> bool:
  556. """Set/update a K-profile on the printer.
  557. Args:
  558. filament_id: Bambu filament identifier
  559. name: Profile name
  560. k_value: Pressure advance value (e.g., "0.020000")
  561. nozzle_diameter: Nozzle diameter (e.g., "0.4")
  562. nozzle_id: Nozzle identifier (e.g., "HS00-0.4")
  563. extruder_id: Extruder ID (0 or 1 for dual nozzle)
  564. setting_id: Existing setting ID for updates, None for new
  565. slot_id: Calibration index (cali_idx) for the profile
  566. Returns:
  567. True if command was sent, False otherwise
  568. """
  569. if not self._client or not self.state.connected:
  570. logger.warning(f"[{self.serial_number}] Cannot set K-profile: not connected")
  571. return False
  572. self._sequence_id += 1
  573. # Build the filament entry - printer uses cali_idx for profile identification
  574. # For new profiles (slot_id=0), use cali_idx=-1 to tell printer to create new slot
  575. cali_idx = -1 if slot_id == 0 else slot_id
  576. # Generate a setting_id for new profiles (required by printer)
  577. # Format: "PF" + 17 random digits
  578. import random
  579. if not setting_id and slot_id == 0:
  580. setting_id = f"PF{random.randint(10000000000000000, 99999999999999999)}"
  581. filament_entry = {
  582. "ams_id": 0,
  583. "cali_idx": cali_idx,
  584. "extruder_id": extruder_id,
  585. "filament_id": filament_id,
  586. "k_value": k_value,
  587. "n_coef": "0.000000",
  588. "name": name,
  589. "nozzle_diameter": nozzle_diameter,
  590. "nozzle_id": nozzle_id,
  591. "setting_id": setting_id, # Always include setting_id
  592. "tray_id": -1,
  593. }
  594. command = {
  595. "print": {
  596. "command": "extrusion_cali_set",
  597. "filaments": [filament_entry],
  598. "nozzle_diameter": nozzle_diameter,
  599. "sequence_id": str(self._sequence_id),
  600. }
  601. }
  602. command_json = json.dumps(command)
  603. logger.info(f"[{self.serial_number}] Setting K-profile: {name} = {k_value} (cali_idx={cali_idx}, new={slot_id==0})")
  604. logger.debug(f"[{self.serial_number}] K-profile command: {command_json}")
  605. self._client.publish(self.topic_publish, command_json)
  606. return True
  607. def delete_kprofile(
  608. self,
  609. cali_idx: int,
  610. filament_id: str,
  611. nozzle_id: str,
  612. nozzle_diameter: str = "0.4",
  613. extruder_id: int = 0,
  614. ) -> bool:
  615. """Delete a K-profile from the printer.
  616. Args:
  617. cali_idx: The calibration index (slot_id) of the profile to delete
  618. filament_id: Bambu filament identifier
  619. nozzle_id: Nozzle identifier (e.g., "HH00-0.4")
  620. nozzle_diameter: Nozzle diameter (e.g., "0.4")
  621. extruder_id: Extruder ID (0 or 1 for dual nozzle)
  622. Returns:
  623. True if command was sent, False otherwise
  624. """
  625. if not self._client or not self.state.connected:
  626. logger.warning(f"[{self.serial_number}] Cannot delete K-profile: not connected")
  627. return False
  628. self._sequence_id += 1
  629. command = {
  630. "print": {
  631. "command": "extrusion_cali_del",
  632. "sequence_id": str(self._sequence_id),
  633. "extruder_id": extruder_id,
  634. "nozzle_id": nozzle_id,
  635. "filament_id": filament_id,
  636. "cali_idx": cali_idx,
  637. "nozzle_diameter": nozzle_diameter,
  638. }
  639. }
  640. command_json = json.dumps(command)
  641. logger.info(f"[{self.serial_number}] Deleting K-profile: cali_idx={cali_idx}, filament={filament_id}")
  642. logger.debug(f"[{self.serial_number}] K-profile delete command: {command_json}")
  643. self._client.publish(self.topic_publish, command_json)
  644. return True