feat(bin): add mode parameter to agent._restart & and call `agent._…

…restart` from `USR1` handler instead of `shikigotchi.restart()`
andatoshiki · Sep 17, 2024 · 50382f4 · 50382f4
1 parent 06f336b
commit 50382f4
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 3 deletions.
diff --git a/bin/shikigotchi b/bin/shikigotchi
@@ -324,7 +324,7 @@ def shikigotchi_cli():
 
     def usr1_handler(*unused):
         logging.info('Received USR1 singal. Restart process ...')
-        restart("MANU" if args.do_manual else "AUTO")
+        agent.restart("MANU" if args.do_manual else "AUTO")
 
     signal.signal(signal.SIGUSR1, usr1_handler)
 

diff --git a/shikigotchi/agent.py b/shikigotchi/agent.py
@@ -269,9 +269,9 @@ def _reboot(self):
         self._save_recovery_data()
         shikigotchi.reboot()
 
-    def _restart(self):
+    def _restart(self, mode='AUTO'):
         self._save_recovery_data()
-        shikigotchi.restart("AUTO")
+        shikigotchi.restart(mode)
 
     def _save_recovery_data(self):
         logging.warning("writing recovery data to %s ...", RECOVERY_DATA_FILE)