fix the crash in a case of non-existent filament data

agrif · May 3, 2024 · 0806c6c · 0806c6c
1 parent c2a4e8b
commit 0806c6c
Showing 1 changed file with 11 additions and 10 deletions.
diff --git a/octoprint_influxdb/__init__.py b/octoprint_influxdb/__init__.py
@@ -303,16 +303,17 @@ def add_to(d, k, x):
 			if job.get('file', {}).get('name'):
 				# a file is loaded...
 				filename = job['file']['name']
-				filaments = job.get('filament')
-				for tool, filval in filaments.items():
-					fields = { }
-					tags = {}
-					add_to(fields, 'length', filval.get('length'))
-					add_to(fields, 'volume', filval.get('volume'))
-					add_to(tags, 'filename', filename)
-					add_to(tags, 'tool', tool)
-					if fields:
-						self.influx_emit('filament', fields, tags)
+				if job.get('filament'):
+					filaments = job.get('filament')
+					for tool, filval in filaments.items():
+						fields = { }
+						tags = {}
+						add_to(fields, 'length', filval.get('length'))
+						add_to(fields, 'volume', filval.get('volume'))
+						add_to(tags, 'filename', filename)
+						add_to(tags, 'tool', tool)
+						if fields:
+							self.influx_emit('filament', fields, tags)
 
 	##~~ EventHandlerPlugin mixin