feat (trivial): use ops to tell juju to reboot the machine. (#146)

* Use ops to tell juju to reboot the machine. * Fix version requirement.
canonical · Nov 14, 2023 · d5afde0 · d5afde0
1 parent 07e0cbd
commit d5afde0
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 7 deletions.
diff --git a/requirements.txt b/requirements.txt
@@ -1,6 +1,6 @@
 ghapi
 jinja2
-ops
+ops>=2.8
 pylxd @ git+https://github.com/canonical/pylxd
 requests
 typing-extensions

diff --git a/src/charm.py b/src/charm.py
@@ -385,12 +385,7 @@ def _update_kernel(self, now: bool = False) -> None:
         _, exit_code = execute_command(["ls", "/var/run/reboot-required"], check_exit=False)
         if exit_code == 0:
             logger.info("Rebooting system...")
-
-            # The juju-reboot is inject to PATH by juju.
-            cmd = ["juju-reboot"]
-            if now:
-                cmd += ["--now"]
-            execute_command(cmd)
+            self.unit.reboot(now=now)
 
     @catch_charm_errors
     def _on_upgrade_charm(self, _event: UpgradeCharmEvent) -> None: