delete controller when terminate

src/drunc/process_manager/interface/commands.py

@@ -86,6 +86,8 @@ async def terminate(obj:ProcessManagerContext) -> None:
     from drunc.process_manager.utils import tabulate_process_instance_list
     obj.print(tabulate_process_instance_list(result.data, 'Terminated process', False))
 
+    obj.delete_driver('controller')
+
 @click.command('kill')
 @add_query_options(at_least_one=True)
 @click.pass_obj
@@ -100,6 +102,7 @@ async def kill(obj:ProcessManagerContext, query:ProcessQuery) -> None:
     from drunc.process_manager.utils import tabulate_process_instance_list
     obj.print(tabulate_process_instance_list(result.data, 'Killed process', False))
 
+    obj.delete_driver('controller')
 
 @click.command('flush')
 @add_query_options(at_least_one=False, all_processes_by_default=True)

src/drunc/utils/shell_utils.py

@@ -276,6 +276,14 @@ def get_driver(self, name:str=None) -> GRPCDriver:
             self._log.error(f'Controller-specific commands cannot be sent until the session is booted')
             raise SystemExit(1) # used to avoid having to catch multiple Attribute errors when this function gets called
 
+    def delete_driver(self, name: str) -> None:
+        if name in self._drivers:
+            del self._drivers[name]
+            self._log.info(f"Driver '{name}' has been deleted.")
+        else:
+            self._log.warn(f"Driver '{name}' does not exist in this context.")
+
+
     def get_token(self) -> Token:
         return self._token