Handle KeyErrors separately

turtlebot · Sep 12, 2024 · 6519d9b · 6519d9b
1 parent 72f04a3
commit 6519d9b
Showing 1 changed file with 4 additions and 0 deletions.
diff --git a/turtlebot4_setup/ros_setup.py b/turtlebot4_setup/ros_setup.py
@@ -370,6 +370,8 @@ def install(self):
 
             self.daemon_reload()
 
+        except KeyError as err:
+            ErrorPrompt(f'Failed to install systemd job:\n{err} is not defined').show()
         except Exception as err:
             ErrorPrompt(f'Failed to install systemd job:\n{err}').show()
 
@@ -391,6 +393,8 @@ def uninstall(self):
                 discovery_job.uninstall(Provider=TurtleBot4Extras)
 
             self.daemon_reload()
+        except KeyError as err:
+            ErrorPrompt(f'Failed to uninstall existing systemd job:\n{err} is not defined').show()
         except Exception as err:
             ErrorPrompt(f'Failed to uninstall existing systemd job:\n{err}').show()