Option to disable all keys by using --disable_all_keys.

dstatic/dstatic.py

@@ -191,9 +191,9 @@ def static(screen, color_mode: str, args: argparse.Namespace):
         ch = screen.getch()
         if args.screen_saver and ch != -1:
             break
-        if ch in [81, 113]:  # q, Q
+        if ch in [81, 113] and not args.disable_all_keys:  # q, Q
             break
-        if args.disable_keys:
+        if args.disable_keys or args.disable_all_keys:
             pass
         else:
             if ch == 98:  # b
@@ -336,6 +336,9 @@ def argument_parser(argv: Optional[Sequence[str]] = None) -> argparse.Namespace:
     parser.add_argument("-D", dest="disable_keys", action="store_true",
                         help="Disable keys while running except for 'Q' or 'q' and "
                              "for ctrl-c. Does not affect screensaver mode.")
+    parser.add_argument("--disable_all_keys", action="store_true",
+                        help="Disable all keys while running including 'Q' and 'q.'"
+                        "Use ctrl-c to quit. Does not affect screensaver mode.")
     parser.add_argument("--list_colors", action="store_true",
                         help="List available colors and exit.")
     parser.add_argument("--list_commands", action="store_true",

tests/test_dstatic.py

@@ -3,6 +3,7 @@
 from unittest import mock
 
 import pytest
+import time
 from hecate import Runner
 
 from dstatic import dstatic
@@ -276,3 +277,34 @@ def test_dstatic_disable_keys_quit_works(test_keys):
         h.write(test_keys)
         h.press("Enter")
         h.await_exit()
+
+
+@pytest.mark.parametrize("test_keys", ["Q", "q"])
+def test_dstatic_disable_all_keys_check_quit(test_keys):
+    with Runner(*dstatic_cmd("--disable_all_keys", "--test_mode")) as h:
+        h.default_timeout = 1
+        h.await_text("0")
+        h.write(test_keys)
+        h.press("Enter")
+        time.sleep(1)
+        h.await_text("0")
+        sc = h.screenshot()
+        assert "$" not in sc
+
+
+def test_dstatic_disable_all_keys_run_timer_exit():
+    """ Test auto exit when using run timer. """
+    with Runner(*dstatic_cmd("-r2", "--disable_all_keys")) as h:
+        h.default_timeout = 3
+        h.await_text(chr(9617))
+        h.await_exit()
+
+
+@pytest.mark.parametrize("test_keys", ["Q", "q", "1", "u", "+", "d", "a", " "])
+def test_dstatic_disable_all_keys_screensaver_works(test_keys):
+    with Runner(*dstatic_cmd("--disable_all_keys", "-S")) as h:
+        h.default_timeout = 3
+        h.await_text(chr(9617))
+        h.write(test_keys)
+        h.press("Enter")
+        h.await_exit()

tests/test_dstatic_arguments.py

@@ -157,3 +157,11 @@ def test_argument_parsing_additive_color_mode(test_values, expected_result):
 def test_argument_parsing_disable_keys(test_values, expected_result):
     result = dstatic.argument_parser(test_values)
     assert result.disable_keys is expected_result
+
+
+@pytest.mark.parametrize("test_values, expected_result", [
+    ([], False), (["--disable_all_keys"], True),
+])
+def test_argument_parsing_disable_all_keys(test_values, expected_result):
+    result = dstatic.argument_parser(test_values)
+    assert result.disable_all_keys is expected_result