Check for sleep ability at startup, not each run

bulletmark · bulletmark · commit 7efc5dadec9f · 2023-06-06T09:56:31.000+10:00
diff --git a/setup.py b/setup.py
@@ -11,7 +11,7 @@
 
 setup(
     name=name,
-    version='1.19',
+    version='1.20',
     description='Program to run plugins to inhibit system '
     'sleep/suspend/hibernate',
     long_description=here.joinpath('README.md').read_text(),
diff --git a/sleep_inhibitor.py b/sleep_inhibitor.py
@@ -101,7 +101,7 @@ async def run(self):
             await proc.wait()
 
             while proc.returncode == SUSP_CODE:
-                if not self.is_inhibiting:
+                if self.is_inhibiting is not True:
                     self.is_inhibiting = True
                     print(f'{self.name} is inhibiting '
                           f'suspend (return={proc.returncode})')
@@ -128,10 +128,6 @@ def init():
     opt.add_argument('-i', '--inhibit', help=argparse.SUPPRESS)
     args = opt.parse_args()
 
-    # Don't run if this system does not support sleep
-    if not Path('/sys/power/state').read_text():
-        sys.exit('System does not support any sleep states, quitting.')
-
     # This instance may be a child invocation merely to run and check
     # the plugin while it is inhibiting.
     if args.inhibit:
@@ -142,6 +138,11 @@ def init():
             if res.returncode != SUSP_CODE:
                 sys.exit(res.returncode)
 
+    # Don't run if this system does not support sleep
+    pstate = Path('/sys/power/state')
+    if not pstate.exists() or not pstate.read_text().strip():
+        sys.exit('System does not support any sleep states, quitting.')
+
     prog = Path(sys.argv[0]).resolve()
     progname = prog.stem.replace('_', '-')
 
